fix(tracing): add system prompt, model to google genai (#18424)

Related to getsentry/testing-ai-sdk-integrations#10

Fix JS-1218

Author: isaacs

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

@@ -16,14 +16,16 @@ import {
   GEN_AI_REQUEST_TEMPERATURE_ATTRIBUTE,
   GEN_AI_REQUEST_TOP_K_ATTRIBUTE,
   GEN_AI_REQUEST_TOP_P_ATTRIBUTE,
+  GEN_AI_RESPONSE_MODEL_ATTRIBUTE,
   GEN_AI_RESPONSE_TEXT_ATTRIBUTE,
   GEN_AI_RESPONSE_TOOL_CALLS_ATTRIBUTE,
   GEN_AI_SYSTEM_ATTRIBUTE,
   GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE,
   GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE,
   GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE,
 } from '../ai/gen-ai-attributes';
-import { buildMethodPath, getFinalOperationName, getSpanOperation, getTruncatedJsonString } from '../ai/utils';
+import { truncateGenAiMessages } from '../ai/messageTruncation';
+import { buildMethodPath, getFinalOperationName, getSpanOperation } from '../ai/utils';
 import { CHAT_PATH, CHATS_CREATE_METHOD, GOOGLE_GENAI_SYSTEM_NAME } from './constants';
 import { instrumentStream } from './streaming';
 import type {
@@ -134,27 +136,52 @@ function extractRequestAttributes(
  * Handles different parameter formats for different Google GenAI methods.
  */
 function addPrivateRequestAttributes(span: Span, params: Record<string, unknown>): void {
+  const messages: unknown[] = [];
+
+  // include system prompt if present in config param
+  if (
+    'config' in params &&
+    params.config &&
+    typeof params.config === 'object' &&
+    'systemInstruction' in params.config
+  ) {
+    messages.push({
+      role: 'system',
+      content: params.config.systemInstruction,
+    });
+  }
+
   // For models.generateContent: ContentListUnion: Content | Content[] | PartUnion | PartUnion[]
   if ('contents' in params) {
-    const contents = params.contents;
-    // For models.generateContent: ContentListUnion: Content | Content[] | PartUnion | PartUnion[]
-    const truncatedContents = getTruncatedJsonString(contents);
-    span.setAttributes({ [GEN_AI_REQUEST_MESSAGES_ATTRIBUTE]: truncatedContents });
+    if (Array.isArray(params.contents)) {
+      messages.push(...params.contents);
+    } else {
+      messages.push({ role: 'user', content: params.contents });
+    }
   }
 
   // For chat.sendMessage: message can be string or Part[]
   if ('message' in params) {
-    const message = params.message;
-    const truncatedMessage = getTruncatedJsonString(message);
-    span.setAttributes({ [GEN_AI_REQUEST_MESSAGES_ATTRIBUTE]: truncatedMessage });
+    if (Array.isArray(params.message)) {
+      messages.push(...params.message);
+    } else {
+      messages.push({ role: 'user', content: params.message });
+    }
   }
 
   // For chats.create: history contains the conversation history
   if ('history' in params) {
-    const history = params.history;
-    const truncatedHistory = getTruncatedJsonString(history);
-    span.setAttributes({ [GEN_AI_REQUEST_MESSAGES_ATTRIBUTE]: truncatedHistory });
+    if (Array.isArray(params.history)) {
+      messages.push(...params.history);
+    } else {
+      messages.push({ role: 'user', content: params.history });
+    }
   }
+
+  span.setAttributes({
+    [GEN_AI_REQUEST_MESSAGES_ATTRIBUTE]: JSON.stringify(truncateGenAiMessages(messages)),
+    [GEN_AI_RESPONSE_MODEL_ATTRIBUTE]: String(params.model),
+  });
 }
 
 /**