upstream/mi-gpt
1
0
Fork 0
mirror of https://github.com/idootop/mi-gpt.git synced 2024-09-20 14:56:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 支持 Microsoft Azure OpenAI， fix #7

Browse Source
This commit is contained in:
WJG 2024-05-26 18:09:38 +08:00
parent 8025ee4fe9
commit 3bd37b2a9d
No known key found for this signature in database
GPG Key ID: 258474EF8590014A
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/services/openai.ts
View File

@ -1,4 +1,4 @@
import OpenAI from "openai"; import OpenAI, { AzureOpenAI } from "openai";
import { import {
ChatCompletionMessageParam, ChatCompletionMessageParam,
ChatCompletionTool, ChatCompletionTool,
@ -28,10 +28,9 @@ class OpenAIClient {
private _client?: OpenAI; private _client?: OpenAI;
private _init() { private _init() {
if (!this._client) { if (!this._client) {
this._client = new OpenAI({ this._client = kEnvs.AZURE_OPENAI_API_KEY
httpAgent: kProxyAgent, ? new AzureOpenAI({ httpAgent: kProxyAgent })
apiKey: kEnvs.OPENAI_API_KEY!, : new OpenAI({ httpAgent: kProxyAgent });
});
} }
} }

1
src/utils/env.ts
View File

@ -4,4 +4,5 @@ export const kEnvs: Partial<{
MI_DID: string; MI_DID: string;
OPENAI_MODEL: string; OPENAI_MODEL: string;
OPENAI_API_KEY: string; OPENAI_API_KEY: string;
AZURE_OPENAI_API_KEY: string;
}> = process.env as any; }> = process.env as any;