Commit 3914ec2

benny-dou <60535774+benny-dou@users.noreply.github.com>
2025-04-03 09:24:19
style(gpt): improve model name
main
1 parent 8203459
Changed files (1)
src
llm
utils.py
src/llm/utils.py
@@ -79,6 +79,10 @@ def beautify_model_name(name: str) -> str:
     name = name.split("/")[-1]  # gpt-4o
     # remove "-latest"
     name = name.replace("-latest", "")
+    # remove "-exp"
+    name = name.replace("-experimental", "").replace("-exp", "")
+    # remove "-preview"
+    name = name.replace("-preview", "")
 
     return name.replace("gpt", "GPT").replace("gemini", "Gemini").replace("deepseek", "DeepSeek")  # GPT-4o