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');