@dewtech/dare-cli 3.8.2 → 3.10.0

package/dist/__tests__/semantic-regression.test.js

@@ -0,0 +1,248 @@
+import { createHash } from 'node:crypto';
+import { readFileSync, readdirSync } from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import fs from 'fs-extra';
+import { describe, expect, it, vi } from 'vitest';
+import { EmbeddingModelMissingError } from '../graphrag/embeddings.js';
+import * as embeddingsModule from '../graphrag/embeddings.js';
+import { GraphRAG } from '../graphrag/graph-rag.js';
+import { hybridSearch } from '../graphrag/hybrid.js';
+import { JsonGraph } from '../graphrag/json-graph.js';
+const TEST_DIR = path.dirname(fileURLToPath(import.meta.url));
+const FIXTURE_PATH = path.join(TEST_DIR, '__fixtures__', 'semantic-queries.json');
+const PKG_ROOT = path.resolve(TEST_DIR, '..', '..');
+const ALLOWED_REL = path.join('src', 'graphrag', 'embeddings.ts');
+const FORBIDDEN_IMPORT_RE = /(?:import|require)\s*\(?['"](?:onnxruntime|@xenova\/transformers)|from\s+['"](?:onnxruntime|@xenova\/transformers)/;
+const EMBEDDING_DIM = 8;
+function canonicalSemanticKey(text) {
+    const normalized = text.toLowerCase();
+    if (normalized.includes('refund'))
+        return 'concept:refund_policy';
+    if (normalized.includes('billing') || normalized.includes('invoice'))
+        return 'concept:billing_invoice';
+    if (normalized.includes('cancel') ||
+        normalized.includes('termination') ||
+        normalized.includes('stop recurring')) {
+        return 'concept:cancel_subscription';
+    }
+    if (normalized.includes('downgrade') ||
+        normalized.includes('cheaper tier') ||
+        normalized.includes('lower tier')) {
+        return 'concept:downgrade_plan';
+    }
+    return `concept:${normalized.replace(/\s+/g, '_')}`;
+}
+function deterministicVector(text) {
+    const digest = createHash('sha256').update(canonicalSemanticKey(text)).digest();
+    const vector = new Float32Array(EMBEDDING_DIM);
+    for (let i = 0; i < EMBEDDING_DIM; i++) {
+        const left = digest[i] ?? 0;
+        const right = digest[i + EMBEDDING_DIM] ?? 0;
+        vector[i] = (left - right) / 255;
+    }
+    return vector;
+}
+function createDeterministicEmbedder() {
+    return {
+        dim: EMBEDDING_DIM,
+        async embed(text) {
+            return deterministicVector(text);
+        },
+    };
+}
+function buildFixtureNodes() {
+    return [
+        {
+            id: 'node:refund',
+            type: 'task',
+            label: 'refund policy handbook',
+            description: 'official refund policy for digital orders',
+            vector: Array.from(deterministicVector('refund policy')),
+        },
+        {
+            id: 'node:billing',
+            type: 'task',
+            label: 'billing invoice reconciliation',
+            description: 'billing invoice closing steps',
+            vector: Array.from(deterministicVector('billing invoice')),
+        },
+        {
+            id: 'node:cancel',
+            type: 'task',
+            label: 'subscription termination playbook',
+            description: 'stop recurring charge workflow',
+            vector: Array.from(deterministicVector('cancel subscription')),
+        },
+        {
+            id: 'node:downgrade',
+            type: 'task',
+            label: 'plan tier adjustment guide',
+            description: 'switch to a lower tier safely',
+            vector: Array.from(deterministicVector('downgrade plan')),
+        },
+        {
+            id: 'node:noise',
+            type: 'task',
+            label: 'engineering onboarding notes',
+            description: 'team process and meeting cadence',
+            vector: Array.from(deterministicVector('engineering onboarding')),
+        },
+    ];
+}
+function readLabeledQueries() {
+    return JSON.parse(readFileSync(FIXTURE_PATH, 'utf8'));
+}
+function reciprocalRank(results, relevantNodeIds) {
+    for (let i = 0; i < results.length; i++) {
+        const item = results[i];
+        if (item && relevantNodeIds.has(item.node.id)) {
+            return 1 / (i + 1);
+        }
+    }
+    return 0;
+}
+function recallAtK(results, relevantNodeIds, k) {
+    if (relevantNodeIds.size === 0)
+        return 0;
+    const matched = new Set(results.slice(0, k).filter((result) => relevantNodeIds.has(result.node.id)).map((result) => result.node.id));
+    return matched.size / relevantNodeIds.size;
+}
+function walkTsFiles(dir, base = PKG_ROOT) {
+    const entries = readdirSync(dir, { withFileTypes: true });
+    const files = [];
+    for (const entry of entries) {
+        const full = path.join(dir, entry.name);
+        if (entry.isDirectory()) {
+            if (entry.name === '__tests__' || entry.name === 'node_modules')
+                continue;
+            files.push(...walkTsFiles(full, base));
+            continue;
+        }
+        if (!entry.name.endsWith('.ts'))
+            continue;
+        if (entry.name.endsWith('.test.ts') || entry.name.endsWith('.spec.ts'))
+            continue;
+        files.push(path.relative(base, full));
+    }
+    return files;
+}
+describe('semantic regression audit (task-809)', () => {
+    it('hybrid_not_worse_than_keyword', async () => {
+        const graphPath = path.join(os.tmpdir(), `semantic-regression-${Date.now()}-${Math.random()}.json`);
+        const graph = new JsonGraph(graphPath);
+        await graph.init();
+        for (const node of buildFixtureNodes()) {
+            graph.addNode(node);
+        }
+        try {
+            const queries = readLabeledQueries();
+            const embedder = createDeterministicEmbedder();
+            let semanticImproved = false;
+            for (const query of queries) {
+                const relevant = new Set(query.relevantNodeIds);
+                const keywordRanked = graph.searchNodes(query.query, 5);
+                const hybridRanked = await hybridSearch(graph, embedder, query.query, { k: 5, rrfK: 60 });
+                const keywordMrr = reciprocalRank(keywordRanked, relevant);
+                const hybridMrr = reciprocalRank(hybridRanked, relevant);
+                const keywordRecall = recallAtK(keywordRanked, relevant, 5);
+                const hybridRecall = recallAtK(hybridRanked, relevant, 5);
+                expect(hybridMrr, `MRR regressed for query "${query.id}" (${query.query})`).toBeGreaterThanOrEqual(keywordMrr);
+                expect(hybridRecall, `Recall@5 regressed for query "${query.id}" (${query.query})`).toBeGreaterThanOrEqual(keywordRecall);
+                if (query.kind === 'semantic' &&
+                    (hybridMrr > keywordMrr || hybridRecall > keywordRecall)) {
+                    semanticImproved = true;
+                }
+            }
+            expect(semanticImproved).toBe(true);
+        }
+        finally {
+            graph.close();
+            await fs.remove(graphPath).catch(() => undefined);
+        }
+    });
+    it('fallback_without_model', async () => {
+        const projectRoot = await fs.mkdtemp(path.join(os.tmpdir(), 'semantic-fallback-'));
+        const dbPath = path.join(projectRoot, '.dare', 'graph.db');
+        await fs.ensureDir(path.dirname(dbPath));
+        await fs.writeJson(path.join(projectRoot, 'dare.config.json'), {
+            graphrag: {
+                backend: 'sqlite',
+                semantic: {
+                    enabled: true,
+                    model: 'all-MiniLM-L6-v2',
+                    modelHash: 'sha256:missing',
+                    rrfK: 60,
+                },
+            },
+        });
+        const loadEmbedderSpy = vi
+            .spyOn(embeddingsModule, 'loadEmbedder')
+            .mockRejectedValue(new EmbeddingModelMissingError('missing-runtime'));
+        const infoSpy = vi.spyOn(console, 'info').mockImplementation(() => undefined);
+        const graph = new GraphRAG(dbPath);
+        await graph.init();
+        graph.addNode({
+            id: 'node:keyword-fallback',
+            type: 'task',
+            label: 'fallback keyword target',
+            description: 'keyword fallback should still resolve this node',
+        });
+        try {
+            const keywordRanked = graph.searchNodes('fallback keyword', 5);
+            const hybridRanked = await graph.searchNodesHybrid('fallback keyword', 5);
+            expect(hybridRanked.map((entry) => entry.node.id)).toEqual(keywordRanked.map((entry) => entry.node.id));
+            expect(loadEmbedderSpy).toHaveBeenCalled();
+            expect(infoSpy).toHaveBeenCalledWith(expect.stringContaining('semantic disabled, using keyword fallback'));
+        }
+        finally {
+            vi.restoreAllMocks();
+            await Promise.resolve(graph.close());
+            await fs.remove(projectRoot).catch(() => undefined);
+        }
+    });
+    it('no_heavy_dep_in_core', () => {
+        const sourceFiles = walkTsFiles(path.join(PKG_ROOT, 'src')).filter((rel) => path.normalize(rel) !== path.normalize(ALLOWED_REL));
+        const offenders = sourceFiles.filter((rel) => {
+            const content = readFileSync(path.join(PKG_ROOT, rel), 'utf8');
+            return FORBIDDEN_IMPORT_RE.test(content);
+        });
+        expect(offenders).toEqual([]);
+    });
+    it('legacy_graph_without_vectors', async () => {
+        const graphPath = path.join(os.tmpdir(), `legacy-semantic-${Date.now()}-${Math.random()}.json`);
+        const graph = new JsonGraph(graphPath);
+        await graph.init();
+        graph.addNode({
+            id: 'legacy:1',
+            type: 'task',
+            label: 'legacy keyword anchor',
+            description: 'legacy graph has no vectors',
+        });
+        graph.addNode({
+            id: 'legacy:2',
+            type: 'task',
+            label: 'legacy keyword secondary',
+            description: 'secondary keyword fallback candidate',
+        });
+        try {
+            const keywordRanked = graph.searchNodes('legacy keyword', 5);
+            const hybridRanked = await hybridSearch(graph, createDeterministicEmbedder(), 'legacy keyword', { k: 5, rrfK: 60 });
+            expect(hybridRanked.map((entry) => entry.node.id)).toEqual(keywordRanked.map((entry) => entry.node.id));
+        }
+        finally {
+            graph.close();
+            await fs.remove(graphPath).catch(() => undefined);
+        }
+    });
+    it('embeddings_deterministic', async () => {
+        const embedder = createDeterministicEmbedder();
+        const first = await embedder.embed('cancel my plan');
+        const second = await embedder.embed('cancel my plan');
+        const different = await embedder.embed('move to cheaper tier');
+        expect(Array.from(first)).toEqual(Array.from(second));
+        expect(Array.from(first)).not.toEqual(Array.from(different));
+    });
+});
+//# sourceMappingURL=semantic-regression.test.js.map

package/dist/__tests__/semantic-regression.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"semantic-regression.test.js","sourceRoot":"","sources":["../../src/__tests__/semantic-regression.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACpD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,0BAA0B,EAAiB,MAAM,2BAA2B,CAAC;AACtF,OAAO,KAAK,gBAAgB,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAUtD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAC9D,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,uBAAuB,CAAC,CAAC;AAClF,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AACpD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;AAClE,MAAM,mBAAmB,GACvB,oHAAoH,CAAC;AACvH,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,SAAS,oBAAoB,CAAC,IAAY;IACxC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACtC,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,uBAAuB,CAAC;IAClE,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,yBAAyB,CAAC;IACvG,IACE,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC7B,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC;QAClC,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EACrC,CAAC;QACD,OAAO,6BAA6B,CAAC;IACvC,CAAC;IACD,IACE,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC;QAChC,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC;QACnC,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,EACjC,CAAC;QACD,OAAO,wBAAwB,CAAC;IAClC,CAAC;IACD,OAAO,WAAW,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;AACtD,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY;IACvC,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAChF,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,aAAa,CAAC,CAAC;IAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC;IACnC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,2BAA2B;IAClC,OAAO;QACL,GAAG,EAAE,aAAa;QAClB,KAAK,CAAC,KAAK,CAAC,IAAY;YACtB,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB;IACxB,OAAO;QACL;YACE,EAAE,EAAE,aAAa;YACjB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,wBAAwB;YAC/B,WAAW,EAAE,2CAA2C;YACxD,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;SACzD;QACD;YACE,EAAE,EAAE,cAAc;YAClB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,gCAAgC;YACvC,WAAW,EAAE,+BAA+B;YAC5C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;SAC3D;QACD;YACE,EAAE,EAAE,aAAa;YACjB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,mCAAmC;YAC1C,WAAW,EAAE,gCAAgC;YAC7C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,CAAC;SAC/D;QACD;YACE,EAAE,EAAE,gBAAgB;YACpB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,4BAA4B;YACnC,WAAW,EAAE,+BAA+B;YAC5C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;SAC1D;QACD;YACE,EAAE,EAAE,YAAY;YAChB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,8BAA8B;YACrC,WAAW,EAAE,kCAAkC;YAC/C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,wBAAwB,CAAC,CAAC;SAClE;KACF,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB;IACzB,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAmB,CAAC;AAC1E,CAAC;AAED,SAAS,cAAc,CAAC,OAAgC,EAAE,eAAoC;IAC5F,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,IAAI,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,SAAS,CAChB,OAAgC,EAChC,eAAoC,EACpC,CAAS;IAET,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,GAAG,CACrB,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAC5G,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC;AAC7C,CAAC;AAED,SAAS,WAAW,CAAC,GAAW,EAAE,IAAI,GAAG,QAAQ;IAC/C,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACxB,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc;gBAAE,SAAS;YAC1E,KAAK,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;YACvC,SAAS;QACX,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,SAAS;QAC1C,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;YAAE,SAAS;QACjF,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpG,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,EAAE,CAAC;YACvC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,2BAA2B,EAAE,CAAC;YAC/C,IAAI,gBAAgB,GAAG,KAAK,CAAC;YAE7B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;gBAChD,MAAM,aAAa,GAAG,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,YAAY,GAAG,MAAM,YAAY,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;gBAE1F,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;gBAC3D,MAAM,SAAS,GAAG,cAAc,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACzD,MAAM,aAAa,GAAG,SAAS,CAAC,aAAa,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;gBAC5D,MAAM,YAAY,GAAG,SAAS,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;gBAE1D,MAAM,CACJ,SAAS,EACT,4BAA4B,KAAK,CAAC,EAAE,MAAM,KAAK,CAAC,KAAK,GAAG,CACzD,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBACrC,MAAM,CACJ,YAAY,EACZ,iCAAiC,KAAK,CAAC,EAAE,MAAM,KAAK,CAAC,KAAK,GAAG,CAC9D,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;gBAExC,IACE,KAAK,CAAC,IAAI,KAAK,UAAU;oBACzB,CAAC,SAAS,GAAG,UAAU,IAAI,YAAY,GAAG,aAAa,CAAC,EACxD,CAAC;oBACD,gBAAgB,GAAG,IAAI,CAAC;gBAC1B,CAAC;YACH,CAAC;YAED,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;gBAAS,CAAC;YACT,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACpD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAC;QACnF,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAC3D,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QACzC,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,EAAE;YAC7D,QAAQ,EAAE;gBACR,OAAO,EAAE,QAAQ;gBACjB,QAAQ,EAAE;oBACR,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,kBAAkB;oBACzB,SAAS,EAAE,gBAAgB;oBAC3B,IAAI,EAAE,EAAE;iBACT;aACF;SACF,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,EAAE;aACvB,KAAK,CAAC,gBAAgB,EAAE,cAAc,CAAC;aACvC,iBAAiB,CAAC,IAAI,0BAA0B,CAAC,iBAAiB,CAAC,CAAC,CAAC;QACxE,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAE9E,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,CAAC,OAAO,CAAC;YACZ,EAAE,EAAE,uBAAuB;YAC3B,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,yBAAyB;YAChC,WAAW,EAAE,iDAAiD;SAC/D,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,KAAK,CAAC,WAAW,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC/D,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC1E,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CACxD,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAC5C,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,MAAM,CAAC,gBAAgB,CAAC,2CAA2C,CAAC,CACrE,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,EAAE,CAAC,eAAe,EAAE,CAAC;YACrB,MAAM,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;YACrC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACtD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,MAAM,CAChE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAC7D,CAAC;QACF,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;YAC3C,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;YAC/D,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChG,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,CAAC,OAAO,CAAC;YACZ,EAAE,EAAE,UAAU;YACd,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,uBAAuB;YAC9B,WAAW,EAAE,6BAA6B;SAC3C,CAAC,CAAC;QACH,KAAK,CAAC,OAAO,CAAC;YACZ,EAAE,EAAE,UAAU;YACd,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,0BAA0B;YACjC,WAAW,EAAE,sCAAsC;SACpD,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;YAC7D,MAAM,YAAY,GAAG,MAAM,YAAY,CACrC,KAAK,EACL,2BAA2B,EAAE,EAC7B,gBAAgB,EAChB,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CACnB,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CACxD,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAC5C,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACpD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,QAAQ,GAAG,2BAA2B,EAAE,CAAC;QAC/C,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACtD,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAE/D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/budget.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=budget.test.d.ts.map

package/dist/agent/__tests__/budget.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/budget.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/budget.test.js

@@ -0,0 +1,36 @@
+import { describe, it, expect } from 'vitest';
+import { BudgetTracker } from '../budget.js';
+const usage = (input, output, cost = 0) => ({
+    inputTokens: input,
+    outputTokens: output,
+    costUsd: cost,
+    model: 'mock',
+});
+describe('BudgetTracker', () => {
+    it('should_sum_all_best_of_n_candidates', () => {
+        const budget = new BudgetTracker(1000);
+        budget.add(usage(100, 50, 0.01));
+        budget.add(usage(200, 100, 0.02));
+        budget.add(usage(50, 25, 0.005));
+        expect(budget.spent.tokens).toBe(525);
+        expect(budget.spent.usd).toBeCloseTo(0.035);
+    });
+    it('should_be_unlimited_when_null', () => {
+        const budget = new BudgetTracker(null);
+        budget.add(usage(999999, 999999));
+        expect(budget.remaining()).toBe(Number.POSITIVE_INFINITY);
+        expect(budget.exhausted()).toBe(false);
+    });
+    it('should_exhaust_at_limit', () => {
+        const budget = new BudgetTracker(100);
+        budget.add(usage(60, 40));
+        expect(budget.exhausted()).toBe(true);
+        expect(budget.remaining()).toBe(0);
+    });
+    it('should_not_go_negative', () => {
+        const budget = new BudgetTracker(50);
+        budget.add(usage(100, 100));
+        expect(budget.remaining()).toBe(0);
+    });
+});
+//# sourceMappingURL=budget.test.js.map

package/dist/agent/__tests__/budget.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/budget.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAG7C,MAAM,KAAK,GAAG,CAAC,KAAa,EAAE,MAAc,EAAE,IAAI,GAAG,CAAC,EAAc,EAAE,CAAC,CAAC;IACtE,WAAW,EAAE,KAAK;IAClB,YAAY,EAAE,MAAM;IACpB,OAAO,EAAE,IAAI;IACb,KAAK,EAAE,MAAM;CACd,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAO,EAAE,MAAO,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/claude-driver.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=claude-driver.test.d.ts.map

package/dist/agent/__tests__/claude-driver.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude-driver.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/claude-driver.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/claude-driver.test.js

@@ -0,0 +1,88 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { AgentSdkMissingError, createClaudeDriver, setClaudeSdkImporterForTests, } from '../drivers/claude.js';
+function makeInput(signal) {
+    return {
+        taskId: 'task-605',
+        spec: '# spec',
+        steering: [],
+        worktree: '/tmp/worktree',
+        budgetRemaining: 1000,
+        signal,
+    };
+}
+describe('claude driver', () => {
+    const originalKey = process.env.ANTHROPIC_API_KEY;
+    beforeEach(() => {
+        vi.restoreAllMocks();
+        setClaudeSdkImporterForTests(null);
+        process.env.ANTHROPIC_API_KEY = 'sk-ant-test-key';
+    });
+    afterEach(() => {
+        setClaudeSdkImporterForTests(null);
+        if (originalKey === undefined) {
+            delete process.env.ANTHROPIC_API_KEY;
+        }
+        else {
+            process.env.ANTHROPIC_API_KEY = originalKey;
+        }
+    });
+    it('throws_AgentSdkMissingError_when_absent', async () => {
+        setClaudeSdkImporterForTests(async () => {
+            throw new Error('module missing');
+        });
+        await expect(createClaudeDriver({ model: 'claude-sonnet-4-5' })).rejects.toBeInstanceOf(AgentSdkMissingError);
+    });
+    it('does_not_log_api_key', async () => {
+        const key = 'sk-ant-secret-should-not-leak';
+        process.env.ANTHROPIC_API_KEY = key;
+        const createSpy = vi.fn(async () => ({
+            usage: { input_tokens: 5, output_tokens: 3 },
+            content: [{ type: 'text', text: 'ok' }],
+        }));
+        class MockAnthropic {
+            constructor() {
+                this.messages = { create: createSpy };
+            }
+        }
+        setClaudeSdkImporterForTests(async () => ({ default: MockAnthropic }));
+        const logSpy = vi.spyOn(console, 'log').mockImplementation(() => undefined);
+        const infoSpy = vi.spyOn(console, 'info').mockImplementation(() => undefined);
+        const warnSpy = vi.spyOn(console, 'warn').mockImplementation(() => undefined);
+        const errorSpy = vi.spyOn(console, 'error').mockImplementation(() => undefined);
+        const driver = await createClaudeDriver({ model: 'claude-sonnet-4-5' });
+        await driver.run(makeInput(new AbortController().signal));
+        const output = [...logSpy.mock.calls, ...infoSpy.mock.calls, ...warnSpy.mock.calls, ...errorSpy.mock.calls]
+            .flat()
+            .map((value) => String(value))
+            .join('\n');
+        expect(output).not.toContain(key);
+    });
+    it('run_returns_usage', async () => {
+        const createSpy = vi.fn(async () => ({
+            usage: { input_tokens: 120, output_tokens: 60 },
+            content: [{ type: 'text', text: 'implemented patch' }],
+        }));
+        let receivedKey = '';
+        class MockAnthropic {
+            constructor(options) {
+                this.messages = { create: createSpy };
+                receivedKey = options.apiKey;
+            }
+        }
+        setClaudeSdkImporterForTests(async () => ({ default: MockAnthropic }));
+        const driver = await createClaudeDriver({ model: 'claude-sonnet-4-5', maxTokens: 512 });
+        const signal = new AbortController().signal;
+        const result = await driver.run(makeInput(signal));
+        expect(receivedKey).toBe('sk-ant-test-key');
+        expect(createSpy).toHaveBeenCalledWith(expect.objectContaining({
+            model: 'claude-sonnet-4-5',
+            max_tokens: 512,
+        }), expect.objectContaining({ signal }));
+        expect(result.status).toBe('implemented');
+        expect(result.usage.inputTokens).toBe(120);
+        expect(result.usage.outputTokens).toBe(60);
+        expect(result.usage.costUsd).toBeCloseTo(0.00126, 6);
+        expect(result.usage.model).toBe('claude-sonnet-4-5');
+    });
+});
+//# sourceMappingURL=claude-driver.test.js.map

package/dist/agent/__tests__/claude-driver.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude-driver.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/claude-driver.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EACL,oBAAoB,EACpB,kBAAkB,EAClB,4BAA4B,GAC7B,MAAM,sBAAsB,CAAC;AAE9B,SAAS,SAAS,CAAC,MAAmB;IACpC,OAAO;QACL,MAAM,EAAE,UAAU;QAClB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,eAAe;QACzB,eAAe,EAAE,IAAI;QACrB,MAAM;KACP,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAElD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,4BAA4B,CAAC,IAAI,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,4BAA4B,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,OAAO,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,WAAW,CAAC;QAC9C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,4BAA4B,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CACrF,oBAAoB,CACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,GAAG,GAAG,+BAA+B,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,GAAG,CAAC;QAEpC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACnC,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;YAC5C,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;SACxC,CAAC,CAAC,CAAC;QAEJ,MAAM,aAAa;YAAnB;gBACW,aAAQ,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;YAC5C,CAAC;SAAA;QAED,4BAA4B,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;QAEvE,MAAM,MAAM,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5E,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC9E,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC9E,MAAM,QAAQ,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAEhF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;QACxE,MAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QAE1D,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC;aACxG,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;aAC7B,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACnC,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE;YAC/C,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;SACvD,CAAC,CAAC,CAAC;QACJ,IAAI,WAAW,GAAG,EAAE,CAAC;QAErB,MAAM,aAAa;YAGjB,YAAY,OAAoC;gBAFvC,aAAQ,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;gBAGxC,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;YAC/B,CAAC;SACF;QAED,4BAA4B,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;QAEvE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QACxF,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;QAC5C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;QAEnD,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,MAAM,CAAC,gBAAgB,CAAC;YACtB,KAAK,EAAE,mBAAmB;YAC1B,UAAU,EAAE,GAAG;SAChB,CAAC,EACF,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC,CACpC,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/cost-telemetry.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=cost-telemetry.test.d.ts.map

package/dist/agent/__tests__/cost-telemetry.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-telemetry.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/cost-telemetry.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/cost-telemetry.test.js

@@ -0,0 +1,61 @@
+import os from 'node:os';
+import path from 'node:path';
+import fs from 'fs-extra';
+import { beforeEach, afterEach, describe, expect, it } from 'vitest';
+import { JsonGraph } from '../../graphrag/json-graph.js';
+import { recordCostTelemetry } from '../telemetry.js';
+const sampleUsage = {
+    inputTokens: 120,
+    outputTokens: 45,
+    costUsd: 0.03125,
+    model: 'claude-sonnet',
+};
+describe('recordCostTelemetry', () => {
+    let filePath;
+    let graph;
+    beforeEach(async () => {
+        filePath = path.join(os.tmpdir(), `dare-cost-telemetry-${Date.now()}-${Math.random().toString(36).slice(2)}.json`);
+        graph = new JsonGraph(filePath);
+        await graph.init();
+    });
+    afterEach(async () => {
+        graph.close();
+        await fs.remove(filePath).catch(() => undefined);
+    });
+    it('writes_cost_metadata_on_task_node', () => {
+        graph.addNode({
+            id: 'task:task-607',
+            type: 'task',
+            label: 'task-607',
+            metadata: { status: 'DONE' },
+        });
+        recordCostTelemetry(graph, 'task-607', sampleUsage, 3);
+        const taskNode = graph.getNode('task:task-607');
+        expect(taskNode?.metadata?.status).toBe('DONE');
+        expect(taskNode?.metadata?.inputTokens).toBe(120);
+        expect(taskNode?.metadata?.outputTokens).toBe(45);
+        expect(taskNode?.metadata?.costUsd).toBe(0.03125);
+        expect(taskNode?.metadata?.model).toBe('claude-sonnet');
+        expect(taskNode?.metadata?.attempts).toBe(3);
+    });
+    it('is_queryable_via_graph', () => {
+        graph.addNode({
+            id: 'task:task-608',
+            type: 'task',
+            label: 'task-608',
+        });
+        recordCostTelemetry(graph, 'task-608', sampleUsage, 1);
+        const fetched = graph.getNode('task:task-608');
+        expect(fetched?.metadata?.costUsd).toBe(0.03125);
+        const matches = graph.searchNodes('task-608', 5);
+        expect(matches.some((entry) => entry.node.id === 'task:task-608')).toBe(true);
+        const queried = matches.find((entry) => entry.node.id === 'task:task-608');
+        expect(queried?.node.metadata?.inputTokens).toBe(120);
+    });
+    it('no_throw_on_missing_node', () => {
+        expect(() => {
+            recordCostTelemetry(graph, 'task-missing', sampleUsage, 2);
+        }).not.toThrow();
+    });
+});
+//# sourceMappingURL=cost-telemetry.test.js.map

package/dist/agent/__tests__/cost-telemetry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-telemetry.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/cost-telemetry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAErE,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEtD,MAAM,WAAW,GAAe;IAC9B,WAAW,EAAE,GAAG;IAChB,YAAY,EAAE,EAAE;IAChB,OAAO,EAAE,OAAO;IAChB,KAAK,EAAE,eAAe;CACvB,CAAC;AAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,QAAgB,CAAC;IACrB,IAAI,KAAgB,CAAC;IAErB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,QAAQ,GAAG,IAAI,CAAC,IAAI,CAClB,EAAE,CAAC,MAAM,EAAE,EACX,uBAAuB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAChF,CAAC;QACF,KAAK,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,KAAK,CAAC,OAAO,CAAC;YACZ,EAAE,EAAE,eAAe;YACnB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,UAAU;YACjB,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;SAC7B,CAAC,CAAC;QAEH,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QAEvD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAChD,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACxD,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,KAAK,CAAC,OAAO,CAAC;YACZ,EAAE,EAAE,eAAe;YACnB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,UAAU;SAClB,CAAC,CAAC;QAEH,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QAEvD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9E,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC;QAC3E,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,CAAC,GAAG,EAAE;YACV,mBAAmB,CAAC,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/driver.types.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=driver.types.test.d.ts.map

package/dist/agent/__tests__/driver.types.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"driver.types.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/driver.types.test.ts"],"names":[],"mappings":""}

package/dist/agent/__tests__/driver.types.test.js

@@ -0,0 +1,33 @@
+import { describe, it, expect } from 'vitest';
+describe('agent/driver types', () => {
+    it('exports_all_symbols', () => {
+        const usage = {
+            inputTokens: 0,
+            outputTokens: 0,
+            costUsd: 0,
+            model: 'mock',
+        };
+        const steering = [];
+        const input = {
+            taskId: 'task-601',
+            spec: '# spec',
+            steering,
+            worktree: '/tmp/wt',
+            budgetRemaining: 1000,
+            signal: new AbortController().signal,
+        };
+        const result = {
+            status: 'implemented',
+            worktree: input.worktree,
+            summary: 'done',
+            usage,
+        };
+        const driver = {
+            id: 'mock',
+            requiresNetwork: false,
+            run: async () => result,
+        };
+        expect(driver.id).toBe('mock');
+    });
+});
+//# sourceMappingURL=driver.types.test.js.map

package/dist/agent/__tests__/driver.types.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"driver.types.test.js","sourceRoot":"","sources":["../../../src/agent/__tests__/driver.types.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAS9C,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,MAAM,KAAK,GAAe;YACxB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,OAAO,EAAE,CAAC;YACV,KAAK,EAAE,MAAM;SACd,CAAC;QACF,MAAM,QAAQ,GAAmC,EAAE,CAAC;QACpD,MAAM,KAAK,GAAkB;YAC3B,MAAM,EAAE,UAAU;YAClB,IAAI,EAAE,QAAQ;YACd,QAAQ;YACR,QAAQ,EAAE,SAAS;YACnB,eAAe,EAAE,IAAI;YACrB,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACrC,CAAC;QACF,MAAM,MAAM,GAAmB;YAC7B,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,OAAO,EAAE,MAAM;YACf,KAAK;SACN,CAAC;QACF,MAAM,MAAM,GAAgB;YAC1B,EAAE,EAAE,MAAM;YACV,eAAe,EAAE,KAAK;YACtB,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;SACxB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agent/__tests__/mock-driver.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=mock-driver.test.d.ts.map

package/dist/agent/__tests__/mock-driver.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-driver.test.d.ts","sourceRoot":"","sources":["../../../src/agent/__tests__/mock-driver.test.ts"],"names":[],"mappings":""}