reformat

Author: matengm1

app/cache.py

@@ -18,7 +18,9 @@ def __init__(self):
         Initialize the LLM cache.
         """
         self.cache_dir = config.cache_config.cache_dir
-        self.cache_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), self.cache_dir)
+        self.cache_dir = os.path.join(
+            os.path.dirname(os.path.dirname(__file__)), self.cache_dir
+        )
         os.makedirs(self.cache_dir, exist_ok=True)
 
         self.ttl = config.cache_config.ttl

app/config.py

@@ -56,12 +56,12 @@ class BrowserSettings(BaseModel):
         None, description="Proxy settings for the browser"
     )
 
+
 class CacheSettings(BaseModel):
     enabled: bool = Field(False, description="Whether to enable caching")
     ttl: int = Field(86400, description="Cache time-to-live in seconds")
-    cache_dir: str = Field(
-        "cache", description="Path to the cache directory"
-    )
+    cache_dir: str = Field("cache", description="Path to the cache directory")
+
 
 class AppConfig(BaseModel):
     llm: Dict[str, LLMSettings]
@@ -199,7 +199,7 @@ def browser_config(self) -> Optional[BrowserSettings]:
     @property
     def search_config(self) -> Optional[SearchSettings]:
         return self._config.search_config
-    
+
     @property
     def cache_config(self) -> Optional[CacheSettings]:
         return self._config.cache_config

app/llm.py

@@ -164,7 +164,9 @@ async def ask(
                 params["max_completion_tokens"] = self.max_tokens
             else:
                 params["max_tokens"] = self.max_tokens
-                params["temperature"] = temperature if temperature is not None else self.temperature
+                params["temperature"] = (
+                    temperature if temperature is not None else self.temperature
+                )
 
             if not stream:
                 # Non-streaming request
@@ -292,7 +294,9 @@ async def ask_tool(
                 params["max_completion_tokens"] = self.max_tokens
             else:
                 params["max_tokens"] = self.max_tokens
-                params["temperature"] = temperature if temperature is not None else self.temperature
+                params["temperature"] = (
+                    temperature if temperature is not None else self.temperature
+                )
 
             response = await self.client.chat.completions.create(**params)