pi-harness-runtime 0.9.4 → 0.9.6

package/packages/test-generator/test/generator.test.ts

@@ -0,0 +1,156 @@
+/**
+ * Test Generator Tests (RFC-0013)
+ */
+
+import { describe, it, expect } from "bun:test";
+import {
+	detectAssertions,
+	generateUnitTests,
+	generateE2ETests,
+	generateMockFixtures,
+} from "../src/index.js";
+import type { SourceFile, TestScenario } from "../src/index.js";
+
+const src = (content: string): SourceFile => ({
+	path: "src/math.ts",
+	content,
+	language: "typescript",
+});
+
+describe("detectAssertions", () => {
+	it("detects .toBe", () => {
+		const s = src("expect(result).toBe(42);");
+		const assertions = detectAssertions(s);
+		expect(assertions.length).toBeGreaterThan(0);
+		expect(assertions[0].type).toBe("equality");
+	});
+
+	it("detects .toEqual", () => {
+		const s = src("expect(result).toEqual({ ok: true });");
+		const assertions = detectAssertions(s);
+		expect(assertions.some((a) => a.type === "equality")).toBe(true);
+	});
+
+	it("detects .toThrow", () => {
+		const s = src("expect(() => run()).toThrow();");
+		const assertions = detectAssertions(s);
+		expect(assertions.some((a) => a.type === "exception")).toBe(true);
+	});
+
+	it("detects truthiness", () => {
+		const s = src("expect(value).toBeTruthy();");
+		const assertions = detectAssertions(s);
+		expect(assertions.some((a) => a.type === "truthiness")).toBe(true);
+	});
+
+	it("returns empty for clean code", () => {
+		const s = src("const x = 1; const y = 2;");
+		expect(detectAssertions(s).length).toBe(0);
+	});
+
+	it("includes line number", () => {
+		const s = src("const x = 1;\nexpect(x).toBe(1);");
+		const assertions = detectAssertions(s);
+		expect(assertions[0].line).toBe(2);
+	});
+});
+
+describe("generateUnitTests", () => {
+	it("generates test content", () => {
+		const s = src(
+			"export function add(a: number, b: number): number { return a + b; }",
+		);
+		const test = generateUnitTests(s);
+		expect(test.content).toContain("describe(");
+		expect(test.content).toContain("it(");
+		expect(test.fileName).toContain(".test.ts");
+	});
+
+	it("uses bun test imports by default", () => {
+		const s = src("function fn() {}");
+		const test = generateUnitTests(s);
+		expect(test.content).toContain('from "bun:test"');
+	});
+
+	it("uses vitest for vitest framework", () => {
+		const s = src("function fn() {}");
+		const test = generateUnitTests(s, "vitest");
+		expect(test.content).toContain("vitest");
+	});
+
+	it("uses playwright for playwright framework", () => {
+		const s = src("function fn() {}");
+		const test = generateUnitTests(s, "playwright");
+		expect(test.content).toContain("playwright");
+		expect(test.testCount).toBeGreaterThan(0);
+	});
+
+	it("extracts function name", () => {
+		const s = src("export const multiply = (a: number, b: number) => a * b;");
+		const test = generateUnitTests(s);
+		expect(test.content).toContain("multiply");
+	});
+
+	it("handles anonymous function", () => {
+		const s = src("const fn = function() {};");
+		const test = generateUnitTests(s);
+		expect(test.content).toContain("describe(");
+	});
+});
+
+describe("generateE2ETests", () => {
+	it("generates playwright test from scenario", () => {
+		const scenario: TestScenario = {
+			name: "Login Flow",
+			description: "User logs in",
+			steps: [
+				"navigate:https://example.com/login",
+				"click:#username",
+				"assert:.error",
+			],
+			expectedResult: "User sees login form",
+			mockFixtures: [],
+		};
+		const tests = generateE2ETests([scenario]);
+		expect(tests).toHaveLength(1);
+		expect(tests[0].content).toContain("Login Flow");
+		expect(tests[0].framework).toBe("playwright");
+	});
+
+	it("generates correct step content", () => {
+		const scenario: TestScenario = {
+			name: "Navigate",
+			description: "",
+			steps: ["navigate:https://example.com"],
+			expectedResult: "",
+			mockFixtures: [],
+		};
+		const tests = generateE2ETests([scenario]);
+		expect(tests[0].content).toContain('page.goto("https://example.com")');
+	});
+});
+
+describe("generateMockFixtures", () => {
+	it("extracts typed variables", () => {
+		const s = src("const name: string = 'Alice'; const count: number = 42;");
+		const fixtures = generateMockFixtures(s);
+		expect(fixtures.some((f) => f.name === "name" && f.type === "string")).toBe(
+			true,
+		);
+		expect(
+			fixtures.some((f) => f.name === "count" && f.type === "number"),
+		).toBe(true);
+	});
+
+	it("maps types to values", () => {
+		const s = src("const flag: boolean = true;");
+		const fixtures = generateMockFixtures(s);
+		expect(fixtures[0].value).toBe("true");
+	});
+
+	it("defaults unknown types to null", () => {
+		const s = src("const custom: CustomType = x;");
+		const fixtures = generateMockFixtures(s);
+		expect(fixtures[0].value).toBe("null");
+	});
+});

