contextos-mcp 0.4.0

package/dist/agents/direct-llm.js

@@ -0,0 +1,254 @@
+/**
+ * ContextOS Direct LLM agent backend.
+ *
+ * Enhanced version of the original direct-llm provider.
+ * Key improvements:
+ *   1. Accepts a ContextOS system prompt (project rules/standards)
+ *   2. Parses code blocks from LLM responses and writes them to the worktree
+ *   3. Builds a coding-oriented prompt that instructs the LLM to produce file edits
+ *   4. Reports files changed based on parsed code blocks
+ *
+ * Uses @mariozechner/pi-ai for unified access to OpenAI, Anthropic, and Gemini.
+ */
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import * as path from "node:path";
+import { isPathAllowed } from "../security/file-policy.js";
+import { containsSecrets, isBlockedPath, redactSecrets } from "../security/secret-filter.js";
+import { registerAgent } from "./provider.js";
+/**
+ * Parse code blocks from LLM output that include file paths.
+ *
+ * Supports formats:
+ *   ```language:path/to/file.ts
+ *   ```language filepath="path/to/file.ts"
+ *   <!-- FILE: path/to/file.ts -->
+ *   // FILE: path/to/file.ts
+ */
+function parseCodeBlocks(output) {
+    const blocks = [];
+    const codeBlockRegex = /```(\w+)?(?::([^\n]+)|[^\S\n]+filepath="([^"]+)")?\n([\s\S]*?)```/g;
+    let match;
+    while ((match = codeBlockRegex.exec(output)) !== null) {
+        const language = match[1] || "";
+        const filePath = match[2] || match[3] || "";
+        const content = match[4] || "";
+        if (filePath && content.trim()) {
+            blocks.push({
+                filePath: filePath.trim(),
+                content: content,
+                language,
+            });
+        }
+    }
+    // Also try FILE: comment pattern
+    const fileCommentRegex = /(?:<!--\s*FILE:\s*(.+?)\s*-->|\/\/\s*FILE:\s*(.+))\n```\w*\n([\s\S]*?)```/g;
+    while ((match = fileCommentRegex.exec(output)) !== null) {
+        const filePath = (match[1] || match[2] || "").trim();
+        const content = match[3] || "";
+        if (filePath && content.trim()) {
+            // Avoid duplicates
+            if (!blocks.some((b) => b.filePath === filePath)) {
+                blocks.push({ filePath, content, language: "" });
+            }
+        }
+    }
+    return blocks;
+}
+/**
+ * Write parsed code blocks to the worktree directory.
+ * Creates directories as needed. Returns list of written file paths.
+ */
+function writeCodeBlocks(workDir, blocks) {
+    const written = [];
+    for (const block of blocks) {
+        // Security: prevent path traversal and blocked/forbidden paths
+        const normalized = path.normalize(block.filePath);
+        if (normalized.startsWith("..") || path.isAbsolute(normalized)) {
+            process.stderr.write(`[contextos-agent] Skipping unsafe path (traversal): ${block.filePath}\n`);
+            continue;
+        }
+        if (isBlockedPath(normalized) || !isPathAllowed(normalized, workDir)) {
+            process.stderr.write(`[contextos-agent] Skipping forbidden/policy-blocked path: ${block.filePath}\n`);
+            continue;
+        }
+        const fullPath = path.join(workDir, normalized);
+        // Ensure parent directory exists
+        const dir = path.dirname(fullPath);
+        if (!existsSync(dir)) {
+            mkdirSync(dir, { recursive: true });
+        }
+        writeFileSync(fullPath, block.content, "utf-8");
+        written.push(normalized);
+        process.stderr.write(`[contextos-agent] Wrote: ${normalized}\n`);
+    }
+    return written;
+}
+// ── Prompt Builder ─────────────────────────────────────────────────────────
+/**
+ * Build a coding-oriented prompt that tells the LLM to produce file edits
+ * in a parseable format.
+ */
+function buildCodingPrompt(task, workDir, files, contextosPrompt) {
+    const sections = [];
+    // ContextOS project standards (if provided)
+    if (contextosPrompt) {
+        sections.push(contextosPrompt);
+    }
+    // Coding instructions
+    sections.push(`
+## Your Task
+
+You are a coding agent. Your job is to implement the following task by writing code.
+
+**Task**: ${task}
+
+## Output Format
+
+For EVERY file you create or modify, output it in this exact format:
+
+\`\`\`language:path/to/file.ts
+// file contents here
+\`\`\`
+
+The path must be relative to the project root. Include the COMPLETE file contents.
+Do NOT use placeholders like "// ... rest of the code". Write the full file.
+
+## Important Rules
+- Write production-quality code
+- Include proper error handling
+- Follow existing code style and patterns in the project
+- Do not modify files that are not related to the task
+`);
+    // Include existing file contents for context
+    if (files && files.length > 0) {
+        sections.push("\n## Existing Files (for context)\n");
+        for (const file of files.slice(0, 10)) {
+            // Cap at 10 files
+            if (!isPathAllowed(file, workDir) || isBlockedPath(file)) {
+                process.stderr.write(`[direct-llm] Skipping blocked or out-of-boundary file: ${file}\n`);
+                continue;
+            }
+            const fullPath = path.resolve(workDir, file);
+            if (existsSync(fullPath)) {
+                try {
+                    const rawContent = readFileSync(fullPath, "utf-8");
+                    if (containsSecrets(rawContent)) {
+                        process.stderr.write(`[direct-llm] Redacting detected secrets in: ${file}\n`);
+                    }
+                    const content = redactSecrets(rawContent);
+                    // Cap per-file content at 3000 chars
+                    const truncated = content.length > 3000 ? `${content.slice(0, 3000)}\n// ... [truncated]` : content;
+                    sections.push(`### ${file}\n\`\`\`\n${truncated}\n\`\`\`\n`);
+                }
+                catch {
+                    // Skip unreadable files
+                }
+            }
+        }
+    }
+    return sections.join("\n");
+}
+const contextosDirectLlm = {
+    name: "direct-llm",
+    async isAvailable() {
+        return !!(process.env.ANTHROPIC_API_KEY ||
+            process.env.OPENAI_API_KEY ||
+            process.env.GEMINI_API_KEY ||
+            process.env.OPENROUTER_API_KEY);
+    },
+    async run(options) {
+        const startTime = Date.now();
+        const extOptions = options;
+        try {
+            const { completeSimple, getModels, getProviders } = await import("@mariozechner/pi-ai");
+            const modelId = options.model || process.env.CONTEXTOS_DEFAULT_MODEL || process.env.RLM_MODEL || "claude-sonnet-4-6";
+            // Find model
+            let model;
+            for (const provider of getProviders()) {
+                for (const m of getModels(provider)) {
+                    if (m.id === modelId) {
+                        model = m;
+                        break;
+                    }
+                }
+                if (model)
+                    break;
+            }
+            if (!model) {
+                return {
+                    success: false,
+                    output: "",
+                    filesChanged: [],
+                    diff: "",
+                    durationMs: Date.now() - startTime,
+                    error: `Model "${modelId}" not found. Available models depend on your API keys (OPENAI_API_KEY, ANTHROPIC_API_KEY, GEMINI_API_KEY).`,
+                };
+            }
+            // Build the coding prompt with ContextOS rules
+            const codingPrompt = buildCodingPrompt(options.task, options.workDir, options.files, extOptions.contextosPrompt);
+            // Abort-aware LLM call — race against cancellation signal to stop
+            // spending money on abandoned requests (P1.9 fix)
+            const llmPromise = completeSimple(model, {
+                systemPrompt: "You are an expert coding agent. You write complete, production-ready code. Follow all project standards provided to you.",
+                messages: [
+                    {
+                        role: "user",
+                        content: codingPrompt,
+                        timestamp: Date.now(),
+                    },
+                ],
+            });
+            let response;
+            const { signal } = options;
+            if (signal) {
+                const abortPromise = new Promise((_, reject) => {
+                    if (signal.aborted) {
+                        reject(new DOMException("LLM request aborted", "AbortError"));
+                        return;
+                    }
+                    signal.addEventListener("abort", () => {
+                        reject(new DOMException("LLM request aborted", "AbortError"));
+                    }, { once: true });
+                });
+                response = await Promise.race([llmPromise, abortPromise]);
+            }
+            else {
+                response = await llmPromise;
+            }
+            const output = response.content
+                .filter((b) => b.type === "text")
+                .map((b) => b.text)
+                .join("\n");
+            // Parse code blocks and write to worktree
+            const codeBlocks = parseCodeBlocks(output);
+            let filesChanged = [];
+            if (codeBlocks.length > 0 && options.workDir) {
+                filesChanged = writeCodeBlocks(options.workDir, codeBlocks);
+                process.stderr.write(`[contextos-agent] Parsed ${codeBlocks.length} code blocks, wrote ${filesChanged.length} files\n`);
+            }
+            else {
+                process.stderr.write(`[contextos-agent] No code blocks found in LLM response (${output.length} chars)\n`);
+            }
+            return {
+                success: true,
+                output,
+                filesChanged,
+                diff: "", // Will be captured by WorktreeManager.getDiff()
+                durationMs: Date.now() - startTime,
+            };
+        }
+        catch (err) {
+            return {
+                success: false,
+                output: "",
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+                error: err instanceof Error ? err.message : String(err),
+            };
+        }
+    },
+};
+registerAgent(contextosDirectLlm);
+export default contextosDirectLlm;
+//# sourceMappingURL=direct-llm.js.map

