From 2fbe0bff0d9c5035ccf9fcc41e9a8e60c8339c95 Mon Sep 17 00:00:00 2001
From: penurin <71948-penurin@users.noreply.gitgud.io>
Date: Fri, 27 Dec 2024 17:17:39 +0000
Subject: [PATCH] Check for o1 instead of o1-preview for the o1 family

---
 src/shared/models.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/shared/models.ts b/src/shared/models.ts
index 2a013d6..80baf1c 100644
--- a/src/shared/models.ts
+++ b/src/shared/models.ts
@@ -154,7 +154,7 @@ export const OPENAI_MODEL_FAMILY_MAP: { [regex: string]: OpenAIModelFamily } = {
   "^text-embedding-ada-002$": "turbo",
   "^dall-e-\\d{1}$": "dall-e",
   "^o1-mini(-\\d{4}-\\d{2}-\\d{2})?$": "o1-mini",
-  "^o1(-preview)?(-\\d{4}-\\d{2}-\\d{2})?$": "o1",
+  "^o1(-\\d{4}-\\d{2}-\\d{2})?$": "o1",
 };
 
 export function getOpenAIModelFamily(