Merge branch 'develop' into rpatel/linear-webhook-example

Author: rushilpatel0

codegen-examples/examples/codegen-mcp-server/README.md

@@ -0,0 +1,67 @@
+# Codegen MCP Server
+
+<p align="center">
+  <a href="https://docs.codegen.com">
+    <img src="https://i.imgur.com/6RF9W0z.jpeg" />
+  </a>
+</p>
+
+<h2 align="center">
+  An MCP server implementation that integrates the codegen sdk.
+
+</h2>
+
+<div align="center">
+
+[![Documentation](https://img.shields.io/badge/Docs-docs.codegen.com-purple?style=flat-square)](https://docs.codegen.com)
+[![License](https://img.shields.io/badge/Code%20License-Apache%202.0-gray?&color=gray)](https://github.com/codegen-sh/codegen-sdk/tree/develop?tab=Apache-2.0-1-ov-file)
+
+</div>
+
+This example demonstrates how to run a Model Control Protocol (MCP) server that integrates with Codegen. The server provides:
+
+1. A standardized interface for model inference
+1. Integration with Codegen's core functionality, parsing codebases and executing codemods
+1. Support for various LLM providers through the MCP protocol
+
+## Quick Start
+
+### Prerequisites
+
+- [uv](https://docs.astral.sh/uv/getting-started/installation/)
+- [python](https://www.python.org/downloads/) 3.12+
+
+### Direct Execution
+
+1. No installation is necessary, with the following command. Run this command directly or add it your `.json` mcp config file.
+
+```bash
+ uvx --from 'git+https://github.com/codegen-sh/codegen-sdk.git#egg=codegen-mcp-server&subdirectory=codegen-examples/examples/codegen-mcp-server' codegen-mcp-server
+```
+
+### Example MCP Config
+
+Here is an example mcp config that can be used with Cline or Claude desktop to integrate this MCP server
+
+```json
+{
+  "mcpServers": {
+    "codegen-mcp": {
+      "command": "uvx",
+      "args": [
+        "--from",
+        "git+https://github.com/codegen-sh/codegen-sdk.git#egg=codegen-mcp-server&subdirectory=codegen-examples/examples/codegen-mcp-server",
+        "codegen-mcp-server"
+      ],
+      "disabled": false,
+      "autoApprove": []
+    }
+  }
+}
+```
+
+### MCP Tools:
+
+- `parse_codebase`: Parses a codebase located at the provided path.
+- `check_parse_status`: Provides the current parsing status for the provided codebase.
+- `execute_codemod`: Executes a codemod script on a parsed codebase. This is where the codegen sdk leveraged to run simple or sophisticated codemods on the codebase.

codegen-examples/examples/codegen-mcp-server/llms-install.md

@@ -0,0 +1,17 @@
+Please read two things before trying to install this codegen MCP server:
+
+1. [Readme](https://raw.githubusercontent.com/codegen-sh/codegen-sdk/refs/heads/develop/codegen-examples/examples/codegen-mcp-server/README.md)
+
+- This contains the instructions for setting up the mcp server
+
+2. [Documentation Index for Agents](https://docs.codegen.com/llms.txt)
+
+- This contains a list of urls to the documentation for topics related to creating codegen codemods
+
+Please use a CURL command to retrieve the contents of the readme and documentation and read both before starting to install the server.
+
+1. curl -sL "https://raw.githubusercontent.com/codegen-sh/codegen-sdk/refs/heads/develop/README.md" | cat
+
+1. curl -sL "https://docs.codegen.com/llms.txt" | cat
+
+This should give you all the context you need to successfully install and utilize the server.

codegen-examples/examples/codegen-mcp-server/pyproject.toml

@@ -0,0 +1,20 @@
+[project]
+name = "codegen-mcp-server"
+version = "0.1.0"
+description = "MCP server implementation adding tools for the Codegen SDK"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+  "codegen>=0.14.5",
+  "uvicorn>=0.27.1",
+  "fastapi>=0.110.0",
+  "pydantic>=2.6.3",
+  "rich>=13.7.1",
+  "click>=8.1.7",
+  "mcp[cli]>=1.2.1",
+]
+
+[tool.poetry.scripts]
+codegen-mcp-server = "server:main"
+[project.scripts]
+codegen-mcp-server = "server:main"

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

@@ -0,0 +1,105 @@
+import asyncio
+from dataclasses import dataclass, field
+from typing import Annotated, Optional, Dict, Any, List
+from mcp.server.fastmcp import FastMCP
+from codegen import Codebase
+
+
+@dataclass
+class CodebaseState:
+    """Class to manage codebase state and parsing."""
+
+    parse_task: Optional[asyncio.Future] = None
+    parsed_codebase: Optional[Codebase] = None
+    log_buffer: List[str] = field(default_factory=list)
+
+    def parse(self, path: str) -> Codebase:
+        """Parse the codebase at the given path."""
+        codebase = Codebase(path)
+        self.parsed_codebase = codebase
+        return codebase
+
+    def reset(self) -> None:
+        """Reset the state."""
+        self.log_buffer.clear()
+
+
+# Initialize FastMCP server
+mcp = FastMCP(
+    "codegen-mcp-server",
+    instructions="""This server provides tools to parse and modify a codebase using codemods.
+    It can initiate parsing, check parsing status, and execute codemods.""",
+)
+
+# Initialize state
+state = CodebaseState()
+
+
+def capture_output(*args, **kwargs) -> None:
+    """Capture and log output messages."""
+    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:
+        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 is already being parsed.", "status": "error"}
+
+
+@mcp.tool(name="check_parse_status", description="Check if codebase parsing has completed")
+async def check_parse_status() -> Dict[str, str]:
+    if not state.parse_task:
+        return {"message": "No codebase provided to parse."}
+    if state.parse_task.done():
+        return {"message": "Codebase parsing completed."}
+    return {"message": "Codebase parsing in progress."}
+
+
+@mcp.tool(name="execute_codemod", description="Execute a codemod on the codebase")
+async def execute_codemod(codemod: Annotated[str, "The python codemod code to execute on the codebase"]) -> Dict[str, Any]:
+    if not state.parse_task or not state.parse_task.done():
+        return {"error": "Codebase is not ready for codemod execution."}
+
+    try:
+        await state.parse_task
+        if state.parsed_codebase is None:
+            return {"error": "Codebase path is not set."}
+        else:
+            # TODO: Implement proper sandboxing for code execution
+            context = {
+                "codebase": state.parsed_codebase,
+                "print": capture_output,
+            }
+            exec(codemod, context)
+
+        logs = "\n".join(state.log_buffer)
+        state.reset()
+        return {"message": "Codemod executed, view the logs for any output and your source code for any resulting updates.", "logs": logs}
+    except Exception as e:
+        return {"error": f"Error executing codemod: {str(e)}", "details": {"type": type(e).__name__, "message": str(e)}}
+
+
+def main():
+    print("starting codegen-mcp-server")
+    run = mcp.run_stdio_async()
+    print("codegen-mcp-server started")
+    asyncio.get_event_loop().run_until_complete(run)
+
+
+if __name__ == "__main__":
+    main()