feat: 支持azure openai (#81)

Co-authored-by: qiaocc <[email protected]>
xcatliu · Mar 28, 2024 · 319aa13 · 319aa13
1 parent bd1432e
commit 319aa13
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/app/api/chat/route.ts b/app/api/chat/route.ts
@@ -28,11 +28,24 @@ export async function POST(req: NextRequest) {
 
   const apiKey = getApiKey(cookies().get('apiKey')?.value);
 
-  const fetchResult = await fetch(`https://${env.CHATGPT_NEXT_API_HOST}/v1/chat/completions`, {
+  let apiURL = '';
+  switch (env.CHATGPT_NEXT_API_PROVIDER) {
+    case 'openai':
+      apiURL = `https://${env.CHATGPT_NEXT_API_HOST}/v1/chat/completions`;
+      break;
+    case 'azure':
+      apiURL = `${env.CHATGPT_NEXT_API_AzureAPIURL}/${env.CHATGPT_NEXT_API_AzureAPIURLPath}`;
+      break;
+    default:
+      apiURL = 'unknown';
+  }
+
+  const fetchResult = await fetch(apiURL, {
     method: HttpMethod.POST,
     headers: {
       ...HttpHeaderJson,
       Authorization: `Bearer ${apiKey}`,
+      'api-key': apiKey ?? '',
     },
     // 直接透传，组装逻辑完全由前端实现
     body: req.body,

diff --git a/app/utils/env.ts b/app/utils/env.ts
@@ -9,4 +9,11 @@ export const env = {
 
   /** 配置 API 请求的 host（包含端口） */
   CHATGPT_NEXT_API_HOST: process.env.CHATGPT_NEXT_API_HOST ?? 'api.openai.com',
+
+  /** API提供商: openai/azure */
+  CHATGPT_NEXT_API_PROVIDER: process.env.CHATGPT_NEXT_API_PROVIDER ?? 'openai',
+  /** azure api url */
+  CHATGPT_NEXT_API_AzureAPIURL: process.env.CHATGPT_NEXT_API_AzureAPIURL,
+  /** azure api url path */
+  CHATGPT_NEXT_API_AzureAPIURLPath: process.env.CHATGPT_NEXT_API_AzureAPIURLPath,
 };