func EmbedContent()

in go/embed.go [13:44]

• 28 lines of code
• 4 McCabe index (conditional complexity)

func EmbedContent() error {
	// [START embed_content]
	ctx := context.Background()
	client, err := genai.NewClient(ctx, &genai.ClientConfig{
		APIKey:  os.Getenv("GEMINI_API_KEY"),
		Backend: genai.BackendGeminiAPI,
	})
	if err != nil {
		log.Fatal(err)
	}

	text := "Hello World!"
	outputDim := int32(10)
	contents := []*genai.Content{
		genai.NewContentFromText(text, genai.RoleUser),
	}
	result, err := client.Models.EmbedContent(ctx, "text-embedding-004", 
		contents, &genai.EmbedContentConfig{
			OutputDimensionality: &outputDim,
	})
	if err != nil {
		log.Fatal(err)
	}

	embeddings, err := json.MarshalIndent(result.Embeddings, "", "  ")
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(string(embeddings))
	// [END embed_content]
	return err
}