streaming type support

dexaai · Sep 7, 2024 · 0c40688 · 0c40688
1 parent f82f647
commit 0c40688
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/openai-client.ts b/src/openai-client.ts
@@ -207,9 +207,9 @@ export class OpenAIClient<T extends ConfigOpts = ConfigOpts> {
   }
 
   /** Create a chat completion and stream back partial progress. */
-  async streamChatCompletion(
-    params: ChatStreamParams<InferProvider<T>>,
-    opts?: RequestOpts
+  async streamChatCompletion<R extends RequestOpts>(
+    params: ChatStreamParams<InferProvider<R['headers'] & T>>,
+    opts?: R
   ): Promise<ChatStreamResponse> {
     if (this.getProvider(opts?.headers?.baseUrl) === 'anthropic') {
       const anthropicParams = this.convertChatParamsToAnthropicFormat(params as ChatParams<'anthropic'>);