Automated pre-commit update

Author: rushilpatel0

src/codegen/cli/api/client.py

@@ -11,11 +11,11 @@
     DOCS_ENDPOINT,
     EXPERT_ENDPOINT,
     IDENTIFY_ENDPOINT,
+    IMPROVE_ENDPOINT,
     LOOKUP_ENDPOINT,
     PR_LOOKUP_ENDPOINT,
     RUN_ENDPOINT,
     RUN_ON_PR_ENDPOINT,
-    IMPROVE_ENDPOINT,
 )
 from codegen.cli.api.schemas import (
     AskExpertInput,
@@ -28,6 +28,8 @@
     DocsInput,
     DocsResponse,
     IdentifyResponse,
+    ImproveCodemodInput,
+    ImproveCodemodResponse,
     LookupInput,
     LookupOutput,
     PRLookupInput,
@@ -37,8 +39,6 @@
     RunCodemodOutput,
     RunOnPRInput,
     RunOnPRResponse,
-    ImproveCodemodInput,
-    ImproveCodemodResponse,
 )
 from codegen.cli.auth.session import CodegenSession
 from codegen.cli.codemod.convert import convert_to_ui
@@ -253,12 +253,11 @@ def lookup_pr(self, repo_full_name: str, github_pr_number: int) -> PRSchema:
             PRLookupResponse,
         )
 
-
     def improve_codemod(self, codemod: str, task: str, concerns: list[str], context: dict[str, str], language: ProgrammingLanguage) -> ImproveCodemodResponse:
         """Improve a codemod."""
         return self._make_request(
             "GET",
             IMPROVE_ENDPOINT,
             ImproveCodemodInput(input=ImproveCodemodInput.BaseImproveCodemodInput(codemod=codemod, task=task, concerns=concerns, context=context, language=language)),
-            ImproveCodemodResponse
-        )
+            ImproveCodemodResponse,
+        )

src/codegen/cli/api/endpoints.py

@@ -10,4 +10,4 @@
 RUN_ON_PR_ENDPOINT = f"https://{MODAL_PREFIX}--cli-run-on-pull-request.modal.run"
 PR_LOOKUP_ENDPOINT = f"https://{MODAL_PREFIX}--cli-pr-lookup.modal.run"
 CODEGEN_SYSTEM_PROMPT_URL = "https://gist.githubusercontent.com/jayhack/15681a2ceaccd726f19e6fdb3a44738b/raw/17c08054e3931b3b7fdf424458269c9e607541e8/codegen-system-prompt.txt"
-IMPROVE_ENDPOINT = f"https://{MODAL_PREFIX}--cli-improve.modal.run"
+IMPROVE_ENDPOINT = f"https://{MODAL_PREFIX}--cli-improve.modal.run"

src/codegen/cli/api/schemas.py

@@ -240,6 +240,7 @@ class RunOnPRResponse(BaseModel):
 # IMPROVE
 ###########################################################################
 
+
 class ImproveCodemodInput(BaseModel):
     class BaseImproveCodemodInput(BaseModel):
         codemod: str = Field(..., description="Source code of the codemod to improve")
@@ -250,6 +251,7 @@ class BaseImproveCodemodInput(BaseModel):
 
     input: BaseImproveCodemodInput = Field(..., description="Input data for improvement")
 
+
 class ImproveCodemodResponse(BaseModel):
     success: bool = Field(..., description="Whether the improvement was successful")
     codemod_source: str = Field(..., description="Source code of the improved codemod")

src/codegen/cli/mcp/README.md

@@ -8,9 +8,12 @@ A MCP server implementation that provides tools and resources for using and work
 
 ## Usage
 
-Most AI Agents that support MCP will have some way to configure the server startup. 
+Most AI Agents that support MCP will have some way to configure the server startup.
+
 ### Cline
+
 Add this to your `cline_mcp_settings.json` file to get started:
+
 ```
 {
   "mcpServers": {
@@ -35,8 +38,3 @@ Name: codegen-mcp
 Type: Command
 Command: uv --directory <path to codegen installation>/codegen-sdk/src/codegen/cli/mcp run server.py
 ```
-
-
-
-
-

src/codegen/cli/mcp/resources/system_prompt.py

