@sourcepress/ai 0.1.0

package/.omc/state/last-tool-error.json

@@ -0,0 +1,7 @@
+{
+  "tool_name": "Bash",
+  "tool_input_preview": "{\"command\":\"find packages/ai/src -name \\\"*.ts\\\" -not -path \\\"*__tests__*\\\" -not -path \\\"*node_modules*\\\" | while read f; do\\n  sed -i '' 's/usage?.promptTokens/usage?.inputTokens/g; s/usage\\\\.promptTo...",
+  "error": "Exit code 1\nfind: packages/ai/src: No such file or directory\nFixed\n(eval):cd:4: no such file or directory: packages/ai",
+  "timestamp": "2026-04-04T21:54:17.369Z",
+  "retry_count": 2
+}

package/.turbo/turbo-build.log

@@ -0,0 +1,4 @@
+
+> @sourcepress/ai@0.1.0 build /Users/fabianvontiedemann/Developer/org/sourcepress/packages/ai
+> tsc
+

package/.turbo/turbo-test.log

@@ -0,0 +1,24 @@
+
+> @sourcepress/ai@0.1.0 test /Users/fabianvontiedemann/Developer/org/sourcepress/packages/ai
+> vitest run
+
+
+ RUN  v3.2.4 /Users/fabianvontiedemann/Developer/org/sourcepress/packages/ai
+
+ ✓ src/__tests__/classify.test.ts (4 tests) 12ms
+ ✓ src/__tests__/score.test.ts (2 tests) 27ms
+ ✓ src/__tests__/staleness.test.ts (2 tests) 15ms
+ ✓ src/__tests__/budget.test.ts (7 tests) 2ms
+ ✓ src/__tests__/improve-prompt.test.ts (1 test) 16ms
+ ✓ src/__tests__/find-gaps.test.ts (2 tests) 7ms
+ ✓ src/__tests__/intent-impact.test.ts (2 tests) 15ms
+ ✓ src/__tests__/judge.test.ts (3 tests) 7ms
+ ✓ src/__tests__/extract.test.ts (3 tests) 61ms
+ ✓ src/__tests__/generate.test.ts (3 tests) 72ms
+ ✓ src/__tests__/eval-runner.test.ts (5 tests) 162ms
+
+ Test Files  11 passed (11)
+      Tests  34 passed (34)
+   Start at  13:45:48
+   Duration  1.43s (transform 377ms, setup 0ms, collect 1.14s, tests 397ms, environment 1ms, prepare 1.84s)
+

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 SourcePress Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

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

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

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

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

package/dist/__tests__/budget.test.js

