dual-brain 0.2.29 → 0.3.0

package/dist/mcp-server/index.js

@@ -0,0 +1,359 @@
+#!/usr/bin/env node
+/**
+ * mcp-server/index.ts — TypeScript MCP server for dual-brain orchestration.
+ *
+ * Exposes dual-brain orchestration as tools via JSON-RPC 2.0 over stdio
+ * using the MCP protocol (Content-Length framing).
+ *
+ * Tools:
+ *   dual_brain_status   — Provider health, routing stats, active rooms
+ *   dual_brain_dispatch — Route a prompt to best provider/model/tier
+ *   dual_brain_history  — Recent routing decisions and outcomes
+ *   dual_brain_health   — Per-provider health scores and cooldown status
+ *
+ * Usage:
+ *   node dist/mcp-server/index.js
+ *
+ * MCP client config:
+ *   {
+ *     "mcpServers": {
+ *       "dual-brain": {
+ *         "command": "node",
+ *         "args": ["node_modules/dual-brain/mcp-server/index.mjs"]
+ *       }
+ *     }
+ *   }
+ */
+// Static imports — resolve to compiled siblings in dist/src/ after tsc.
+// @ts-ignore — these modules use dynamic exports
+import { detectTask } from '../src/detect.js';
+// @ts-ignore
+import { decideRoute } from '../src/decide.js';
+// @ts-ignore
+import { loadProfile } from '../src/profile.js';
+// @ts-ignore
+import { getHealth, getProviderScore, getSessionStats, checkCooldown } from '../src/health.js';
+// ─── Tool definitions ────────────────────────────────────────────────────────
+const TOOLS = [
+    {
+        name: 'dual_brain_status',
+        description: 'Get current provider health, routing stats, active rooms, and session overview. Returns which providers are enabled, their plan tier, health states, and active routing preferences.',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+        },
+    },
+    {
+        name: 'dual_brain_dispatch',
+        description: 'Accept a prompt and options, detect task type, and return the routing decision: which provider, model, tier, and explanation of why.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                prompt: {
+                    type: 'string',
+                    description: 'The task description or prompt to route.',
+                },
+                files: {
+                    type: 'array',
+                    items: { type: 'string' },
+                    description: 'Optional file paths involved in the task (for risk classification).',
+                },
+                profile: {
+                    type: 'string',
+                    enum: ['auto', 'balanced', 'cost-saver', 'quality-first'],
+                    description: 'Optional profile mode override.',
+                },
+            },
+            required: ['prompt'],
+        },
+    },
+    {
+        name: 'dual_brain_history',
+        description: 'Return recent routing decisions and outcomes from the current session. Shows dispatches made, providers used, and token consumption.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                limit: {
+                    type: 'number',
+                    description: 'Maximum number of recent entries to return (default 10).',
+                },
+            },
+        },
+    },
+    {
+        name: 'dual_brain_health',
+        description: 'Return per-provider health scores and cooldown status. Shows whether each provider/model is healthy, degraded, hot, or in cooldown.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                provider: {
+                    type: 'string',
+                    enum: ['claude', 'openai'],
+                    description: 'Optional: filter to a specific provider.',
+                },
+            },
+        },
+    },
+];
+// ─── Tool handlers ───────────────────────────────────────────────────────────
+function handleStatus() {
+    const cwd = process.cwd();
+    const profile = loadProfile(cwd);
+    const health = getHealth(cwd);
+    const providers = {};
+    for (const prov of ['claude', 'openai']) {
+        const cfg = profile?.providers?.[prov];
+        if (!cfg) {
+            providers[prov] = { enabled: false };
+            continue;
+        }
+        const models = {
+            search: prov === 'claude' ? 'haiku' : 'gpt-4.1-mini',
+            execute: prov === 'claude' ? 'sonnet' : 'gpt-5.4',
+            think: prov === 'claude' ? 'opus' : 'gpt-5.5',
+        };
+        const scores = {};
+        const states = {};
+        for (const [tier, modelClass] of Object.entries(models)) {
+            scores[tier] = getProviderScore(prov, modelClass, cwd);
+            const key = `${prov}:${modelClass}`;
+            states[modelClass] = health.states[key] ?? { status: 'healthy' };
+        }
+        providers[prov] = {
+            enabled: cfg.enabled ?? false,
+            plan: cfg.plan ?? null,
+            scores,
+            states,
+        };
+    }
+    const session = getSessionStats(cwd);
+    return {
+        providers,
+        session,
+        profile: {
+            mode: profile?.mode || 'auto',
+            preferences: (profile?.preferences || [])
+                .filter((p) => p.enabled)
+                .map((p) => p.text),
+        },
+    };
+}
+function handleDispatch(args) {
+    const prompt = args.prompt;
+    const files = args.files || [];
+    const profileOverride = args.profile;
+    const cwd = process.cwd();
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    let profile = loadProfile(cwd);
+    if (profileOverride) {
+        profile = { ...profile, mode: profileOverride };
+    }
+    const detection = detectTask({ prompt, files });
+    const decision = decideRoute({ profile, detection, cwd });
+    return {
+        provider: decision.provider,
+        model: decision.model,
+        effort: decision.effort,
+        tier: decision.tier,
+        dualBrain: decision.dualBrain,
+        modes: decision.modes,
+        sandbox: decision.sandbox,
+        explanation: decision.explanation,
+        detection: {
+            intent: detection.intent,
+            risk: detection.risk,
+            complexity: detection.complexity,
+            tier: detection.tier,
+        },
+    };
+}
+function handleHistory(args) {
+    const limit = args.limit || 10;
+    const cwd = process.cwd();
+    const health = getHealth(cwd);
+    const session = health.session;
+    if (!session || !session.dispatches || session.dispatches.length === 0) {
+        return { dispatches: [], count: 0, message: 'No dispatches recorded in current session.' };
+    }
+    const dispatches = session.dispatches.slice(-limit);
+    const stats = getSessionStats(cwd);
+    return {
+        dispatches,
+        count: dispatches.length,
+        total: session.dispatches.length,
+        startedAt: session.startedAt ?? null,
+        aggregated: stats,
+    };
+}
+function handleHealth(args) {
+    const providerFilter = args.provider;
+    const cwd = process.cwd();
+    const health = getHealth(cwd);
+    const result = {};
+    for (const [key, state] of Object.entries(health.states)) {
+        const [prov] = key.split(':');
+        if (providerFilter && prov !== providerFilter)
+            continue;
+        const modelClass = key.split(':')[1];
+        const cooldownExpired = checkCooldown(prov, modelClass, cwd);
+        result[key] = {
+            ...state,
+            cooldownExpired,
+            score: getProviderScore(prov, modelClass, cwd),
+        };
+    }
+    return {
+        states: result,
+        filter: providerFilter ?? 'all',
+        stateCount: Object.keys(result).length,
+    };
+}
+// ─── JSON-RPC dispatch ───────────────────────────────────────────────────────
+function dispatchTool(name, args) {
+    let result;
+    switch (name) {
+        case 'dual_brain_status':
+            result = handleStatus();
+            break;
+        case 'dual_brain_dispatch':
+            result = handleDispatch(args);
+            break;
+        case 'dual_brain_history':
+            result = handleHistory(args);
+            break;
+        case 'dual_brain_health':
+            result = handleHealth(args);
+            break;
+        default:
+            return {
+                content: [{ type: 'text', text: `Unknown tool: ${name}` }],
+                isError: true,
+            };
+    }
+    return {
+        content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
+    };
+}
+// ─── MCP protocol handlers ───────────────────────────────────────────────────
+function makeResponse(id, result) {
+    return { jsonrpc: '2.0', id, result };
+}
+function makeError(id, code, message, data) {
+    const error = { code, message };
+    if (data !== undefined)
+        error.data = data;
+    return { jsonrpc: '2.0', id, error };
+}
+function handleMessage(msg) {
+    const { id, method, params } = msg;
+    try {
+        switch (method) {
+            case 'initialize':
+                return makeResponse(id ?? null, {
+                    protocolVersion: '2024-11-05',
+                    capabilities: { tools: {} },
+                    serverInfo: { name: 'dual-brain', version: '0.2.30' },
+                });
+            case 'initialized':
+            case 'notifications/initialized':
+                // Notifications — no response
+                return null;
+            case 'tools/list':
+                return makeResponse(id ?? null, { tools: TOOLS });
+            case 'tools/call': {
+                const toolName = params?.name;
+                const toolArgs = (params?.arguments) ?? {};
+                if (!toolName) {
+                    return makeError(id ?? null, -32602, 'Missing tool name in params');
+                }
+                const result = dispatchTool(toolName, toolArgs);
+                return makeResponse(id ?? null, result);
+            }
+            case 'ping':
+                return makeResponse(id ?? null, {});
+            default:
+                return makeError(id ?? null, -32601, `Method not found: ${method}`);
+        }
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : 'Internal error';
+        return makeError(id ?? null, -32000, message);
+    }
+}
+// ─── MCP stdio transport (Content-Length framing) ────────────────────────────
+function sendMessage(response) {
+    const body = JSON.stringify(response);
+    const header = `Content-Length: ${Buffer.byteLength(body)}\r\n\r\n`;
+    process.stdout.write(header + body);
+}
+/**
+ * Reads Content-Length framed messages from stdin.
+ * Also supports bare newline-delimited JSON for compatibility.
+ */
+function startTransport() {
+    let buffer = '';
+    process.stdin.setEncoding('utf8');
+    process.stdin.on('data', (chunk) => {
+        buffer += chunk;
+        processBuffer();
+    });
+    process.stdin.on('end', () => {
+        if (buffer.trim()) {
+            processLine(buffer.trim());
+        }
+        process.exit(0);
+    });
+    function processBuffer() {
+        while (true) {
+            const headerEnd = buffer.indexOf('\r\n\r\n');
+            if (headerEnd === -1) {
+                // Fall back to newline-delimited mode
+                processNewlineDelimited();
+                return;
+            }
+            const headerSection = buffer.slice(0, headerEnd);
+            const match = headerSection.match(/Content-Length:\s*(\d+)/i);
+            if (!match) {
+                processNewlineDelimited();
+                return;
+            }
+            const contentLength = parseInt(match[1], 10);
+            const bodyStart = headerEnd + 4;
+            const available = Buffer.byteLength(buffer.slice(bodyStart), 'utf8');
+            if (available < contentLength) {
+                return;
+            }
+            const bodyBuf = Buffer.from(buffer.slice(bodyStart));
+            const body = bodyBuf.slice(0, contentLength).toString('utf8');
+            const consumed = bodyStart + body.length;
+            buffer = buffer.slice(consumed);
+            processLine(body);
+        }
+    }
+    function processNewlineDelimited() {
+        let newlineIdx;
+        while ((newlineIdx = buffer.indexOf('\n')) !== -1) {
+            const line = buffer.slice(0, newlineIdx).trim();
+            buffer = buffer.slice(newlineIdx + 1);
+            if (line)
+                processLine(line);
+        }
+    }
+    function processLine(raw) {
+        let msg;
+        try {
+            msg = JSON.parse(raw);
+        }
+        catch {
+            sendMessage(makeError(null, -32700, 'Parse error'));
+            return;
+        }
+        const response = handleMessage(msg);
+        if (response !== null) {
+            sendMessage(response);
+        }
+    }
+}
+// ─── Entry point ─────────────────────────────────────────────────────────────
+startTransport();
+//# sourceMappingURL=index.js.map

