Formatting

agent/llm/ollama.py

@@ -66,7 +66,9 @@ class OllamaClient:
 
         logger.info(f"Ollama client initialized with model: {self.model}")
 
-    def complete(self, messages: list[dict[str, Any]], tools: list[dict[str, Any]] | None = None) -> dict[str, Any]:
+    def complete(
+        self, messages: list[dict[str, Any]], tools: list[dict[str, Any]] | None = None
+    ) -> dict[str, Any]:
         """
         Generate a completion from the LLM.
 
@@ -95,7 +97,9 @@ class OllamaClient:
                 raise ValueError(f"Invalid role: {msg['role']}")
             # Content is optional for tool messages (they may have tool_call_id instead)
             if msg["role"] != "tool" and "content" not in msg:
-                raise ValueError(f"Non-tool message must have 'content' key, got {msg.keys()}")
+                raise ValueError(
+                    f"Non-tool message must have 'content' key, got {msg.keys()}"
+                )
 
         url = f"{self.base_url}/api/chat"
         payload = {
@@ -106,13 +110,15 @@ class OllamaClient:
                 "temperature": self.temperature,
             },
         }
-        
+
         # Add tools if provided
         if tools:
             payload["tools"] = tools
 
         try:
-            logger.debug(f"Sending request to {url} with {len(messages)} messages and {len(tools) if tools else 0} tools")
+            logger.debug(
+                f"Sending request to {url} with {len(messages)} messages and {len(tools) if tools else 0} tools"
+            )
             response = requests.post(url, json=payload, timeout=self.timeout)
             response.raise_for_status()
             data = response.json()