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 (optional): Default workspace directory (default: "./workspace")

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 (default: "./workspace")

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 (optional): Custom commit message
  • force_push (optional): Force push the branch (use with caution; default: false)

Example:

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

What it does:

  1. Stages all changes in the working directory (git add -A)
  2. Commits with the provided or auto-generated 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"
}

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.2.6.tar.gz (9.5 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.2.6-py3-none-any.whl (9.8 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: ai_coding_gym_mcp-0.2.6.tar.gz
  • Upload date:
  • Size: 9.5 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.2.6.tar.gz
Algorithm Hash digest
SHA256 0cf362b123353a38499201919925ea6cdc2e12d8e77965fcc678fdeab0ae7af0
MD5 ec1336407e1f2d9d66fc596365cae7ad
BLAKE2b-256 1e6d0aab7057df8d6523a01b1665f1188ac09e990e0b9cedd1631ecbadd0f73a

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ai_coding_gym_mcp-0.2.6-py3-none-any.whl
Algorithm Hash digest
SHA256 68e8b2207994d38ad7ba5bb1b86b431474ff87bd22fabe9cc240be4854052d87
MD5 000ca19b2fc244b6676a68c5f08e16cc
BLAKE2b-256 8a14d2e4275362062eb02c8fa1ea824887ac176af251db188ad472f93395ba65

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