package/dist/agents/direct-llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct-llm.js","sourceRoot":"","sources":["../../src/agents/direct-llm.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7F,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAU9C;;;;;;;;GAQG;AACH,SAAS,eAAe,CAAC,MAAc;IACtC,MAAM,MAAM,GAAsB,EAAE,CAAC;IACrC,MAAM,cAAc,GAAG,oEAAoE,CAAC;IAE5F,IAAI,KAA6B,CAAC;IAClC,OAAO,CAAC,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE/B,IAAI,QAAQ,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC;gBACX,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE;gBACzB,OAAO,EAAE,OAAO;gBAChB,QAAQ;aACR,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,iCAAiC;IACjC,MAAM,gBAAgB,GAAG,4EAA4E,CAAC;IACtG,OAAO,CAAC,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACzD,MAAM,QAAQ,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE/B,IAAI,QAAQ,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YAChC,mBAAmB;YACnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,EAAE,CAAC;gBAClD,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAClD,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,OAAe,EAAE,MAAyB;IAClE,MAAM,OAAO,GAAa,EAAE,CAAC;IAE7B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC5B,+DAA+D;QAC/D,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAChE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,uDAAuD,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC;YAChG,SAAS;QACV,CAAC;QAED,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,CAAC;YACtE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,6DAA6D,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC;YACtG,SAAS;QACV,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAEhD,iCAAiC;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACtB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC;QAED,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAChD,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,UAAU,IAAI,CAAC,CAAC;IAClE,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,8EAA8E;AAE9E;;;GAGG;AACH,SAAS,iBAAiB,CAAC,IAAY,EAAE,OAAe,EAAE,KAAgB,EAAE,eAAwB;IACnG,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,4CAA4C;IAC5C,IAAI,eAAe,EAAE,CAAC;QACrB,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChC,CAAC;IAED,sBAAsB;IACtB,QAAQ,CAAC,IAAI,CAAC;;;;;YAKH,IAAI;;;;;;;;;;;;;;;;;;CAkBf,CAAC,CAAC;IAEF,6CAA6C;IAC7C,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,QAAQ,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;QACrD,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;YACvC,kBAAkB;YAClB,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1D,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,0DAA0D,IAAI,IAAI,CAAC,CAAC;gBACzF,SAAS;YACV,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACJ,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBACnD,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;wBACjC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,+CAA+C,IAAI,IAAI,CAAC,CAAC;oBAC/E,CAAC;oBACD,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;oBAC1C,qCAAqC;oBACrC,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC,OAAO,CAAC;oBACpG,QAAQ,CAAC,IAAI,CAAC,OAAO,IAAI,aAAa,SAAS,YAAY,CAAC,CAAC;gBAC9D,CAAC;gBAAC,MAAM,CAAC;oBACR,wBAAwB;gBACzB,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAYD,MAAM,kBAAkB,GAAkB;IACzC,IAAI,EAAE,YAAY;IAElB,KAAK,CAAC,WAAW;QAChB,OAAO,CAAC,CAAC,CACR,OAAO,CAAC,GAAG,CAAC,iBAAiB;YAC7B,OAAO,CAAC,GAAG,CAAC,cAAc;YAC1B,OAAO,CAAC,GAAG,CAAC,cAAc;YAC1B,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAC9B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAwB;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,OAAmC,CAAC;QAEvD,IAAI,CAAC;YACJ,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,CAAC;YACxF,MAAM,OAAO,GACZ,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,mBAAmB,CAAC;YAEtG,aAAa;YACb,IAAI,KAAK,CAAC;YACV,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,EAAE,CAAC;gBACvC,KAAK,MAAM,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrC,IAAI,CAAC,CAAC,EAAE,KAAK,OAAO,EAAE,CAAC;wBACtB,KAAK,GAAG,CAAC,CAAC;wBACV,MAAM;oBACP,CAAC;gBACF,CAAC;gBACD,IAAI,KAAK;oBAAE,MAAM;YAClB,CAAC;YAED,IAAI,CAAC,KAAK,EAAE,CAAC;gBACZ,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,EAAE;oBACV,YAAY,EAAE,EAAE;oBAChB,IAAI,EAAE,EAAE;oBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBAClC,KAAK,EAAE,UAAU,OAAO,4GAA4G;iBACpI,CAAC;YACH,CAAC;YAED,+CAA+C;YAC/C,MAAM,YAAY,GAAG,iBAAiB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,EAAE,UAAU,CAAC,eAAe,CAAC,CAAC;YAEjH,kEAAkE;YAClE,kDAAkD;YAClD,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,EAAE;gBACxC,YAAY,EACX,0HAA0H;gBAC3H,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE,YAAY;wBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACrB;iBACD;aACD,CAAC,CAAC;YAEH,IAAI,QAAQ,CAAC;YACb,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;YAC3B,IAAI,MAAM,EAAE,CAAC;gBACZ,MAAM,YAAY,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;oBACrD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;wBACpB,MAAM,CAAC,IAAI,YAAY,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC,CAAC;wBAC9D,OAAO;oBACR,CAAC;oBACD,MAAM,CAAC,gBAAgB,CACtB,OAAO,EACP,GAAG,EAAE;wBACJ,MAAM,CAAC,IAAI,YAAY,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC,CAAC;oBAC/D,CAAC,EACD,EAAE,IAAI,EAAE,IAAI,EAAE,CACd,CAAC;gBACH,CAAC,CAAC,CAAC;gBACH,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACP,QAAQ,GAAG,MAAM,UAAU,CAAC;YAC7B,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO;iBAC7B,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBACrE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,0CAA0C;YAC1C,MAAM,UAAU,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,YAAY,GAAa,EAAE,CAAC;YAEhC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC9C,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC5D,OAAO,CAAC,MAAM,CAAC,KAAK,CACnB,4BAA4B,UAAU,CAAC,MAAM,uBAAuB,YAAY,CAAC,MAAM,UAAU,CACjG,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,2DAA2D,MAAM,CAAC,MAAM,WAAW,CAAC,CAAC;YAC3G,CAAC;YAED,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,MAAM;gBACN,YAAY;gBACZ,IAAI,EAAE,EAAE,EAAE,gDAAgD;gBAC1D,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aAClC,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,EAAE;gBACV,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACvD,CAAC;QACH,CAAC;IACF,CAAC;CACD,CAAC;AAEF,aAAa,CAAC,kBAAkB,CAAC,CAAC;AAClC,eAAe,kBAAkB,CAAC"}