package/dist/mcp-server/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../mcp-server/index.ts"],"names":[],"mappings":";AACA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAEH,wEAAwE;AACxE,iDAAiD;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,aAAa;AACb,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,aAAa;AACb,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,aAAa;AACb,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAsC/F,gFAAgF;AAEhF,MAAM,KAAK,GAAwB;IACjC;QACE,IAAI,EAAE,mBAAmB;QACzB,WAAW,EACT,sLAAsL;QACxL,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf;KACF;IACD;QACE,IAAI,EAAE,qBAAqB;QAC3B,WAAW,EACT,sIAAsI;QACxI,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0CAA0C;iBACxD;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBACzB,WAAW,EAAE,qEAAqE;iBACnF;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,eAAe,CAAC;oBACzD,WAAW,EAAE,iCAAiC;iBAC/C;aACF;YACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;SACrB;KACF;IACD;QACE,IAAI,EAAE,oBAAoB;QAC1B,WAAW,EACT,sIAAsI;QACxI,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0DAA0D;iBACxE;aACF;SACF;KACF;IACD;QACE,IAAI,EAAE,mBAAmB;QACzB,WAAW,EACT,qIAAqI;QACvI,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;oBAC1B,WAAW,EAAE,0CAA0C;iBACxD;aACF;SACF;KACF;CACF,CAAC;AAEF,gFAAgF;AAEhF,SAAS,YAAY;IACnB,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1B,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAE9B,MAAM,SAAS,GAA4B,EAAE,CAAC;IAE9C,KAAK,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAU,EAAE,CAAC;QACjD,MAAM,GAAG,GAAG,OAAO,EAAE,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YACrC,SAAS;QACX,CAAC;QAED,MAAM,MAAM,GAA2B;YACrC,MAAM,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc;YACpD,OAAO,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YACjD,KAAK,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;SAC9C,CAAC;QAEF,MAAM,MAAM,GAA2B,EAAE,CAAC;QAC1C,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;YACvD,MAAM,GAAG,GAAG,GAAG,IAAI,IAAI,UAAU,EAAE,CAAC;YACpC,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QACnE,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,GAAG;YAChB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,KAAK;YAC7B,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,IAAI;YACtB,MAAM;YACN,MAAM;SACP,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;IAErC,OAAO;QACL,SAAS;QACT,OAAO;QACP,OAAO,EAAE;YACP,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,MAAM;YAC7B,WAAW,EAAE,CAAC,OAAO,EAAE,WAAW,IAAI,EAAE,CAAC;iBACtC,MAAM,CAAC,CAAC,CAAuB,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;iBAC9C,GAAG,CAAC,CAAC,CAAmB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SACxC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,IAA6B;IACnD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAgB,CAAC;IACrC,MAAM,KAAK,GAAI,IAAI,CAAC,KAAkB,IAAI,EAAE,CAAC;IAC7C,MAAM,eAAe,GAAG,IAAI,CAAC,OAA6B,CAAC;IAE3D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1B,8DAA8D;IAC9D,IAAI,OAAO,GAAQ,WAAW,CAAC,GAAG,CAAC,CAAC;IAEpC,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;IAClD,CAAC;IAED,MAAM,SAAS,GAAG,UAAU,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,WAAW,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IAE1D,OAAO;QACL,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,SAAS,EAAE;YACT,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,UAAU,EAAE,SAAS,CAAC,UAAU;YAChC,IAAI,EAAE,SAAS,CAAC,IAAI;SACrB;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,IAA6B;IAClD,MAAM,KAAK,GAAI,IAAI,CAAC,KAAgB,IAAI,EAAE,CAAC;IAE3C,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAC9B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAE/B,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,4CAA4C,EAAE,CAAC;IAC7F,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;IACpD,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;IAEnC,OAAO;QACL,UAAU;QACV,KAAK,EAAE,UAAU,CAAC,MAAM;QACxB,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,MAAM;QAChC,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;QACpC,UAAU,EAAE,KAAK;KAClB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAA6B;IACjD,MAAM,cAAc,GAAG,IAAI,CAAC,QAA8B,CAAC;IAE3D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAE9B,MAAM,MAAM,GAA4B,EAAE,CAAC;IAE3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAiC,CAAC,EAAE,CAAC;QACpF,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,cAAc,IAAI,IAAI,KAAK,cAAc;YAAE,SAAS;QAExD,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,eAAe,GAAG,aAAa,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,CAAC,GAAG,CAAC,GAAG;YACZ,GAAI,KAAgB;YACpB,eAAe;YACf,KAAK,EAAE,gBAAgB,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC;SAC/C,CAAC;IACJ,CAAC;IAED,OAAO;QACL,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,cAAc,IAAI,KAAK;QAC/B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM;KACvC,CAAC;AACJ,CAAC;AAED,gFAAgF;AAEhF,SAAS,YAAY,CAAC,IAAY,EAAE,IAA6B;IAC/D,IAAI,MAAe,CAAC;IAEpB,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,mBAAmB;YACtB,MAAM,GAAG,YAAY,EAAE,CAAC;YACxB,MAAM;QACR,KAAK,qBAAqB;YACxB,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM;QACR,KAAK,oBAAoB;YACvB,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM;QACR,KAAK,mBAAmB;YACtB,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;YAC5B,MAAM;QACR;YACE,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;gBAC1D,OAAO,EAAE,IAAI;aACd,CAAC;IACN,CAAC;IAED,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;KACnE,CAAC;AACJ,CAAC;AAED,gFAAgF;AAEhF,SAAS,YAAY,CAAC,EAA0B,EAAE,MAAe;IAC/D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;AACxC,CAAC;AAED,SAAS,SAAS,CAAC,EAA0B,EAAE,IAAY,EAAE,OAAe,EAAE,IAAc;IAC1F,MAAM,KAAK,GAAsD,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IACnF,IAAI,IAAI,KAAK,SAAS;QAAE,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;IAC1C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,aAAa,CAAC,GAAmB;IACxC,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC;IAEnC,IAAI,CAAC;QACH,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,YAAY;gBACf,OAAO,YAAY,CAAC,EAAE,IAAI,IAAI,EAAE;oBAC9B,eAAe,EAAE,YAAY;oBAC7B,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC3B,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,QAAQ,EAAE;iBACtD,CAAC,CAAC;YAEL,KAAK,aAAa,CAAC;YACnB,KAAK,2BAA2B;gBAC9B,8BAA8B;gBAC9B,OAAO,IAAI,CAAC;YAEd,KAAK,YAAY;gBACf,OAAO,YAAY,CAAC,EAAE,IAAI,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YAEpD,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,QAAQ,GAAI,MAA4B,EAAE,IAAI,CAAC;gBACrD,MAAM,QAAQ,GAAG,CAAE,MAAkD,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC;gBAExF,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,OAAO,SAAS,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,KAAK,EAAE,6BAA6B,CAAC,CAAC;gBACtE,CAAC;gBAED,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;gBAChD,OAAO,YAAY,CAAC,EAAE,IAAI,IAAI,EAAE,MAAM,CAAC,CAAC;YAC1C,CAAC;YAED,KAAK,MAAM;gBACT,OAAO,YAAY,CAAC,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;YAEtC;gBACE,OAAO,SAAS,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,KAAK,EAAE,qBAAqB,MAAM,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAAC,OAAO,GAAY,EAAE,CAAC;QACtB,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,CAAC;QACtE,OAAO,SAAS,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;AACH,CAAC;AAED,gFAAgF;AAEhF,SAAS,WAAW,CAAC,QAAyB;IAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACtC,MAAM,MAAM,GAAG,mBAAmB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC;IACpE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;AACtC,CAAC;AAED;;;GAGG;AACH,SAAS,cAAc;IACrB,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAClC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;QACzC,MAAM,IAAI,KAAK,CAAC;QAChB,aAAa,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;QAC3B,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAClB,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,SAAS,aAAa;QACpB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC7C,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;gBACrB,sCAAsC;gBACtC,uBAAuB,EAAE,CAAC;gBAC1B,OAAO;YACT,CAAC;YAED,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YACjD,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC9D,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,uBAAuB,EAAE,CAAC;gBAC1B,OAAO;YACT,CAAC;YAED,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,SAAS,GAAG,CAAC,CAAC;YAChC,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC,CAAC;YAErE,IAAI,SAAS,GAAG,aAAa,EAAE,CAAC;gBAC9B,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;YACrD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC9D,MAAM,QAAQ,GAAG,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC;YACzC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEhC,WAAW,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED,SAAS,uBAAuB;QAC9B,IAAI,UAAkB,CAAC;QACvB,OAAO,CAAC,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAClD,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;YAChD,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;YACtC,IAAI,IAAI;gBAAE,WAAW,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,SAAS,WAAW,CAAC,GAAW;QAC9B,IAAI,GAAmB,CAAC;QACxB,IAAI,CAAC;YACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAmB,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC;YACpD,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACtB,WAAW,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;AACH,CAAC;AAED,gFAAgF;AAEhF,cAAc,EAAE,CAAC"}

