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',
|