Automated pre-commit update

Author: rushilpatel0

codegen-examples/examples/codegen-mcp-server/server.py

@@ -1,8 +1,6 @@
 import asyncio
 from dataclasses import dataclass, field
-import json
-from types import coroutine
-from typing import Annotated, Coroutine, Optional, Dict, Any, List
+from typing import Annotated, Optional, Dict, Any, List
 from mcp.server.fastmcp import FastMCP
 from codegen import Codebase
 
@@ -44,26 +42,24 @@ def capture_output(*args, **kwargs) -> None:
     for arg in args:
         state.log_buffer.append(str(arg))
 
+
 def update_codebase(future: asyncio.Future):
     try:
         result = future.result()
         if result is not None:
             state.parsed_codebase = result
         else:
             state.parsed_codebase = None
-    except Exception as e:
+    except Exception:
         pass
 
 
-
 @mcp.tool(name="parse_codebase", description="Initiate codebase parsing")
 async def parse_codebase(codebase_path: Annotated[str, "path to the codebase to be parsed"]) -> Dict[str, str]:
     if not state.parse_task or state.parse_task.done():
         state.parse_task = asyncio.get_event_loop().run_in_executor(None, lambda: state.parse(codebase_path))
         state.parse_task.add_done_callback(update_codebase)
-        return {
-            "message": "Codebase parsing initiated, this may take some time depending on the size of the codebase. Use the `check_parsing_status` tool to check if the parse has completed."
-        }
+        return {"message": "Codebase parsing initiated, this may take some time depending on the size of the codebase. Use the `check_parsing_status` tool to check if the parse has completed."}
     return {"message": "Codebase is already being parsed.", "status": "error"}
 
 
@@ -83,7 +79,7 @@ async def execute_codemod(codemod: Annotated[str, "The python codemod code to ex
 
     try:
         await state.parse_task
-        if (state.parsed_codebase is None):
+        if state.parsed_codebase is None:
             return {"error": "Codebase path is not set."}
         else:
             # TODO: Implement proper sandboxing for code execution