Fix #34 ai-sdk integration fails to work with other platforms

Author: seratch

examples/ai-sdk/ai-sdk-model.ts

@@ -10,35 +10,34 @@ const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
 const getWeatherTool = tool({
   name: 'get_weather',
   description: 'Get the weather for a given city',
-  parameters: z.object({
-    demo: z.string(),
-  }),
+  parameters: z.object({ city: z.string() }),
   execute: async (input) => {
     await sleep(300);
-    return `The weather in ${input.demo} is sunny`;
+    return `The weather in ${input.city} is sunny`;
   },
 });
 
-const dataAgentTwo = new Agent({
-  name: 'Data agent',
-  instructions: 'You are a data agent',
-  handoffDescription: 'You know everything about the weather',
+const dataAgent = new Agent({
+  name: 'Weather Data Agent',
+  instructions: 'You are a weather data agent.',
+  handoffDescription:
+    'When you are asked about the weather, you will use tools to get the weather.',
   tools: [getWeatherTool],
   model, // Using the AI SDK model for this agent
 });
 
 const agent = new Agent({
-  name: 'Basic test agent',
-  instructions: 'You are a basic agent',
-  handoffs: [dataAgentTwo],
+  name: 'Helpful Assistant',
+  instructions:
+    'You are a helpful assistant. When you need to get the weather, you can hand off the task to the Weather Data Agent.',
+  handoffs: [dataAgent],
 });
 
 async function main() {
   const result = await run(
     agent,
     'Hello what is the weather in San Francisco and oakland?',
   );
-
   console.log(result.finalOutput);
 }
 

packages/agents-extensions/src/aiSdk.ts

@@ -5,7 +5,6 @@ import type {
   LanguageModelV1FunctionTool,
   LanguageModelV1Message,
   LanguageModelV1Prompt,
-  LanguageModelV1TextPart,
   LanguageModelV1ProviderDefinedTool,
   LanguageModelV1ToolCallPart,
   LanguageModelV1ToolResultPart,
@@ -147,7 +146,7 @@ export function itemsToLanguageV1Messages(
           type: 'tool-call',
           toolCallId: item.callId,
           toolName: item.name,
-          args: item.arguments,
+          args: JSON.parse(item.arguments),
         };
         currentAssistantMessage.content.push(content);
       }
@@ -365,13 +364,8 @@ export class AiSdkModel implements Model {
           input = [
             {
               role: 'system',
-              content: [
-                {
-                  type: 'text',
-                  text: request.systemInstructions,
-                } as LanguageModelV1TextPart,
-              ],
-            } as unknown as LanguageModelV1Message,
+              content: request.systemInstructions,
+            },
             ...input,
           ];
         }
@@ -428,7 +422,11 @@ export class AiSdkModel implements Model {
           });
         });
 
-        if (result.text) {
+        // Some of other platforms may return both tool calls and text.
+        // Putting a text message here will let the agent loop to complete,
+        // so adding this item only when the tool calls are empty.
+        // Note that the same support is not available for streaming mode.
+        if (!result.toolCalls && result.text) {
           output.push({
             type: 'message',
             content: [{ type: 'output_text', text: result.text }],
@@ -515,13 +513,8 @@ export class AiSdkModel implements Model {
         input = [
           {
             role: 'system',
-            content: [
-              {
-                type: 'text',
-                text: request.systemInstructions,
-              } as LanguageModelV1TextPart,
-            ],
-          } as unknown as LanguageModelV1Message,
+            content: request.systemInstructions,
+          },
           ...input,
         ];
       }