ai-coding-gym-mcp 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

{ai_coding_gym_mcp-0.1.4.dist-info → ai_coding_gym_mcp-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-coding-gym-mcp
-Version: 0.1.4
+Version: 0.1.5
 Summary: MCP server for AI Coding Gym - fetch and submit coding challenges
 Home-page: https://github.com/yourusername/ai-coding-gym-mcp
 Author: AICodingGym Team

ai_coding_gym_mcp-0.1.5.dist-info/RECORD

@@ -0,0 +1,6 @@
+server.py,sha256=EpEFN-XHdE0x2GFMMaA9SYuTY6ulSFezI8i7A_g8gB0,13980
+ai_coding_gym_mcp-0.1.5.dist-info/METADATA,sha256=7M0rJOZdNzyWJ0nCEshptavra2hi41_ec2AnFa6IXPc,6526
+ai_coding_gym_mcp-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ai_coding_gym_mcp-0.1.5.dist-info/entry_points.txt,sha256=8RFDakwz8EhbAs4nFxFMAFuiH2NBJsZygTNgWIs-nV4,82
+ai_coding_gym_mcp-0.1.5.dist-info/top_level.txt,sha256=StKOSmRhvWS5IPcvhsDRbtxUTEofJgYFGOu5AAJdSWo,7
+ai_coding_gym_mcp-0.1.5.dist-info/RECORD,,

server.py

@@ -4,23 +4,20 @@ MCP Server for AI Coding Gym
 Provides /fetch and /submit tools for local problem solving workflow
 """
 
-import asyncio
-import json
 import os
 import subprocess
-import tempfile
 from datetime import datetime
 from pathlib import Path
 from typing import Any, Dict, Optional
 import requests
 
-from mcp.server import Server
-from mcp.types import Tool, TextContent, ErrorData
-import mcp.server.stdio
+from mcp.server.fastmcp import FastMCP
 
 
+# Initialize FastMCP server
+mcp = FastMCP("ai-coding-gym")
+
 # Global state
-app = Server("ai-coding-gym")
 credentials_store: Dict[str, Dict[str, Any]] = {}  # problem_id -> {deploy_key, repo_url, branch, etc}
 config_store: Dict[str, str] = {}  # Global config: user_id, server_url, api_key
 
@@ -28,78 +25,6 @@ config_store: Dict[str, str] = {}  # Global config: user_id, server_url, api_key
 DEFAULT_SERVER_URL = "http://ai-coding-gym.cis240515.projects.jetstream-cloud.org:4000"
 
 
-# Tool definitions
-CONFIGURE_TOOL = Tool(
-    name="configure",
-    description="Configure global settings for the MCP server. Generates an SSH key pair locally, "
-                "sends the public key to the server, and receives your repository name. "
-                "Set your user_id once. "
-                "These values will be used as defaults for all fetch and submit operations.",
-    inputSchema={
-        "type": "object",
-        "properties": {
-            "user_id": {
-                "type": "string",
-                "description": "Your user ID for authentication with the backend server"
-            },
-            "workspace_dir": {
-                "type": "string",
-                "description": "Default workspace directory for cloning repositories (optional)"
-            }
-        },
-        "required": ["user_id"]
-    }
-)
-
-FETCH_TOOL = Tool(
-    name="fetch",
-    description="Fetch a coding problem from the server and clone the repository locally. "
-                "Uses your SSH key from /configure to access the repository.",
-    inputSchema={
-        "type": "object",
-        "properties": {
-            "problem_id": {
-                "type": "string",
-                "description": "The unique identifier for the problem (e.g., 'django__django-10097')"
-            },
-            "user_id": {
-                "type": "string",
-                "description": "Your user ID (optional if configured globally via /configure)"
-            },
-            "workspace_dir": {
-                "type": "string",
-                "description": "Local directory to clone the repository into (optional)"
-            }
-        },
-        "required": ["problem_id"]
-    }
-)
-
-SUBMIT_TOOL = Tool(
-    name="submit",
-    description="Submit your solution by committing all changes and pushing to the remote repository. "
-                "Notifies the backend server that a submission has been made.",
-    inputSchema={
-        "type": "object",
-        "properties": {
-            "problem_id": {
-                "type": "string",
-                "description": "The unique identifier for the problem"
-            },
-            "user_id": {
-                "type": "string",
-                "description": "Your user ID (optional if configured globally via /configure)"
-            },
-            "commit_message": {
-                "type": "string",
-                "description": "Custom commit message (optional)"
-            }
-        },
-        "required": ["problem_id"]
-    }
-)
-
-
 # Helper functions for Git SSH key management
 def infer_user_id_from_keys() -> Optional[str]:
     """
@@ -190,11 +115,23 @@ def run_git_command(cmd: str, cwd: str, key_path: Optional[Path] = None) -> subp
 
 
 # Tool implementations
-async def configure(
+@mcp.tool()
+def configure(
     user_id: str,
-    workspace_dir: Optional[str] = None
+    workspace_dir: str = None
 ) -> str:
-    """Configure global settings for the MCP server."""
+    """
+    Configure global settings for the MCP server. Generates an SSH key pair locally, 
+    sends the public key to the server, and receives your repository name. 
+    Set your user_id once. These values will be used as defaults for all fetch and submit operations.
+    
+    :param user_id: Your user ID for authentication with the backend server
+    :type user_id: str
+    :param workspace_dir: Default workspace directory for cloning repositories (optional)
+    :type workspace_dir: str
+    :return: Configuration status message
+    :rtype: str
+    """
     server_url = DEFAULT_SERVER_URL
     try:
         # Generate SSH key pair
@@ -240,13 +177,24 @@ You can now use /fetch and /submit without passing these parameters.
         return f"Error: Configuration failed: {str(e)}"
 
 
-async def fetch_problem(
+@mcp.tool()
+def fetch_problem(
     problem_id: str,
-    user_id: Optional[str] = None,
-    workspace_dir: Optional[str] = None
+    user_id: str = None,
+    workspace_dir: str = None
 ) -> str:
     """
-    Fetch problem by cloning only the specific branch from the forked repository.
+    Fetch a coding problem from the server and clone the repository locally. 
+    Uses your SSH key from /configure to access the repository.
+    
+    :param problem_id: The unique identifier for the problem (e.g., 'django__django-10097')
+    :type problem_id: str
+    :param user_id: Your user ID (optional if configured globally via /configure)
+    :type user_id: str
+    :param workspace_dir: Local directory to clone the repository into (optional)
+    :type workspace_dir: str
+    :return: Fetch status message
+    :rtype: str
     """
     try:
         # Use configured values if not provided
@@ -333,13 +281,24 @@ You can now start working on the solution!
         return f"Error: Unexpected error: {str(e)}"
 
 
-async def submit_solution(
+@mcp.tool()
+def submit_solution(
     problem_id: str,
-    user_id: Optional[str] = None,
-    commit_message: Optional[str] = None
+    user_id: str = None,
+    commit_message: str = None
 ) -> str:
     """
-    Submit solution by committing and pushing changes, then notify backend.
+    Submit your solution by committing all changes and pushing to the remote repository. 
+    Notifies the backend server that a submission has been made.
+    
+    :param problem_id: The unique identifier for the problem
+    :type problem_id: str
+    :param user_id: Your user ID (optional if configured globally via /configure)
+    :type user_id: str
+    :param commit_message: Custom commit message (optional)
+    :type commit_message: str
+    :return: Submission status message
+    :rtype: str
     """
     try:
         # Use configured values if not provided
@@ -433,58 +392,6 @@ Your solution has been submitted for evaluation!
         return f"Error: Unexpected error: {str(e)}"
 
 
-# MCP Server handlers
-@app.list_tools()
-async def list_tools() -> list[Tool]:
-    """List available tools."""
-    return [CONFIGURE_TOOL, FETCH_TOOL, SUBMIT_TOOL]
-
-
-@app.call_tool()
-async def call_tool(name: str, arguments: Any) -> list[TextContent]:
-    """Handle tool calls."""
-    try:
-        if name == "configure":
-            result = await configure(
-                user_id=arguments["user_id"],
-                workspace_dir=arguments.get("workspace_dir")
-            )
-            return [TextContent(type="text", text=result)]
-        
-        elif name == "fetch":
-            result = await fetch_problem(
-                problem_id=arguments["problem_id"],
-                user_id=arguments.get("user_id"),
-                workspace_dir=arguments.get("workspace_dir")
-            )
-            return [TextContent(type="text", text=result)]
-        
-        elif name == "submit":
-            result = await submit_solution(
-                problem_id=arguments["problem_id"],
-                user_id=arguments.get("user_id"),
-                commit_message=arguments.get("commit_message")
-            )
-            return [TextContent(type="text", text=result)]
-        
-        else:
-            raise ValueError(f"Unknown tool: {name}")
-    
-    except KeyError as e:
-        return [TextContent(type="text", text=f"Error: Missing required parameter: {str(e)}")]
-    except Exception as e:
-        return [TextContent(type="text", text=f"Error: Error executing tool: {str(e)}")]
-
-
-async def main():
-    """Run the MCP server."""
-    async with mcp.server.stdio.stdio_server() as (read_stream, write_stream):
-        await app.run(
-            read_stream,
-            write_stream,
-            app.create_initialization_options()
-        )
-
-
 if __name__ == "__main__":
-    asyncio.run(main())
+    # Initialize and run the server
+    mcp.run(transport='stdio')

ai_coding_gym_mcp-0.1.4.dist-info/RECORD

@@ -1,6 +0,0 @@
-server.py,sha256=CPUOvS60mwvBa7iSFGkUhFvfi2sB8DAJr1eHqB1MJYQ,16939
-ai_coding_gym_mcp-0.1.4.dist-info/METADATA,sha256=36X6NU8ecK7txC6d8KxQkkQExz-0t0_6RszSgoayJgw,6526
-ai_coding_gym_mcp-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ai_coding_gym_mcp-0.1.4.dist-info/entry_points.txt,sha256=8RFDakwz8EhbAs4nFxFMAFuiH2NBJsZygTNgWIs-nV4,82
-ai_coding_gym_mcp-0.1.4.dist-info/top_level.txt,sha256=StKOSmRhvWS5IPcvhsDRbtxUTEofJgYFGOu5AAJdSWo,7
-ai_coding_gym_mcp-0.1.4.dist-info/RECORD,,