@@ -0,0 +1,96 @@
+import { beforeEach, describe, expect, it } from "vitest";
+import { BudgetTracker } from "../budget.js";
+describe("BudgetTracker", () => {
+    let tracker;
+    beforeEach(() => {
+        tracker = new BudgetTracker({ daily_limit_usd: 5.0, warn_at_usd: 3.0 });
+    });
+    it("starts with zero spent", () => {
+        const status = tracker.getStatus();
+        expect(status.spent_today_usd).toBe(0);
+        expect(status.remaining_usd).toBe(5.0);
+        expect(status.is_over_limit).toBe(false);
+        expect(status.is_warned).toBe(false);
+    });
+    it("tracks token usage", () => {
+        tracker.record({
+            input_tokens: 1000,
+            output_tokens: 500,
+            estimated_cost_usd: 0.01,
+            function_name: "classify",
+            timestamp: new Date().toISOString(),
+        });
+        const status = tracker.getStatus();
+        expect(status.spent_today_usd).toBe(0.01);
+        expect(status.remaining_usd).toBe(4.99);
+    });
+    it("warns when approaching limit", () => {
+        tracker.record({
+            input_tokens: 100000,
+            output_tokens: 50000,
+            estimated_cost_usd: 3.5,
+            function_name: "extract",
+            timestamp: new Date().toISOString(),
+        });
+        const status = tracker.getStatus();
+        expect(status.is_warned).toBe(true);
+        expect(status.is_over_limit).toBe(false);
+    });
+    it("detects over-limit", () => {
+        tracker.record({
+            input_tokens: 200000,
+            output_tokens: 100000,
+            estimated_cost_usd: 5.5,
+            function_name: "judge",
+            timestamp: new Date().toISOString(),
+        });
+        expect(tracker.getStatus().is_over_limit).toBe(true);
+    });
+    it("can check if operation is allowed", () => {
+        expect(tracker.canSpend(4.0)).toBe(true);
+        expect(tracker.canSpend(6.0)).toBe(false);
+        tracker.record({
+            input_tokens: 100000,
+            output_tokens: 50000,
+            estimated_cost_usd: 4.0,
+            function_name: "score",
+            timestamp: new Date().toISOString(),
+        });
+        expect(tracker.canSpend(0.5)).toBe(true);
+        expect(tracker.canSpend(2.0)).toBe(false);
+    });
+    it("returns usage history", () => {
+        tracker.record({
+            input_tokens: 1000,
+            output_tokens: 500,
+            estimated_cost_usd: 0.01,
+            function_name: "classify",
+            timestamp: new Date().toISOString(),
+        });
+        tracker.record({
+            input_tokens: 2000,
+            output_tokens: 1000,
+            estimated_cost_usd: 0.02,
+            function_name: "extract",
+            timestamp: new Date().toISOString(),
+        });
+        const history = tracker.getHistory();
+        expect(history).toHaveLength(2);
+        expect(history[0].function_name).toBe("classify");
+        expect(history[1].function_name).toBe("extract");
+    });
+    it("resets daily spending", () => {
+        tracker.record({
+            input_tokens: 100000,
+            output_tokens: 50000,
+            estimated_cost_usd: 3.0,
+            function_name: "judge",
+            timestamp: new Date().toISOString(),
+        });
+        tracker.resetDaily();
+        const status = tracker.getStatus();
+        expect(status.spent_today_usd).toBe(0);
+        expect(status.remaining_usd).toBe(5.0);
+    });
+});
+//# sourceMappingURL=budget.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"budget.test.js","sourceRoot":"","sources":["../../src/__tests__/budget.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC9B,IAAI,OAAsB,CAAC;IAC3B,UAAU,CAAC,GAAG,EAAE;QACf,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACjC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC7B,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,IAAI;YAClB,aAAa,EAAE,GAAG;YAClB,kBAAkB,EAAE,IAAI;YACxB,aAAa,EAAE,UAAU;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACvC,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,MAAM;YACpB,aAAa,EAAE,KAAK;YACpB,kBAAkB,EAAE,GAAG;YACvB,aAAa,EAAE,SAAS;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC7B,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,MAAM;YACpB,aAAa,EAAE,MAAM;YACrB,kBAAkB,EAAE,GAAG;YACvB,aAAa,EAAE,OAAO;YACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC5C,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,MAAM;YACpB,aAAa,EAAE,KAAK;YACpB,kBAAkB,EAAE,GAAG;YACvB,aAAa,EAAE,OAAO;YACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAChC,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,IAAI;YAClB,aAAa,EAAE,GAAG;YAClB,kBAAkB,EAAE,IAAI;YACxB,aAAa,EAAE,UAAU;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,IAAI;YAClB,aAAa,EAAE,IAAI;YACnB,kBAAkB,EAAE,IAAI;YACxB,aAAa,EAAE,SAAS;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAChC,OAAO,CAAC,MAAM,CAAC;YACd,YAAY,EAAE,MAAM;YACpB,aAAa,EAAE,KAAK;YACpB,kBAAkB,EAAE,GAAG;YACvB,aAAa,EAAE,OAAO;YACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACnC,CAAC,CAAC;QACH,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/__tests__/classify.test.d.ts

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

package/dist/__tests__/classify.test.d.ts.map

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

package/dist/__tests__/classify.test.js

