Skip to main content

MCP server for AI Coding Gym - fetch and submit coding challenges

Project description

AI Coding Gym - MCP Server

Local MCP server for interacting with the AI Coding Gym platform. Provides tools to fetch coding problems and submit solutions.

Features

  • /fetch: Fetch a coding problem and clone the repository to your local machine
  • /submit: Submit your solution by committing and pushing changes

Quick Start

Installation

Option 1: Install from PyPI

pip install ai-coding-gym-mcp

Option 2: Install from GitHub

pip install git+https://github.com/yourusername/ai-coding-gym-mcp.git

Option 3: Install from source

git clone https://github.com/yourusername/ai-coding-gym-mcp.git
cd ai-coding-gym-mcp
pip install -e .

Configure Claude Desktop

Add to your Claude Desktop config (~/Library/Application Support/Claude/claude_desktop_config.json on Mac):

{
  "mcpServers": {
    "ai-coding-gym": {
      "command": "ai-coding-gym-mcp"
    }
  }
}

Configure VS Code (Copilot with MCP)

Add to your VS Code settings (.vscode/settings.json or User Settings):

{
  "github.copilot.chat.codeGeneration.instructions": [
    {
      "text": "Use AI Coding Gym MCP tools for problem solving"
    }
  ],
  "mcp.servers": {
    "ai-coding-gym": {
      "command": "python",
      "args": ["-m", "server"],
      "cwd": "/path/to/ai-coding-gym-mcp"
    }
  }
}

Or if installed via pip, use the executable directly:

{
  "mcp.servers": {
    "ai-coding-gym": {
      "command": "ai-coding-gym-mcp"
    }
  }
}

Usage

Running the MCP Server

The server uses stdio for communication with MCP clients:

python server.py

Or configure it in your MCP client settings (e.g., Claude Desktop).

Tool: /configure

Configure the MCP server with your user ID. This generates an SSH key pair and registers it with the server.

Parameters:

  • user_id (required): Your user ID for authentication
  • workspace_dir (required): Default workspace directory for cloning repositories

Example:

{
  "user_id": "user_123",
  "workspace_dir": "./workspace"
}

What it does:

  1. Generates an SSH key pair locally (stored in ~/.mcp-keys/)
  2. Sends the public key to the server
  3. Receives your repository name
  4. Stores configuration for future use

Tool: /fetch

Fetches a problem from the backend and clones the repository locally. This tool is single-purpose: it does not install dependencies, run tests, or perform any other follow-up actions after fetching.

Parameters:

  • problem_id (required): Problem identifier (e.g., "django__django-10097")
  • user_id (optional): Your user ID (uses configured value if not provided)
  • workspace_dir (optional): Local workspace directory (defaults to configured workspace_dir)

Example:

{
  "problem_id": "django__django-10097"
}

What it does:

  1. Uses your SSH key from /configure to access the repository
  2. Clones only the specific problem branch (shallow clone)
  3. Sets up the workspace at workspace/{problem_id}/
  4. Does not run installs, tests, or any other post-fetch steps

Tool: /submit

Submits your solution by committing changes and pushing to the remote repository.

Parameters:

  • problem_id (required): Problem identifier
  • user_id (optional): Your user ID (uses configured value if not provided)
  • commit_message (required): Custom commit message
  • workspace_dir (optional): Local workspace directory where the repo was cloned (defaults to configured workspace_dir)
  • force_push (optional): Force push the branch (use with caution; default: false)

Example:

{
  "problem_id": "django__django-10097",
  "commit_message": "Fixed the authentication bug",
  "workspace_dir": "./workspace",
  "force_push": false
}

What it does:

  1. Stages all changes in the working directory (git add -A)
  2. Commits with the provided message
  3. Pushes to the remote branch using deployment key (optionally force-pushes)
  4. Notifies the backend server about the submission

Tool: /reset

Resets a problem repository back to the original setup commit.

Parameters:

  • problem_id (required): Problem identifier
  • user_id (optional): Your user ID (uses configured value if not provided)

Example:

{
  "problem_id": "django__django-10097"
}

What it does:

  1. Finds the original setup commit with a message that starts with Setup SWE-bench instance:
  2. Resets the repo to that commit (git reset --hard)
  3. Removes untracked files (git clean -fd)

Backend API Endpoints

The MCP server connects to the hardcoded AI Coding Gym server and uses the following endpoints:

POST /api/configure

Request:

{
  "user_id": "user_123",
  "public_key": "ssh-rsa AAAAB3..."
}

Response:

{
  "repo_name": "user_123-swebench",
  "repo_url": "git@github.com:AICodingGym/submission-user_123.git"
}

POST /api/submit

Request:

{
  "problem_id": "django__django-10097",
  "user_id": "user_123",
  "commit_hash": "abc123def456...",
  "branch": "django__django-10097-user_123",
  "commit_message": "Fixed the authentication bug",
  "timestamp": "2026-02-03T10:30:00"
}

Response:

{
  "status": "success",
  "message": "Submission received"
}

Security

  • User SSH keys are stored in ~/.mcp-keys/ with 600 permissions
  • Keys are generated locally and public key is shared with the server
  • SSH host key checking is disabled for convenience (consider enabling in production)
  • Configuration is cached in memory during the MCP server session

Troubleshooting

"No credentials found for problem_id"

  • Run /configure first to set up your credentials
  • Then run /fetch to download the problem

"Git clone/push failed"

  • Check network connectivity
  • Verify deployment key is valid
  • Ensure SSH agent isn't interfering

"Directory already exists"

  • Remove the existing directory or use a different workspace location

Development

The server uses:

  • mcp: Model Context Protocol SDK
  • requests: HTTP client for backend API calls
  • subprocess: Git command execution with SSH key management

Local Development

# Clone the repository
git clone https://github.com/yourusername/ai-coding-gym-mcp.git
cd ai-coding-gym-mcp

# Install in development mode
pip install -e .

# Run tests (if available)
pytest

# Test the server locally
python server.py

Contributing

Contributions welcome! Please:

  1. Fork the repository
  2. Create a feature branch
  3. Make your changes
  4. Submit a pull request

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ai_coding_gym_mcp-0.3.0.tar.gz (10.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ai_coding_gym_mcp-0.3.0-py3-none-any.whl (10.4 kB view details)

Uploaded Python 3

File details

Details for the file ai_coding_gym_mcp-0.3.0.tar.gz.

File metadata

  • Download URL: ai_coding_gym_mcp-0.3.0.tar.gz
  • Upload date:
  • Size: 10.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.12

File hashes

Hashes for ai_coding_gym_mcp-0.3.0.tar.gz
Algorithm Hash digest
SHA256 13ad22eef202ddcd9afc8171b6d02e5c1d380ca24a41caf246b1a8d05640b436
MD5 be402df7db0bd266a2d574a1945b0b0a
BLAKE2b-256 5ab287ae05577d2e36d4be77d03cc0fd713dd2a85b07d4e603801382ae87e74e

See more details on using hashes here.

File details

Details for the file ai_coding_gym_mcp-0.3.0-py3-none-any.whl.

File metadata

File hashes

Hashes for ai_coding_gym_mcp-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 bf2fa555ffd2207866d5e1c45ba5c99e5c239a89f39c7a19c86184cce66e25f1
MD5 7b63c583c4fa0185cd3d5a47e712118d
BLAKE2b-256 9d903d6125150f0da028cea76868161bd77eb9cb0bfc0f30de1537243809517c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page