Fix pre-commit issues: remove unused import and fix line length

- Remove unused Codebase import from mcp/server.py
- Fix line length issue in test_server_startup.py by splitting long string
- Apply ruff formatting to all files

Author: codegen-sh[bot]

src/codegen/cli/commands/mcp/main.py

@@ -26,20 +26,20 @@
 )
 def mcp_command(host: str, port: int | None, transport: str):
     """Start the Codegen MCP server.
-    
+
     The MCP server exposes Codegen API endpoints as MCP tools,
     allowing MCP clients to interact with the Codegen platform.
     """
     console.print("🚀 Starting Codegen MCP server...", style="bold green")
-    
+
     if transport == "stdio":
         console.print("📡 Using stdio transport", style="dim")
     else:
         console.print(f"📡 Using HTTP transport on {host}:{port}", style="dim")
-    
+
     # Import here to avoid circular imports and ensure dependencies are available
     from codegen.cli.mcp.server import run_server
-    
+
     try:
         run_server(transport=transport, host=host, port=port)
     except KeyboardInterrupt:

src/codegen/cli/mcp/server.py

@@ -1,6 +1,6 @@
-from typing import Annotated, Any, Optional
-import os
 import json
+import os
+from typing import Annotated, Any
 
 from fastmcp import Context, FastMCP
 
@@ -10,8 +10,8 @@
 # Optional imports for existing functionality
 try:
     from codegen.cli.api.client import RestAPI
-    from codegen.sdk.core.codebase import Codebase
     from codegen.shared.enums.programming_language import ProgrammingLanguage
+
     LEGACY_IMPORTS_AVAILABLE = True
 except ImportError:
     LEGACY_IMPORTS_AVAILABLE = False
@@ -21,14 +21,15 @@
     from codegen_api_client import ApiClient, Configuration
     from codegen_api_client.api import AgentsApi, OrganizationsApi, UsersApi
     from codegen_api_client.models import CreateAgentRunInput
+
     API_CLIENT_AVAILABLE = True
 except ImportError:
     API_CLIENT_AVAILABLE = False
 
 # Initialize FastMCP server
 mcp = FastMCP(
-    "codegen-mcp", 
-    instructions="MCP server for the Codegen platform. Use the tools and resources to interact with Codegen APIs, setup codegen in your environment, and create/improve Codegen Codemods."
+    "codegen-mcp",
+    instructions="MCP server for the Codegen platform. Use the tools and resources to interact with Codegen APIs, setup codegen in your environment, and create/improve Codegen Codemods.",
 )
 
 # Global API client instances
@@ -41,30 +42,32 @@
 def get_api_client():
     """Get or create the API client instance."""
     global _api_client, _agents_api, _organizations_api, _users_api
-    
+
     if not API_CLIENT_AVAILABLE:
-        raise RuntimeError("codegen-api-client is not available")
-    
+        msg = "codegen-api-client is not available"
+        raise RuntimeError(msg)
+
     if _api_client is None:
         # Configure the API client
         configuration = Configuration()
-        
+
         # Set base URL from environment or use default
         base_url = os.getenv("CODEGEN_API_BASE_URL", "https://api.codegen.com")
         configuration.host = base_url
-        
+
         # Set authentication
         api_key = os.getenv("CODEGEN_API_KEY")
         if api_key:
             configuration.api_key = {"Authorization": f"Bearer {api_key}"}
-        
+
         _api_client = ApiClient(configuration)
         _agents_api = AgentsApi(_api_client)
         _organizations_api = OrganizationsApi(_api_client)
         _users_api = UsersApi(_api_client)
-    
+
     return _api_client, _agents_api, _organizations_api, _users_api
 
+
 # ----- RESOURCES -----
 
 
@@ -95,6 +98,7 @@ def get_service_config() -> dict[str, Any]:
 
 # Legacy SDK tool (only available if imports are available)
 if LEGACY_IMPORTS_AVAILABLE:
+
     @mcp.tool()
     def ask_codegen_sdk(query: Annotated[str, "Ask a question to an expert agent for details about any aspect of the codegen sdk core set of classes and utilities"]):
         """Ask questions about the Codegen SDK (requires legacy imports)."""
