Parse tool call arguments for onToolCall callback. (#1113)

vercel · Mar 6, 2024 · f6ae717 · f6ae717
1 parent bb94b2b
commit f6ae717
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/packages/core/streams/openai-stream.test.ts b/packages/core/streams/openai-stream.test.ts
@@ -352,22 +352,24 @@ describe('OpenAIStream', () => {
       });
 
       const response = new StreamingTextResponse(stream);
-      const client = createClient(response);
-      const chunks = await client.readAll();
+
+      await createClient(response).readAll(); // consume stream
 
       expect(toolCalls).toEqual({
         tools: [
           {
             func: {
-              arguments: '{}',
+              arguments: {},
               name: 'get_date_time',
             },
             id: 'call_NPkY32jNUOb3Kkm7v9cOgmVg',
             type: 'function',
           },
           {
             func: {
-              arguments: '{"url": "https://www.linkedin.com/in/jessepascoe"}',
+              arguments: {
+                url: 'https://www.linkedin.com/in/jessepascoe',
+              },
               name: 'open_webpage',
             },
             id: 'call_pOyOtXFQltSjUGsF7gnLAEcD',

diff --git a/packages/core/streams/openai-stream.ts b/packages/core/streams/openai-stream.ts
@@ -605,7 +605,7 @@ function createFunctionCallTransformer(
                 type: 'function',
                 func: {
                   name: tool.function.name,
-                  arguments: tool.function.arguments,
+                  arguments: JSON.parse(tool.function.arguments),
                 },
               });
             }