From c96c12b6e4410da5834fa9188e85de6b0a107719 Mon Sep 17 00:00:00 2001
From: kwaroran <amasio1234@proton.me>
Date: Mon, 27 May 2024 13:25:26 +0900
Subject: [PATCH] refactor: Update getModelShortName function to handle GPT-4o
 model

---
 src/ts/model/names.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/ts/model/names.ts b/src/ts/model/names.ts
index b7504e8b..c5c9bbbb 100644
--- a/src/ts/model/names.ts
+++ b/src/ts/model/names.ts
@@ -131,6 +131,9 @@ export function getModelShortName(model:string){
     if(model.startsWith("cohere-")){
         return model.replace("cohere-", "")
     }
+    if(model.startsWith("gpt4o")){
+        return "GPT-4o"
+    }
     if(model.startsWith("gpt4")){
         return "GPT-4"
     }