OpenAI: fix accessing choices when using completion

vercel · Aug 18, 2023 · 7b389a7 · 7b389a7
1 parent 4da195b
commit 7b389a7
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/.changeset/red-toes-prove.md b/.changeset/red-toes-prove.md
@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+fix: improve safety for type check in openai-stream
diff --git a/packages/core/streams/openai-stream.ts b/packages/core/streams/openai-stream.ts
@@ -322,11 +322,21 @@ type OpenAIStreamReturnTypes = ExtractType<AsyncIterableOpenAIStreamReturnTypes>
 function isChatCompletionChunk(
   data: OpenAIStreamReturnTypes
 ): data is ChatCompletionChunk {
-  return 'choices' in data && 'delta' in data.choices[0]
+  return (
+    'choices' in data &&
+    data.choices &&
+    data.choices[0] &&
+    'delta' in data.choices[0]
+  )
 }
 
 function isCompletion(data: OpenAIStreamReturnTypes): data is Completion {
-  return 'choices' in data && 'text' in data.choices[0]
+  return (
+    'choices' in data &&
+    data.choices &&
+    data.choices[0] &&
+    'text' in data.choices[0]
+  )
 }
 
 export function OpenAIStream(