matthoffner commited on
Commit
9594a7a
·
1 Parent(s): e26e708

Update types/openai.ts

Browse files
Files changed (1) hide show
  1. 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.GPT_3_5;
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',