Skip to content

Commit f844bd2

Browse files
committed
Reformat code after merging #11
1 parent dfd6773 commit f844bd2

File tree

5 files changed

+18
-11
lines changed

5 files changed

+18
-11
lines changed

os_computer_use/config.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,15 @@
55
grounding_model = providers.OSAtlasProvider()
66
# grounding_model = providers.ShowUIProvider()
77

8-
#vision_model = providers.FireworksProvider("llama3.2")
8+
# vision_model = providers.FireworksProvider("llama3.2")
99
# vision_model = providers.OpenAIProvider("gpt-4o")
1010
# vision_model = providers.AnthropicProvider("claude-3.5-sonnet")
1111
vision_model = providers.GroqProvider("llama3.2")
12-
#vision_model = providers.MistralProvider("pixtral") # pixtral-large-latest has vision capabilities
12+
# vision_model = providers.MistralProvider("pixtral") # pixtral-large-latest has vision capabilities
1313

1414

15-
#action_model = providers.FireworksProvider("llama3.3")
15+
# action_model = providers.FireworksProvider("llama3.3")
1616
# action_model = providers.OpenAIProvider("gpt-4o")
1717
# action_model = providers.AnthropicProvider("claude-3.5-sonnet")
1818
action_model = providers.GroqProvider("llama3.3")
19-
#action_model = providers.MistralProvider("large") # mistral-large-latest for non-vision tasks
19+
# action_model = providers.MistralProvider("large") # mistral-large-latest for non-vision tasks

os_computer_use/llm_provider.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,12 +211,14 @@ def create_function_def(self, name, details, properties, required):
211211
if isinstance(details.get("description"), dict):
212212
details["description"] = details["description"].get("description", "")
213213
return super().create_function_def(name, details, properties, required)
214-
214+
215215
def call(self, messages, functions=None):
216216
if messages and messages[-1].get("role") == "assistant":
217217
prefix = messages.pop()["content"]
218218
if messages and messages[-1].get("role") == "user":
219-
messages[-1]["content"] = prefix + "\n" + messages[-1].get("content", "")
219+
messages[-1]["content"] = (
220+
prefix + "\n" + messages[-1].get("content", "")
221+
)
220222
else:
221223
messages.append({"role": "user", "content": prefix})
222224
return super().call(messages, functions)

os_computer_use/osatlas_provider.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ class OSAtlasProvider:
1818
"""
1919

2020
def __init__(self):
21-
self.client = Client(OSATLAS_HUGGINGFACE_SOURCE, hf_token=HF_TOKEN)
21+
self.client = Client(OSATLAS_HUGGINGFACE_SOURCE, hf_token=HF_TOKEN)
2222

2323
def call(self, prompt, image_data):
2424
result = self.client.predict(

os_computer_use/providers.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
11
import os
22
from dotenv import load_dotenv
3-
from os_computer_use.llm_provider import OpenAIBaseProvider, AnthropicBaseProvider, MistralBaseProvider
3+
from os_computer_use.llm_provider import (
4+
OpenAIBaseProvider,
5+
AnthropicBaseProvider,
6+
MistralBaseProvider,
7+
)
48
from os_computer_use.osatlas_provider import OSAtlasProvider
59
from os_computer_use.showui_provider import ShowUIProvider
610

@@ -63,12 +67,13 @@ class GroqProvider(OpenAIBaseProvider):
6367
"llama3.3": "llama-3.3-70b-versatile",
6468
}
6569

70+
6671
class MistralProvider(MistralBaseProvider):
6772
base_url = "https://api.mistral.ai/v1"
6873
api_key = os.getenv("MISTRAL_API_KEY")
6974
aliases = {
7075
"small": "mistral-small-latest",
7176
"medium": "mistral-medium-latest",
7277
"large": "mistral-large-latest",
73-
"pixtral": "pixtral-large-latest"
74-
}
78+
"pixtral": "pixtral-large-latest",
79+
}

os_computer_use/sandbox_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,4 +219,4 @@ def run(self, instruction):
219219

220220
self.messages.append(
221221
Message(logger.log(f"OBSERVATION: {result}", "yellow"))
222-
)
222+
)

0 commit comments

Comments
 (0)