@@ -118,6 +122,7 @@ def generate_codemod(
 
 # Legacy improve codemod tool (only available if imports are available)
 if LEGACY_IMPORTS_AVAILABLE:
+
     @mcp.tool()
     def improve_codemod(
         codemod_source: Annotated[str, "The source code of the codemod to improve"],
@@ -138,40 +143,37 @@ def improve_codemod(
 
 # ----- CODEGEN API TOOLS -----
 
+
 @mcp.tool()
 def create_agent_run(
     org_id: Annotated[int, "Organization ID"],
     prompt: Annotated[str, "The prompt/task for the agent to execute"],
-    repo_name: Annotated[Optional[str], "Repository name (optional)"] = None,
-    branch_name: Annotated[Optional[str], "Branch name (optional)"] = None,
+    repo_name: Annotated[str | None, "Repository name (optional)"] = None,
+    branch_name: Annotated[str | None, "Branch name (optional)"] = None,
     ctx: Context = None,
 ) -> str:
     """Create a new agent run in the specified organization."""
     try:
         _, agents_api, _, _ = get_api_client()
-        
+
         # Create the input object
-        agent_input = CreateAgentRunInput(
-            prompt=prompt,
-            repo_name=repo_name,
-            branch_name=branch_name
-        )
-        
+        agent_input = CreateAgentRunInput(prompt=prompt, repo_name=repo_name, branch_name=branch_name)
+
         # Make the API call
-        response = agents_api.create_agent_run_v1_organizations_org_id_agent_run_post(
-            org_id=org_id,
-            create_agent_run_input=agent_input
+        response = agents_api.create_agent_run_v1_organizations_org_id_agent_run_post(org_id=org_id, create_agent_run_input=agent_input)
+
+        return json.dumps(
+            {
+                "id": response.id,
+                "status": response.status,
+                "created_at": response.created_at.isoformat() if response.created_at else None,
+                "prompt": response.prompt,
+                "repo_name": response.repo_name,
+                "branch_name": response.branch_name,
+            },
+            indent=2,
         )
-        
-        return json.dumps({
-            "id": response.id,
-            "status": response.status,
-            "created_at": response.created_at.isoformat() if response.created_at else None,
-            "prompt": response.prompt,
-            "repo_name": response.repo_name,
-            "branch_name": response.branch_name
-        }, indent=2)
-        
+
     except Exception as e:
         return f"Error creating agent run: {e}"
 
@@ -185,23 +187,23 @@ def get_agent_run(
     """Get details of a specific agent run."""
     try:
         _, agents_api, _, _ = get_api_client()
-        
-        response = agents_api.get_agent_run_v1_organizations_org_id_agent_run_agent_run_id_get(
-            org_id=org_id,
-            agent_run_id=agent_run_id
+
+        response = agents_api.get_agent_run_v1_organizations_org_id_agent_run_agent_run_id_get(org_id=org_id, agent_run_id=agent_run_id)
+
+        return json.dumps(
+            {
+                "id": response.id,
+                "status": response.status,
+                "created_at": response.created_at.isoformat() if response.created_at else None,
+                "updated_at": response.updated_at.isoformat() if response.updated_at else None,
+                "prompt": response.prompt,
+                "repo_name": response.repo_name,
+                "branch_name": response.branch_name,
+                "result": response.result,
+            },
+            indent=2,
         )
-        
-        return json.dumps({
-            "id": response.id,
-            "status": response.status,
-            "created_at": response.created_at.isoformat() if response.created_at else None,
-            "updated_at": response.updated_at.isoformat() if response.updated_at else None,
-            "prompt": response.prompt,
-            "repo_name": response.repo_name,
-            "branch_name": response.branch_name,
-            "result": response.result
-        }, indent=2)
-        
+
     except Exception as e:
         return f"Error getting agent run: {e}"
 
@@ -215,26 +217,16 @@ def get_organizations(
     """Get list of organizations the user has access to."""
     try:
         _, _, organizations_api, _ = get_api_client()
-        
+
         response = organizations_api.get_organizations_v1_organizations_get()
-        
+
         # Format the response
         organizations = []
         for org in response.items:
-            organizations.append({
-                "id": org.id,
-                "name": org.name,
-                "slug": org.slug,
-                "created_at": org.created_at.isoformat() if org.created_at else None
-            })
-        
-        return json.dumps({
-            "organizations": organizations,
-            "total": response.total,
-            "page": response.page,
-            "limit": response.limit
-        }, indent=2)
-        
+            organizations.append({"id": org.id, "name": org.name, "slug": org.slug, "created_at": org.created_at.isoformat() if org.created_at else None})
+
+        return json.dumps({"organizations": organizations, "total": response.total, "page": response.page, "limit": response.limit}, indent=2)
+
     except Exception as e:
         return f"Error getting organizations: {e}"
 
@@ -249,28 +241,16 @@ def get_users(
     """Get list of users in an organization."""
     try:
         _, _, _, users_api = get_api_client()
-        
-        response = users_api.get_users_v1_organizations_org_id_users_get(
-            org_id=org_id
-        )
-        
+
+        response = users_api.get_users_v1_organizations_org_id_users_get(org_id=org_id)
+
         # Format the response
         users = []
         for user in response.items:
-            users.append({
-                "id": user.id,
-                "email": user.email,
-                "name": user.name,
-                "created_at": user.created_at.isoformat() if user.created_at else None
-            })
-        
-        return json.dumps({
-            "users": users,
-            "total": response.total,
-            "page": response.page,
-            "limit": response.limit
-        }, indent=2)
-        
+            users.append({"id": user.id, "email": user.email, "name": user.name, "created_at": user.created_at.isoformat() if user.created_at else None})
+
+        return json.dumps({"users": users, "total": response.total, "page": response.page, "limit": response.limit}, indent=2)
+
     except Exception as e:
         return f"Error getting users: {e}"
 
@@ -284,20 +264,20 @@ def get_user(
     """Get details of a specific user in an organization."""
     try:
         _, _, _, users_api = get_api_client()
-        
-        response = users_api.get_user_v1_organizations_org_id_users_user_id_get(
-            org_id=org_id,
-            user_id=user_id
+
+        response = users_api.get_user_v1_organizations_org_id_users_user_id_get(org_id=org_id, user_id=user_id)
+
+        return json.dumps(
+            {
+                "id": response.id,
+                "email": response.email,
+                "name": response.name,
+                "created_at": response.created_at.isoformat() if response.created_at else None,
+                "updated_at": response.updated_at.isoformat() if response.updated_at else None,
+            },
+            indent=2,
         )
-        
-        return json.dumps({
-            "id": response.id,
-            "email": response.email,
-            "name": response.name,
-            "created_at": response.created_at.isoformat() if response.created_at else None,
-            "updated_at": response.updated_at.isoformat() if response.updated_at else None
-        }, indent=2)
-        
+
     except Exception as e:
         return f"Error getting user: {e}"
 
@@ -314,7 +294,8 @@ def run_server(transport: str = "stdio", host: str = "localhost", port: int | No
         print(f"HTTP transport not yet implemented. Would run on {host}:{port}")
         mcp.run(transport="stdio")  # Fallback to stdio for now
     else:
-        raise ValueError(f"Unsupported transport: {transport}")
+        msg = f"Unsupported transport: {transport}"
+        raise ValueError(msg)
 
 
 if __name__ == "__main__":

src/codegen/shared/compilation/exception_utils.py

@@ -45,10 +45,7 @@ def get_local_frame(exc_type: type[BaseException], exc_value: BaseException, exc
 
 def get_local_frame_context(frame: FrameType):
     local_vars = {k: v for k, v in frame.f_locals.items() if not k.startswith("__")}
-    if "print" in local_vars:
-        del local_vars["print"]
-    if "codebase" in local_vars:
-        del local_vars["codebase"]
-    if "pr_options" in local_vars:
-        del local_vars["pr_options"]
+    local_vars.pop("print", None)
+    local_vars.pop("codebase", None)
+    local_vars.pop("pr_options", None)
     return local_vars