fix: resolve typescript issue

openai · Jun 4, 2024 · 1129707 · 1129707
1 parent dba4ffb
commit 1129707
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/lib/AbstractChatCompletionRunner.ts b/src/lib/AbstractChatCompletionRunner.ts
@@ -232,7 +232,12 @@ export abstract class AbstractChatCompletionRunner<
     while (i-- > 0) {
       const message = this.messages[i];
       if (isAssistantMessage(message)) {
-        return { ...message, content: message.content ?? null };
+        const { function_call, ...rest } = message;
+        const ret: ChatCompletionMessage = { ...rest, content: message.content ?? null };
+        if (function_call) {
+          ret.function_call = function_call;
+        }
+        return ret;
       }
     }
     throw new OpenAIError('stream ended without producing a ChatCompletionMessage with role=assistant');