greed-compute-mcp 0.1.0

package/README.md

@@ -0,0 +1,73 @@
+# greed-compute-mcp
+
+MCP server that gives LLMs stateful Python execution. Connect it to Claude, GPT, or any MCP-compatible agent and they can create sessions, run code, checkpoint state, fork workers, and run parallel MapReduce — all through natural tool calls.
+
+## Setup
+
+```bash
+npm install -g greed-compute-mcp
+```
+
+Set your API key:
+
+```bash
+export GREED_API_KEY=gc_your_key_here
+```
+
+## Usage with Claude Desktop
+
+Add to your `claude_desktop_config.json`:
+
+```json
+{
+  "mcpServers": {
+    "greed-compute": {
+      "command": "greed-compute-mcp",
+      "env": {
+        "GREED_API_KEY": "gc_your_key_here"
+      }
+    }
+  }
+}
+```
+
+## Usage with Claude Code
+
+```bash
+claude mcp add greed-compute -- greed-compute-mcp
+```
+
+Set the env var in your shell or `.env`.
+
+## Available Tools
+
+| Tool | What it does |
+|------|-------------|
+| `create_session` | Spin up a Python interpreter (blank, data-science, ML, or scraping template) |
+| `execute_code` | Run Python in a session. State persists between calls |
+| `session_status` | Check TTL and activity |
+| `terminate_session` | Kill a session |
+| `install_packages` | pip install into a running session |
+| `create_checkpoint` | Snapshot all interpreter state |
+| `restore_checkpoint` | Load a checkpoint into a session |
+| `list_checkpoints` | See all saved checkpoints |
+| `delete_checkpoint` | Free checkpoint storage |
+| `create_swarm` | Parallel MapReduce across N workers |
+| `get_swarm` | Poll swarm progress |
+| `create_workspace` | Persistent shared environment |
+| `execute_in_workspace` | Run code in shared workspace |
+| `list_workspaces` | See your workspaces |
+| `upload_file` | Push files into a session |
+| `download_file` | Pull files out of a session |
+| `get_usage` | Check plan limits and usage |
+
+## Environment Variables
+
+| Variable | Required | Default |
+|----------|----------|---------|
+| `GREED_API_KEY` | Yes | — |
+| `GREED_COMPUTE_URL` | No | `https://compute.deep-ml.com` |
+
+## License
+
+MIT

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/index.js

