Merge pull request #32 from askorama/feat/updates-models

feat: updates openai models
oramasearch · Aug 20, 2024 · f69158c · f69158c
2 parents 864be9a + 5833a26
commit f69158c
Showing 1 changed file with 12 additions and 5 deletions.
diff --git 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 {