package/packages/test-generator/tsconfig.json

@@ -0,0 +1,6 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "compilerOptions": { "outDir": "./dist", "rootDir": "./src", "types": ["node"], "lib": ["ES2022"] },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "test"]
+}

package/packages/token-estimation/package.json

@@ -0,0 +1,22 @@
+{
+  "name": "@pi-harness/token-estimation",
+  "version": "0.9.6",
+  "description": "pi-harness-runtime — token-estimation",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "test": "bun test"
+  },
+  "dependencies": {}
+}

package/packages/token-estimation/test/token-estimation.test.ts

@@ -0,0 +1,207 @@
+/**
+ * Token Estimation Tests
+ */
+
+import { describe, it, expect } from "bun:test";
+import {
+	roughTokenCount,
+	roughMessageTokens,
+	roughMessagesTokens,
+	roughSystemPromptTokens,
+	roughToolDefinitionTokens,
+	estimateRequestTokens,
+} from "../src/index";
+
+describe("roughTokenCount", () => {
+	it("returns 0 for empty string", () => expect(roughTokenCount("")).toBe(0));
+
+	it("returns 0 for null-like input", () =>
+		expect(roughTokenCount("   ")).toBeGreaterThanOrEqual(0));
+
+	it("returns positive for non-empty text", () =>
+		expect(roughTokenCount("hello world")).toBeGreaterThan(0));
+
+	it("handles unicode text", () =>
+		expect(roughTokenCount("สวัสดี")).toBeGreaterThan(0));
+
+	it("handles long text proportionally", () => {
+		const short = roughTokenCount("hi");
+		const long = roughTokenCount("hi ".repeat(100));
+		expect(long).toBeGreaterThan(short * 10);
+	});
+});
+
+describe("roughMessageTokens", () => {
+	it("adds role overhead", () => {
+		const tokens = roughMessageTokens({ role: "user", content: null });
+		expect(tokens).toBeGreaterThanOrEqual(4);
+	});
+
+	it("counts string content", () => {
+		const tokens = roughMessageTokens({
+			role: "user",
+			content: "hello world",
+		});
+		expect(tokens).toBeGreaterThanOrEqual(4 + roughTokenCount("hello world"));
+	});
+
+	it("counts text blocks in content array", () => {
+		const tokens = roughMessageTokens({
+			role: "assistant",
+			content: [{ type: "text", text: "the answer is 42" }],
+		});
+		expect(tokens).toBeGreaterThanOrEqual(
+			4 + roughTokenCount("the answer is 42"),
+		);
+	});
+
+	it("estimates tool_use blocks", () => {
+		const tokens = roughMessageTokens({
+			role: "assistant",
+			content: [
+				{
+					type: "tool_use",
+					name: "get_weather",
+					input: { city: "Bangkok" },
+				},
+			],
+		});
+		expect(tokens).toBeGreaterThan(4 + roughTokenCount("get_weather"));
+	});
+
+	it("adds overhead for tool_result blocks", () => {
+		const tokens = roughMessageTokens({
+			role: "user",
+			content: [{ type: "tool_result" }],
+		});
+		expect(tokens).toBeGreaterThanOrEqual(4 + 50);
+	});
+
+	it("estimates image blocks as 2000 tokens", () => {
+		const tokens = roughMessageTokens({
+			role: "user",
+			content: [{ type: "image" }],
+		});
+		expect(tokens).toBeGreaterThanOrEqual(4 + 2000);
+	});
+
+	it("counts tool_calls outside content", () => {
+		const tokens = roughMessageTokens({
+			role: "assistant",
+			content: null,
+			tool_calls: [{ id: "call-1", name: "search", input: { q: "test" } }],
+		});
+		expect(tokens).toBeGreaterThan(4 + roughTokenCount("search"));
+	});
+
+	it("returns 0 for null message", () =>
+		expect(roughMessageTokens(null as any)).toBe(0));
+
+	it("counts thinking blocks", () => {
+		const tokens = roughMessageTokens({
+			role: "assistant",
+			content: null,
+			thinking: "I should use the search tool here",
+		});
+		expect(tokens).toBeGreaterThanOrEqual(
+			4 + roughTokenCount("I should use the search tool here"),
+		);
+	});
+});
+
+describe("roughMessagesTokens", () => {
+	it("sums tokens from multiple messages", () => {
+		const messages = [
+			{ role: "user", content: "hello" },
+			{ role: "assistant", content: "hi there" },
+		];
+		const total = roughMessagesTokens(messages);
+		const individual =
+			roughMessageTokens({ role: "user", content: "hello" }) +
+			roughMessageTokens({ role: "assistant", content: "hi there" });
+		expect(total).toBe(individual);
+	});
+
+	it("returns 0 for empty array", () =>
+		expect(roughMessagesTokens([])).toBe(0));
+});
+
+describe("roughSystemPromptTokens", () => {
+	it("includes overhead plus content", () => {
+		const tokens = roughSystemPromptTokens("You are a helpful assistant");
+		expect(tokens).toBeGreaterThanOrEqual(
+			15 + roughTokenCount("You are a helpful assistant"),
+		);
+	});
+
+	it("handles empty system prompt", () =>
+		expect(roughSystemPromptTokens("")).toBe(15));
+});
+
+describe("roughToolDefinitionTokens", () => {
+	it("counts name, description, and input_schema", () => {
+		const tokens = roughToolDefinitionTokens({
+			name: "get_weather",
+			description: "Get current weather",
+			input_schema: {
+				type: "object",
+				properties: { city: { type: "string" } },
+			},
+		});
+		expect(tokens).toBeGreaterThanOrEqual(
+			roughTokenCount("get_weather") +
+				roughTokenCount("Get current weather") +
+				roughTokenCount(
+					'{"type":"object","properties":{"city":{"type":"string"}}}',
+				),
+		);
+	});
+
+	it("handles minimal tool with only name", () => {
+		const tokens = roughToolDefinitionTokens({ name: "ping" });
+		expect(tokens).toBeGreaterThan(0);
+	});
+});
+
+describe("estimateRequestTokens", () => {
+	it("sums system + messages + tools", () => {
+		const estimate = estimateRequestTokens(
+			"You are a helpful assistant",
+			[{ role: "user", content: "hello" }],
+			[{ name: "search", description: "Search the web" }],
+			100_000,
+		);
+		expect(estimate.total).toBeGreaterThan(estimate.systemPrompt);
+		expect(estimate.total).toBeGreaterThan(estimate.messages);
+		expect(estimate.total).toBeGreaterThan(estimate.tools);
+		expect(estimate.total).toBe(
+			estimate.systemPrompt + estimate.messages + estimate.tools,
+		);
+	});
+
+	it("computes availableForContext correctly", () => {
+		const estimate = estimateRequestTokens(
+			"system",
+			[{ role: "user", content: "msg" }],
+			[],
+			100_000,
+		);
+		expect(estimate.availableForContext).toBe(100_000 - estimate.total);
+	});
+
+	it("returns 0 available when over limit", () => {
+		const estimate = estimateRequestTokens(
+			"x".repeat(200_000),
+			[],
+			[],
+			100_000,
+		);
+		expect(estimate.availableForContext).toBeLessThan(100_000);
+	});
+
+	it("handles empty inputs", () => {
+		const estimate = estimateRequestTokens("", [], [], 100_000);
+		expect(estimate.total).toBe(15); // system overhead only
+		expect(estimate.availableForContext).toBe(100_000 - 15);
+	});
+});