package/dist/agents/mock.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Mock agent backend — for integration testing without real API keys.
+ *
+ * Simulates a coding agent by writing actual files in the work directory.
+ * Registered as "mock" and "mock-reviewer" in the agent registry.
+ */
+import type { AgentProvider } from "../core/types.js";
+declare const mockProvider: AgentProvider;
+export declare const mockReviewerProvider: AgentProvider;
+export default mockProvider;

package/dist/agents/mock.js

@@ -0,0 +1,170 @@
+/**
+ * Mock agent backend — for integration testing without real API keys.
+ *
+ * Simulates a coding agent by writing actual files in the work directory.
+ * Registered as "mock" and "mock-reviewer" in the agent registry.
+ */
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { registerAgent } from "./provider.js";
+const mockProvider = {
+    name: "mock",
+    async isAvailable() {
+        return true;
+    },
+    async run(options) {
+        const { task, workDir, signal } = options;
+        const startTime = Date.now();
+        // Check for abort
+        if (signal?.aborted) {
+            return {
+                success: false,
+                output: "Aborted",
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+                error: "Aborted before start",
+            };
+        }
+        // Simulate work delay (20ms)
+        await new Promise((resolve) => setTimeout(resolve, 20));
+        // Check for forced failure mode
+        if (task.includes("__FAIL__")) {
+            return {
+                success: false,
+                output: "Simulated failure",
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+                error: "Mock agent forced failure",
+            };
+        }
+        // Handle review evaluation tasks cleanly if invoked as reviewer
+        if (task.includes("Independent Code Reviewer") || task.includes("specCompliance") || task.includes("codeQuality")) {
+            if (task.includes("__MALFORMED__")) {
+                return {
+                    success: true,
+                    output: "Unparseable non-json response",
+                    filesChanged: [],
+                    diff: "",
+                    durationMs: Date.now() - startTime,
+                };
+            }
+            const specCompliance = task.includes("__FAIL_SPEC__") ? "FAIL" : "PASS";
+            const codeQuality = task.includes("__FAIL_QUALITY__") ? "FAIL" : "PASS";
+            return {
+                success: true,
+                output: JSON.stringify({
+                    specCompliance,
+                    codeQuality,
+                    summary: `Review completed: spec=${specCompliance}, quality=${codeQuality}`,
+                }),
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+                usage: { inputTokens: 500, outputTokens: 200, totalTokens: 700 },
+            };
+        }
+        // Write actual files in the work directory to simulate agent edits
+        const filesChanged = [];
+        // Create or modify a file based on the task
+        const targetFile = path.join(workDir, "hello.ts");
+        const content = `// Generated by mock agent\n// Task: ${task}\nexport function hello(): string {\n\treturn "Hello from swarm thread!";\n}\n`;
+        fs.writeFileSync(targetFile, content, "utf-8");
+        filesChanged.push("hello.ts");
+        // If task mentions "multi", create additional files
+        if (task.toLowerCase().includes("multi")) {
+            const extraFile = path.join(workDir, "utils.ts");
+            fs.writeFileSync(extraFile, '// Utility module\nexport const VERSION = "1.0.0";\n', "utf-8");
+            filesChanged.push("utils.ts");
+        }
+        const output = [
+            "Applied edit to hello.ts",
+            `Created ${filesChanged.length} file(s)`,
+            "Result: Successfully completed task",
+        ].join("\n");
+        return {
+            success: true,
+            output,
+            filesChanged,
+            diff: "", // Diff is captured by worktree manager
+            durationMs: Date.now() - startTime,
+            // Simulate token usage for testing budget tracking
+            usage: {
+                inputTokens: 1500,
+                outputTokens: 800,
+                totalTokens: 2300,
+            },
+        };
+    },
+};
+export const mockReviewerProvider = {
+    name: "mock-reviewer",
+    async isAvailable() {
+        return true;
+    },
+    async run(options) {
+        const { task, signal } = options;
+        const startTime = Date.now();
+        if (signal?.aborted) {
+            return {
+                success: false,
+                output: "Aborted",
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+                error: "Aborted before start",
+            };
+        }
+        if (task.includes("__TIMEOUT__")) {
+            await new Promise((resolve) => setTimeout(resolve, 35000));
+        }
+        if (task.includes("__MALFORMED__")) {
+            return {
+                success: true,
+                output: "This is completely unparseable non-JSON text from the reviewer.",
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+            };
+        }
+        if (task.includes("__ERROR__") || task.includes("__FAIL__")) {
+            return {
+                success: false,
+                output: "",
+                filesChanged: [],
+                diff: "",
+                durationMs: Date.now() - startTime,
+                error: "Independent reviewer model execution failed",
+            };
+        }
+        const specCompliance = task.includes("__FAIL_SPEC__") ? "FAIL" : "PASS";
+        const codeQuality = task.includes("__FAIL_QUALITY__") ? "FAIL" : "PASS";
+        return {
+            success: true,
+            output: JSON.stringify({
+                specCompliance,
+                codeQuality,
+                summary: specCompliance === "PASS" && codeQuality === "PASS"
+                    ? "Independent code review passed"
+                    : `Independent code review verdict: spec=${specCompliance}, quality=${codeQuality}`,
+            }),
+            filesChanged: [],
+            diff: "",
+            durationMs: Date.now() - startTime,
+            usage: {
+                inputTokens: 600,
+                outputTokens: 250,
+                totalTokens: 850,
+            },
+        };
+    },
+};
+registerAgent(mockProvider);
+registerAgent(mockReviewerProvider);
+registerAgent({
+    ...mockReviewerProvider,
+    name: "reviewer-independent",
+});
+export default mockProvider;
+//# sourceMappingURL=mock.js.map