@@ -2389,7 +2389,7 @@ def validate_data(data: dict) -> bool:
 results = function.find_string_literals(["error"], fuzzy_match=True)
 
 # Search with regex
-matches = function.search(r"data\['[^']*'\]")  # Find dict access
+matches = function.search(r"data\\['[^']*'\\]")  # Find dict access
 matches = function.search("TODO:", include_comments=True)
 
 # Find specific patterns
@@ -6327,9 +6327,9 @@ def migrate_api_v1_to_v2(codebase):
 4. **Document Changes**:
    ```python
    # Add clear deprecation messages
-   old_api.add_decorator(\'\'\@deprecated(
+   old_api.add_decorator(\'\'\\@deprecated(
        "Use new_api() instead. Migration guide: docs/migrations/v2.md"
-   )\'\'\)
+   )\'\'\\)
    ```
 
 <Note>
@@ -9909,4 +9909,4 @@ def list_users(request: Request):
 - [Codegen API Reference](/api-reference)
 - [Moving Symbols Guide](/building-with-codegen/moving-symbols)
 - [Dependencies and Usages](/building-with-codegen/dependencies-and-usages)
-'''
+'''

src/codegen/cli/mcp/server.py

@@ -1,15 +1,19 @@
 from typing import Annotated, Any
-from codegen.cli.mcp.resources.system_setup_instructions import SETUP_INSTRUCTIONS
-from mcp.server.fastmcp import FastMCP, Context
+
+from mcp.server.fastmcp import Context, FastMCP
+
 from codegen.cli.api.client import RestAPI
 from codegen.cli.mcp.resources.system_prompt import SYSTEM_PROMPT
+from codegen.cli.mcp.resources.system_setup_instructions import SETUP_INSTRUCTIONS
 from codegen.cli.utils.constants import ProgrammingLanguage
+
 # Initialize FastMCP server
 
 mcp = FastMCP("codegen-mcp", instructions="MCP server for the Codegen SDK. Use the tools and resources to setup codegen in your environment and to create and improve your Codegen Codemods.")
 
 # ----- RESOURCES -----
 
+
 @mcp.resource("system://agent_prompt", description="Provides all the information the agent needs to know about Codegen SDK", mime_type="text/plain")
 def get_docs() -> str:
     """Get the sdk doc url."""
@@ -31,25 +35,33 @@ def get_service_config() -> dict[str, Any]:
         "description": "The MCP server for assisting with creating/writing/improving codegen codemods.",
     }
 
+
 # ----- TOOLS -----
 
+
 @mcp.tool()
-def generate_codemod(title: Annotated[str, "The title of the codemod (hyphenated)"], task: Annotated[str, "The task to which the codemod should implement to solve"], codebase_path: Annotated[str, "The absolute path to the codebase directory"], ctx: Context) -> str:
+def generate_codemod(
+    title: Annotated[str, "The title of the codemod (hyphenated)"],
+    task: Annotated[str, "The task to which the codemod should implement to solve"],
+    codebase_path: Annotated[str, "The absolute path to the codebase directory"],
+    ctx: Context,
+) -> str:
     """Generate a codemod for the given task and codebase."""
     return f'''
     Use the codegen cli to generate a codemod. If you need to intall the cli the command to do so is `uv tool install codegen`. Once installed, run the following command to generate the codemod:
-          
+
     codegen create {title} -d "{task}"
     '''
 
+
 @mcp.tool()
 def improve_codemod(
     codemod_source: Annotated[str, "The source code of the codemod to improve"],
     task: Annotated[str, "The task to which the codemod should implement to solve"],
     concerns: Annotated[list[str], "A list of issues that were discovered with the current codemod that need to be considered in the next iteration"],
     context: Annotated[dict[str, Any], "Additional context for the codemod this can be a list of files that are related, additional information about the task, etc."],
     language: Annotated[ProgrammingLanguage, "The language of the codebase, i.e ALL CAPS PYTHON or TYPESCRIPT "],
-    ctx: Context
+    ctx: Context,
 ) -> str:
     """Improve the codemod."""
     try:
@@ -63,4 +75,4 @@ def improve_codemod(
 if __name__ == "__main__":
     # Initialize and run the server
     print("Starting codegen server...")
-    mcp.run(transport='stdio')
+    mcp.run(transport="stdio")