duocode 0.1.0

package/dist/context/project_context.js

@@ -0,0 +1,292 @@
+/**
+ * Smart context gathering — scans the project, builds a tree summary,
+ * and identifies relevant files based on the task prompt.
+ */
+import fs from "node:fs";
+import path from "node:path";
+const DEFAULT_IGNORE_DIRS = new Set([
+    "node_modules", ".git", ".duocode", "dist", "build", "out",
+    ".next", ".nuxt", ".svelte-kit", "__pycache__", ".venv",
+    "venv", "env", ".tox", "coverage", ".nyc_output", ".cache",
+    ".parcel-cache", ".turbo", ".vercel", ".output",
+]);
+const DEFAULT_IGNORE_PATTERNS = [
+    /\.lock$/,
+    /package-lock\.json$/,
+    /yarn\.lock$/,
+    /pnpm-lock\.yaml$/,
+    /\.min\.(js|css)$/,
+    /\.map$/,
+    /\.d\.ts$/,
+    /\.(png|jpg|jpeg|gif|ico|svg|woff|woff2|ttf|eot|mp4|mp3|webm|webp)$/i,
+    /\.(zip|tar|gz|bz2|7z|rar)$/i,
+    /\.(exe|dll|so|dylib|bin)$/i,
+];
+/**
+ * Scan a project directory and return a flat list of files.
+ * Respects .gitignore-style patterns and depth limits.
+ */
+export function scanProjectTree(rootDir, config = {}) {
+    const maxDepth = config.maxDepth ?? 4;
+    const maxFiles = config.maxTreeFiles ?? 200;
+    const ignoreDirs = new Set([...DEFAULT_IGNORE_DIRS, ...(config.ignoreDirs ?? [])]);
+    const ignorePatterns = [
+        ...DEFAULT_IGNORE_PATTERNS,
+        ...(config.ignorePatterns ?? []).map((p) => new RegExp(p)),
+    ];
+    const files = [];
+    let truncated = false;
+    function walk(dir, depth) {
+        if (depth > maxDepth || files.length >= maxFiles) {
+            truncated = true;
+            return;
+        }
+        let entries;
+        try {
+            entries = fs.readdirSync(dir, { withFileTypes: true });
+        }
+        catch {
+            return; // Permission denied or similar
+        }
+        // Sort for deterministic output
+        entries.sort((a, b) => a.name.localeCompare(b.name));
+        for (const entry of entries) {
+            if (files.length >= maxFiles) {
+                truncated = true;
+                return;
+            }
+            if (entry.name.startsWith(".") && entry.name !== ".env.example")
+                continue;
+            const fullPath = path.join(dir, entry.name);
+            const relativePath = path.relative(rootDir, fullPath);
+            if (entry.isDirectory()) {
+                if (ignoreDirs.has(entry.name))
+                    continue;
+                walk(fullPath, depth + 1);
+            }
+            else if (entry.isFile()) {
+                if (ignorePatterns.some((p) => p.test(entry.name)))
+                    continue;
+                let stat;
+                try {
+                    stat = fs.statSync(fullPath);
+                }
+                catch {
+                    continue;
+                }
+                files.push({
+                    relativePath,
+                    absolutePath: fullPath,
+                    sizeBytes: stat.size,
+                    ext: path.extname(entry.name).toLowerCase(),
+                    mtimeMs: stat.mtimeMs,
+                });
+            }
+        }
+    }
+    walk(rootDir, 0);
+    return {
+        root: rootDir,
+        files,
+        totalFiles: files.length,
+        truncated,
+    };
+}
+// ── Tree Rendering ────────────────────────────────────────────────
+/**
+ * Render a directory tree as a concise indented string.
+ */
+export function renderTree(tree) {
+    const byDir = new Map();
+    for (const file of tree.files) {
+        const dir = path.dirname(file.relativePath);
+        const list = byDir.get(dir) ?? [];
+        list.push(path.basename(file.relativePath));
+        byDir.set(dir, list);
+    }
+    const lines = [`${path.basename(tree.root)}/`];
+    const sortedDirs = [...byDir.keys()].sort();
+    for (const dir of sortedDirs) {
+        const depth = dir === "." ? 0 : dir.split(path.sep).length;
+        const indent = "  ".repeat(depth);
+        const filesList = byDir.get(dir);
+        if (dir !== ".") {
+            lines.push(`${indent}${path.basename(dir)}/`);
+        }
+        const fileIndent = "  ".repeat(depth + 1);
+        for (const f of filesList) {
+            lines.push(`${fileIndent}${f}`);
+        }
+    }
+    if (tree.truncated) {
+        lines.push(`  ... (truncated at ${tree.totalFiles} files)`);
+    }
+    return lines.join("\n");
+}
+// ── Relevance Detection ───────────────────────────────────────────
+/**
+ * Score files by relevance to a prompt using keyword matching.
+ * Returns files sorted by relevance score (highest first).
+ */
+export function rankFilesByRelevance(files, prompt) {
+    const words = extractKeywords(prompt);
+    if (words.length === 0)
+        return [];
+    return files
+        .map((file) => {
+        let score = 0;
+        const fileParts = file.relativePath.toLowerCase().split(/[/\\.]/);
+        for (const word of words) {
+            // Exact filename match
+            if (fileParts.some((p) => p === word))
+                score += 3;
+            // Partial filename match
+            else if (fileParts.some((p) => p.includes(word)))
+                score += 1;
+        }
+        // Boost source files over configs
+        if (isSrcFile(file.ext))
+            score += 0.5;
+        // Boost test files when prompt mentions testing
+        if (isTestFile(file.relativePath) && words.some((w) => /test|spec|assert/.test(w))) {
+            score += 2;
+        }
+        return { ...file, relevanceScore: score };
+    })
+        .filter((f) => f.relevanceScore > 0)
+        .sort((a, b) => b.relevanceScore - a.relevanceScore);
+}
+function extractKeywords(prompt) {
+    const stopWords = new Set([
+        "the", "a", "an", "is", "are", "was", "were", "be", "been",
+        "being", "have", "has", "had", "do", "does", "did", "will",
+        "would", "could", "should", "may", "might", "shall", "can",
+        "to", "of", "in", "for", "on", "with", "at", "by", "from",
+        "as", "into", "through", "during", "before", "after", "this",
+        "that", "these", "those", "it", "its", "and", "or", "but",
+        "not", "no", "if", "then", "else", "when", "up", "out",
+        "about", "each", "which", "their", "what", "so", "some",
+        "all", "any", "most", "other", "than", "too", "very",
+        "just", "only", "also", "how", "where", "why", "who",
+        "add", "create", "make", "build", "write", "implement",
+        "fix", "update", "change", "modify", "please", "need",
+        "want", "file", "code",
+    ]);
+    return prompt
+        .toLowerCase()
+        .replace(/[^a-z0-9_-]/g, " ")
+        .split(/\s+/)
+        .filter((w) => w.length > 2 && !stopWords.has(w));
+}
+function isSrcFile(ext) {
+    return [".ts", ".tsx", ".js", ".jsx", ".py", ".go", ".rs", ".java", ".c", ".cpp", ".h"].includes(ext);
+}
+function isTestFile(relativePath) {
+    return /\.(test|spec)\.(ts|tsx|js|jsx|py)$/.test(relativePath)
+        || relativePath.includes("__tests__")
+        || relativePath.startsWith("tests/")
+        || relativePath.startsWith("test/");
+}
+/**
+ * Build smart context from a project directory and task prompt.
+ * Includes:
+ * 1. Project tree summary
+ * 2. Contents of relevant files (within token budget)
+ * 3. Explicit @ mentions
+ */
+export function buildProjectContext(rootDir, prompt, config = {}) {
+    const maxChars = config.maxContextChars ?? 60000;
+    const maxFileSize = config.maxFileSizeBytes ?? 100000;
+    // 1. Scan tree
+    const tree = scanProjectTree(rootDir, config);
+    const treeSummary = renderTree(tree);
+    // 2. Extract @ mentions
+    const mentions = extractMentions(prompt, rootDir);
+    // 3. Rank remaining files by relevance
+    const ranked = rankFilesByRelevance(tree.files.filter((f) => !mentions.some((m) => m.absolutePath === f.absolutePath)), prompt);
+    // 4. Assemble context within budget
+    const parts = [];
+    const includedFiles = [];
+    const droppedFiles = [];
+    // Project tree (always included, usually ~2-4KB)
+    const treeSection = `## Project Structure\n\`\`\`\n${treeSummary}\n\`\`\`\n`;
+    parts.push(treeSection);
+    let charBudget = maxChars - treeSection.length;
+    // @ mentioned files first (highest priority)
+    for (const mention of mentions) {
+        if (charBudget <= 0)
+            break;
+        const content = safeReadFile(mention.absolutePath, maxFileSize);
+        if (!content)
+            continue;
+        if (content.length > charBudget) {
+            droppedFiles.push(mention.relativePath);
+            continue;
+        }
+        const section = `## ${mention.relativePath}\n\`\`\`\n${content}\n\`\`\`\n`;
+        parts.push(section);
+        includedFiles.push({ path: mention.relativePath, content, reason: "@ mention" });
+        charBudget -= section.length;
+    }
+    // Relevant files next
+    for (const file of ranked) {
+        if (charBudget <= 0)
+            break;
+        const content = safeReadFile(file.absolutePath, maxFileSize);
+        if (!content)
+            continue;
+        if (content.length > charBudget) {
+            droppedFiles.push(file.relativePath);
+            continue;
+        }
+        const section = `## ${file.relativePath}\n\`\`\`\n${content}\n\`\`\`\n`;
+        parts.push(section);
+        includedFiles.push({
+            path: file.relativePath,
+            content,
+            reason: `relevance: ${file.relevanceScore.toFixed(1)}`,
+        });
+        charBudget -= section.length;
+    }
+    const systemContext = parts.join("\n");
+    return {
+        systemContext,
+        includedFiles,
+        totalChars: systemContext.length,
+        droppedFiles,
+    };
+}
+// ── @ Mention Parsing ─────────────────────────────────────────────
+export function extractMentions(prompt, rootDir) {
+    const mentions = [];
+    // Match @path/to/file.ext or @"path with spaces/file.ext"
+    const regex = /@(?:"([^"]+)"|(\S+\.\w+))/g;
+    let match;
+    while ((match = regex.exec(prompt)) !== null) {
+        const relativePath = match[1] ?? match[2];
+        const absolutePath = path.resolve(rootDir, relativePath);
+        if (fs.existsSync(absolutePath) && fs.statSync(absolutePath).isFile()) {
+            mentions.push({ relativePath, absolutePath });
+        }
+    }
+    return mentions;
+}
+/**
+ * Strip @mentions from prompt text so the model sees clean input.
+ * Returns the prompt with @path references removed.
+ */
+export function stripMentions(prompt) {
+    return prompt.replace(/@(?:"[^"]+"|(\S+\.\w+))/g, "").replace(/\s{2,}/g, " ").trim();
+}
+function safeReadFile(filePath, maxBytes) {
+    try {
+        const stat = fs.statSync(filePath);
+        if (stat.size > maxBytes)
+            return null;
+        return fs.readFileSync(filePath, "utf8");
+    }
+    catch {
+        return null;
+    }
+}
+//# sourceMappingURL=project_context.js.map

package/dist/context/project_context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"project_context.js","sourceRoot":"","sources":["../../src/context/project_context.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAmB7B,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC;IAClC,cAAc,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK;IAC1D,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,OAAO;IACvD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,QAAQ;IAC1D,eAAe,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS;CAChD,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG;IAC9B,SAAS;IACT,qBAAqB;IACrB,aAAa;IACb,kBAAkB;IAClB,kBAAkB;IAClB,QAAQ;IACR,UAAU;IACV,qEAAqE;IACrE,6BAA6B;IAC7B,4BAA4B;CAC7B,CAAC;AAwBF;;;GAGG;AACH,MAAM,UAAU,eAAe,CAC7B,OAAe,EACf,SAAwB,EAAE;IAE1B,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,MAAM,CAAC,YAAY,IAAI,GAAG,CAAC;IAC5C,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,mBAAmB,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACnF,MAAM,cAAc,GAAG;QACrB,GAAG,uBAAuB;QAC1B,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC;KAC3D,CAAC;IAEF,MAAM,KAAK,GAAgB,EAAE,CAAC;IAC9B,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,SAAS,IAAI,CAAC,GAAW,EAAE,KAAa;QACtC,IAAI,KAAK,GAAG,QAAQ,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;YACjD,SAAS,GAAG,IAAI,CAAC;YACjB,OAAO;QACT,CAAC;QAED,IAAI,OAAoB,CAAC;QACzB,IAAI,CAAC;YACH,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,+BAA+B;QACzC,CAAC;QAED,gCAAgC;QAChC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAErD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;gBAC7B,SAAS,GAAG,IAAI,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc;gBAAE,SAAS;YAE1E,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAEtD,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACzC,IAAI,CAAC,QAAQ,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YAC5B,CAAC;iBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC1B,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAAE,SAAS;gBAE7D,IAAI,IAAc,CAAC;gBACnB,IAAI,CAAC;oBACH,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBAC/B,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;gBAED,KAAK,CAAC,IAAI,CAAC;oBACT,YAAY;oBACZ,YAAY,EAAE,QAAQ;oBACtB,SAAS,EAAE,IAAI,CAAC,IAAI;oBACpB,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE;oBAC3C,OAAO,EAAE,IAAI,CAAC,OAAO;iBACtB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEjB,OAAO;QACL,IAAI,EAAE,OAAO;QACb,KAAK;QACL,UAAU,EAAE,KAAK,CAAC,MAAM;QACxB,SAAS;KACV,CAAC;AACJ,CAAC;AAED,qEAAqE;AAErE;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,IAAmB;IAC5C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAoB,CAAC;IAE1C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QAC5C,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACvB,CAAC;IAED,MAAM,KAAK,GAAa,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzD,MAAM,UAAU,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAE5C,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;QAElC,IAAI,GAAG,KAAK,GAAG,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QAC1C,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,KAAK,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,UAAU,SAAS,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,qEAAqE;AAErE;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAClC,KAAkB,EAClB,MAAc;IAEd,MAAM,KAAK,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACtC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAElC,OAAO,KAAK;SACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAElE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,uBAAuB;YACvB,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC;gBAAE,KAAK,IAAI,CAAC,CAAC;YAClD,yBAAyB;iBACpB,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAAE,KAAK,IAAI,CAAC,CAAC;QAC/D,CAAC;QAED,kCAAkC;QAClC,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC;YAAE,KAAK,IAAI,GAAG,CAAC;QACtC,gDAAgD;QAChD,IAAI,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACnF,KAAK,IAAI,CAAC,CAAC;QACb,CAAC;QAED,OAAO,EAAE,GAAG,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;IAC5C,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC;SACnC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,eAAe,CAAC,MAAc;IACrC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC;QACxB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;QAC1D,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;QAC1D,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK;QAC1D,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM;QACzD,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM;QAC5D,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK;QACzD,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK;QACtD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;QACvD,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;QACpD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK;QACpD,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW;QACtD,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM;QACrD,MAAM,EAAE,MAAM,EAAE,MAAM;KACvB,CAAC,CAAC;IAEH,OAAO,MAAM;SACV,WAAW,EAAE;SACb,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC;SAC5B,KAAK,CAAC,KAAK,CAAC;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,SAAS,CAAC,GAAW;IAC5B,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;AACxG,CAAC;AAED,SAAS,UAAU,CAAC,YAAoB;IACtC,OAAO,oCAAoC,CAAC,IAAI,CAAC,YAAY,CAAC;WACzD,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC;WAClC,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC;WACjC,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;AACxC,CAAC;AAeD;;;;;;GAMG;AACH,MAAM,UAAU,mBAAmB,CACjC,OAAe,EACf,MAAc,EACd,SAAwB,EAAE;IAE1B,MAAM,QAAQ,GAAG,MAAM,CAAC,eAAe,IAAI,KAAM,CAAC;IAClD,MAAM,WAAW,GAAG,MAAM,CAAC,gBAAgB,IAAI,MAAO,CAAC;IAEvD,eAAe;IACf,MAAM,IAAI,GAAG,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC9C,MAAM,WAAW,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAErC,wBAAwB;IACxB,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAElD,uCAAuC;IACvC,MAAM,MAAM,GAAG,oBAAoB,CACjC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,EAClF,MAAM,CACP,CAAC;IAEF,oCAAoC;IACpC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,aAAa,GAAmC,EAAE,CAAC;IACzD,MAAM,YAAY,GAAa,EAAE,CAAC;IAElC,iDAAiD;IACjD,MAAM,WAAW,GAAG,iCAAiC,WAAW,YAAY,CAAC;IAC7E,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACxB,IAAI,UAAU,GAAG,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC;IAE/C,6CAA6C;IAC7C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,UAAU,IAAI,CAAC;YAAE,MAAM;QAC3B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;QAChE,IAAI,CAAC,OAAO;YAAE,SAAS;QAEvB,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;YAChC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACxC,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,YAAY,aAAa,OAAO,YAAY,CAAC;QAC3E,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QACjF,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,sBAAsB;IACtB,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAC1B,IAAI,UAAU,IAAI,CAAC;YAAE,MAAM;QAC3B,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;QAC7D,IAAI,CAAC,OAAO;YAAE,SAAS;QAEvB,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;YAChC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACrC,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,aAAa,OAAO,YAAY,CAAC;QACxE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,aAAa,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,OAAO;YACP,MAAM,EAAE,cAAc,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;SACvD,CAAC,CAAC;QACH,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvC,OAAO;QACL,aAAa;QACb,aAAa;QACb,UAAU,EAAE,aAAa,CAAC,MAAM;QAChC,YAAY;KACb,CAAC;AACJ,CAAC;AAED,qEAAqE;AAErE,MAAM,UAAU,eAAe,CAC7B,MAAc,EACd,OAAe;IAEf,MAAM,QAAQ,GAA0D,EAAE,CAAC;IAC3E,0DAA0D;IAC1D,MAAM,KAAK,GAAG,4BAA4B,CAAC;IAC3C,IAAI,KAA6B,CAAC;IAElC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC7C,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAEzD,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YACtE,QAAQ,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,MAAc;IAC1C,OAAO,MAAM,CAAC,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AACvF,CAAC;AAED,SAAS,YAAY,CAAC,QAAgB,EAAE,QAAgB;IACtD,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,IAAI,CAAC,IAAI,GAAG,QAAQ;YAAE,OAAO,IAAI,CAAC;QACtC,OAAO,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/context/token_budget.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Token budget estimation and context truncation.
+ *
+ * Uses a simple chars/3.5 heuristic for token estimation.
+ * Provides truncation strategies: oldest history → lowest relevance files → individual file trimming.
+ */
+export declare function estimateTokens(text: string): number;
+export interface TokenBudgetInfo {
+    contextWindow: number;
+    maxOutputTokens: number;
+    systemPromptTokens: number;
+    conversationTokens: number;
+    fileContextTokens: number;
+    availableForUser: number;
+    utilizationPct: number;
+}
+export declare function computeBudget(opts: {
+    contextWindow: number;
+    maxOutputTokens: number;
+    systemPrompt: string;
+    conversationHistory: string;
+    fileContext: string;
+}): TokenBudgetInfo;
+export declare function truncateConversationHistory(history: string, maxTokens: number): string;
+export interface FileContextEntry {
+    path: string;
+    content: string;
+    relevance: number;
+}
+export declare function truncateFileContext(files: FileContextEntry[], maxTokens: number): {
+    kept: FileContextEntry[];
+    dropped: string[];
+};
+export declare function truncateFileContent(content: string, maxLines: number): string;
+export declare function formatBudgetStatus(budget: TokenBudgetInfo): string;

package/dist/context/token_budget.js

@@ -0,0 +1,81 @@
+/**
+ * Token budget estimation and context truncation.
+ *
+ * Uses a simple chars/3.5 heuristic for token estimation.
+ * Provides truncation strategies: oldest history → lowest relevance files → individual file trimming.
+ */
+// ── Estimation ──────────────────────────────────────────────────
+export function estimateTokens(text) {
+    return Math.ceil(text.length / 3.5);
+}
+export function computeBudget(opts) {
+    const systemPromptTokens = estimateTokens(opts.systemPrompt);
+    const conversationTokens = estimateTokens(opts.conversationHistory);
+    const fileContextTokens = estimateTokens(opts.fileContext);
+    const used = systemPromptTokens + conversationTokens + fileContextTokens + opts.maxOutputTokens;
+    const availableForUser = Math.max(0, opts.contextWindow - used);
+    const utilizationPct = Math.min(100, (used / opts.contextWindow) * 100);
+    return {
+        contextWindow: opts.contextWindow,
+        maxOutputTokens: opts.maxOutputTokens,
+        systemPromptTokens,
+        conversationTokens,
+        fileContextTokens,
+        availableForUser,
+        utilizationPct,
+    };
+}
+// ── Truncation: conversation history (oldest turns first) ───────
+export function truncateConversationHistory(history, maxTokens) {
+    if (!history)
+        return history;
+    const currentTokens = estimateTokens(history);
+    if (currentTokens <= maxTokens)
+        return history;
+    // Split by turn markers and drop oldest
+    const turns = history.split(/(?=### Turn \d+)/);
+    while (turns.length > 1 && estimateTokens(turns.join("")) > maxTokens) {
+        turns.shift();
+    }
+    const result = turns.join("");
+    if (estimateTokens(result) > maxTokens) {
+        const targetChars = Math.floor(maxTokens * 3.5);
+        return "[...earlier context truncated]\n" + result.slice(-targetChars);
+    }
+    return result;
+}
+export function truncateFileContext(files, maxTokens) {
+    const sorted = [...files].sort((a, b) => b.relevance - a.relevance);
+    const kept = [];
+    const dropped = [];
+    let usedTokens = 0;
+    for (const file of sorted) {
+        const fileTokens = estimateTokens(file.content);
+        if (usedTokens + fileTokens <= maxTokens) {
+            kept.push(file);
+            usedTokens += fileTokens;
+        }
+        else {
+            dropped.push(file.path);
+        }
+    }
+    return { kept, dropped };
+}
+// ── Truncation: individual file content (keep head + tail) ──────
+export function truncateFileContent(content, maxLines) {
+    const lines = content.split("\n");
+    if (lines.length <= maxLines)
+        return content;
+    const keepEach = Math.floor(maxLines / 2);
+    const head = lines.slice(0, keepEach);
+    const tail = lines.slice(-keepEach);
+    const dropped = lines.length - keepEach * 2;
+    return [...head, `\n... (${dropped} lines omitted) ...\n`, ...tail].join("\n");
+}
+// ── Display ─────────────────────────────────────────────────────
+export function formatBudgetStatus(budget) {
+    const used = budget.systemPromptTokens + budget.conversationTokens + budget.fileContextTokens;
+    return (`~${used.toLocaleString()} input tokens / ${budget.contextWindow.toLocaleString()} window ` +
+        `(${budget.utilizationPct.toFixed(0)}% used)`);
+}
+//# sourceMappingURL=token_budget.js.map

package/dist/context/token_budget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token_budget.js","sourceRoot":"","sources":["../../src/context/token_budget.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,mEAAmE;AAEnE,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;AACtC,CAAC;AAcD,MAAM,UAAU,aAAa,CAAC,IAM7B;IACC,MAAM,kBAAkB,GAAG,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC7D,MAAM,kBAAkB,GAAG,cAAc,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACpE,MAAM,iBAAiB,GAAG,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3D,MAAM,IAAI,GAAG,kBAAkB,GAAG,kBAAkB,GAAG,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC;IAChG,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC;IAChE,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC;IAExE,OAAO;QACL,aAAa,EAAE,IAAI,CAAC,aAAa;QACjC,eAAe,EAAE,IAAI,CAAC,eAAe;QACrC,kBAAkB;QAClB,kBAAkB;QAClB,iBAAiB;QACjB,gBAAgB;QAChB,cAAc;KACf,CAAC;AACJ,CAAC;AAED,mEAAmE;AAEnE,MAAM,UAAU,2BAA2B,CACzC,OAAe,EACf,SAAiB;IAEjB,IAAI,CAAC,OAAO;QAAE,OAAO,OAAO,CAAC;IAC7B,MAAM,aAAa,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAI,aAAa,IAAI,SAAS;QAAE,OAAO,OAAO,CAAC;IAE/C,wCAAwC;IACxC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAChD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,GAAG,SAAS,EAAE,CAAC;QACtE,KAAK,CAAC,KAAK,EAAE,CAAC;IAChB,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9B,IAAI,cAAc,CAAC,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;QAChD,OAAO,kCAAkC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC;IACzE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAUD,MAAM,UAAU,mBAAmB,CACjC,KAAyB,EACzB,SAAiB;IAEjB,MAAM,MAAM,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IACpE,MAAM,IAAI,GAAuB,EAAE,CAAC;IACpC,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAC1B,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,IAAI,UAAU,GAAG,UAAU,IAAI,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,UAAU,IAAI,UAAU,CAAC;QAC3B,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AAC3B,CAAC;AAED,mEAAmE;AAEnE,MAAM,UAAU,mBAAmB,CACjC,OAAe,EACf,QAAgB;IAEhB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ;QAAE,OAAO,OAAO,CAAC;IAE7C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;IAC1C,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtC,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,GAAG,QAAQ,GAAG,CAAC,CAAC;IAE5C,OAAO,CAAC,GAAG,IAAI,EAAE,UAAU,OAAO,uBAAuB,EAAE,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACjF,CAAC;AAED,mEAAmE;AAEnE,MAAM,UAAU,kBAAkB,CAAC,MAAuB;IACxD,MAAM,IAAI,GAAG,MAAM,CAAC,kBAAkB,GAAG,MAAM,CAAC,kBAAkB,GAAG,MAAM,CAAC,iBAAiB,CAAC;IAC9F,OAAO,CACL,IAAI,IAAI,CAAC,cAAc,EAAE,mBAAmB,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,UAAU;QAC3F,IAAI,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAC9C,CAAC;AACJ,CAAC"}

package/dist/db/queries.d.ts

@@ -0,0 +1,121 @@
+import type Database from "better-sqlite3";
+import type { TaskPriority, TaskStatus } from "./schema.js";
+import type { DuoRole, GradingResult } from "../duo/duo_types.js";
+export interface TaskRow {
+    id: string;
+    title: string;
+    description: string;
+    priority: TaskPriority;
+    required_capabilities: string;
+    status: TaskStatus;
+    worktree_path: string | null;
+    created_at: string;
+    updated_at: string;
+}
+export interface NegotiationSessionRow {
+    id: string;
+    task_id: string | null;
+    file_path: string;
+    language: "typescript" | "python" | "go";
+    phase: string;
+    conflict_count: number;
+    resolution_method: string | null;
+    winner_agent_id: string | null;
+    confidence: number | null;
+    explanation: string | null;
+    merged_source: string | null;
+    created_at: string;
+    updated_at: string;
+}
+export interface ProviderUsageRow {
+    request_id: string;
+    task_id: string | null;
+    provider: "openai" | "anthropic" | "openrouter";
+    model: string;
+    input_tokens: number;
+    output_tokens: number;
+    cache_read_input_tokens: number;
+    cache_write_input_tokens: number;
+    cost_usd: number;
+    created_at: string;
+}
+export interface SharedContextRow {
+    id: number;
+    task_id: string;
+    file_path: string;
+    ownership_path: string | null;
+    publisher_agent_id: string;
+    content: string;
+    visibility_csv: string;
+    created_at: string;
+}
+export interface NegotiationLearningRow {
+    id: number;
+    negotiation_id: string;
+    task_id: string | null;
+    file_path: string;
+    resolution_method: "auto_compose" | "debate_winner" | "escalated";
+    winner_agent_id: string | null;
+    confidence: number;
+    conflict_count: number;
+    round_count: number;
+    consensus_trend: number;
+    recommended_action: string;
+    created_at: string;
+}
+export declare function getTaskById(db: Database.Database, taskId: string): TaskRow | null;
+export declare function listTasksByStatus(db: Database.Database, status?: TaskStatus): TaskRow[];
+export declare function getNegotiationSessionById(db: Database.Database, negotiationId: string): NegotiationSessionRow | null;
+export declare function listNegotiationConflicts(db: Database.Database, negotiationId: string): Array<{
+    ownership_path: string;
+    left_type: string;
+    right_type: string;
+}>;
+export declare function listNegotiationRounds(db: Database.Database, negotiationId: string): Array<{
+    round_number: number;
+    left_score: number;
+    right_score: number;
+    consensus: number;
+    reasoning: string;
+}>;
+export declare function listProviderUsageByTask(db: Database.Database, taskId: string): ProviderUsageRow[];
+export declare function getProviderSpendTotals(db: Database.Database): {
+    total_usd: number;
+    openai_usd: number;
+    anthropic_usd: number;
+    openrouter_usd: number;
+};
+export declare function listSharedContextByTask(db: Database.Database, taskId: string): SharedContextRow[];
+export declare function getNegotiationLearningBySession(db: Database.Database, negotiationId: string): NegotiationLearningRow | null;
+export declare function listNegotiationLearningByTask(db: Database.Database, taskId: string): NegotiationLearningRow[];
+export interface EloRatingRow {
+    id: number;
+    model: string;
+    role: DuoRole;
+    capability_bucket: string;
+    elo: number;
+    match_count: number;
+    last_updated: string;
+}
+export declare function getEloRating(db: Database.Database, model: string, role: DuoRole, capabilityBucket: string): EloRatingRow | null;
+export declare function upsertEloRating(db: Database.Database, model: string, role: DuoRole, capabilityBucket: string, elo: number, matchCount: number): void;
+export declare function listEloRatingsByRole(db: Database.Database, role: DuoRole): EloRatingRow[];
+export declare function getAggregateElo(db: Database.Database, model: string, role: DuoRole): number;
+export interface GradingResultRow {
+    id: number;
+    task_id: string | null;
+    model: string;
+    role: DuoRole;
+    correctness: number;
+    security: number;
+    efficiency: number;
+    robustness: number;
+    maintainability: number;
+    usability: number;
+    composite_score: number;
+    graded_by: string;
+    created_at: string;
+}
+export declare function insertGradingResult(db: Database.Database, result: GradingResult): void;
+export declare function listGradingResultsByModel(db: Database.Database, model: string): GradingResultRow[];
+export declare function getAggregateMatchCount(db: Database.Database, model: string, role: DuoRole): number;

package/dist/db/queries.js

@@ -0,0 +1,109 @@
+export function getTaskById(db, taskId) {
+    return db.prepare(`SELECT * FROM tasks WHERE id = ?`).get(taskId);
+}
+export function listTasksByStatus(db, status) {
+    if (!status) {
+        return db.prepare(`SELECT * FROM tasks ORDER BY created_at DESC`).all();
+    }
+    return db
+        .prepare(`SELECT * FROM tasks WHERE status = ? ORDER BY created_at DESC`)
+        .all(status);
+}
+export function getNegotiationSessionById(db, negotiationId) {
+    return db
+        .prepare(`SELECT * FROM negotiation_sessions WHERE id = ?`)
+        .get(negotiationId);
+}
+export function listNegotiationConflicts(db, negotiationId) {
+    return db
+        .prepare(`SELECT ownership_path, left_type, right_type FROM negotiation_conflicts WHERE negotiation_id = ? ORDER BY id ASC`)
+        .all(negotiationId);
+}
+export function listNegotiationRounds(db, negotiationId) {
+    return db
+        .prepare(`SELECT round_number, left_score, right_score, consensus, reasoning
+       FROM negotiation_rounds
+       WHERE negotiation_id = ?
+       ORDER BY round_number ASC`)
+        .all(negotiationId);
+}
+export function listProviderUsageByTask(db, taskId) {
+    return db
+        .prepare(`SELECT * FROM provider_usage WHERE task_id = ? ORDER BY created_at DESC`)
+        .all(taskId);
+}
+export function getProviderSpendTotals(db) {
+    return db
+        .prepare(`
+      SELECT
+        COALESCE(SUM(cost_usd), 0) AS total_usd,
+        COALESCE(SUM(CASE WHEN provider = 'openai' THEN cost_usd ELSE 0 END), 0) AS openai_usd,
+        COALESCE(SUM(CASE WHEN provider = 'anthropic' THEN cost_usd ELSE 0 END), 0) AS anthropic_usd,
+        COALESCE(SUM(CASE WHEN provider = 'openrouter' THEN cost_usd ELSE 0 END), 0) AS openrouter_usd
+      FROM provider_usage
+    `)
+        .get();
+}
+export function listSharedContextByTask(db, taskId) {
+    return db
+        .prepare(`SELECT * FROM shared_context WHERE task_id = ? ORDER BY created_at ASC`)
+        .all(taskId);
+}
+export function getNegotiationLearningBySession(db, negotiationId) {
+    return db
+        .prepare(`SELECT * FROM negotiation_learning WHERE negotiation_id = ?`)
+        .get(negotiationId);
+}
+export function listNegotiationLearningByTask(db, taskId) {
+    return db
+        .prepare(`SELECT * FROM negotiation_learning WHERE task_id = ? ORDER BY created_at DESC`)
+        .all(taskId);
+}
+export function getEloRating(db, model, role, capabilityBucket) {
+    return db
+        .prepare(`SELECT * FROM elo_ratings WHERE model = ? AND role = ? AND capability_bucket = ?`)
+        .get(model, role, capabilityBucket);
+}
+export function upsertEloRating(db, model, role, capabilityBucket, elo, matchCount) {
+    db.prepare(`
+    INSERT INTO elo_ratings (model, role, capability_bucket, elo, match_count, last_updated)
+    VALUES (?, ?, ?, ?, ?, datetime('now'))
+    ON CONFLICT(model, role, capability_bucket) DO UPDATE SET
+      elo = excluded.elo,
+      match_count = excluded.match_count,
+      last_updated = datetime('now')
+  `).run(model, role, capabilityBucket, elo, matchCount);
+}
+export function listEloRatingsByRole(db, role) {
+    return db
+        .prepare(`SELECT * FROM elo_ratings WHERE role = ? ORDER BY elo DESC`)
+        .all(role);
+}
+export function getAggregateElo(db, model, role) {
+    const row = db
+        .prepare(`SELECT AVG(elo) as avg_elo FROM elo_ratings WHERE model = ? AND role = ?`)
+        .get(model, role);
+    return row?.avg_elo ?? 1500;
+}
+export function insertGradingResult(db, result) {
+    const dims = Object.fromEntries(result.dimensions.map((d) => [d.dimension, d.score]));
+    db.prepare(`
+    INSERT INTO grading_results (
+      task_id, model, role,
+      correctness, security, efficiency, robustness, maintainability, usability,
+      composite_score, graded_by
+    ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(result.taskId ?? null, result.model, result.role, dims.correctness ?? 0, dims.security ?? 0, dims.efficiency ?? 0, dims.robustness ?? 0, dims.maintainability ?? 0, dims.usability ?? 0, result.compositeScore, result.gradedBy);
+}
+export function listGradingResultsByModel(db, model) {
+    return db
+        .prepare(`SELECT * FROM grading_results WHERE model = ? ORDER BY created_at DESC`)
+        .all(model);
+}
+export function getAggregateMatchCount(db, model, role) {
+    const row = db
+        .prepare(`SELECT COALESCE(SUM(match_count), 0) as total FROM elo_ratings WHERE model = ? AND role = ?`)
+        .get(model, role);
+    return row?.total ?? 0;
+}
+//# sourceMappingURL=queries.js.map

package/dist/db/queries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queries.js","sourceRoot":"","sources":["../../src/db/queries.ts"],"names":[],"mappings":"AAuEA,MAAM,UAAU,WAAW,CAAC,EAAqB,EAAE,MAAc;IAC/D,OAAO,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAmB,CAAC;AACtF,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,EAAqB,EAAE,MAAmB;IAC1E,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC,OAAO,CAAC,8CAA8C,CAAC,CAAC,GAAG,EAAe,CAAC;IACvF,CAAC;IAED,OAAO,EAAE;SACN,OAAO,CAAC,+DAA+D,CAAC;SACxE,GAAG,CAAC,MAAM,CAAc,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,EAAqB,EACrB,aAAqB;IAErB,OAAO,EAAE;SACN,OAAO,CAAC,iDAAiD,CAAC;SAC1D,GAAG,CAAC,aAAa,CAAiC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,EAAqB,EACrB,aAAqB;IAErB,OAAO,EAAE;SACN,OAAO,CACN,kHAAkH,CACnH;SACA,GAAG,CAAC,aAAa,CAA6E,CAAC;AACpG,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,EAAqB,EACrB,aAAqB;IAErB,OAAO,EAAE;SACN,OAAO,CACN;;;iCAG2B,CAC5B;SACA,GAAG,CAAC,aAAa,CAAmH,CAAC;AAC1I,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,EAAqB,EAAE,MAAc;IAC3E,OAAO,EAAE;SACN,OAAO,CAAC,yEAAyE,CAAC;SAClF,GAAG,CAAC,MAAM,CAAuB,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,EAAqB;IAM1D,OAAO,EAAE;SACN,OAAO,CAAC;;;;;;;KAOR,CAAC;SACD,GAAG,EAA8F,CAAC;AACvG,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,EAAqB,EAAE,MAAc;IAC3E,OAAO,EAAE;SACN,OAAO,CAAC,wEAAwE,CAAC;SACjF,GAAG,CAAC,MAAM,CAAuB,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC7C,EAAqB,EACrB,aAAqB;IAErB,OAAO,EAAE;SACN,OAAO,CAAC,6DAA6D,CAAC;SACtE,GAAG,CAAC,aAAa,CAAkC,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,EAAqB,EACrB,MAAc;IAEd,OAAO,EAAE;SACN,OAAO,CAAC,+EAA+E,CAAC;SACxF,GAAG,CAAC,MAAM,CAA6B,CAAC;AAC7C,CAAC;AAcD,MAAM,UAAU,YAAY,CAC1B,EAAqB,EACrB,KAAa,EACb,IAAa,EACb,gBAAwB;IAExB,OAAO,EAAE;SACN,OAAO,CAAC,kFAAkF,CAAC;SAC3F,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,gBAAgB,CAAwB,CAAC;AAC/D,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,EAAqB,EACrB,KAAa,EACb,IAAa,EACb,gBAAwB,EACxB,GAAW,EACX,UAAkB;IAElB,EAAE,CAAC,OAAO,CAAC;;;;;;;GAOV,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,EAAqB,EACrB,IAAa;IAEb,OAAO,EAAE;SACN,OAAO,CAAC,4DAA4D,CAAC;SACrE,GAAG,CAAC,IAAI,CAAmB,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,EAAqB,EACrB,KAAa,EACb,IAAa;IAEb,MAAM,GAAG,GAAG,EAAE;SACX,OAAO,CAAC,0EAA0E,CAAC;SACnF,GAAG,CAAC,KAAK,EAAE,IAAI,CAAsC,CAAC;IACzD,OAAO,GAAG,EAAE,OAAO,IAAI,IAAI,CAAC;AAC9B,CAAC;AAoBD,MAAM,UAAU,mBAAmB,CACjC,EAAqB,EACrB,MAAqB;IAErB,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAC7B,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CACrD,CAAC;IAEF,EAAE,CAAC,OAAO,CAAC;;;;;;GAMV,CAAC,CAAC,GAAG,CACJ,MAAM,CAAC,MAAM,IAAI,IAAI,EACrB,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,IAAI,EACX,IAAI,CAAC,WAAW,IAAI,CAAC,EACrB,IAAI,CAAC,QAAQ,IAAI,CAAC,EAClB,IAAI,CAAC,UAAU,IAAI,CAAC,EACpB,IAAI,CAAC,UAAU,IAAI,CAAC,EACpB,IAAI,CAAC,eAAe,IAAI,CAAC,EACzB,IAAI,CAAC,SAAS,IAAI,CAAC,EACnB,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,QAAQ,CAChB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,EAAqB,EACrB,KAAa;IAEb,OAAO,EAAE;SACN,OAAO,CAAC,wEAAwE,CAAC;SACjF,GAAG,CAAC,KAAK,CAAuB,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,EAAqB,EACrB,KAAa,EACb,IAAa;IAEb,MAAM,GAAG,GAAG,EAAE;SACX,OAAO,CAAC,6FAA6F,CAAC;SACtG,GAAG,CAAC,KAAK,EAAE,IAAI,CAA6B,CAAC;IAChD,OAAO,GAAG,EAAE,KAAK,IAAI,CAAC,CAAC;AACzB,CAAC"}