slight refactor of handling the final token counts chunk

openai · Sep 13, 2024 · 61e9377 · 61e9377
1 parent 51f42e3
commit 61e9377
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/tool-calls-stream.ts b/examples/tool-calls-stream.ts
@@ -185,10 +185,12 @@ function messageReducer(previous: ChatCompletionMessage, item: ChatCompletionChu
     return acc;
   };
 
-  if (item.choices.length == 0) {
+  const choice = item.choices[0];
+  if (!choice) {
+    // chunk contains information about usage and token counts
     return previous;
   }
-  return reduce(previous, item.choices[0]!.delta) as ChatCompletionMessage;
+  return reduce(previous, choice.delta) as ChatCompletionMessage;
 }
 
 function lineRewriter() {