package/dist/agents/mock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock.js","sourceRoot":"","sources":["../../src/agents/mock.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,MAAM,YAAY,GAAkB;IACnC,IAAI,EAAE,MAAM;IAEZ,KAAK,CAAC,WAAW;QAChB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAwB;QACjC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,kBAAkB;QAClB,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACrB,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,SAAS;gBACjB,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,sBAAsB;aAC7B,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAExD,gCAAgC;QAChC,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,mBAAmB;gBAC3B,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,2BAA2B;aAClC,CAAC;QACH,CAAC;QAED,gEAAgE;QAChE,IAAI,IAAI,CAAC,QAAQ,CAAC,2BAA2B,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACnH,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;gBACpC,OAAO;oBACN,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,+BAA+B;oBACvC,YAAY,EAAE,EAAE;oBAChB,IAAI,EAAE,EAAE;oBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBAClC,CAAC;YACH,CAAC;YACD,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;YACxE,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;YACxE,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC;oBACtB,cAAc;oBACd,WAAW;oBACX,OAAO,EAAE,0BAA0B,cAAc,aAAa,WAAW,EAAE;iBAC3E,CAAC;gBACF,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE;aAChE,CAAC;QACH,CAAC;QAED,mEAAmE;QACnE,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,4CAA4C;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,wCAAwC,IAAI,gFAAgF,CAAC;QAC7I,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC/C,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE9B,oDAAoD;QACpD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACjD,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,sDAAsD,EAAE,OAAO,CAAC,CAAC;YAC7F,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,MAAM,GAAG;YACd,0BAA0B;YAC1B,WAAW,YAAY,CAAC,MAAM,UAAU;YACxC,qCAAqC;SACrC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,OAAO;YACN,OAAO,EAAE,IAAI;YACb,MAAM;YACN,YAAY;YACZ,IAAI,EAAE,EAAE,EAAE,uCAAuC;YACjD,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,mDAAmD;YACnD,KAAK,EAAE;gBACN,WAAW,EAAE,IAAI;gBACjB,YAAY,EAAE,GAAG;gBACjB,WAAW,EAAE,IAAI;aACjB;SACD,CAAC;IACH,CAAC;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAkB;IAClD,IAAI,EAAE,eAAe;IAErB,KAAK,CAAC,WAAW;QAChB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAwB;QACjC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACrB,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,SAAS;gBACjB,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,sBAAsB;aAC7B,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACpC,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,iEAAiE;gBACzE,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aAClC,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7D,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,EAAE;gBACV,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,6CAA6C;aACpD,CAAC;QACH,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QACxE,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QAExE,OAAO;YACN,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC;gBACtB,cAAc;gBACd,WAAW;gBACX,OAAO,EACN,cAAc,KAAK,MAAM,IAAI,WAAW,KAAK,MAAM;oBAClD,CAAC,CAAC,gCAAgC;oBAClC,CAAC,CAAC,yCAAyC,cAAc,aAAa,WAAW,EAAE;aACrF,CAAC;YACF,YAAY,EAAE,EAAE;YAChB,IAAI,EAAE,EAAE;YACR,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,KAAK,EAAE;gBACN,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,GAAG;gBACjB,WAAW,EAAE,GAAG;aAChB;SACD,CAAC;IACH,CAAC;CACD,CAAC;AAEF,aAAa,CAAC,YAAY,CAAC,CAAC;AAC5B,aAAa,CAAC,oBAAoB,CAAC,CAAC;AACpC,aAAa,CAAC;IACb,GAAG,oBAAoB;IACvB,IAAI,EAAE,sBAAsB;CAC5B,CAAC,CAAC;AAEH,eAAe,YAAY,CAAC"}

package/dist/agents/opencode.d.ts

@@ -0,0 +1,23 @@
+/**
+ * OpenCode agent backend.
+ *
+ * Three execution modes (in priority order):
+ *   1. Server mode: Connects to a managed `opencode serve` instance via HTTP API.
+ *      Avoids cold-start overhead. Server is shared across threads.
+ *   2. Attach mode: Uses `opencode run --attach <url>` to connect to a running server.
+ *      Falls back here if direct HTTP API call fails.
+ *   3. Subprocess mode: Cold-starts `opencode run --format json` per invocation.
+ *      Always-available fallback.
+ *
+ * The server pool manages lifecycle of `opencode serve` processes, starting them
+ * lazily and shutting them down when the session ends.
+ */
+import type { AgentProvider } from "../core/types.js";
+/** Enable server mode for the OpenCode agent. Call before spawning threads. */
+export declare function enableServerMode(): void;
+/** Disable server mode and shut down all managed servers. */
+export declare function disableServerMode(): Promise<void>;
+/** Get the number of active server instances. */
+export declare function getActiveServerCount(): number;
+declare const openCodeProvider: AgentProvider;
+export default openCodeProvider;