package/packages/token-optimizer/dist/index.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Token Optimizer — Main Entry (RFC-0011)
+ */
+export { estimateTokens, estimateMessageTokens, buildBudget, splitByPriority, optimizeBudget, calculateCost, trimToTokens, DEFAULT_PRICING, } from "./optimizer.js";
+export type * from "./types.js";
+//# sourceMappingURL=index.d.ts.map

package/packages/token-optimizer/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACN,cAAc,EACd,qBAAqB,EACrB,WAAW,EACX,eAAe,EACf,cAAc,EACd,aAAa,EACb,YAAY,EACZ,eAAe,GACf,MAAM,gBAAgB,CAAC;AACxB,mBAAmB,YAAY,CAAC"}

package/packages/token-optimizer/dist/index.js

@@ -0,0 +1,5 @@
+/**
+ * Token Optimizer — Main Entry (RFC-0011)
+ */
+export { estimateTokens, estimateMessageTokens, buildBudget, splitByPriority, optimizeBudget, calculateCost, trimToTokens, DEFAULT_PRICING, } from "./optimizer.js";
+//# sourceMappingURL=index.js.map

package/packages/token-optimizer/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACN,cAAc,EACd,qBAAqB,EACrB,WAAW,EACX,eAAe,EACf,cAAc,EACd,aAAa,EACb,YAAY,EACZ,eAAe,GACf,MAAM,gBAAgB,CAAC"}

