Add support for Azure OpenAI (#495)
This commit is contained in:
@@ -1,3 +1,5 @@
|
||||
import { OPENAI_API_TYPE } from '../utils/app/const';
|
||||
|
||||
export interface OpenAIModel {
|
||||
id: string;
|
||||
name: string;
|
||||
@@ -7,7 +9,9 @@ export interface OpenAIModel {
|
||||
|
||||
export enum OpenAIModelID {
|
||||
GPT_3_5 = 'gpt-3.5-turbo',
|
||||
GPT_3_5_AZ = 'gpt-35-turbo',
|
||||
GPT_4 = 'gpt-4',
|
||||
GPT_4_32K = 'gpt-4-32k',
|
||||
}
|
||||
|
||||
// in case the `DEFAULT_MODEL` environment variable is not set or set to an unsupported model
|
||||
@@ -20,10 +24,22 @@ export const OpenAIModels: Record<OpenAIModelID, OpenAIModel> = {
|
||||
maxLength: 12000,
|
||||
tokenLimit: 4000,
|
||||
},
|
||||
[OpenAIModelID.GPT_3_5_AZ]: {
|
||||
id: OpenAIModelID.GPT_3_5_AZ,
|
||||
name: 'GPT-3.5',
|
||||
maxLength: 12000,
|
||||
tokenLimit: 4000,
|
||||
},
|
||||
[OpenAIModelID.GPT_4]: {
|
||||
id: OpenAIModelID.GPT_4,
|
||||
name: 'GPT-4',
|
||||
maxLength: 24000,
|
||||
tokenLimit: 8000,
|
||||
},
|
||||
[OpenAIModelID.GPT_4_32K]: {
|
||||
id: OpenAIModelID.GPT_4_32K,
|
||||
name: 'GPT-4-32K',
|
||||
maxLength: 96000,
|
||||
tokenLimit: 32000,
|
||||
},
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user