@@ -0,0 +1,161 @@
+#!/usr/bin/env node
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+const BASE = process.env.GREED_COMPUTE_URL || 'https://compute.deep-ml.com';
+const API_KEY = process.env.GREED_API_KEY || '';
+if (!API_KEY) {
+    console.error('greed-compute-mcp: set GREED_API_KEY environment variable');
+    process.exit(1);
+}
+// ── HTTP helper ──────────────────────────────────────────────────────────────
+async function api(method, path, body) {
+    const res = await fetch(`${BASE}/v1${path}`, {
+        method,
+        headers: {
+            'Content-Type': 'application/json',
+            'X-API-Key': API_KEY,
+        },
+        body: body ? JSON.stringify(body) : undefined,
+    });
+    const text = await res.text();
+    try {
+        return JSON.parse(text);
+    }
+    catch {
+        return text;
+    }
+}
+function text(data) {
+    return { content: [{ type: 'text', text: typeof data === 'string' ? data : JSON.stringify(data, null, 2) }] };
+}
+// ── Server ───────────────────────────────────────────────────────────────────
+const server = new McpServer({
+    name: 'greed-compute',
+    version: '0.1.0',
+});
+// ── Session tools ────────────────────────────────────────────────────────────
+server.tool('create_session', 'Create a stateful Python session. State persists between execute calls.', {
+    template: z.enum(['blank', 'data-science', 'machine-learning', 'web-scraping']).default('blank').describe('Pre-loaded library template'),
+    ttl_seconds: z.number().optional().describe('Session lifetime in seconds (default 120)'),
+}, async ({ template, ttl_seconds }) => {
+    const result = await api('POST', '/session/create', { template, ttl_seconds });
+    return text(result);
+});
+server.tool('execute_code', 'Run Python code in a session. Variables persist between calls. Returns stdout, result, errors, and plots.', {
+    session_id: z.string().describe('Session ID from create_session'),
+    code: z.string().describe('Python code to execute'),
+}, async ({ session_id, code }) => {
+    const result = await api('POST', `/session/${session_id}/execute`, { code });
+    return text(result);
+});
+server.tool('session_status', 'Check if a session is alive and how much TTL remains.', {
+    session_id: z.string().describe('Session ID'),
+}, async ({ session_id }) => {
+    const result = await api('GET', `/session/${session_id}/status`);
+    return text(result);
+});
+server.tool('terminate_session', 'Kill a session and free its resources.', {
+    session_id: z.string().describe('Session ID'),
+}, async ({ session_id }) => {
+    const result = await api('DELETE', `/session/${session_id}`);
+    return text(result);
+});
+server.tool('install_packages', 'Install pip packages into a running session without restarting it.', {
+    session_id: z.string().describe('Session ID'),
+    packages: z.array(z.string()).describe('Package names to install'),
+}, async ({ session_id, packages }) => {
+    const result = await api('POST', `/session/${session_id}/install`, { packages });
+    return text(result);
+});
+// ── Checkpoint tools ─────────────────────────────────────────────────────────
+server.tool('create_checkpoint', 'Snapshot the entire Python interpreter state (all variables, imports, models). Can be restored later or forked into multiple sessions.', {
+    session_id: z.string().describe('Session ID to checkpoint'),
+    name: z.string().optional().describe('Human-readable name for the checkpoint'),
+}, async ({ session_id, name }) => {
+    const result = await api('POST', `/session/${session_id}/checkpoint`, { name });
+    return text(result);
+});
+server.tool('restore_checkpoint', 'Load a previously saved checkpoint into a running session. All variables are restored.', {
+    session_id: z.string().describe('Session to restore into'),
+    checkpoint_id: z.string().describe('Checkpoint ID to restore'),
+}, async ({ session_id, checkpoint_id }) => {
+    const result = await api('POST', `/session/${session_id}/restore/${checkpoint_id}`);
+    return text(result);
+});
+server.tool('list_checkpoints', 'List all saved checkpoints for your API key.', {}, async () => {
+    const result = await api('GET', '/checkpoints');
+    return text(result);
+});
+server.tool('delete_checkpoint', 'Delete a checkpoint and free the storage it uses.', {
+    checkpoint_id: z.string().describe('Checkpoint ID to delete'),
+}, async ({ checkpoint_id }) => {
+    const result = await api('DELETE', `/checkpoints/${checkpoint_id}`);
+    return text(result);
+});
+// ── Swarm tools ──────────────────────────────────────────────────────────────
+server.tool('create_swarm', 'Run parallel MapReduce across N workers. Setup code runs once and is cloned to all workers. Each worker processes one data partition.', {
+    template_code: z.string().optional().describe('Setup code that runs once and gets cloned to all workers (imports, model loading)'),
+    map_fn: z.string().describe('Code each worker runs. Has access to `partition` variable.'),
+    data: z.array(z.any()).describe('Array of items, one per worker. Each item is injected as `partition`.'),
+    reduce_fn: z.string().optional().describe('Code to combine results. Has access to `results` list.'),
+    webhook_url: z.string().optional().describe('URL to POST results when swarm finishes'),
+}, async ({ template_code, map_fn, data, reduce_fn, webhook_url }) => {
+    const result = await api('POST', '/swarm', { template_code, map_fn, data, reduce_fn, webhook_url });
+    return text(result);
+});
+server.tool('get_swarm', 'Check swarm progress and get results.', {
+    swarm_id: z.string().describe('Swarm ID'),
+}, async ({ swarm_id }) => {
+    const result = await api('GET', `/swarm/${swarm_id}`);
+    return text(result);
+});
+// ── Workspace tools ──────────────────────────────────────────────────────────
+server.tool('create_workspace', 'Create a persistent shared Python environment. Multiple API keys can execute in the same workspace. State auto-saves.', {
+    name: z.string().describe('Workspace name'),
+}, async ({ name }) => {
+    const result = await api('POST', '/workspaces', { name });
+    return text(result);
+});
+server.tool('execute_in_workspace', 'Run code in a shared workspace. All members see the same state.', {
+    workspace_id: z.string().describe('Workspace ID'),
+    code: z.string().describe('Python code to execute'),
+}, async ({ workspace_id, code }) => {
+    const result = await api('POST', `/workspaces/${workspace_id}/execute`, { code });
+    return text(result);
+});
+server.tool('list_workspaces', 'List all workspaces you own or have access to.', {}, async () => {
+    const result = await api('GET', '/workspaces');
+    return text(result);
+});
+// ── File tools ───────────────────────────────────────────────────────────────
+server.tool('upload_file', 'Upload a file to a session workspace. Your code can then read it by filename.', {
+    session_id: z.string().describe('Session ID'),
+    filename: z.string().describe('Filename in the workspace'),
+    content_base64: z.string().describe('File content, base64 encoded'),
+}, async ({ session_id, filename, content_base64 }) => {
+    const result = await api('POST', `/session/${session_id}/files`, { filename, content: content_base64 });
+    return text(result);
+});
+server.tool('download_file', 'Download a file from a session workspace. Returns base64 encoded content.', {
+    session_id: z.string().describe('Session ID'),
+    filename: z.string().describe('Filename to download'),
+}, async ({ session_id, filename }) => {
+    const result = await api('GET', `/session/${session_id}/output/${filename}`);
+    return text(result);
+});
+// ── Usage tool ───────────────────────────────────────────────────────────────
+server.tool('get_usage', 'Check your current API usage, plan limits, and billing status.', {}, async () => {
+    const result = await api('GET', '/usage');
+    return text(result);
+});
+// ── Start ────────────────────────────────────────────────────────────────────
+async function main() {
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error('greed-compute MCP server running on stdio');
+}
+main().catch((err) => {
+    console.error('Fatal:', err);
+    process.exit(1);
+});

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "greed-compute-mcp",
+  "version": "0.1.0",
+  "description": "MCP server for greed-compute — gives LLMs stateful Python execution",
+  "bin": {
+    "greed-compute-mcp": "./dist/index.js"
+  },
+  "main": "./dist/index.js",
+  "type": "module",
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "start": "node dist/index.js"
+  },
+  "keywords": ["mcp", "greed-compute", "python", "ai-agents", "code-execution"],
+  "author": "Aditya Khalkar",
+  "license": "MIT",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.1"
+  },
+  "devDependencies": {
+    "typescript": "^5.7.0",
+    "@types/node": "^22.0.0"
+  }
+}