package/packages/token-optimizer/dist/optimizer.d.ts

@@ -0,0 +1,29 @@
+/**
+ * Token Optimizer Core — budget optimization and cost calculation (RFC-0011)
+ */
+import type { PromptMessage, TokenBudget, OptimizationResult, CostEstimate } from "./types.js";
+declare const DEFAULT_PRICING: Record<string, {
+    inputPer1M: number;
+    outputPer1M: number;
+    currency: string;
+}>;
+export { DEFAULT_PRICING };
+/** Estimate token count from string. */
+export declare function estimateTokens(text: string): number;
+/** Estimate tokens in a list of messages. */
+export declare function estimateMessageTokens(messages: PromptMessage[]): number;
+/** Build a token budget from components. */
+export declare function buildBudget(maxTokens: number, systemTokens?: number, reservedTokens?: number): TokenBudget;
+/** Split messages by priority for progressive trimming. */
+export declare function splitByPriority(messages: PromptMessage[]): {
+    high: PromptMessage[];
+    medium: PromptMessage[];
+    low: PromptMessage[];
+};
+/** Optimize messages to fit within a token budget. */
+export declare function optimizeBudget(messages: PromptMessage[], budget: TokenBudget): OptimizationResult;
+/** Calculate cost for a set of messages. */
+export declare function calculateCost(messages: PromptMessage[], outputTokens: number, model?: string): CostEstimate;
+/** Trim a string to a maximum token count. */
+export declare function trimToTokens(text: string, maxTokens: number): string;
+//# sourceMappingURL=optimizer.d.ts.map

