Bug:fix issue with vercel sdk model noth validating schema properly

Oneirocom · Jun 11, 2024 · 4ef9ca6 · 4ef9ca6
1 parent 79f6ee2
commit 4ef9ca6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/packages/server/vercel/core/src/lib/magick-openai/src/openai-chat-language-model.ts b/packages/server/vercel/core/src/lib/magick-openai/src/openai-chat-language-model.ts
@@ -376,7 +376,7 @@ const openAIChatResponseSchema = z.object({
     z.object({
       message: z.object({
         role: z.literal('assistant'),
-        content: z.string().nullable().optional(),
+        content: z.string().nullable().nullish(),
         tool_calls: z
           .array(
             z.object({
@@ -388,7 +388,7 @@ const openAIChatResponseSchema = z.object({
               }),
             })
           )
-          .optional(),
+          .nullish(),
       }),
       index: z.number(),
       logprobs: z
@@ -409,7 +409,7 @@ const openAIChatResponseSchema = z.object({
             .nullable(),
         })
         .nullable()
-        .optional(),
+        .nullish(),
       finish_reason: z.string().optional().nullable(),
     })
   ),
@@ -425,7 +425,7 @@ const openaiChatChunkSchema = z.object({
   choices: z.array(
     z.object({
       delta: z.object({
-        role: z.enum(['assistant']).optional(),
+        role: z.enum(['assistant']).nullish(),
         content: z.string().nullable().optional(),
         tool_calls: z
           .array(
@@ -439,7 +439,7 @@ const openaiChatChunkSchema = z.object({
               }),
             })
           )
-          .optional(),
+          .nullish(),
       }),
       logprobs: z
         .object({
@@ -456,7 +456,7 @@ const openaiChatChunkSchema = z.object({
             })
           ),
         })
-        .optional(),
+        .nullish(),
       finish_reason: z.string().nullable().optional(),
       index: z.number(),
     })

diff --git a/packages/server/vercel/core/src/lib/magick-openai/src/openai-completion-language-model.ts b/packages/server/vercel/core/src/lib/magick-openai/src/openai-completion-language-model.ts
@@ -290,7 +290,7 @@ const openAICompletionResponseSchema = z.object({
           top_logprobs: z.array(z.record(z.string(), z.number())).nullable(),
         })
         .nullable()
-        .optional(),
+        .nullish(),
     })
   ),
   usage: z.object({
@@ -317,7 +317,7 @@ const openaiCompletionChunkSchema = z.object({
           top_logprobs: z.array(z.record(z.string(), z.number())).nullable(),
         })
         .nullable()
-        .optional(),
+        .nullish(),
     })
   ),
   usage: z