105 lines
4.0 KiB
JavaScript
105 lines
4.0 KiB
JavaScript
#!/usr/bin/env node
|
|
// =============================================================================
|
|
// vibn-agent-mcp
|
|
// -----------------------------------------------------------------------------
|
|
// Stdio MCP server exposing the vibn-agent-runner sub-agent orchestration API.
|
|
// This lets any MCP-speaking client (Goose, Claude Desktop, Cursor, etc.)
|
|
// spawn Coder / PM / Marketing jobs against the vibn-agent-runner HTTP service
|
|
// and poll their status.
|
|
//
|
|
// Config (env):
|
|
// AGENT_RUNNER_URL (default: http://localhost:3333) — URL of the runner
|
|
// =============================================================================
|
|
|
|
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
|
|
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
|
|
|
|
import * as api from '../tools/agent-api';
|
|
import type { AgentRunnerConfig } from '../tools/agent-api';
|
|
|
|
function loadConfig(): AgentRunnerConfig {
|
|
const runnerUrl = process.env.AGENT_RUNNER_URL?.trim() || 'http://localhost:3333';
|
|
return { runnerUrl };
|
|
}
|
|
|
|
const TOOL_DEFINITIONS = [
|
|
{
|
|
name: 'spawn_agent',
|
|
description: 'Dispatch a sub-agent job to run in the background on the vibn-agent-runner. Returns a job ID.',
|
|
inputSchema: {
|
|
type: 'object' as const,
|
|
properties: {
|
|
agent: { type: 'string', description: '"Coder", "PM", or "Marketing"' },
|
|
task: { type: 'string', description: 'Detailed task description for the agent' },
|
|
repo: { type: 'string', description: 'Gitea repo in "owner/name" format' },
|
|
},
|
|
required: ['agent', 'task', 'repo'],
|
|
},
|
|
},
|
|
{
|
|
name: 'get_job_status',
|
|
description: 'Check the status of a previously spawned agent job.',
|
|
inputSchema: {
|
|
type: 'object' as const,
|
|
properties: {
|
|
job_id: { type: 'string', description: 'Job ID returned by spawn_agent' },
|
|
},
|
|
required: ['job_id'],
|
|
},
|
|
},
|
|
];
|
|
|
|
async function dispatch(cfg: AgentRunnerConfig, name: string, args: Record<string, unknown>): Promise<unknown> {
|
|
switch (name) {
|
|
case 'spawn_agent':
|
|
return api.spawnAgent(cfg, {
|
|
agent: String(args.agent),
|
|
task: String(args.task),
|
|
repo: String(args.repo),
|
|
});
|
|
case 'get_job_status':
|
|
return api.getJobStatus(cfg, String(args.job_id));
|
|
default:
|
|
throw new Error(`Unknown tool: ${name}`);
|
|
}
|
|
}
|
|
|
|
function buildServer(cfg: AgentRunnerConfig): Server {
|
|
const server = new Server(
|
|
{ name: 'vibn-agent-mcp', version: '0.1.0' },
|
|
{ capabilities: { tools: {} } },
|
|
);
|
|
|
|
server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOL_DEFINITIONS }));
|
|
|
|
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
const name = request.params.name;
|
|
const args = (request.params.arguments ?? {}) as Record<string, unknown>;
|
|
try {
|
|
const result = await dispatch(cfg, name, args);
|
|
return { content: [{ type: 'text', text: typeof result === 'string' ? result : JSON.stringify(result, null, 2) }] };
|
|
} catch (err) {
|
|
const message = err instanceof Error ? err.message : String(err);
|
|
return { isError: true, content: [{ type: 'text', text: `Error: ${message}` }] };
|
|
}
|
|
});
|
|
|
|
return server;
|
|
}
|
|
|
|
async function main(): Promise<void> {
|
|
const cfg = loadConfig();
|
|
const server = buildServer(cfg);
|
|
const transport = new StdioServerTransport();
|
|
await server.connect(transport);
|
|
// eslint-disable-next-line no-console
|
|
console.error(`[vibn-agent-mcp] ready — ${TOOL_DEFINITIONS.length} tools exposed (runner=${cfg.runnerUrl})`);
|
|
}
|
|
|
|
main().catch((err) => {
|
|
// eslint-disable-next-line no-console
|
|
console.error('[vibn-agent-mcp] fatal:', err);
|
|
process.exit(1);
|
|
});
|