package/packages/token-optimizer/dist/optimizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"optimizer.d.ts","sourceRoot":"","sources":["../src/optimizer.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EACX,aAAa,EACb,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,MAAM,YAAY,CAAC;AAIpB,QAAA,MAAM,eAAe,EAAE,MAAM,CAC5B,MAAM,EACN;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAQ7D,CAAC;AAEF,OAAO,EAAE,eAAe,EAAE,CAAC;AAE3B,wCAAwC;AACxC,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAEnD;AAED,6CAA6C;AAC7C,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,MAAM,CAKvE;AAED,4CAA4C;AAC5C,wBAAgB,WAAW,CAC1B,SAAS,EAAE,MAAM,EACjB,YAAY,SAAI,EAChB,cAAc,SAAI,GAChB,WAAW,CAYb;AAED,2DAA2D;AAC3D,wBAAgB,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG;IAC3D,IAAI,EAAE,aAAa,EAAE,CAAC;IACtB,MAAM,EAAE,aAAa,EAAE,CAAC;IACxB,GAAG,EAAE,aAAa,EAAE,CAAC;CACrB,CAUA;AAED,sDAAsD;AACtD,wBAAgB,cAAc,CAC7B,QAAQ,EAAE,aAAa,EAAE,EACzB,MAAM,EAAE,WAAW,GACjB,kBAAkB,CAoCpB;AAED,4CAA4C;AAC5C,wBAAgB,aAAa,CAC5B,QAAQ,EAAE,aAAa,EAAE,EACzB,YAAY,EAAE,MAAM,EACpB,KAAK,SAAY,GACf,YAAY,CAWd;AAED,8CAA8C;AAC9C,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAYpE"}

package/packages/token-optimizer/dist/optimizer.js

@@ -0,0 +1,106 @@
+/**
+ * Token Optimizer Core — budget optimization and cost calculation (RFC-0011)
+ */
+const TOKENS_PER_CHAR = 0.25;
+const DEFAULT_PRICING = {
+    "gpt-4o": { inputPer1M: 2.5, outputPer1M: 10.0, currency: "USD" },
+    "gpt-4o-mini": { inputPer1M: 0.15, outputPer1M: 0.6, currency: "USD" },
+    "claude-sonnet": { inputPer1M: 3.0, outputPer1M: 15.0, currency: "USD" },
+    "claude-haiku": { inputPer1M: 0.25, outputPer1M: 1.25, currency: "USD" },
+    "gemini-2.5": { inputPer1M: 0.125, outputPer1M: 0.5, currency: "USD" },
+    default: { inputPer1M: 1.0, outputPer1M: 3.0, currency: "USD" },
+};
+export { DEFAULT_PRICING };
+/** Estimate token count from string. */
+export function estimateTokens(text) {
+    return Math.ceil(text.length * TOKENS_PER_CHAR);
+}
+/** Estimate tokens in a list of messages. */
+export function estimateMessageTokens(messages) {
+    return messages.reduce((sum, msg) => sum + estimateTokens(msg.content) + 4, 0);
+}
+/** Build a token budget from components. */
+export function buildBudget(maxTokens, systemTokens = 0, reservedTokens = 0) {
+    const availableTokens = Math.max(0, maxTokens - systemTokens - reservedTokens);
+    return {
+        maxTokens,
+        systemTokens,
+        promptTokens: 0,
+        reservedTokens,
+        availableTokens,
+    };
+}
+/** Split messages by priority for progressive trimming. */
+export function splitByPriority(messages) {
+    const sorted = [...messages].sort((a, b) => (a.priority ?? 5) - (b.priority ?? 5));
+    const third = Math.ceil(sorted.length / 3);
+    return {
+        high: sorted.slice(0, third),
+        medium: sorted.slice(third, third * 2),
+        low: sorted.slice(third * 2),
+    };
+}
+/** Optimize messages to fit within a token budget. */
+export function optimizeBudget(messages, budget) {
+    if (budget.availableTokens <= 0) {
+        return {
+            kept: [],
+            removed: messages,
+            finalBudget: budget,
+            compressionRatio: 1,
+        };
+    }
+    const prioritized = splitByPriority(messages);
+    const all = [...prioritized.high, ...prioritized.medium, ...prioritized.low];
+    const kept = [];
+    const removed = [];
+    let usedTokens = 0;
+    for (const msg of all) {
+        const msgTokens = estimateTokens(msg.content) + 4;
+        if (usedTokens + msgTokens <= budget.availableTokens) {
+            kept.push(msg);
+            usedTokens += msgTokens;
+        }
+        else {
+            removed.push(msg);
+        }
+    }
+    const originalTokens = estimateMessageTokens(messages);
+    const compressionRatio = originalTokens > 0 ? (originalTokens - usedTokens) / originalTokens : 0;
+    return {
+        kept,
+        removed,
+        finalBudget: { ...budget, promptTokens: usedTokens },
+        compressionRatio,
+    };
+}
+/** Calculate cost for a set of messages. */
+export function calculateCost(messages, outputTokens, model = "default") {
+    const pricing = DEFAULT_PRICING[model] ?? DEFAULT_PRICING["default"];
+    const inputTokens = estimateMessageTokens(messages);
+    const inputCost = (inputTokens / 1_000_000) * pricing.inputPer1M;
+    const outputCost = (outputTokens / 1_000_000) * pricing.outputPer1M;
+    return {
+        inputTokens,
+        outputTokens,
+        totalCost: Math.round((inputCost + outputCost) * 1_000_000) / 1_000_000,
+        currency: pricing.currency,
+    };
+}
+/** Trim a string to a maximum token count. */
+export function trimToTokens(text, maxTokens) {
+    const words = text.split(/\s+/);
+    let result = "";
+    let tokens = 0;
+    for (const word of words) {
+        const wordTokens = estimateTokens(word);
+        if (tokens + wordTokens <= maxTokens) {
+            result += (result ? " " : "") + word;
+            tokens += wordTokens;
+        }
+        else
+            break;
+    }
+    return result;
+}
+//# sourceMappingURL=optimizer.js.map

