diff --git a/src/proxy.ts b/src/proxy.ts index 1ec8f86..9370778 100644 --- a/src/proxy.ts +++ b/src/proxy.ts @@ -28,13 +28,17 @@ const ORAMA_EMBEDDINGS_MODEL_GTE_SMALL = 'gte-small' const ORAMA_EMBEDDINGS_MODEL_GTE_MEDIUM = 'gte-medium' const ORAMA_EMBEDDINGS_MODEL_GTE_LARGE = 'gte-large' + +const OPENAI_CHAT_MODE_GPT_4O = 'gpt-4o' +const OPENAI_CHAT_MODE_GPT_4O_MINI = 'gpt-4o-mini' +const OPENAI_CHAT_MODE_GPT_4_TURBO = 'gpt-4-turbo' +const OPENAI_CHAT_MODEL_GPT_4_TURBO_2024_04_09 = 'gpt-4-turbo-2024-04-09' const OPENAI_CHAT_MODE_GPT_4_0125_PREVIEW = 'gpt-4-0125-preview' const OPENAI_CHAT_MODEL_GPT4_1106_PREVIEW = 'gpt-4-1106-preview' -const OPENAI_CHAT_MODEL_GPT4_TURBO_PREVIEW = 'gpt-4-turbo-preview ' +const OPENAI_CHAT_MODEL_GPT4_TURBO_PREVIEW = 'gpt-4-turbo-preview' const OPENAI_CHAT_MODEL_GPT4 = 'gpt-4' const OPENAI_CHAT_MODEL_GPT3_5_TURBO = 'gpt-3.5-turbo' -const OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_16K = 'gpt-3.5-turbo-16k' -const OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_0125 = 'gpt-3.5-turbo-0125 ' +const OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_0125 = 'gpt-3.5-turbo-0125' const embeddingsModels = { [`openai/${OPENAI_EMBEDDINGS_MODEL_ADA}`]: `openai/${OPENAI_EMBEDDINGS_MODEL_ADA}`, @@ -48,11 +52,14 @@ const embeddingsModels = { const chatModels = { [`openai/${OPENAI_CHAT_MODEL_GPT3_5_TURBO}`]: OPENAI_CHAT_MODEL_GPT3_5_TURBO, [`openai/${OPENAI_CHAT_MODEL_GPT4}`]: OPENAI_CHAT_MODEL_GPT4, - [`openai/${OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_16K}`]: OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_16K, [`openai/${OPENAI_CHAT_MODEL_GPT4_1106_PREVIEW}`]: OPENAI_CHAT_MODEL_GPT4_1106_PREVIEW, [`openai/${OPENAI_CHAT_MODE_GPT_4_0125_PREVIEW}`]: OPENAI_CHAT_MODE_GPT_4_0125_PREVIEW, [`openai/${OPENAI_CHAT_MODEL_GPT4_TURBO_PREVIEW}`]: OPENAI_CHAT_MODEL_GPT4_TURBO_PREVIEW, - [`openai/${OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_0125}`]: OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_0125 + [`openai/${OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_0125}`]: OPENAI_CHAT_MODEL_GPT3_3_5_TURBO_0125, + [`openai/${OPENAI_CHAT_MODEL_GPT_4_TURBO_2024_04_09}`]: OPENAI_CHAT_MODEL_GPT_4_TURBO_2024_04_09, + [`openai/${OPENAI_CHAT_MODE_GPT_4O}`]: OPENAI_CHAT_MODE_GPT_4O, + [`openai/${OPENAI_CHAT_MODE_GPT_4O_MINI}`]: OPENAI_CHAT_MODE_GPT_4O_MINI, + [`openai/${OPENAI_CHAT_MODE_GPT_4_TURBO}`]: OPENAI_CHAT_MODE_GPT_4_TURBO } export class OramaProxy {