From 5e646b1c86ca58703b2369161bfa10aae068da0a Mon Sep 17 00:00:00 2001
From: nai-degen <khoners@protonmail.com>
Date: Thu, 25 Jan 2024 13:27:03 -0600
Subject: [PATCH] adds gpt-4-0125-preview and gpt-4-turbo-preview alias

---
 src/proxy/openai.ts  | 2 ++
 src/shared/models.ts | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/proxy/openai.ts b/src/proxy/openai.ts
index a20c6d3..0b42a54 100644
--- a/src/proxy/openai.ts
+++ b/src/proxy/openai.ts
@@ -28,6 +28,8 @@ import {
 
 // https://platform.openai.com/docs/models/overview
 export const KNOWN_OPENAI_MODELS = [
+  "gpt-4-turbo-preview",
+  "gpt-4-0125-preview",
   "gpt-4-1106-preview",
   "gpt-4-vision-preview",
   "gpt-4",
diff --git a/src/shared/models.ts b/src/shared/models.ts
index 3121f2e..96ea377 100644
--- a/src/shared/models.ts
+++ b/src/shared/models.ts
@@ -73,7 +73,8 @@ export const LLM_SERVICES = (<A extends readonly LLMService[]>(
 ] as const);
 
 export const OPENAI_MODEL_FAMILY_MAP: { [regex: string]: OpenAIModelFamily } = {
-  "^gpt-4-1106(-preview)?$": "gpt4-turbo",
+  "^gpt-4-turbo(-preview)?$": "gpt4-turbo",
+  "^gpt-4-(0125|1106)(-preview)?$": "gpt4-turbo",
   "^gpt-4(-\\d{4})?-vision(-preview)?$": "gpt4-turbo",
   "^gpt-4-32k-\\d{4}$": "gpt4-32k",
   "^gpt-4-32k$": "gpt4-32k",