package/packages/token-optimizer/dist/optimizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"optimizer.js","sourceRoot":"","sources":["../src/optimizer.ts"],"names":[],"mappings":"AAAA;;GAEG;AASH,MAAM,eAAe,GAAG,IAAI,CAAC;AAE7B,MAAM,eAAe,GAGjB;IACH,QAAQ,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IACjE,aAAa,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE;IACtE,eAAe,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IACxE,cAAc,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE;IACxE,YAAY,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE;IACtE,OAAO,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE;CAC/D,CAAC;AAEF,OAAO,EAAE,eAAe,EAAE,CAAC;AAE3B,wCAAwC;AACxC,MAAM,UAAU,cAAc,CAAC,IAAY;IAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,eAAe,CAAC,CAAC;AACjD,CAAC;AAED,6CAA6C;AAC7C,MAAM,UAAU,qBAAqB,CAAC,QAAyB;IAC9D,OAAO,QAAQ,CAAC,MAAM,CACrB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,EACnD,CAAC,CACD,CAAC;AACH,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,WAAW,CAC1B,SAAiB,EACjB,YAAY,GAAG,CAAC,EAChB,cAAc,GAAG,CAAC;IAElB,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAC/B,CAAC,EACD,SAAS,GAAG,YAAY,GAAG,cAAc,CACzC,CAAC;IACF,OAAO;QACN,SAAS;QACT,YAAY;QACZ,YAAY,EAAE,CAAC;QACf,cAAc;QACd,eAAe;KACf,CAAC;AACH,CAAC;AAED,2DAA2D;AAC3D,MAAM,UAAU,eAAe,CAAC,QAAyB;IAKxD,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAC/C,CAAC;IACF,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC3C,OAAO;QACN,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;QAC5B,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC;QACtC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;KAC5B,CAAC;AACH,CAAC;AAED,sDAAsD;AACtD,MAAM,UAAU,cAAc,CAC7B,QAAyB,EACzB,MAAmB;IAEnB,IAAI,MAAM,CAAC,eAAe,IAAI,CAAC,EAAE,CAAC;QACjC,OAAO;YACN,IAAI,EAAE,EAAE;YACR,OAAO,EAAE,QAAQ;YACjB,WAAW,EAAE,MAAM;YACnB,gBAAgB,EAAE,CAAC;SACnB,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC9C,MAAM,GAAG,GAAG,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,GAAG,WAAW,CAAC,MAAM,EAAE,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;IAC7E,MAAM,IAAI,GAAoB,EAAE,CAAC;IACjC,MAAM,OAAO,GAAoB,EAAE,CAAC;IACpC,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClD,IAAI,UAAU,GAAG,SAAS,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YACtD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACf,UAAU,IAAI,SAAS,CAAC;QACzB,CAAC;aAAM,CAAC;YACP,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;IACF,CAAC;IAED,MAAM,cAAc,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IACvD,MAAM,gBAAgB,GACrB,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;IAEzE,OAAO;QACN,IAAI;QACJ,OAAO;QACP,WAAW,EAAE,EAAE,GAAG,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE;QACpD,gBAAgB;KAChB,CAAC;AACH,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,aAAa,CAC5B,QAAyB,EACzB,YAAoB,EACpB,KAAK,GAAG,SAAS;IAEjB,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,eAAe,CAAC,SAAS,CAAC,CAAC;IACrE,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC;IACjE,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC;IACpE,OAAO;QACN,WAAW;QACX,YAAY;QACZ,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,SAAS,CAAC,GAAG,SAAS;QACvE,QAAQ,EAAE,OAAO,CAAC,QAAQ;KAC1B,CAAC;AACH,CAAC;AAED,8CAA8C;AAC9C,MAAM,UAAU,YAAY,CAAC,IAAY,EAAE,SAAiB;IAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,MAAM,UAAU,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,MAAM,GAAG,UAAU,IAAI,SAAS,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;YACrC,MAAM,IAAI,UAAU,CAAC;QACtB,CAAC;;YAAM,MAAM;IACd,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC"}

