genai: add uploading audio example, clarify comment in uploadFile (#184)

google · Jul 19, 2024 · 17dab4e · 17dab4e
1 parent 0fdd360
commit 17dab4e
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 0 deletions.
diff --git a/genai/example_test.go b/genai/example_test.go
@@ -1104,6 +1104,32 @@ func ExampleClient_UploadFile_video() {
 
 }
 
+func ExampleClient_UploadFile_audio() {
+	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, "sample.mp3"))
+	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.FileData{URI: file.URI},
+		genai.Text("Describe this audio clip"))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	printResponse(resp)
+
+}
+
 func ExampleClient_GetFile() {
 	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
@@ -1129,6 +1129,33 @@ func ExampleClient_UploadFile_video() {
 	// [END files_create_video]
 }
 
+func ExampleClient_UploadFile_audio() {
+	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_audio]
+	file, err := uploadFile(ctx, client, filepath.Join(testDataDir, "sample.mp3"))
+	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.FileData{URI: file.URI},
+		genai.Text("Describe this audio clip"))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	printResponse(resp)
+	// [END files_create_audio]
+}
+
 func ExampleClient_GetFile() {
 	ctx := context.Background()
 	client, err := genai.NewClient(ctx, option.WithAPIKey(os.Getenv("GEMINI_API_KEY")))