From 38bc63f0ac633535436ee3e4aa59e8da3f0fcafb Mon Sep 17 00:00:00 2001
From: Jing Hua <tohjinghua123@gmail.com>
Date: Thu, 6 Apr 2023 02:48:12 +0800
Subject: [PATCH] feat: Azure OpenAI endpoint support

fixes #208
---
 src/api/api.ts   | 11 +++++++++--
 src/utils/api.ts |  3 +++
 2 files changed, 12 insertions(+), 2 deletions(-)
 create mode 100644 src/utils/api.ts

diff --git a/src/api/api.ts b/src/api/api.ts
index 5948796..153de0f 100644
--- a/src/api/api.ts
+++ b/src/api/api.ts
@@ -1,16 +1,20 @@
 import { ShareGPTSubmitBodyInterface } from '@type/api';
 import { ConfigInterface, MessageInterface } from '@type/chat';
+import { isAzureEndpoint } from '@utils/api';
 
 export const getChatCompletion = async (
   endpoint: string,
   messages: MessageInterface[],
   config: ConfigInterface,
-  apiKey?: string
+  apiKey?: string,
+  customHeaders?: Record<string, string>
 ) => {
   const headers: HeadersInit = {
     'Content-Type': 'application/json',
+    ...customHeaders,
   };
   if (apiKey) headers.Authorization = `Bearer ${apiKey}`;
+  if (isAzureEndpoint(endpoint) && apiKey) headers['api-key'] = apiKey;
 
   const response = await fetch(endpoint, {
     method: 'POST',
@@ -31,12 +35,15 @@ export const getChatCompletionStream = async (
   endpoint: string,
   messages: MessageInterface[],
   config: ConfigInterface,
-  apiKey?: string
+  apiKey?: string,
+  customHeaders?: Record<string, string>
 ) => {
   const headers: HeadersInit = {
     'Content-Type': 'application/json',
+    ...customHeaders,
   };
   if (apiKey) headers.Authorization = `Bearer ${apiKey}`;
+  if (isAzureEndpoint(endpoint) && apiKey) headers['api-key'] = apiKey;
 
   const response = await fetch(endpoint, {
     method: 'POST',
diff --git a/src/utils/api.ts b/src/utils/api.ts
new file mode 100644
index 0000000..d66c22a
--- /dev/null
+++ b/src/utils/api.ts
@@ -0,0 +1,3 @@
+export const isAzureEndpoint = (endpoint: string) => {
+  return endpoint.includes('openai.azure.com');
+};