.

Author: nicohrubec

dev-packages/browser-integration-tests/suites/tracing/ai-providers/google-genai/test.ts

@@ -29,3 +29,26 @@ sentryTest('manual Google GenAI instrumentation sends gen_ai transactions', asyn
     'gen_ai.request.model': 'gemini-1.5-pro',
   });
 });
+
+sentryTest('manual Google GenAI instrumentation sends embeddings transactions', async ({ getLocalTestUrl, page }) => {
+  const transactionPromise = waitForTransactionRequest(page, event => {
+    return !!event.transaction?.includes('text-embedding-004');
+  });
+
+  const url = await getLocalTestUrl({ testDir: __dirname });
+  await page.goto(url);
+
+  const req = await transactionPromise;
+
+  const eventData = envelopeRequestParser(req);
+
+  // Verify it's a gen_ai embeddings transaction
+  expect(eventData.transaction).toBe('embeddings text-embedding-004');
+  expect(eventData.contexts?.trace?.op).toBe('gen_ai.embeddings');
+  expect(eventData.contexts?.trace?.origin).toBe('auto.ai.google_genai');
+  expect(eventData.contexts?.trace?.data).toMatchObject({
+    'gen_ai.operation.name': 'embeddings',
+    'gen_ai.system': 'google_genai',
+    'gen_ai.request.model': 'text-embedding-004',
+  });
+});

dev-packages/cloudflare-integration-tests/suites/tracing/google-genai/index.ts

@@ -55,7 +55,13 @@ export default Sentry.withSentry(
         ],
       });
 
-      return new Response(JSON.stringify({ chatResponse, modelResponse }));
+      // Test 3: models.embedContent
+      const embedResponse = await client.models.embedContent({
+        model: 'text-embedding-004',
+        contents: 'Hello world',
+      });
+
+      return new Response(JSON.stringify({ chatResponse, modelResponse, embedResponse }));
     },
   },
 );

dev-packages/cloudflare-integration-tests/suites/tracing/google-genai/mocks.ts

@@ -4,6 +4,7 @@ export class MockGoogleGenAI implements GoogleGenAIClient {
   public models: {
     generateContent: (...args: unknown[]) => Promise<GoogleGenAIResponse>;
     generateContentStream: (...args: unknown[]) => Promise<AsyncGenerator<GoogleGenAIResponse, any, unknown>>;
+    embedContent: (...args: unknown[]) => Promise<{ embeddings: { values: number[] }[] }>;
   };
   public chats: {
     create: (...args: unknown[]) => GoogleGenAIChat;
@@ -49,6 +50,20 @@ export class MockGoogleGenAI implements GoogleGenAIClient {
           },
         };
       },
+      embedContent: async (...args: unknown[]) => {
+        const params = args[0] as { model: string; contents?: unknown };
+        await new Promise(resolve => setTimeout(resolve, 10));
+
+        if (params.model === 'error-model') {
+          const error = new Error('Model not found');
+          (error as unknown as { status: number }).status = 404;
+          throw error;
+        }
+
+        return {
+          embeddings: [{ values: [0.1, 0.2, 0.3, 0.4, 0.5] }],
+        };
+      },
       generateContentStream: async () => {
         // Return a promise that resolves to an async generator
         return (async function* (): AsyncGenerator<GoogleGenAIResponse, any, unknown> {

dev-packages/cloudflare-integration-tests/suites/tracing/google-genai/test.ts

@@ -78,6 +78,19 @@ it('traces Google GenAI chat creation and message sending', async () => {
             op: 'gen_ai.generate_content',
             origin: 'auto.ai.google_genai',
           }),
+          // Fourth span - models.embedContent
+          expect.objectContaining({
+            data: expect.objectContaining({
+              [GEN_AI_OPERATION_NAME_ATTRIBUTE]: 'embeddings',
+              [SEMANTIC_ATTRIBUTE_SENTRY_OP]: 'gen_ai.embeddings',
+              [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: 'auto.ai.google_genai',
+              [GEN_AI_SYSTEM_ATTRIBUTE]: 'google_genai',
+              [GEN_AI_REQUEST_MODEL_ATTRIBUTE]: 'text-embedding-004',
+            }),
+            description: 'embeddings text-embedding-004',
+            op: 'gen_ai.embeddings',
+            origin: 'auto.ai.google_genai',
+          }),
         ]),
       );
     })

packages/core/src/tracing/google-genai/index.ts

@@ -1,3 +1,4 @@
+/* eslint-disable max-lines */
 import { captureException } from '../../exports';
 import { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN } from '../../semanticAttributes';
 import { SPAN_STATUS_ERROR } from '../../tracing';

packages/core/src/tracing/google-genai/types.ts

@@ -163,6 +163,8 @@ export interface GoogleGenAIClient {
     // https://googleapis.github.io/js-genai/release_docs/classes/models.Models.html#generatecontentstream
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     generateContentStream: (...args: unknown[]) => Promise<AsyncGenerator<GenerateContentResponse, any, unknown>>;
+    // https://googleapis.github.io/js-genai/release_docs/classes/models.Models.html#embedcontent
+    embedContent: (...args: unknown[]) => Promise<unknown>;
   };
   chats: {
     create: (...args: unknown[]) => GoogleGenAIChat;