fix: separate the mcp server running function

Author: cognitive-glitch

main.py

@@ -136,5 +136,12 @@ def run_ast_grep_yaml(yaml: str, project_folder: str) -> List[dict[str, Any]]:
         print("Command not found")
         return []
 
+def run_mcp_server():
+    """
+    Run the MCP server.
+    This function is used to start the MCP server when this script is run directly.
+    """
+    mcp.run(transport="stdio")
+
 if __name__ == "__main__":
-    mcp.run(transport = "stdio")
+    run_mcp_server()

pyproject.toml

@@ -10,4 +10,4 @@ dependencies = [
 ]
 
 [project.scripts]
-ast-grep-server = "main:mcp.run"
+ast-grep-server = "main:run_mcp_server"