Azure OpenAI API fix for gpt-3.5-turbo-16k (#383)

Signed-off-by: Xtrah <12988459+Xtrah@users.noreply.github.com>
ztjhz · Aug 5, 2023 · 3477321 · 3477321
1 parent d0ddcfe
commit 3477321
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/api/api.ts b/src/api/api.ts
@@ -18,9 +18,8 @@ export const getChatCompletion = async (
   if (isAzureEndpoint(endpoint) && apiKey) {
     headers['api-key'] = apiKey;
 
-    const gpt3forAzure = 'gpt-35-turbo';
-    const model =
-      config.model === 'gpt-3.5-turbo' ? gpt3forAzure : config.model;
+    const model = config.model === 'gpt-3.5-turbo' ? 'gpt-35-turbo' : config.model === 'gpt-3.5-turbo-16k' ? 'gpt-35-turbo-16k' : config.model;
+
     const apiVersion = '2023-03-15-preview';
 
     const path = `openai/deployments/${model}/chat/completions?api-version=${apiVersion}`;
@@ -64,9 +63,8 @@ export const getChatCompletionStream = async (
   if (isAzureEndpoint(endpoint) && apiKey) {
     headers['api-key'] = apiKey;
 
-    const gpt3forAzure = 'gpt-35-turbo';
-    const model =
-      config.model === 'gpt-3.5-turbo' ? gpt3forAzure : config.model;
+    const model = config.model === 'gpt-3.5-turbo' ? 'gpt-35-turbo' : config.model === 'gpt-3.5-turbo-16k' ? 'gpt-35-turbo-16k' : config.model;
+
     const apiVersion = '2023-03-15-preview';
 
     const path = `openai/deployments/${model}/chat/completions?api-version=${apiVersion}`;