Adding example of EmbedContent

google · Dec 13, 2023 · 676dda8 · 676dda8
1 parent 0a2c82e
commit 676dda8
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/genai/example_test.go b/genai/example_test.go
@@ -130,6 +130,22 @@ func ExampleChatSession() {
 	printResponse(res)
 }
 
+func ExampleEmbeddingModel_EmbedContent() {
+	ctx := context.Background()
+	client, err := genai.NewClient(ctx, option.WithAPIKey("your-API-key"))
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer client.Close()
+	em := client.EmbeddingModel("embedding-001")
+	res, err := em.EmbedContent(ctx, genai.Text("cheddar cheese"))
+
+	if err != nil {
+		panic(err)
+	}
+	fmt.Println(res.Embedding.Values)
+}
+
 func ExampleClient_ListModels() {
 	ctx := context.Background()
 	client, err := genai.NewClient(ctx, option.WithAPIKey("your-API-key"))