fix (@ai-sdk/azure): allow for nullish delta (#1915)

vercel · Jun 11, 2024 · fc9552e · fc9552e
1 parent 2b6cc6d
commit fc9552e
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 17 deletions.
diff --git a/.changeset/stale-jokes-try.md b/.changeset/stale-jokes-try.md
@@ -0,0 +1,5 @@
+---
+'@ai-sdk/openai': patch
+---
+
+fix (@ai-sdk/azure): allow for nullish delta
diff --git a/examples/ai-core/src/stream-text/azure.ts b/examples/ai-core/src/stream-text/azure.ts
@@ -6,7 +6,7 @@ dotenv.config();
 
 async function main() {
   const result = await streamText({
-    model: azure('v0-gpt-35-turbo'), // use your own deployment
+    model: azure('gpt-4o'), // use your own deployment
     prompt: 'Invent a new holiday and describe its traditions.',
   });
 

diff --git a/packages/openai/src/openai-chat-language-model.ts b/packages/openai/src/openai-chat-language-model.ts
@@ -458,23 +458,25 @@ const openaiChatChunkSchema = z.union([
   z.object({
     choices: z.array(
       z.object({
-        delta: z.object({
-          role: z.enum(['assistant']).optional(),
-          content: z.string().nullish(),
-          tool_calls: z
-            .array(
-              z.object({
-                index: z.number(),
-                id: z.string().nullish(),
-                type: z.literal('function').optional(),
-                function: z.object({
-                  name: z.string().nullish(),
-                  arguments: z.string().nullish(),
+        delta: z
+          .object({
+            role: z.enum(['assistant']).optional(),
+            content: z.string().nullish(),
+            tool_calls: z
+              .array(
+                z.object({
+                  index: z.number(),
+                  id: z.string().nullish(),
+                  type: z.literal('function').optional(),
+                  function: z.object({
+                    name: z.string().nullish(),
+                    arguments: z.string().nullish(),
+                  }),
                 }),
-              }),
-            )
-            .nullish(),
-        }),
+              )
+              .nullish(),
+          })
+          .nullish(),
         logprobs: z
           .object({
             content: z