Spaces:
Paused
Paused
Commit
·
9594a7a
1
Parent(s):
e26e708
Update types/openai.ts
Browse files- types/openai.ts +8 -1
types/openai.ts
CHANGED
@@ -8,6 +8,7 @@ export interface OpenAIModel {
|
|
8 |
}
|
9 |
|
10 |
export enum OpenAIModelID {
|
|
|
11 |
GPT_3_5 = 'gpt-3.5-turbo',
|
12 |
GPT_3_5_AZ = 'gpt-35-turbo',
|
13 |
GPT_4 = 'gpt-4',
|
@@ -15,9 +16,15 @@ export enum OpenAIModelID {
|
|
15 |
}
|
16 |
|
17 |
// in case the `DEFAULT_MODEL` environment variable is not set or set to an unsupported model
|
18 |
-
export const fallbackModelID = OpenAIModelID.
|
19 |
|
20 |
export const OpenAIModels: Record<OpenAIModelID, OpenAIModel> = {
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
[OpenAIModelID.GPT_3_5]: {
|
22 |
id: OpenAIModelID.GPT_3_5,
|
23 |
name: 'GPT-3.5',
|
|
|
8 |
}
|
9 |
|
10 |
export enum OpenAIModelID {
|
11 |
+
Open_LLaMA = 'open-llama-7b',
|
12 |
GPT_3_5 = 'gpt-3.5-turbo',
|
13 |
GPT_3_5_AZ = 'gpt-35-turbo',
|
14 |
GPT_4 = 'gpt-4',
|
|
|
16 |
}
|
17 |
|
18 |
// in case the `DEFAULT_MODEL` environment variable is not set or set to an unsupported model
|
19 |
+
export const fallbackModelID = OpenAIModelID.Open_LLaMA;
|
20 |
|
21 |
export const OpenAIModels: Record<OpenAIModelID, OpenAIModel> = {
|
22 |
+
[OpenAIModelID.Open_LLaMA]: {
|
23 |
+
id: OpenAIModelID.Open_LLaMA,
|
24 |
+
name: 'open-llama-7b',
|
25 |
+
maxLength: 12000,
|
26 |
+
tokenLimit: 2048,
|
27 |
+
},
|
28 |
[OpenAIModelID.GPT_3_5]: {
|
29 |
id: OpenAIModelID.GPT_3_5,
|
30 |
name: 'GPT-3.5',
|