Azure OpenAI API Fix (ztjhz#297)

* Azure OpenAI API Fix * improve code * improve code --------- Co-authored-by: Jing Hua <[email protected]>
AOMJ2PMP · May 17, 2023 · 52fa56c · 52fa56c
1 parent 82f8761
commit 52fa56c
Showing 1 changed file with 36 additions and 2 deletions.
diff --git a/src/api/api.ts b/src/api/api.ts
@@ -14,7 +14,24 @@ export const getChatCompletion = async (
  ...customHeaders,
  };
  if (apiKey) headers.Authorization = `Bearer ${apiKey}`;
- if (isAzureEndpoint(endpoint) && apiKey) headers['api-key'] = apiKey;
+
+ 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 apiVersion = '2023-03-15-preview';
+
+ const path = `openai/deployments/${model}/chat/completions?api-version=${apiVersion}`;
+
+ if (!endpoint.endsWith(path)) {
+ if (!endpoint.endsWith('/')) {
+ endpoint += '/';
+ }
+ endpoint += path;
+ }
+ }
 
  const response = await fetch(endpoint, {
  method: 'POST',
@@ -43,7 +60,24 @@ export const getChatCompletionStream = async (
  ...customHeaders,
  };
  if (apiKey) headers.Authorization = `Bearer ${apiKey}`;
- if (isAzureEndpoint(endpoint) && apiKey) headers['api-key'] = apiKey;
+
+ 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 apiVersion = '2023-03-15-preview';
+
+ const path = `openai/deployments/${model}/chat/completions?api-version=${apiVersion}`;
+
+ if (!endpoint.endsWith(path)) {
+ if (!endpoint.endsWith('/')) {
+ endpoint += '/';
+ }
+ endpoint += path;
+ }
+ }
 
  const response = await fetch(endpoint, {
  method: 'POST',