Google GenAI `embedContent` not instrumented (wrapper and auto-instrumentation)

[braintrust-bot]

Summary

The @google/genai SDK provides a stable models.embedContent() method for generating vector embeddings, but neither the wrapper (wrapGoogleGenAI) nor the auto-instrumentation plugin instruments it. By contrast, OpenAI's embeddings.create() is fully instrumented in both paths.

What is missing

• Wrapper (js/src/wrappers/google-genai.ts): wrapModels() only proxies generateContent and generateContentStream (lines 88–101). embedContent calls pass through untraced.
• Auto-instrumentation config (js/src/auto-instrumentations/configs/google-genai.ts): Only defines configs for generateContentInternal and generateContentStreamInternal. No config for embedContent.
• Plugin (js/src/instrumentation/plugins/google-genai-plugin.ts): No channel or handler for embeddings.

Upstream reference

• @google/genai SDK — Models.embedContent(): API reference
• Gemini Embeddings guide: https://ai.google.dev/gemini-api/docs/embeddings
• Supports models like text-embedding-004 and gemini-embedding-001, with config options outputDimensionality and taskType.

Braintrust docs status

The Braintrust Google GenAI integration page documents generateContent only. Embeddings are not mentioned (not_found).

• https://www.braintrust.dev/docs/reference/wrappers/google-genai

Precedent in this repo

OpenAI embeddings are instrumented in all three layers:

• Wrapper: js/src/wrappers/oai.ts (lines 101–104, wrapEmbeddings)
• Auto-instrumentation config: js/src/auto-instrumentations/configs/openai.ts (lines 59–72)
• Plugin: js/src/instrumentation/plugins/openai-plugin.ts (lines 70–96)

Local files inspected

• js/src/wrappers/google-genai.ts
• js/src/auto-instrumentations/configs/google-genai.ts
• js/src/instrumentation/plugins/google-genai-plugin.ts
• js/src/instrumentation/plugins/google-genai-channels.ts
• js/src/vendor-sdk-types/google-genai.ts