@@ -0,0 +1,72 @@
+import { describe, expect, it, vi } from "vitest";
+import { BudgetTracker } from "../budget.js";
+import { classify } from "../functions/classify.js";
+vi.mock("ai", () => ({ generateObject: vi.fn() }));
+import { generateObject } from "ai";
+const mockProvider = {
+    provider: "anthropic",
+    model: "claude-sonnet-4-5-20250514",
+};
+describe("classify", () => {
+    it("classifies structured text correctly", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                quality: "structured",
+                quality_score: 8,
+                type: "project-notes",
+                reasoning: "Well-organized document with clear sections",
+            },
+            usage: { promptTokens: 500, completionTokens: 100 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await classify({ text: "Meeting with Acme Corp 2026-04-01. Discussed Next.js migration timeline." }, mockProvider, budget);
+        expect(result.quality).toBe("structured");
+        expect(result.quality_score).toBe(8);
+        expect(result.type).toBe("project-notes");
+        expect(result.usage.function_name).toBe("classify");
+    });
+    it("classifies draft-quality text", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { quality: "draft", quality_score: 5, type: "brainstorm", reasoning: "Rough notes" },
+            usage: { promptTokens: 300, completionTokens: 80 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await classify({ text: "maybe we should try... react? or vue?" }, mockProvider, budget);
+        expect(result.quality).toBe("draft");
+        expect(result.quality_score).toBeLessThanOrEqual(6);
+    });
+    it("respects available_types constraint", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                quality: "structured",
+                quality_score: 7,
+                type: "meeting-notes",
+                reasoning: "Typed as meeting-notes",
+            },
+            usage: { promptTokens: 400, completionTokens: 90 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await classify({
+            text: "Team standup 2026-04-04",
+            available_types: ["meeting-notes", "project-notes", "transcript"],
+        }, mockProvider, budget);
+        expect(result.type).toBe("meeting-notes");
+    });
+    it("records budget usage", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { quality: "structured", quality_score: 8, type: "notes", reasoning: "OK" },
+            usage: { promptTokens: 500, completionTokens: 100 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        await classify({ text: "test" }, mockProvider, budget);
+        const history = budget.getHistory();
+        expect(history).toHaveLength(1);
+        expect(history[0].function_name).toBe("classify");
+        expect(history[0].input_tokens).toBe(500);
+    });
+});
+//# sourceMappingURL=classify.test.js.map

package/dist/__tests__/classify.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classify.test.js","sourceRoot":"","sources":["../../src/__tests__/classify.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAGpD,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEpC,MAAM,YAAY,GAAqB;IACtC,QAAQ,EAAE,WAAW;IACrB,KAAK,EAAE,4BAA4B;CACnC,CAAC;AAEF,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACrD,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,OAAO,EAAE,YAAY;gBACrB,aAAa,EAAE,CAAC;gBAChB,IAAI,EAAE,eAAe;gBACrB,SAAS,EAAE,6CAA6C;aACxD;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACnD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC5B,EAAE,IAAI,EAAE,0EAA0E,EAAE,EACpF,YAAY,EACZ,MAAM,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC9C,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,aAAa,EAAE;YAC5F,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,EAAE,EAAE;YAClD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC5B,EAAE,IAAI,EAAE,uCAAuC,EAAE,EACjD,YAAY,EACZ,MAAM,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,OAAO,EAAE,YAAY;gBACrB,aAAa,EAAE,CAAC;gBAChB,IAAI,EAAE,eAAe;gBACrB,SAAS,EAAE,wBAAwB;aACnC;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,EAAE,EAAE;YAClD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC5B;YACC,IAAI,EAAE,yBAAyB;YAC/B,eAAe,EAAE,CAAC,eAAe,EAAE,eAAe,EAAE,YAAY,CAAC;SACjE,EACD,YAAY,EACZ,MAAM,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACrC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE;YACnF,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACnD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/__tests__/eval-runner.test.d.ts

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

package/dist/__tests__/eval-runner.test.d.ts.map

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

package/dist/__tests__/eval-runner.test.js