package/src/index.ts

@@ -0,0 +1,275 @@
+#!/usr/bin/env node
+
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
+import { z } from 'zod'
+
+const BASE = process.env.GREED_COMPUTE_URL || 'https://compute.deep-ml.com'
+const API_KEY = process.env.GREED_API_KEY || ''
+
+if (!API_KEY) {
+  console.error('greed-compute-mcp: set GREED_API_KEY environment variable')
+  process.exit(1)
+}
+
+// ── HTTP helper ──────────────────────────────────────────────────────────────
+
+async function api(method: string, path: string, body?: unknown): Promise<unknown> {
+  const res = await fetch(`${BASE}/v1${path}`, {
+    method,
+    headers: {
+      'Content-Type': 'application/json',
+      'X-API-Key': API_KEY,
+    },
+    body: body ? JSON.stringify(body) : undefined,
+  })
+  const text = await res.text()
+  try { return JSON.parse(text) } catch { return text }
+}
+
+function text(data: unknown): { content: { type: 'text'; text: string }[] } {
+  return { content: [{ type: 'text' as const, text: typeof data === 'string' ? data : JSON.stringify(data, null, 2) }] }
+}
+
+// ── Server ───────────────────────────────────────────────────────────────────
+
+const server = new McpServer({
+  name: 'greed-compute',
+  version: '0.1.0',
+})
+
+// ── Session tools ────────────────────────────────────────────────────────────
+
+server.tool(
+  'create_session',
+  'Create a stateful Python session. State persists between execute calls.',
+  {
+    template: z.enum(['blank', 'data-science', 'machine-learning', 'web-scraping']).default('blank').describe('Pre-loaded library template'),
+    ttl_seconds: z.number().optional().describe('Session lifetime in seconds (default 120)'),
+  },
+  async ({ template, ttl_seconds }) => {
+    const result = await api('POST', '/session/create', { template, ttl_seconds })
+    return text(result)
+  }
+)
+
+server.tool(
+  'execute_code',
+  'Run Python code in a session. Variables persist between calls. Returns stdout, result, errors, and plots.',
+  {
+    session_id: z.string().describe('Session ID from create_session'),
+    code: z.string().describe('Python code to execute'),
+  },
+  async ({ session_id, code }) => {
+    const result = await api('POST', `/session/${session_id}/execute`, { code })
+    return text(result)
+  }
+)
+
+server.tool(
+  'session_status',
+  'Check if a session is alive and how much TTL remains.',
+  {
+    session_id: z.string().describe('Session ID'),
+  },
+  async ({ session_id }) => {
+    const result = await api('GET', `/session/${session_id}/status`)
+    return text(result)
+  }
+)
+
+server.tool(
+  'terminate_session',
+  'Kill a session and free its resources.',
+  {
+    session_id: z.string().describe('Session ID'),
+  },
+  async ({ session_id }) => {
+    const result = await api('DELETE', `/session/${session_id}`)
+    return text(result)
+  }
+)
+
+server.tool(
+  'install_packages',
+  'Install pip packages into a running session without restarting it.',
+  {
+    session_id: z.string().describe('Session ID'),
+    packages: z.array(z.string()).describe('Package names to install'),
+  },
+  async ({ session_id, packages }) => {
+    const result = await api('POST', `/session/${session_id}/install`, { packages })
+    return text(result)
+  }
+)
+
+// ── Checkpoint tools ─────────────────────────────────────────────────────────
+
+server.tool(
+  'create_checkpoint',
+  'Snapshot the entire Python interpreter state (all variables, imports, models). Can be restored later or forked into multiple sessions.',
+  {
+    session_id: z.string().describe('Session ID to checkpoint'),
+    name: z.string().optional().describe('Human-readable name for the checkpoint'),
+  },
+  async ({ session_id, name }) => {
+    const result = await api('POST', `/session/${session_id}/checkpoint`, { name })
+    return text(result)
+  }
+)
+
+server.tool(
+  'restore_checkpoint',
+  'Load a previously saved checkpoint into a running session. All variables are restored.',
+  {
+    session_id: z.string().describe('Session to restore into'),
+    checkpoint_id: z.string().describe('Checkpoint ID to restore'),
+  },
+  async ({ session_id, checkpoint_id }) => {
+    const result = await api('POST', `/session/${session_id}/restore/${checkpoint_id}`)
+    return text(result)
+  }
+)
+
+server.tool(
+  'list_checkpoints',
+  'List all saved checkpoints for your API key.',
+  {},
+  async () => {
+    const result = await api('GET', '/checkpoints')
+    return text(result)
+  }
+)
+
+server.tool(
+  'delete_checkpoint',
+  'Delete a checkpoint and free the storage it uses.',
+  {
+    checkpoint_id: z.string().describe('Checkpoint ID to delete'),
+  },
+  async ({ checkpoint_id }) => {
+    const result = await api('DELETE', `/checkpoints/${checkpoint_id}`)
+    return text(result)
+  }
+)
+
+// ── Swarm tools ──────────────────────────────────────────────────────────────
+
+server.tool(
+  'create_swarm',
+  'Run parallel MapReduce across N workers. Setup code runs once and is cloned to all workers. Each worker processes one data partition.',
+  {
+    template_code: z.string().optional().describe('Setup code that runs once and gets cloned to all workers (imports, model loading)'),
+    map_fn: z.string().describe('Code each worker runs. Has access to `partition` variable.'),
+    data: z.array(z.any()).describe('Array of items, one per worker. Each item is injected as `partition`.'),
+    reduce_fn: z.string().optional().describe('Code to combine results. Has access to `results` list.'),
+    webhook_url: z.string().optional().describe('URL to POST results when swarm finishes'),
+  },
+  async ({ template_code, map_fn, data, reduce_fn, webhook_url }) => {
+    const result = await api('POST', '/swarm', { template_code, map_fn, data, reduce_fn, webhook_url })
+    return text(result)
+  }
+)
+
+server.tool(
+  'get_swarm',
+  'Check swarm progress and get results.',
+  {
+    swarm_id: z.string().describe('Swarm ID'),
+  },
+  async ({ swarm_id }) => {
+    const result = await api('GET', `/swarm/${swarm_id}`)
+    return text(result)
+  }
+)
+
+// ── Workspace tools ──────────────────────────────────────────────────────────
+
+server.tool(
+  'create_workspace',
+  'Create a persistent shared Python environment. Multiple API keys can execute in the same workspace. State auto-saves.',
+  {
+    name: z.string().describe('Workspace name'),
+  },
+  async ({ name }) => {
+    const result = await api('POST', '/workspaces', { name })
+    return text(result)
+  }
+)
+
+server.tool(
+  'execute_in_workspace',
+  'Run code in a shared workspace. All members see the same state.',
+  {
+    workspace_id: z.string().describe('Workspace ID'),
+    code: z.string().describe('Python code to execute'),
+  },
+  async ({ workspace_id, code }) => {
+    const result = await api('POST', `/workspaces/${workspace_id}/execute`, { code })
+    return text(result)
+  }
+)
+
+server.tool(
+  'list_workspaces',
+  'List all workspaces you own or have access to.',
+  {},
+  async () => {
+    const result = await api('GET', '/workspaces')
+    return text(result)
+  }
+)
+
+// ── File tools ───────────────────────────────────────────────────────────────
+
+server.tool(
+  'upload_file',
+  'Upload a file to a session workspace. Your code can then read it by filename.',
+  {
+    session_id: z.string().describe('Session ID'),
+    filename: z.string().describe('Filename in the workspace'),
+    content_base64: z.string().describe('File content, base64 encoded'),
+  },
+  async ({ session_id, filename, content_base64 }) => {
+    const result = await api('POST', `/session/${session_id}/files`, { filename, content: content_base64 })
+    return text(result)
+  }
+)
+
+server.tool(
+  'download_file',
+  'Download a file from a session workspace. Returns base64 encoded content.',
+  {
+    session_id: z.string().describe('Session ID'),
+    filename: z.string().describe('Filename to download'),
+  },
+  async ({ session_id, filename }) => {
+    const result = await api('GET', `/session/${session_id}/output/${filename}`)
+    return text(result)
+  }
+)
+
+// ── Usage tool ───────────────────────────────────────────────────────────────
+
+server.tool(
+  'get_usage',
+  'Check your current API usage, plan limits, and billing status.',
+  {},
+  async () => {
+    const result = await api('GET', '/usage')
+    return text(result)
+  }
+)
+
+// ── Start ────────────────────────────────────────────────────────────────────
+
+async function main() {
+  const transport = new StdioServerTransport()
+  await server.connect(transport)
+  console.error('greed-compute MCP server running on stdio')
+}
+
+main().catch((err) => {
+  console.error('Fatal:', err)
+  process.exit(1)
+})

package/tsconfig.json

@@ -0,0 +1,14 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "Node16",
+    "moduleResolution": "Node16",
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "declaration": true
+  },
+  "include": ["src/**/*"]
+}