fix (provider/google): add name/content details to tool responses (#3599)

Co-authored-by: EGOIST <[email protected]>

Author: lgrammel

.changeset/yellow-books-develop.md

@@ -0,0 +1,5 @@
+---
+'@ai-sdk/google': patch
+---
+
+fix (provider/google): add name/content details to tool responses

examples/ai-core/src/stream-text/google-chatbot.ts

@@ -1,7 +1,8 @@
 import { google } from '@ai-sdk/google';
-import { CoreMessage, streamText } from 'ai';
+import { CoreMessage, streamText, tool } from 'ai';
 import 'dotenv/config';
 import * as readline from 'node:readline/promises';
+import { z } from 'zod';
 
 const terminal = readline.createInterface({
   input: process.stdin,
@@ -12,25 +13,35 @@ const messages: CoreMessage[] = [];
 
 async function main() {
   while (true) {
-    const userInput = await terminal.question('You: ');
-
-    messages.push({ role: 'user', content: userInput });
+    messages.push({ role: 'user', content: await terminal.question('You: ') });
 
     const result = await streamText({
       model: google('gemini-1.5-pro-latest'),
-      system: `You are a helpful, respectful and honest assistant.`,
+      tools: {
+        weather: tool({
+          description: 'Get the weather in a location',
+          parameters: z.object({
+            location: z
+              .string()
+              .describe('The location to get the weather for'),
+          }),
+          execute: async ({ location }) => ({
+            location,
+            temperature: 72 + Math.floor(Math.random() * 21) - 10,
+          }),
+        }),
+      },
+      maxSteps: 5,
       messages,
     });
 
-    let fullResponse = '';
     process.stdout.write('\nAssistant: ');
     for await (const delta of result.textStream) {
-      fullResponse += delta;
       process.stdout.write(delta);
     }
     process.stdout.write('\n\n');
 
-    messages.push({ role: 'assistant', content: fullResponse });
+    messages.push(...(await result.response).messages);
   }
 }
 

packages/google/src/convert-to-google-generative-ai-messages.test.ts

@@ -83,3 +83,41 @@ describe('user messages', () => {
     });
   });
 });
+
+describe('tool messages', () => {
+  it('should convert tool result messages to function responses', async () => {
+    const result = convertToGoogleGenerativeAIMessages([
+      {
+        role: 'tool',
+        content: [
+          {
+            type: 'tool-result',
+            toolName: 'testFunction',
+            toolCallId: 'testCallId',
+            result: { someData: 'test result' },
+          },
+        ],
+      },
+    ]);
+
+    expect(result).toEqual({
+      systemInstruction: undefined,
+      contents: [
+        {
+          role: 'user',
+          parts: [
+            {
+              functionResponse: {
+                name: 'testFunction',
+                response: {
+                  name: 'testFunction',
+                  content: { someData: 'test result' },
+                },
+              },
+            },
+          ],
+        },
+      ],
+    });
+  });
+});

packages/google/src/convert-to-google-generative-ai-messages.ts

@@ -133,7 +133,10 @@ export function convertToGoogleGenerativeAIMessages(
           parts: content.map(part => ({
             functionResponse: {
               name: part.toolName,
-              response: part.result,
+              response: {
+                name: part.toolName,
+                content: part.result,
+              },
             },
           })),
         });