Reformat code after merging #11

Author: jamesmurdza

os_computer_use/config.py

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

os_computer_use/llm_provider.py

@@ -211,12 +211,14 @@ def create_function_def(self, name, details, properties, required):
         if isinstance(details.get("description"), dict):
             details["description"] = details["description"].get("description", "")
         return super().create_function_def(name, details, properties, required)
-    
+
     def call(self, messages, functions=None):
         if messages and messages[-1].get("role") == "assistant":
             prefix = messages.pop()["content"]
             if messages and messages[-1].get("role") == "user":
-                messages[-1]["content"] = prefix + "\n" + messages[-1].get("content", "")
+                messages[-1]["content"] = (
+                    prefix + "\n" + messages[-1].get("content", "")
+                )
             else:
                 messages.append({"role": "user", "content": prefix})
         return super().call(messages, functions)

os_computer_use/osatlas_provider.py

@@ -18,7 +18,7 @@ class OSAtlasProvider:
     """
 
     def __init__(self):
-        self.client = Client(OSATLAS_HUGGINGFACE_SOURCE,  hf_token=HF_TOKEN)
+        self.client = Client(OSATLAS_HUGGINGFACE_SOURCE, hf_token=HF_TOKEN)
 
     def call(self, prompt, image_data):
         result = self.client.predict(

os_computer_use/providers.py

@@ -1,6 +1,10 @@
 import os
 from dotenv import load_dotenv
-from os_computer_use.llm_provider import OpenAIBaseProvider, AnthropicBaseProvider, MistralBaseProvider
+from os_computer_use.llm_provider import (
+    OpenAIBaseProvider,
+    AnthropicBaseProvider,
+    MistralBaseProvider,
+)
 from os_computer_use.osatlas_provider import OSAtlasProvider
 from os_computer_use.showui_provider import ShowUIProvider
 
@@ -63,12 +67,13 @@ class GroqProvider(OpenAIBaseProvider):
         "llama3.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",
-        "pixtral": "pixtral-large-latest"
-    }
+        "pixtral": "pixtral-large-latest",
+    }

os_computer_use/sandbox_agent.py

@@ -219,4 +219,4 @@ def run(self, instruction):
 
                 self.messages.append(
                     Message(logger.log(f"OBSERVATION: {result}", "yellow"))
-                )
+                )