fix(azure-openai): streaming

pezzolabs · Oct 22, 2023 · 4963d8f · 4963d8f
1 parent b75bd2d
commit 4963d8f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/unillm-node/package.json b/packages/unillm-node/package.json
@@ -1,6 +1,6 @@
 {
   "name": "unillm",
-  "version": "0.1.0-alpha.2",
+  "version": "0.1.0",
   "main": "./index.ts",
   "type": "commonjs",
   "license": "MIT",

diff --git a/packages/unillm-node/providers/azure-openai.ts b/packages/unillm-node/providers/azure-openai.ts
@@ -92,10 +92,12 @@ export class AzureOpenAIProvider extends BaseProvider<Providers.AzureOpenAI> {
   }
 
   async createChatCompletionStreaming(
-    model: ModelTypes[Providers.AzureOpenAI],
+    _model: ModelTypes[Providers.AzureOpenAI],
     params: UnifiedCreateChatCompletionParamsStreaming,
   ): Promise<UnifiedCreateChatCompletionStreamResult> {
+    const [, model] = _model.split("/");
     const { baseParams } = this.processUnifiedParamsToAzureOpenAIFormat(params);
+
     const originalStreamResponse = this.client.listChatCompletions(
       model,
       params.messages,