package/packages/token-optimizer/dist/types.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Token Optimizer Types (RFC-0011)
+ */
+export interface PromptMessage {
+    role: "system" | "user" | "assistant";
+    content: string;
+    priority?: number;
+}
+export interface TokenBudget {
+    maxTokens: number;
+    systemTokens: number;
+    promptTokens: number;
+    reservedTokens: number;
+    availableTokens: number;
+}
+export interface OptimizationResult {
+    kept: PromptMessage[];
+    removed: PromptMessage[];
+    finalBudget: TokenBudget;
+    compressionRatio: number;
+}
+export interface CostEstimate {
+    inputTokens: number;
+    outputTokens: number;
+    totalCost: number;
+    currency: string;
+}
+export type ModelId = "gpt-4o" | "gpt-4o-mini" | "claude-sonnet" | "claude-haiku" | "gemini-2.5" | "default";
+export interface ModelPricing {
+    [model: string]: {
+        inputPer1M: number;
+        outputPer1M: number;
+        currency: string;
+    };
+}
+//# sourceMappingURL=types.d.ts.map

package/packages/token-optimizer/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,WAAW,aAAa;IAC7B,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,kBAAkB;IAClC,IAAI,EAAE,aAAa,EAAE,CAAC;IACtB,OAAO,EAAE,aAAa,EAAE,CAAC;IACzB,WAAW,EAAE,WAAW,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,YAAY;IAC5B,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,OAAO,GAChB,QAAQ,GACR,aAAa,GACb,eAAe,GACf,cAAc,GACd,YAAY,GACZ,SAAS,CAAC;AAEb,MAAM,WAAW,YAAY;IAC5B,CAAC,KAAK,EAAE,MAAM,GAAG;QAChB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;KACjB,CAAC;CACF"}

package/packages/token-optimizer/dist/types.js

@@ -0,0 +1,5 @@
+/**
+ * Token Optimizer Types (RFC-0011)
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/packages/token-optimizer/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/packages/token-optimizer/package.json

@@ -0,0 +1,21 @@
+{
+  "name": "@pi-harness/token-optimizer",
+  "version": "0.9.6",
+  "description": "Token Budget Optimizer — maximize prompt value within token limits (RFC-0011)",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "bun test"
+  },
+  "devDependencies": {
+    "bun-types": "^1.0.0"
+  }
+}

package/packages/token-optimizer/src/index.ts

@@ -0,0 +1,15 @@
+/**
+ * Token Optimizer — Main Entry (RFC-0011)
+ */
+
+export {
+	estimateTokens,
+	estimateMessageTokens,
+	buildBudget,
+	splitByPriority,
+	optimizeBudget,
+	calculateCost,
+	trimToTokens,
+	DEFAULT_PRICING,
+} from "./optimizer.js";
+export type * from "./types.js";