chore: ruff and lint

Author: Parth220

examples/run_evaluation.py

@@ -121,7 +121,7 @@ async def run_single_task(
             model=model,
             allowed_tools=allowed_tools,
         )
-        print( "Task prompt: ", task.prompt)
+        print("Task prompt: ", task.prompt)
         result = await agent.run(task, max_steps=max_steps)
         print("✅ Reward:", result.reward)
 
@@ -269,7 +269,10 @@ def parse_args() -> argparse.Namespace:  # type: ignore[valid-type]
         "--verbose", "-v", action="store_true", help="Show detailed agent step logs"
     )
     parser.add_argument(
-        "--very-verbose", "-vv", action="store_true", help="Show debug-level logs for maximum visibility"
+        "--very-verbose",
+        "-vv",
+        action="store_true",
+        help="Show debug-level logs for maximum visibility",
     )
 
     return parser.parse_args()

hud/agents/base.py

@@ -556,7 +556,6 @@ async def _filter_tools(self) -> None:
 
         all_tools = await self.mcp_client.list_tools()
 
-
         response_tools_by_server: dict[str, str] = {}  # server_name -> tool_name
         for tool in all_tools:
             if "response" in tool.name or tool.name == "response":
@@ -579,7 +578,9 @@ async def _filter_tools(self) -> None:
             for server_name in reversed(server_names):
                 if server_name in response_tools_by_server:
                     response_tool_name = response_tools_by_server[server_name]
-                    self.console.debug(f"Found response tool '{response_tool_name}' from server '{server_name}'")
+                    self.console.debug(
+                        f"Found response tool '{response_tool_name}' from server '{server_name}'"
+                    )
                     break
 
             # Fallback to any response tool
@@ -593,10 +594,12 @@ async def _filter_tools(self) -> None:
                 self.response_tool_name = response_tool_name
                 self.lifecycle_tools.append(response_tool_name)
             elif response_tool_name:
-                self.console.debug(f"Response tool '{response_tool_name}' already in lifecycle_tools")
+                self.console.debug(
+                    f"Response tool '{response_tool_name}' already in lifecycle_tools"
+                )
                 self.response_tool_name = response_tool_name
         else:
-            self.console.debug(f"No response tools found or no mcp_config")
+            self.console.debug("No response tools found or no mcp_config")
 
         # Filter tools
         self._available_tools = []
@@ -610,7 +613,7 @@ async def _filter_tools(self) -> None:
         for tool in all_tools:
             # Lifecycle tools (setup, evaluate, response) should always be included
             is_lifecycle = tool.name in self.lifecycle_tools
-            
+
             # Check if tool should be included
             if not is_lifecycle:
                 if self.allowed_tools and tool.name not in self.allowed_tools:
@@ -620,7 +623,9 @@ async def _filter_tools(self) -> None:
                     self.console.debug(f"Skipping tool '{tool.name}' - in disallowed_tools")
                     continue
 
-            self.console.debug(f"Adding tool '{tool.name}' to available tools (lifecycle={is_lifecycle})")
+            self.console.debug(
+                f"Adding tool '{tool.name}' to available tools (lifecycle={is_lifecycle})"
+            )
             self._available_tools.append(tool)
             self._tool_map[tool.name] = tool
 

hud/agents/misc/response_agent.py

@@ -34,7 +34,7 @@ def __init__(self, api_key: str | None = None, model: str = "gpt-4o") -> None:
         
         - STOP: If the agent indicates it has successfully completed a task, even if phrased as a question
           like "I have entered the right values into this form. Would you like me to do anything else?"
-          or "Here is the website. Is there any other information you need?" or if the agent has 
+          or "Here is the website. Is there any other information you need?" or if the agent has
           strongly determined it wants to stop the task.
         
         - CONTINUE: If the agent is asking for clarification before proceeding with a task