test(omo-codex): batch 104 (21 files)

2026-05-30 19:12:21 +09:00
parent 89ac00f0e7
commit e9c9142afe
21 changed files with 3653 additions and 0 deletions
@@ -0,0 +1,172 @@
+import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
+import { tmpdir } from "node:os";
+import path from "node:path";
+import { afterEach, describe, expect, it } from "vitest";
+
+import { withPostCompactBudget } from "../src/post-compact-budget.js";
+import type { PiRulesConfig } from "../src/rules/types.js";
+
+const tempDirectories: string[] = [];
+const CONFIG: PiRulesConfig = {
+	disabled: false,
+	mode: "both",
+	maxRuleChars: 30_000,
+	maxResultChars: 50_000,
+	postCompactMaxRuleChars: 12_000,
+	postCompactMaxResultChars: 20_000,
+	enabledSources: "auto",
+};
+
+afterEach(() => {
+	for (const directory of tempDirectories.splice(0)) {
+		rmSync(directory, { recursive: true, force: true });
+	}
+});
+
+describe("post-compact context budget", () => {
+	it("#given known model near its context window #when resolving post-compact budget #then shrinks projected rule injection", () => {
+		// given
+		const transcriptPath = writeCompactedTranscript("A".repeat(760_000));
+
+		// when
+		const budget = withPostCompactBudget(CONFIG, { model: "gpt-5.5", transcriptPath });
+
+		// then
+		expect(budget.maxResultChars).toBeLessThan(1_000);
+		expect(budget.maxRuleChars).toBeLessThanOrEqual(budget.maxResultChars);
+	});
+
+	it("#given unknown model near its context window #when resolving post-compact budget #then shrinks projected rule injection conservatively", () => {
+		// given
+		const transcriptPath = writeCompactedTranscript("A".repeat(760_000));
+
+		// when
+		const budget = withPostCompactBudget(CONFIG, { model: "unknown-model", transcriptPath });
+
+		// then
+		expect(budget.maxResultChars).toBeLessThan(1_000);
+		expect(budget.maxRuleChars).toBeLessThanOrEqual(budget.maxResultChars);
+	});
+
+	it("#given known roomy model #when resolving post-compact budget #then keeps configured post-compact cap", () => {
+		// given
+		const transcriptPath = writeCompactedTranscript("small compacted summary");
+
+		// when
+		const budget = withPostCompactBudget(CONFIG, { model: "openai.gpt-5.5", transcriptPath });
+
+		// then
+		expect(budget.maxRuleChars).toBe(CONFIG.postCompactMaxRuleChars);
+		expect(budget.maxResultChars).toBe(CONFIG.postCompactMaxResultChars);
+	});
+
+	it("#given context pressure marker after compaction #when resolving post-compact budget #then shrinks projected rule injection", () => {
+		// given
+		const transcriptPath = writeCompactedPressureTranscript("small compacted summary");
+
+		// when
+		const budget = withPostCompactBudget(CONFIG, { model: "gpt-5.5", transcriptPath });
+
+		// then
+		expect(budget.maxResultChars).toBeLessThan(1_000);
+		expect(budget.maxRuleChars).toBeLessThanOrEqual(budget.maxResultChars);
+	});
+
+	it("#given Codex canonical context-window marker after compaction #when resolving post-compact budget #then shrinks projected rule injection", () => {
+		// given
+		const transcriptPath = writeCompactedCodexContextWindowTranscript("small compacted summary");
+
+		// when
+		const budget = withPostCompactBudget(CONFIG, { model: "gpt-5.5", transcriptPath });
+
+		// then
+		expect(budget.maxResultChars).toBeLessThan(1_000);
+		expect(budget.maxRuleChars).toBeLessThanOrEqual(budget.maxResultChars);
+	});
+});
+
+function writeCompactedTranscript(retainedText: string): string {
+	const root = mkdtempSync(path.join(tmpdir(), "post-compact-budget-"));
+	tempDirectories.push(root);
+	const transcriptPath = path.join(root, "transcript.jsonl");
+	writeFileSync(
+		transcriptPath,
+		`${JSON.stringify({
+			type: "compacted",
+			payload: {
+				message: "summary",
+				replacement_history: [{ type: "message", role: "user", content: retainedText }],
+			},
+		})}\n`,
+	);
+	return transcriptPath;
+}
+
+function writeCompactedPressureTranscript(retainedText: string): string {
+	const root = mkdtempSync(path.join(tmpdir(), "post-compact-budget-"));
+	tempDirectories.push(root);
+	const transcriptPath = path.join(root, "transcript-pressure.jsonl");
+	writeFileSync(
+		transcriptPath,
+		[
+			JSON.stringify({
+				type: "compacted",
+				payload: {
+					message: "summary",
+					replacement_history: [{ type: "message", role: "user", content: retainedText }],
+				},
+			}),
+			JSON.stringify({
+				type: "message",
+				payload: {
+					content: {
+						error: {
+							code: "context_too_large",
+							message:
+								"Your input exceeds the context window of this model. Please adjust your input and try again.",
+						},
+					},
+				},
+			}),
+			"",
+		].join("\n"),
+	);
+	return transcriptPath;
+}
+
+function writeCompactedCodexContextWindowTranscript(retainedText: string): string {
+	const root = mkdtempSync(path.join(tmpdir(), "post-compact-budget-"));
+	tempDirectories.push(root);
+	const transcriptPath = path.join(root, "transcript-codex-context-window.jsonl");
+	writeFileSync(
+		transcriptPath,
+		[
+			JSON.stringify({
+				type: "compacted",
+				payload: {
+					message: "summary",
+					replacement_history: [{ type: "message", role: "user", content: retainedText }],
+				},
+			}),
+			JSON.stringify({
+				type: "message",
+				payload: {
+					content: {
+						error: {
+							code: "context_length_exceeded",
+						},
+					},
+				},
+			}),
+			JSON.stringify({
+				type: "message",
+				payload: {
+					content:
+						"Codex ran out of room in the model's context window. Start a new thread or clear earlier history before retrying.",
+				},
+			}),
+			"",
+		].join("\n"),
+	);
+	return transcriptPath;
+}