Added GPT-3.5-turbo-16k model

This commit is contained in:
Heiko Joerg Schick
2023-08-12 22:13:34 +02:00
parent fa3f6e93bb
commit 10733fab7d
+7
View File
@@ -10,6 +10,7 @@ export interface OpenAIModel {
export enum OpenAIModelID { export enum OpenAIModelID {
GPT_3_5 = 'gpt-3.5-turbo', GPT_3_5 = 'gpt-3.5-turbo',
GPT_3_5_AZ = 'gpt-35-turbo', GPT_3_5_AZ = 'gpt-35-turbo',
GPT_3_5_16K = 'gpt-3.5-turbo-16k'
GPT_4 = 'gpt-4', GPT_4 = 'gpt-4',
GPT_4_32K = 'gpt-4-32k', GPT_4_32K = 'gpt-4-32k',
} }
@@ -30,6 +31,12 @@ export const OpenAIModels: Record<OpenAIModelID, OpenAIModel> = {
maxLength: 12000, maxLength: 12000,
tokenLimit: 4000, tokenLimit: 4000,
}, },
[OpenAIModelID.GPT_3_5_16K]: {
id: OpenAIModelID.GPT_3_5_16K,
name: 'GPT-3.5-16K',
maxLength: 96000,
tokenLimit: 32000,
},
[OpenAIModelID.GPT_4]: { [OpenAIModelID.GPT_4]: {
id: OpenAIModelID.GPT_4, id: OpenAIModelID.GPT_4,
name: 'GPT-4', name: 'GPT-4',