genai: add basic PDF uploading example (#202)

google · Aug 2, 2024 · fb56fc2 · fb56fc2
1 parent 361f1ce
commit fb56fc2
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 0 deletions.
diff --git a/genai/example_test.go b/genai/example_test.go
@@ -1119,6 +1119,32 @@ func ExampleClient_UploadFile_image() {
 
 }
 
+func ExampleClient_UploadFile_pdf() {
+	ctx := context.Background()
+	client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer client.Close()
+
+	file, err := uploadFile(ctx, client, filepath.Join(testDataDir, "test.pdf"), "")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer client.DeleteFile(ctx, file.Name)
+
+	model := client.GenerativeModel("gemini-1.5-flash")
+	resp, err := model.GenerateContent(ctx,
+		genai.Text("Give me a summary of this pdf file."),
+		genai.FileData{URI: file.URI})
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	printResponse(resp)
+
+}
+
 func ExampleClient_UploadFile_video() {
 	ctx := context.Background()
 	client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))

diff --git a/genai/internal/samples/docs-snippets_test.go b/genai/internal/samples/docs-snippets_test.go
@@ -1150,6 +1150,33 @@ func ExampleClient_UploadFile_image() {
 	// [END files_create_image]
 }
 
+func ExampleClient_UploadFile_pdf() {
+	ctx := context.Background()
+	client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer client.Close()
+
+	// [START files_create_pdf]
+	file, err := uploadFile(ctx, client, filepath.Join(testDataDir, "test.pdf"), "")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer client.DeleteFile(ctx, file.Name)
+
+	model := client.GenerativeModel("gemini-1.5-flash")
+	resp, err := model.GenerateContent(ctx,
+		genai.Text("Give me a summary of this pdf file."),
+		genai.FileData{URI: file.URI})
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	printResponse(resp)
+	// [END files_create_pdf]
+}
+
 func ExampleClient_UploadFile_video() {
 	ctx := context.Background()
 	client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))

diff --git a/genai/testdata/test.pdf b/genai/testdata/test.pdf