formatting + typo fix

agent/core/tools.py

@@ -14,8 +14,8 @@ from mcp.types import EmbeddedResource, ImageContent, TextContent
 
 from agent.config import MCPServerConfig
 from agent.tools.jobs_tool import HF_JOBS_TOOL_SPEC, hf_jobs_handler
-from agent.tools.search_docs_tool import SEARCH_DOCS_TOOL_SPEC, search_docs_handler
 from agent.tools.plan_tool import PLAN_TOOL_SPEC, plan_tool_handler
+from agent.tools.search_docs_tool import SEARCH_DOCS_TOOL_SPEC, search_docs_handler
 
 # Suppress aiohttp deprecation warning
 warnings.filterwarnings(
@@ -204,10 +204,10 @@ def create_builtin_tools() -> list[ToolSpec]:
             parameters=SEARCH_DOCS_TOOL_SPEC["parameters"],
             handler=search_docs_handler,
         ),
-      ToolSpec(
-            ame=PLAN_TOOL_SPEC["name"],
+        ToolSpec(
+            name=PLAN_TOOL_SPEC["name"],
             description=PLAN_TOOL_SPEC["description"],
             parameters=PLAN_TOOL_SPEC["parameters"],
             handler=plan_tool_handler,
-        )
+        ),
     ]

agent/main.py

@@ -71,7 +71,7 @@ async def event_listener(
 
             # Display event
             if event.event_type == "ready":
-                print(format_success("\U0001F917 Agent ready"))
+                print(format_success("\U0001f917 Agent ready"))
                 ready_event.set()
             elif event.event_type == "assistant_message":
                 content = event.data.get("content", "") if event.data else ""
@@ -194,8 +194,7 @@ async def main():
     from agent.utils.terminal_display import Colors
 
     # Clear screen
-    os.system('clear' if os.name != 'nt' else 'cls')
-
+    os.system("clear" if os.name != "nt" else "cls")
 
     banner = r"""
   _   _                   _               _____                   _                    _   
@@ -206,7 +205,6 @@ async def main():
               |___/ |___/         |___/                               |___/
     """
 
-
     print(format_separator())
     print(f"{Colors.YELLOW} {banner}{Colors.RESET}")
     print("Type your messages below. Type 'exit', 'quit', or '/quit' to end.\n")

agent/tools/plan_tool.py

@@ -1,8 +1,8 @@
-import json
 from typing import Any, Dict, List
-from .types import ToolResult
+
 from agent.utils.terminal_display import format_plan_tool_output
 
+from .types import ToolResult
 
 # In-memory storage for the current plan (raw structure from agent)
 _current_plan: List[Dict[str, str]] = []
@@ -86,24 +86,24 @@ PLAN_TOOL_SPEC = {
                     "properties": {
                         "id": {
                             "type": "string",
-                            "description": "Unique identifier for the todo"
+                            "description": "Unique identifier for the todo",
                         },
                         "content": {
                             "type": "string",
-                            "description": "Description of the todo task"
+                            "description": "Description of the todo task",
                         },
                         "status": {
                             "type": "string",
                             "enum": ["pending", "in_progress", "completed"],
-                            "description": "Current status of the todo"
-                        }
+                            "description": "Current status of the todo",
+                        },
                     },
-                    "required": ["id", "content", "status"]
-                }
+                    "required": ["id", "content", "status"],
+                },
             }
         },
-        "required": ["todos"]
-    }
+        "required": ["todos"],
+    },
 }
 
 

agent/tools/search_docs_tool.py

@@ -113,7 +113,9 @@ async def search_docs_handler(arguments: dict[str, Any]) -> tuple[str, bool]:
 
         # Load the search agent's own config file with GitHub MCP server
         search_agent_config_path = (
-            Path(__file__).parent.parent.parent / "configs" / "_subagent_config_search_agent.json"
+            Path(__file__).parent.parent.parent
+            / "configs"
+            / "_subagent_config_search_agent.json"
         )
         search_agent_config = load_config(search_agent_config_path)
 

agent/utils/terminal_display.py

@@ -18,10 +18,13 @@ class Colors:
 
 def truncate_to_lines(text: str, max_lines: int = 6) -> str:
     """Truncate text to max_lines, adding '...' if truncated"""
-    lines = text.split('\n')
+    lines = text.split("\n")
     if len(lines) <= max_lines:
         return text
-    return '\n'.join(lines[:max_lines]) + f'\n{Colors.CYAN}... ({len(lines) - max_lines} more lines){Colors.RESET}'
+    return (
+        "\n".join(lines[:max_lines])
+        + f"\n{Colors.CYAN}... ({len(lines) - max_lines} more lines){Colors.RESET}"
+    )
 
 
 def format_header(text: str, emoji: str = "") -> str:
@@ -65,10 +68,12 @@ def format_plan_display() -> str:
             lines.append(f"  [ ] {todo['id']}. {todo['content']}")
         lines.append("")
 
-    lines.append(f"Total: {len(plan)} todos ({len(completed)} completed, {len(in_progress)} in progress, {len(pending)} pending)")
+    lines.append(
+        f"Total: {len(plan)} todos ({len(completed)} completed, {len(in_progress)} in progress, {len(pending)} pending)"
+    )
     lines.append("=" * 60 + "\n")
 
-    return '\n'.join(lines)
+    return "\n".join(lines)
 
 
 def format_error(message: str) -> str:
@@ -100,7 +105,7 @@ def format_tool_output(output: str, success: bool, truncate: bool = True) -> str
 
 def format_turn_complete() -> str:
     """Format turn complete message in green with hugging face emoji"""
-    return f"{Colors.GREEN}{Colors.BOLD}\U0001F917 Turn complete{Colors.RESET}\n"
+    return f"{Colors.GREEN}{Colors.BOLD}\U0001f917 Turn complete{Colors.RESET}\n"
 
 
 def format_separator(char: str = "=", length: int = 60) -> str:
@@ -138,6 +143,8 @@ def format_plan_tool_output(todos: list) -> str:
             lines.append(f"  [ ] {todo['id']}. {todo['content']}")
         lines.append("")
 
-    lines.append(f"Total: {len(todos)} todos ({len(completed)} completed, {len(in_progress)} in progress, {len(pending)} pending)")
+    lines.append(
+        f"Total: {len(todos)} todos ({len(completed)} completed, {len(in_progress)} in progress, {len(pending)} pending)"
+    )
 
     return "\n".join(lines)