Organize LLMs

virattt · Feb 3, 2025 · c1f6827 · c1f6827
1 parent 9e171d0
commit c1f6827
Showing 1 changed file with 25 additions and 20 deletions.
diff --git a/src/llm/models.py b/src/llm/models.py
@@ -32,44 +32,49 @@ def is_deepseek(self) -> bool:
 # Define available models
 AVAILABLE_MODELS = [
     LLMModel(
-        display_name="gpt-4o [openai]",
-        model_name="gpt-4o",
-        provider=ModelProvider.OPENAI
+        display_name="[anthropic] claude-3.5-haiku",
+        model_name="claude-3-5-haiku-latest",
+        provider=ModelProvider.ANTHROPIC
     ),
     LLMModel(
-        display_name="gpt-4o-mini [openai]",
-        model_name="gpt-4o-mini",
-        provider=ModelProvider.OPENAI
+        display_name="[anthropic] claude-3.5-sonnet",
+        model_name="claude-3-5-sonnet-latest",
+        provider=ModelProvider.ANTHROPIC
     ),
     LLMModel(
-        display_name="o3-mini [openai]",
-        model_name="o3-mini",
-        provider=ModelProvider.OPENAI
+        display_name="[anthropic] claude-3-opus",
+        model_name="claude-3-opus-latest",
+        provider=ModelProvider.ANTHROPIC
     ),
     LLMModel(
-        display_name="deepseek-r1 70b [groq]",
+        display_name="[groq] deepseek-r1 70b",
         model_name="deepseek-r1-distill-llama-70b",
         provider=ModelProvider.GROQ
     ),
     LLMModel(
-        display_name="llama-3.3 70b [groq]",
+        display_name="[groq] llama-3.3 70b",
         model_name="llama-3.3-70b-versatile",
         provider=ModelProvider.GROQ
     ),
     LLMModel(
-        display_name="claude-3.5-sonnet [anthropic]",
-        model_name="claude-3-5-sonnet-latest",
-        provider=ModelProvider.ANTHROPIC
+        display_name="[openai] gpt-4o",
+        model_name="gpt-4o",
+        provider=ModelProvider.OPENAI
     ),
     LLMModel(
-        display_name="claude-3.5-haiku [anthropic]",
-        model_name="claude-3-5-haiku-latest",
-        provider=ModelProvider.ANTHROPIC
+        display_name="[openai] gpt-4o-mini",
+        model_name="gpt-4o-mini",
+        provider=ModelProvider.OPENAI
     ),
     LLMModel(
-        display_name="claude-3-opus [anthropic]",
-        model_name="claude-3-opus-latest",
-        provider=ModelProvider.ANTHROPIC
+        display_name="[openai] o1",
+        model_name="o1",
+        provider=ModelProvider.OPENAI
+    ),
+    LLMModel(
+        display_name="[openai] o3-mini",
+        model_name="o3-mini",
+        provider=ModelProvider.OPENAI
     ),
 ]