package/dist/src/agent-protocol.d.ts

@@ -0,0 +1,163 @@
+/**
+ * agent-protocol.ts — Typed contract system for the dual-brain agent hierarchy.
+ *
+ * Defines how HEAD, Managers, Supervisors, and Workers communicate:
+ * what they receive, what they produce, how they escalate, and how they report.
+ *
+ * Hierarchy:
+ *   HEAD (user's session, lightest context)
+ *     → Manager (project-level, persistent across sessions)
+ *       → Supervisor (wave-level, deploys and monitors workers)
+ *         → Worker (task-level, executes and reports)
+ */
+import type { RoomSummary, RoomInsight, RoomDecision, RoomPlan, WaveTask, WorkerAssignment, WorkerResult, UserInput } from './room.ts';
+export type { RoomSummary, RoomInsight, RoomDecision, RoomPlan, WaveTask, WorkerAssignment, WorkerResult, UserInput, };
+export type AgentLevel = 'head' | 'manager' | 'supervisor' | 'worker';
+export interface HeadInput {
+    userMessage: string;
+    roomSummaries: RoomSummary[];
+    providerHealth: ProviderHealth[];
+    recentInsights: RoomInsight[];
+}
+export interface ManagerInput {
+    roomId: string;
+    taskDescription: string;
+    userInputs: UserInput[];
+    decisions: RoomDecision[];
+    waveResults: WaveResult[];
+    modelProfiles: ModelProfileSummary[];
+    providerHealth: ProviderHealth[];
+}
+export interface SupervisorInput {
+    roomId: string;
+    waveId: string;
+    tasks: WaveTask[];
+    modelProfiles: ModelProfileSummary[];
+    budget: BudgetConstraint;
+    providerHealth: ProviderHealth[];
+}
+export interface WorkerInput {
+    roomId: string;
+    taskId: string;
+    assignment: WorkerAssignment;
+    contextFiles: Record<string, string>;
+}
+export interface HeadOutput {
+    userResponse: string;
+    roomActions: RoomAction[];
+    escalationNeeded: boolean;
+}
+export interface ManagerOutput {
+    plan?: RoomPlan;
+    insights: RoomInsight[];
+    supervisorDeployments: SupervisorDeployment[];
+    status: 'planning' | 'executing' | 'reviewing' | 'completed' | 'blocked';
+    blockedReason?: string;
+}
+export interface SupervisorOutput {
+    workerDeployments: WorkerDeployment[];
+    qualityChecks: QualityCheck[];
+    waveStatus: 'running' | 'completed' | 'failed' | 'needs-replan';
+    replanReason?: string;
+}
+export interface WorkerOutput {
+    result: WorkerResult;
+    selfAssessment: number;
+    issues: string[];
+}
+export interface SupervisorDeployment {
+    waveId: string;
+    tasks: WaveTask[];
+    model: string;
+    budget: BudgetConstraint;
+}
+export interface WorkerDeployment {
+    taskId: string;
+    assignment: WorkerAssignment;
+    model: string;
+    timeout: number;
+}
+export interface BudgetConstraint {
+    maxWorkers: number;
+    maxTotalTokens: number;
+    maxRetries: number;
+    costTier: 'frugal' | 'balanced' | 'quality' | 'maximum';
+}
+export interface QualityCheck {
+    taskId: string;
+    passed: boolean;
+    findings: string[];
+    recommendation: 'accept' | 'retry' | 'escalate' | 'reject';
+}
+export interface RoomAction {
+    type: 'create' | 'close' | 'decide' | 'escalate';
+    roomId?: string;
+    taskDescription?: string;
+    decision?: RoomDecision;
+}
+export interface ModelProfileSummary {
+    id: string;
+    name: string;
+    provider: string;
+    tier: number;
+    available: boolean;
+    strengths: string[];
+}
+export interface ProviderHealth {
+    provider: string;
+    status: 'healthy' | 'degraded' | 'down' | 'rate-limited';
+    remainingCapacity?: number;
+    cooldownUntil?: string;
+}
+export interface WaveResult {
+    waveId: string;
+    status: 'completed' | 'partial' | 'failed';
+    workerResults: WorkerResult[];
+    qualityScore: number;
+}
+/**
+ * Determine how deep the hierarchy should go for this task.
+ * Simple tasks return 'worker' (HEAD dispatches directly).
+ * Medium returns 'supervisor'. Complex returns 'manager'.
+ */
+export declare function classifyDepth(taskDescription: string, complexity?: number): AgentLevel;
+/**
+ * Assemble HEAD's input from room files and provider state.
+ */
+export declare function buildHeadInput(cwd?: string): HeadInput;
+/**
+ * Assemble Manager's input from room files.
+ */
+export declare function buildManagerInput(roomId: string, cwd?: string): ManagerInput;
+/**
+ * Assemble Supervisor's input for a specific wave.
+ */
+export declare function buildSupervisorInput(roomId: string, waveId: string, cwd?: string): SupervisorInput;
+/**
+ * Assemble Worker's input with file contents.
+ */
+export declare function buildWorkerInput(roomId: string, taskId: string, cwd?: string): WorkerInput;
+/**
+ * Type-check agent output before accepting.
+ * Returns true if the output matches the expected shape for the given level.
+ */
+export declare function validateOutput(level: AgentLevel, output: unknown): boolean;
+/**
+ * Estimate token cost for a set of worker deployments.
+ * Returns estimated total tokens.
+ */
+export declare function estimateCost(deployments: WorkerDeployment[], profiles: ModelProfileSummary[]): number;
+/**
+ * Pick the best available model for a task given budget constraints.
+ * Matches task tier to model tier, filtered by availability and budget.
+ */
+export declare function selectModel(task: WaveTask, profiles: ModelProfileSummary[], budget: BudgetConstraint): string;
+/**
+ * Determine if results are bad enough to escalate to a higher level.
+ * Checks failure rate and quality against threshold.
+ */
+export declare function shouldEscalate(results: WorkerResult[], qualityThreshold?: number): boolean;
+/**
+ * Create budget constraints from the user's dial setting and task count.
+ */
+export declare function buildBudget(costTier: string, taskCount: number): BudgetConstraint;