Clean up model names

Author: jamesmurdza

os_computer_use/config.py

@@ -5,17 +5,16 @@
 grounding_model = providers.OSAtlasProvider()
 # grounding_model = providers.ShowUIProvider()
 
-# vision_model = providers.FireworksProvider("llama3.2")
+# vision_model = providers.FireworksProvider("llama-3.2")
 # vision_model = providers.OpenAIProvider("gpt-4o")
 # vision_model = providers.AnthropicProvider("claude-3.5-sonnet")
 # vision_model = providers.MoonshotProvider("moonshot-v1-vision")
-vision_model = providers.GroqProvider("llama3.2")
-# vision_model = providers.MistralProvider("pixtral")  # pixtral-large-latest has vision capabilities
+# vision_model = providers.MistralProvider("pixtral")
+vision_model = providers.GroqProvider("llama-3.2")
 
-
-# action_model = providers.FireworksProvider("llama3.3")
+# action_model = providers.FireworksProvider("llama-3.3")
 # action_model = providers.OpenAIProvider("gpt-4o")
 # action_model = providers.AnthropicProvider("claude-3.5-sonnet")
 # vision_model = providers.MoonshotProvider("moonshot-v1-vision")
-action_model = providers.GroqProvider("llama3.3")
-# action_model = providers.MistralProvider("large")  # mistral-large-latest for non-vision tasks
+# action_model = providers.MistralProvider("mistral")
+action_model = providers.GroqProvider("llama-3.3")

os_computer_use/providers.py

@@ -17,21 +17,21 @@
 class LlamaProvider(OpenAIBaseProvider):
     base_url = "https://api.llama-api.com"
     api_key = os.getenv("LLAMA_API_KEY")
-    aliases = {"llama3.2": "llama3.2-90b-vision", "llama3.3": "llama3.3-70b"}
+    aliases = {"llama-3.2": "llama-3.2-90b-vision", "llama-3.3": "llama-3.3-70b"}
 
 
 class OpenRouterProvider(OpenAIBaseProvider):
     base_url = "https://openrouter.ai/api/v1"
     api_key = os.getenv("OPENROUTER_API_KEY")
-    aliases = {"llama3.2": "meta-llama/llama-3.2-90b-vision-instruct"}
+    aliases = {"llama-3.2": "meta-llama/llama-3.2-90b-vision-instruct"}
 
 
 class FireworksProvider(OpenAIBaseProvider):
     base_url = "https://api.fireworks.ai/inference/v1"
     api_key = os.getenv("FIREWORKS_API_KEY")
     aliases = {
-        "llama3.2": "accounts/fireworks/models/llama-v3p2-90b-vision-instruct",
-        "llama3.3": "accounts/fireworks/models/llama-v3p3-70b-instruct",
+        "llama-3.2": "accounts/fireworks/models/llama-v3p2-90b-vision-instruct",
+        "llama-3.3": "accounts/fireworks/models/llama-v3p3-70b-instruct",
     }
 
 
@@ -63,18 +63,16 @@ class GroqProvider(OpenAIBaseProvider):
     base_url = "https://api.groq.com/openai/v1"
     api_key = os.getenv("GROQ_API_KEY")
     aliases = {
-        "llama3.2": "llama-3.2-90b-vision-preview",
-        "llama3.3": "llama-3.3-70b-versatile",
+        "llama-3.2": "llama-3.2-90b-vision-preview",
+        "llama-3.3": "llama-3.3-70b-versatile",
     }
 
 
 class MistralProvider(MistralBaseProvider):
     base_url = "https://api.mistral.ai/v1"
     api_key = os.getenv("MISTRAL_API_KEY")
     aliases = {
-        "small": "mistral-small-latest",
-        "medium": "mistral-medium-latest",
-        "large": "mistral-large-latest",
+        "mistral": "mistral-large-latest",
         "pixtral": "pixtral-large-latest",
     }
 

tests/llm_provider.py

@@ -65,17 +65,14 @@ def take_screenshot():
 print(fireworks.call(toolcall_messages, tools)[1])
 print(fireworks.call(messages))
 
-
-
 # Pixtral
 mistral = MistralProvider("pixtral") 
 print("\nTesting Mistral :")
 print(mistral.call(toolcall_messages, tools)[1])
 print(mistral.call(messages))
 
-
 # Mistral Large (non-vision) using text-only messages
-mistral_large = MistralProvider("large")  # Using mistral-large-latest for non-vision tasks
+mistral_large = MistralProvider("mistral")
 text_messages = [Message("What is the capital of France?", role="user")]
 print("\nTesting Mistral Large with text-only:")
 print(mistral_large.call(text_messages))