@@ -0,0 +1,171 @@
+import { describe, expect, it, vi } from "vitest";
+import { BudgetTracker } from "../budget.js";
+import { EvalRunner } from "../eval/runner.js";
+vi.mock("ai", () => ({ generateObject: vi.fn() }));
+import { generateObject } from "ai";
+const mockProvider = {
+    provider: "anthropic",
+    model: "claude-sonnet-4-5-20250514",
+};
+const baseConfig = {
+    content_type: "case-study",
+    knowledge_context: "Acme Corp migrated 12 microservices to Next.js in 8 weeks. Load time improved by 40%.",
+    gold_standard: "# Perfect Case Study\n\nDetailed migration with 12 services, 8 weeks, 40% improvement.",
+    judge_prompt: "Score this content 0-100 against the gold standard.",
+    generation_prompt: "Write a compelling case study.",
+    threshold: 70,
+    max_iterations: 3,
+};
+describe("EvalRunner", () => {
+    it("keeps content when score meets threshold on first iteration", async () => {
+        // Generate
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                frontmatter: { title: "Acme Corp Case Study" },
+                body: "# Acme Corp\n\nWe migrated 12 services in 8 weeks, improving load time by 40%.",
+            },
+            usage: { promptTokens: 3000, completionTokens: 500 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        // Judge — score above threshold
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { score: 85, reasoning: "Excellent match. Specific metrics included." },
+            usage: { promptTokens: 2000, completionTokens: 200 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 10.0 });
+        const runner = new EvalRunner(mockProvider, budget);
+        const result = await runner.run(baseConfig);
+        expect(result.final_status).toBe("keep");
+        expect(result.final_score).toBe(85);
+        expect(result.iterations).toHaveLength(1);
+        expect(result.final_content).toBeTruthy();
+        expect(result.prompt_improved).toBe(false);
+    });
+    it("improves prompt and retries when score is below threshold", async () => {
+        // Iteration 1: Generate
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                frontmatter: { title: "Acme" },
+                body: "We helped Acme.",
+            },
+            usage: { promptTokens: 2000, completionTokens: 200 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        // Iteration 1: Judge — below threshold
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { score: 40, reasoning: "Too generic, no metrics." },
+            usage: { promptTokens: 1500, completionTokens: 150 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        // Iteration 1: Improve prompt
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                improved_prompt: "Write a case study with specific metrics and client quotes.",
+                changes_summary: "Added metrics requirement.",
+            },
+            usage: { promptTokens: 1000, completionTokens: 200 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        // Iteration 2: Generate with improved prompt
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                frontmatter: { title: "Acme Corp — Migration to Next.js" },
+                body: "# Acme Corp\n\n12 services migrated in 8 weeks. 40% load time improvement.",
+            },
+            usage: { promptTokens: 2500, completionTokens: 400 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        // Iteration 2: Judge — above threshold
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                score: 78,
+                reasoning: "Good metrics, clear structure. Matches gold standard well.",
+            },
+            usage: { promptTokens: 2000, completionTokens: 180 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 10.0 });
+        const runner = new EvalRunner(mockProvider, budget);
+        const result = await runner.run(baseConfig);
+        expect(result.final_status).toBe("keep");
+        expect(result.final_score).toBe(78);
+        expect(result.iterations).toHaveLength(2);
+        expect(result.prompt_improved).toBe(true);
+        expect(result.final_prompt).toContain("metrics");
+    });
+    it("discards after max iterations if threshold never met", async () => {
+        // Mock 3 iterations, all below threshold
+        for (let i = 0; i < 3; i++) {
+            // Generate
+            vi.mocked(generateObject).mockResolvedValueOnce({
+                object: {
+                    frontmatter: { title: "Attempt" },
+                    body: "Generic content.",
+                },
+                usage: { promptTokens: 2000, completionTokens: 200 },
+                // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+            });
+            // Judge — below threshold
+            vi.mocked(generateObject).mockResolvedValueOnce({
+                object: { score: 30 + i * 10, reasoning: "Still not good enough." },
+                usage: { promptTokens: 1500, completionTokens: 150 },
+                // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+            });
+            // Improve prompt (not on last iteration)
+            if (i < 2) {
+                vi.mocked(generateObject).mockResolvedValueOnce({
+                    object: {
+                        improved_prompt: `Improved prompt v${i + 2}.`,
+                        changes_summary: "Tweaked instructions.",
+                    },
+                    usage: { promptTokens: 1000, completionTokens: 200 },
+                    // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+                });
+            }
+        }
+        const budget = new BudgetTracker({ daily_limit_usd: 10.0 });
+        const runner = new EvalRunner(mockProvider, budget);
+        const result = await runner.run(baseConfig);
+        expect(result.final_status).toBe("discard");
+        expect(result.iterations).toHaveLength(3);
+        expect(result.final_content).toBeUndefined();
+    });
+    it("judgeOnly scores existing content without loop", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { score: 72, reasoning: "Decent quality, matches intent." },
+            usage: { promptTokens: 1500, completionTokens: 100 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 10.0 });
+        const runner = new EvalRunner(mockProvider, budget);
+        const result = await runner.judgeOnly({
+            draft: "# Existing content\n\nSome published text.",
+            gold_standard: "# Perfect example.",
+            judge_prompt: "Score 0-100.",
+        });
+        expect(result.score).toBe(72);
+        expect(result.reasoning).toBeTruthy();
+    });
+    it("accumulates total usage across iterations", async () => {
+        // Generate
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { frontmatter: { title: "T" }, body: "B" },
+            usage: { promptTokens: 1000, completionTokens: 100 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        // Judge — keep
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { score: 90, reasoning: "Great." },
+            usage: { promptTokens: 500, completionTokens: 50 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 10.0 });
+        const runner = new EvalRunner(mockProvider, budget);
+        const result = await runner.run(baseConfig);
+        expect(result.total_usage.input_tokens).toBe(1500);
+        expect(result.total_usage.output_tokens).toBe(150);
+        expect(result.total_usage.estimated_cost_usd).toBeGreaterThan(0);
+    });
+});
+//# sourceMappingURL=eval-runner.test.js.map

package/dist/__tests__/eval-runner.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"eval-runner.test.js","sourceRoot":"","sources":["../../src/__tests__/eval-runner.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAG/C,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEpC,MAAM,YAAY,GAAqB;IACtC,QAAQ,EAAE,WAAW;IACrB,KAAK,EAAE,4BAA4B;CACnC,CAAC;AAEF,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,YAAY;IAC1B,iBAAiB,EAChB,uFAAuF;IACxF,aAAa,EACZ,wFAAwF;IACzF,YAAY,EAAE,qDAAqD;IACnE,iBAAiB,EAAE,gCAAgC;IACnD,SAAS,EAAE,EAAE;IACb,cAAc,EAAE,CAAC;CACjB,CAAC;AAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC5E,WAAW;QACX,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,WAAW,EAAE,EAAE,KAAK,EAAE,sBAAsB,EAAE;gBAC9C,IAAI,EAAE,gFAAgF;aACtF;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,gCAAgC;QAChC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,6CAA6C,EAAE;YAC/E,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE5C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAC1E,wBAAwB;QACxB,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,WAAW,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;gBAC9B,IAAI,EAAE,iBAAiB;aACvB;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,uCAAuC;QACvC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,0BAA0B,EAAE;YAC5D,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,8BAA8B;QAC9B,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,eAAe,EAAE,6DAA6D;gBAC9E,eAAe,EAAE,4BAA4B;aAC7C;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,6CAA6C;QAC7C,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,WAAW,EAAE,EAAE,KAAK,EAAE,kCAAkC,EAAE;gBAC1D,IAAI,EAAE,4EAA4E;aAClF;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,uCAAuC;QACvC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,KAAK,EAAE,EAAE;gBACT,SAAS,EAAE,4DAA4D;aACvE;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE5C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACrE,yCAAyC;QACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,WAAW;YACX,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;gBAC/C,MAAM,EAAE;oBACP,WAAW,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;oBACjC,IAAI,EAAE,kBAAkB;iBACxB;gBACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;gBACpD,yFAAyF;aAClF,CAAC,CAAC;YAEV,0BAA0B;YAC1B,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;gBAC/C,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,GAAG,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,wBAAwB,EAAE;gBACnE,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;gBACpD,yFAAyF;aAClF,CAAC,CAAC;YAEV,yCAAyC;YACzC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACX,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;oBAC/C,MAAM,EAAE;wBACP,eAAe,EAAE,oBAAoB,CAAC,GAAG,CAAC,GAAG;wBAC7C,eAAe,EAAE,uBAAuB;qBACxC;oBACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;oBACpD,yFAAyF;iBAClF,CAAC,CAAC;YACX,CAAC;QACF,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE5C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,aAAa,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,iCAAiC,EAAE;YACnE,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC;YACrC,KAAK,EAAE,4CAA4C;YACnD,aAAa,EAAE,oBAAoB;YACnC,YAAY,EAAE,cAAc;SAC5B,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC1D,WAAW;QACX,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,WAAW,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE;YAClD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QAEV,eAAe;QACf,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE;YAC1C,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,EAAE,EAAE;YAClD,yFAAyF;SAClF,CAAC,CAAC;QAEV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE5C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/__tests__/extract.test.d.ts

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

package/dist/__tests__/extract.test.d.ts.map

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

package/dist/__tests__/extract.test.js

@@ -0,0 +1,79 @@
+import { describe, expect, it, vi } from "vitest";
+import { BudgetTracker } from "../budget.js";
+import { extract } from "../functions/extract.js";
+vi.mock("ai", () => ({ generateObject: vi.fn() }));
+import { generateObject } from "ai";
+const mockProvider = {
+    provider: "anthropic",
+    model: "claude-sonnet-4-5-20250514",
+};
+describe("extract", () => {
+    it("extracts entities and relations from text", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                entities: [
+                    { type: "client", name: "Acme Corp", aliases: ["Acme", "ACME"], confidence: 0.95 },
+                    { type: "technology", name: "Next.js", aliases: ["NextJS"], confidence: 0.99 },
+                ],
+                relations: [
+                    {
+                        from_entity: "Acme Corp",
+                        to_entity: "Next.js",
+                        relation_type: "uses",
+                        confidence: 0.9,
+                        evidence: "Acme Corp is migrating to Next.js",
+                    },
+                ],
+            },
+            usage: { promptTokens: 800, completionTokens: 200 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await extract({
+            text: "Meeting with Acme Corp. They want to migrate to Next.js.",
+            file_path: "knowledge/clients/acme.md",
+        }, mockProvider, budget);
+        expect(result.entities).toHaveLength(2);
+        expect(result.entities[0].name).toBe("Acme Corp");
+        expect(result.relations).toHaveLength(1);
+        expect(result.usage.function_name).toBe("extract");
+    });
+    it("includes existing entities as context", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                entities: [{ type: "technology", name: "React", aliases: [], confidence: 0.95 }],
+                relations: [
+                    {
+                        from_entity: "React",
+                        to_entity: "Next.js",
+                        relation_type: "used_by",
+                        confidence: 0.98,
+                        evidence: "React is the foundation of Next.js",
+                    },
+                ],
+            },
+            usage: { promptTokens: 1000, completionTokens: 150 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await extract({
+            text: "React is great for building UIs.",
+            file_path: "knowledge/tech/react.md",
+            existing_entities: [{ type: "technology", name: "Next.js" }],
+        }, mockProvider, budget);
+        expect(result.entities).toHaveLength(1);
+        expect(result.relations).toHaveLength(1);
+    });
+    it("records budget usage", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: { entities: [], relations: [] },
+            usage: { promptTokens: 400, completionTokens: 50 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        await extract({ text: "Empty text", file_path: "test.md" }, mockProvider, budget);
+        expect(budget.getHistory()).toHaveLength(1);
+        expect(budget.getHistory()[0].function_name).toBe("extract");
+    });
+});
+//# sourceMappingURL=extract.test.js.map

package/dist/__tests__/extract.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extract.test.js","sourceRoot":"","sources":["../../src/__tests__/extract.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAGlD,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEpC,MAAM,YAAY,GAAqB;IACtC,QAAQ,EAAE,WAAW;IACrB,KAAK,EAAE,4BAA4B;CACnC,CAAC;AAEF,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC1D,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,QAAQ,EAAE;oBACT,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;oBAClF,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;iBAC9E;gBACD,SAAS,EAAE;oBACV;wBACC,WAAW,EAAE,WAAW;wBACxB,SAAS,EAAE,SAAS;wBACpB,aAAa,EAAE,MAAM;wBACrB,UAAU,EAAE,GAAG;wBACf,QAAQ,EAAE,mCAAmC;qBAC7C;iBACD;aACD;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACnD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAC3B;YACC,IAAI,EAAE,0DAA0D;YAChE,SAAS,EAAE,2BAA2B;SACtC,EACD,YAAY,EACZ,MAAM,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACtD,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE;gBACP,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;gBAChF,SAAS,EAAE;oBACV;wBACC,WAAW,EAAE,OAAO;wBACpB,SAAS,EAAE,SAAS;wBACpB,aAAa,EAAE,SAAS;wBACxB,UAAU,EAAE,IAAI;wBAChB,QAAQ,EAAE,oCAAoC;qBAC9C;iBACD;aACD;YACD,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE;YACpD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAC3B;YACC,IAAI,EAAE,kCAAkC;YACxC,SAAS,EAAE,yBAAyB;YACpC,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;SAC5D,EACD,YAAY,EACZ,MAAM,CACN,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACrC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC;YAC/C,MAAM,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;YACvC,KAAK,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,gBAAgB,EAAE,EAAE,EAAE;YAClD,yFAAyF;SAClF,CAAC,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAClF,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/__tests__/find-gaps.test.d.ts

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

package/dist/__tests__/find-gaps.test.d.ts.map

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

package/dist/__tests__/find-gaps.test.js

@@ -0,0 +1,82 @@
+import { describe, expect, it, vi } from "vitest";
+import { BudgetTracker } from "../budget.js";
+import { findGaps } from "../functions/find-gaps.js";
+vi.mock("ai", () => ({ generateObject: vi.fn() }));
+import { generateObject } from "ai";
+const mockProvider = {
+    provider: "anthropic",
+    model: "claude-sonnet-4-5-20250514",
+};
+describe("findGaps", () => {
+    it("identifies knowledge without corresponding content", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                gaps: [
+                    {
+                        entity_name: "React",
+                        entity_type: "technology",
+                        suggested_content_type: "blog-post",
+                        priority: "high",
+                        reason: "Core tech with no content page.",
+                    },
+                    {
+                        entity_name: "TypeScript",
+                        entity_type: "technology",
+                        suggested_content_type: "service-page",
+                        priority: "medium",
+                        reason: "Appears in projects but no content.",
+                    },
+                ],
+            },
+            usage: { promptTokens: 2500, completionTokens: 300 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await findGaps({
+            entities: [
+                { type: "technology", name: "React" },
+                { type: "technology", name: "TypeScript" },
+                { type: "technology", name: "Next.js" },
+            ],
+            relations: [{ from: "Next.js", to: "React", type: "built_on" }],
+            existing_content: [
+                {
+                    path: "content/cases/acme.mdx",
+                    title: "Acme Corp — Next.js Migration",
+                    summary: "Case study about Next.js",
+                },
+            ],
+        }, mockProvider, budget);
+        expect(result.gaps).toHaveLength(2);
+        expect(result.gaps[0].entity_name).toBe("React");
+        expect(result.gaps[0].priority).toBe("high");
+        expect(result.usage.function_name).toBe("findGaps");
+    });
+    it("uses business context for prioritization", async () => {
+        vi.mocked(generateObject).mockResolvedValueOnce({
+            object: {
+                gaps: [
+                    {
+                        entity_name: "AI Strategy",
+                        entity_type: "service",
+                        suggested_content_type: "service-page",
+                        priority: "high",
+                        reason: "Core business offering.",
+                    },
+                ],
+            },
+            usage: { promptTokens: 2000, completionTokens: 150 },
+            // biome-ignore lint/suspicious/noExplicitAny: partial mock of generateObject return type
+        });
+        const budget = new BudgetTracker({ daily_limit_usd: 5.0 });
+        const result = await findGaps({
+            entities: [{ type: "service", name: "AI Strategy" }],
+            relations: [],
+            existing_content: [],
+            business_context: "We are a digital agency focused on AI strategy consulting.",
+        }, mockProvider, budget);
+        expect(result.gaps).toHaveLength(1);
+        expect(result.gaps[0].priority).toBe("high");
+    });
+});
+//# sourceMappingURL=find-gaps.test.js.map