@kenkaiiii/ggcoder 4.3.205 → 4.3.207

package/dist/core/goal-worker.test.js

@@ -0,0 +1,224 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { EventEmitter } from "node:events";
+import { PassThrough } from "node:stream";
+import fs from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { getGoalRun, updateGoalTask, upsertGoalRun } from "./goal-store.js";
+const spawnMock = vi.hoisted(() => vi.fn());
+const killProcessTreeMock = vi.hoisted(() => vi.fn());
+vi.mock("node:child_process", () => ({ spawn: spawnMock }));
+vi.mock("../utils/process.js", () => ({ killProcessTree: killProcessTreeMock }));
+class FakeChild extends EventEmitter {
+    stdout = new PassThrough();
+    stderr = new PassThrough();
+    pid = 4242;
+    kill = vi.fn();
+}
+let tmpBase;
+let tmpProject;
+let child;
+async function flushUntil(assertion) {
+    let lastError;
+    for (let i = 0; i < 50; i += 1) {
+        await new Promise((resolve) => setTimeout(resolve, 1));
+        try {
+            await assertion();
+            return;
+        }
+        catch (error) {
+            lastError = error;
+        }
+    }
+    throw lastError;
+}
+async function seedGoal() {
+    await upsertGoalRun(tmpProject, {
+        id: "goal-a",
+        title: "Worker harness",
+        goal: "Exercise worker exits",
+        status: "ready",
+        successCriteria: ["worker covered"],
+        prerequisites: [],
+        harness: [],
+        tasks: [],
+        evidence: [],
+        blockers: [],
+    });
+    await updateGoalTask(tmpProject, "goal-a", "task-a", {
+        id: "task-a",
+        title: "Run worker",
+        prompt: "Do work",
+        status: "pending",
+        attempts: 0,
+    });
+}
+async function start(onComplete = vi.fn()) {
+    const mod = await import("./goal-worker.js");
+    const record = await mod.startGoalWorker({
+        cwd: tmpProject,
+        provider: "anthropic",
+        model: "claude-test",
+        goalRunId: "goal-a",
+        goalTaskId: "task-a",
+        prompt: "Do deterministic work",
+        onComplete,
+    });
+    return { mod, record, onComplete };
+}
+beforeEach(async () => {
+    tmpBase = await fs.mkdtemp(path.join(os.tmpdir(), "goal-worker-test-base-"));
+    tmpProject = await fs.mkdtemp(path.join(os.tmpdir(), "goal-worker-test-project-"));
+    process.env.GG_GOALS_BASE = tmpBase;
+    child = new FakeChild();
+    spawnMock.mockReturnValue(child);
+    killProcessTreeMock.mockReset();
+    process.argv[1] = "/tmp/fake-ggcoder-cli.js";
+    await seedGoal();
+});
+afterEach(async () => {
+    const mod = await import("./goal-worker.js");
+    mod.shutdownGoalWorkers(tmpProject);
+    vi.clearAllMocks();
+    delete process.env.GG_GOALS_BASE;
+    await fs.rm(tmpBase, { recursive: true, force: true });
+    await fs.rm(tmpProject, { recursive: true, force: true });
+});
+describe("goal worker failure propagation", () => {
+    it("prompts workers with explicit durable Goal context before claiming verification", async () => {
+        await start();
+        const args = spawnMock.mock.calls[0]?.[1];
+        const systemPromptIndex = args.indexOf("--system-prompt") + 1;
+        const prompt = args[systemPromptIndex];
+        expect(prompt).toContain(`cwd=${tmpProject}`);
+        expect(prompt).toContain("run_id=goal-a");
+        expect(prompt).toContain("task_id=task-a");
+        expect(prompt).toContain("create needed scripts/fixtures/harnesses");
+        expect(prompt).toContain("source_path/docs/kencode real-code research when relevant");
+        expect(prompt).toContain("command/file evidence");
+        expect(prompt).toContain('goals({ action: "evidence" | "task", run_id: "goal-a", task_id: "task-a"');
+    });
+    it("exports a testable worker system prompt/context helper", async () => {
+        const mod = await import("./goal-worker.js");
+        const prompt = mod.buildGoalWorkerSystemPrompt({
+            cwd: "/repo",
+            goalRunId: "goal-123",
+            goalTaskId: "task-456",
+            taskTitle: "Implement typed handoff",
+        });
+        expect(prompt).toContain("cwd=/repo");
+        expect(prompt).toContain("run_id=goal-123");
+        expect(prompt).toContain("task_id=task-456");
+        expect(prompt).toContain("Implement typed handoff");
+        expect(prompt).toContain("Do not mark the whole goal complete");
+    });
+    it("marks the task done, persists evidence, and notifies callbacks/subscribers for worker success", async () => {
+        const { mod, record, onComplete } = await start();
+        const listener = vi.fn();
+        const unsubscribe = mod.subscribeGoalWorkerCompletions(listener, tmpProject);
+        child.stdout.write(JSON.stringify({ type: "text_delta", text: "implemented" }) + "\n");
+        child.stdout.write(JSON.stringify({
+            type: "tool_call_start",
+            toolCallId: "tool-a",
+            name: "bash",
+            args: { command: "pnpm test" },
+        }) + "\n");
+        child.stdout.write(JSON.stringify({ type: "tool_call_end", toolCallId: "tool-a", isError: false }) + "\n");
+        child.emit("close", 0);
+        await flushUntil(() => expect(onComplete).toHaveBeenCalled());
+        const run = await getGoalRun(tmpProject, "goal-a");
+        expect(record.status).toBe("done");
+        expect(run?.activeWorkerId).toBeUndefined();
+        expect(run?.tasks[0]).toMatchObject({
+            status: "done",
+            workerId: record.id,
+            lastSummary: expect.stringContaining("implemented"),
+        });
+        expect(run?.evidence.some((item) => item.label === `Worker ${record.id} done` && item.content?.includes("implemented"))).toBe(true);
+        expect(onComplete).toHaveBeenCalledWith(expect.objectContaining({
+            status: "done",
+            exitCode: 0,
+            toolsUsed: [{ name: "bash", ok: true }],
+        }));
+        expect(listener).toHaveBeenCalledWith(expect.objectContaining({
+            worker: expect.objectContaining({ id: record.id, goalRunId: "goal-a" }),
+            status: "done",
+            exitCode: 0,
+        }));
+        unsubscribe();
+    });
+    it("replays a pending completion to the next subscriber after a remount gap", async () => {
+        const { mod, record } = await start();
+        child.stdout.write(JSON.stringify({ type: "text_delta", text: "finished during remount" }) + "\n");
+        child.emit("close", 0);
+        await flushUntil(() => expect(record.status).toBe("done"));
+        const listener = vi.fn();
+        const unsubscribe = mod.subscribeGoalWorkerCompletions(listener, tmpProject);
+        await flushUntil(() => expect(listener).toHaveBeenCalledWith(expect.objectContaining({
+            worker: expect.objectContaining({ id: record.id, cwd: tmpProject }),
+            summary: expect.stringContaining("finished during remount"),
+            status: "done",
+        })));
+        unsubscribe();
+    });
+    it("marks the task failed and persists stderr evidence for worker non-zero exit", async () => {
+        const { record, onComplete } = await start();
+        child.stderr.write("boom from stderr");
+        child.emit("close", 1);
+        await flushUntil(() => expect(onComplete).toHaveBeenCalled());
+        const run = await getGoalRun(tmpProject, "goal-a");
+        expect(record.status).toBe("failed");
+        expect(run?.tasks[0]).toMatchObject({
+            status: "failed",
+            workerId: record.id,
+            lastSummary: expect.stringContaining("boom from stderr"),
+        });
+        expect(run?.evidence.some((item) => item.label === `Worker ${record.id} failed` && item.content?.includes("boom"))).toBe(true);
+        expect(onComplete).toHaveBeenCalledWith(expect.objectContaining({ status: "failed", exitCode: 1 }));
+    });
+    it("records spawn crashes as failed task evidence and notifies the orchestrator continuation path", async () => {
+        const { mod, record, onComplete } = await start();
+        const listener = vi.fn();
+        const unsubscribe = mod.subscribeGoalWorkerCompletions(listener, tmpProject);
+        child.emit("error", new Error("spawn exploded"));
+        await flushUntil(async () => {
+            const next = await getGoalRun(tmpProject, "goal-a");
+            expect(next?.tasks[0]?.status).toBe("failed");
+            expect(next?.evidence.some((item) => item.label === `Worker ${record.id} spawn failed`)).toBe(true);
+            expect(onComplete).toHaveBeenCalled();
+        });
+        const run = await getGoalRun(tmpProject, "goal-a");
+        expect(record.status).toBe("failed");
+        expect(run?.tasks[0]).toMatchObject({
+            status: "failed",
+            lastSummary: "Failed to spawn Goal worker: spawn exploded",
+        });
+        expect(run?.evidence.some((item) => item.label === `Worker ${record.id} spawn failed` && item.content === "spawn exploded")).toBe(true);
+        expect(onComplete).toHaveBeenCalledWith(expect.objectContaining({ status: "failed", exitCode: 1, reason: "spawn_error" }));
+        expect(listener).toHaveBeenCalledWith(expect.objectContaining({
+            worker: expect.objectContaining({ id: record.id, cwd: tmpProject }),
+            status: "failed",
+            exitCode: 1,
+            reason: "spawn_error",
+        }));
+        unsubscribe();
+    });
+    it("manual stop blocks the task, clears the active worker, and ignores later close completion", async () => {
+        const { mod, record, onComplete } = await start();
+        const result = await mod.stopGoalWorker(record.id);
+        child.emit("close", 0);
+        await flushUntil(() => expect(record.status).toBe("stopped"));
+        const run = await getGoalRun(tmpProject, "goal-a");
+        expect(result).toBe(`Goal worker ${record.id} stopped.`);
+        expect(record.status).toBe("stopped");
+        expect(killProcessTreeMock).toHaveBeenCalledWith(4242);
+        expect(run?.activeWorkerId).toBeUndefined();
+        expect(run?.tasks[0]).toMatchObject({
+            status: "blocked",
+            lastSummary: "Worker stopped by user.",
+        });
+        expect(run?.evidence.some((item) => item.kind === "summary" && item.label === `Worker ${record.id} stopped`)).toBe(true);
+        expect(onComplete).not.toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=goal-worker.test.js.map

package/dist/core/goal-worker.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"goal-worker.test.js","sourceRoot":"","sources":["../../src/core/goal-worker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAE5E,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAC5C,MAAM,mBAAmB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAEtD,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;AAC5D,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC;AAEjF,MAAM,SAAU,SAAQ,YAAY;IAClC,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;IAC3B,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;IAC3B,GAAG,GAAG,IAAI,CAAC;IACX,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;CAChB;AAED,IAAI,OAAe,CAAC;AACpB,IAAI,UAAkB,CAAC;AACvB,IAAI,KAAgB,CAAC;AAErB,KAAK,UAAU,UAAU,CAAC,SAAqC;IAC7D,IAAI,SAAkB,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACvD,IAAI,CAAC;YACH,MAAM,SAAS,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;IACH,CAAC;IACD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,QAAQ;IACrB,MAAM,aAAa,CAAC,UAAU,EAAE;QAC9B,EAAE,EAAE,QAAQ;QACZ,KAAK,EAAE,gBAAgB;QACvB,IAAI,EAAE,uBAAuB;QAC7B,MAAM,EAAE,OAAO;QACf,eAAe,EAAE,CAAC,gBAAgB,CAAC;QACnC,aAAa,EAAE,EAAE;QACjB,OAAO,EAAE,EAAE;QACX,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,EAAE;KACb,CAAC,CAAC;IACH,MAAM,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE;QACnD,EAAE,EAAE,QAAQ;QACZ,KAAK,EAAE,YAAY;QACnB,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,SAAS;QACjB,QAAQ,EAAE,CAAC;KACZ,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,KAAK,CAAC,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE;IACvC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,eAAe,CAAC;QACvC,GAAG,EAAE,UAAU;QACf,QAAQ,EAAE,WAAW;QACrB,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,QAAQ;QACnB,UAAU,EAAE,QAAQ;QACpB,MAAM,EAAE,uBAAuB;QAC/B,UAAU;KACX,CAAC,CAAC;IACH,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;AACrC,CAAC;AAED,UAAU,CAAC,KAAK,IAAI,EAAE;IACpB,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC,CAAC;IAC7E,UAAU,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;IACnF,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,OAAO,CAAC;IACpC,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;IACxB,SAAS,CAAC,eAAe,CAAC,KAAgC,CAAC,CAAC;IAC5D,mBAAmB,CAAC,SAAS,EAAE,CAAC;IAChC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,0BAA0B,CAAC;IAC7C,MAAM,QAAQ,EAAE,CAAC;AACnB,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;IAC7C,GAAG,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;IACpC,EAAE,CAAC,aAAa,EAAE,CAAC;IACnB,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IACjC,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACvD,MAAM,EAAE,CAAC,EAAE,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC5D,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,KAAK,EAAE,CAAC;QACd,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAa,CAAC;QACtD,MAAM,iBAAiB,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAEvC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,UAAU,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0CAA0C,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,2DAA2D,CAAC,CAAC;QACtF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,0EAA0E,CAC3E,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAE7C,MAAM,MAAM,GAAG,GAAG,CAAC,2BAA2B,CAAC;YAC7C,GAAG,EAAE,OAAO;YACZ,SAAS,EAAE,UAAU;YACrB,UAAU,EAAE,UAAU;YACtB,SAAS,EAAE,yBAAyB;SACrC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+FAA+F,EAAE,KAAK,IAAI,EAAE;QAC7G,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QAClD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,WAAW,GAAG,GAAG,CAAC,8BAA8B,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAE7E,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QACvF,KAAK,CAAC,MAAM,CAAC,KAAK,CAChB,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,iBAAiB;YACvB,UAAU,EAAE,QAAQ;YACpB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;SAC/B,CAAC,GAAG,IAAI,CACV,CAAC;QACF,KAAK,CAAC,MAAM,CAAC,KAAK,CAChB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,GAAG,IAAI,CACvF,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACvB,MAAM,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAE9D,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5C,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAClC,MAAM,EAAE,MAAM;YACd,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,WAAW,EAAE,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC;SACpD,CAAC,CAAC;QACH,MAAM,CACJ,GAAG,EAAE,QAAQ,CAAC,IAAI,CAChB,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,KAAK,KAAK,UAAU,MAAM,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,aAAa,CAAC,CACrF,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM;YACd,QAAQ,EAAE,CAAC;YACX,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;SACxC,CAAC,CACH,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;YACvE,MAAM,EAAE,MAAM;YACd,QAAQ,EAAE,CAAC;SACZ,CAAC,CACH,CAAC;QACF,WAAW,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QAEtC,KAAK,CAAC,MAAM,CAAC,KAAK,CAChB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC,GAAG,IAAI,CAC/E,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAEvB,MAAM,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,WAAW,GAAG,GAAG,CAAC,8BAA8B,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAE7E,MAAM,UAAU,CAAC,GAAG,EAAE,CACpB,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC;YACnE,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,yBAAyB,CAAC;YAC3D,MAAM,EAAE,MAAM;SACf,CAAC,CACH,CACF,CAAC;QACF,WAAW,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QAE7C,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACvC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACvB,MAAM,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAE9D,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAClC,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,WAAW,EAAE,MAAM,CAAC,gBAAgB,CAAC,kBAAkB,CAAC;SACzD,CAAC,CAAC;QACH,MAAM,CACJ,GAAG,EAAE,QAAQ,CAAC,IAAI,CAChB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,KAAK,UAAU,MAAM,CAAC,EAAE,SAAS,IAAI,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CACxF,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAC3D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+FAA+F,EAAE,KAAK,IAAI,EAAE;QAC7G,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QAClD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,WAAW,GAAG,GAAG,CAAC,8BAA8B,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAE7E,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACjD,MAAM,UAAU,CAAC,KAAK,IAAI,EAAE;YAC1B,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,KAAK,UAAU,MAAM,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,IAAI,CAC3F,IAAI,CACL,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAClC,MAAM,EAAE,QAAQ;YAChB,WAAW,EAAE,6CAA6C;SAC3D,CAAC,CAAC;QACH,MAAM,CACJ,GAAG,EAAE,QAAQ,CAAC,IAAI,CAChB,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,KAAK,KAAK,UAAU,MAAM,CAAC,EAAE,eAAe,IAAI,IAAI,CAAC,OAAO,KAAK,gBAAgB,CACzF,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAClF,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC;YACnE,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,aAAa;SACtB,CAAC,CACH,CAAC;QACF,WAAW,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QACzG,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACnD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACvB,MAAM,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QAE9D,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,MAAM,CAAC,EAAE,WAAW,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5C,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAClC,MAAM,EAAE,SAAS;YACjB,WAAW,EAAE,yBAAyB;SACvC,CAAC,CAAC;QACH,MAAM,CACJ,GAAG,EAAE,QAAQ,CAAC,IAAI,CAChB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,KAAK,UAAU,MAAM,CAAC,EAAE,UAAU,CAClF,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/core/model-registry.test.js

@@ -1,5 +1,55 @@
 import { describe, expect, it } from "vitest";
-import { getContextWindow, getDefaultModel, getModelsForProvider, usesOpenAICodexTransport, } from "./model-registry.js";
+import { MODELS, getContextWindow, getDefaultModel, getModelsForProvider, usesOpenAICodexTransport, } from "./model-registry.js";
+const PROVIDERS = [
+    "anthropic",
+    "openai",
+    "gemini",
+    "moonshot",
+    "glm",
+    "minimax",
+    "xiaomi",
+    "deepseek",
+    "openrouter",
+];
+const THINKING_LEVELS = ["low", "medium", "high", "xhigh"];
+const COST_TIERS = ["low", "medium", "high"];
+describe("model registry invariants", () => {
+    it("has unique ids and coherent required metadata for every entry", () => {
+        const ids = new Set();
+        for (const model of MODELS) {
+            expect(ids.has(model.id), `${model.id} is duplicated`).toBe(false);
+            ids.add(model.id);
+            expect(model.id, `${model.id} id`).toEqual(expect.any(String));
+            expect(model.name, `${model.id} name`).toEqual(expect.any(String));
+            expect(PROVIDERS, `${model.id} provider`).toContain(model.provider);
+            expect(model.contextWindow, `${model.id} contextWindow`).toBeGreaterThan(0);
+            expect(Number.isInteger(model.contextWindow), `${model.id} contextWindow integer`).toBe(true);
+            expect(model.maxOutputTokens, `${model.id} maxOutputTokens`).toBeGreaterThan(0);
+            expect(Number.isInteger(model.maxOutputTokens), `${model.id} maxOutputTokens integer`).toBe(true);
+            expect(model.maxOutputTokens, `${model.id} maxOutputTokens <= contextWindow`).toBeLessThanOrEqual(model.contextWindow);
+            expect(typeof model.supportsThinking, `${model.id} supportsThinking`).toBe("boolean");
+            expect(typeof model.supportsImages, `${model.id} supportsImages`).toBe("boolean");
+            expect(COST_TIERS, `${model.id} costTier`).toContain(model.costTier);
+            expect(THINKING_LEVELS, `${model.id} maxThinkingLevel`).toContain(model.maxThinkingLevel);
+            if (!model.supportsThinking) {
+                expect(model.maxThinkingLevel, `${model.id} non-thinking max level`).toBe("low");
+            }
+            if (model.codexContextWindow !== undefined) {
+                expect(model.provider, `${model.id} codexContextWindow provider`).toBe("openai");
+                expect(model.codexContextWindow, `${model.id} codexContextWindow`).toBeGreaterThan(0);
+                expect(Number.isInteger(model.codexContextWindow), `${model.id} codexContextWindow integer`).toBe(true);
+                expect(model.codexContextWindow, `${model.id} codexContextWindow <= contextWindow`).toBeLessThanOrEqual(model.contextWindow);
+            }
+        }
+    });
+    it("returns a registered default model for every provider", () => {
+        for (const provider of PROVIDERS) {
+            const defaultModel = getDefaultModel(provider);
+            expect(defaultModel.provider, `${provider} default provider`).toBe(provider);
+            expect(MODELS, `${provider} default registered`).toContain(defaultModel);
+        }
+    });
+});
 describe("model registry context windows", () => {
     it("uses the public API context window for OpenAI API-key requests", () => {
         expect(getContextWindow("gpt-5.5", { provider: "openai" })).toBe(1_050_000);

package/dist/core/model-registry.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-registry.test.js","sourceRoot":"","sources":["../../src/core/model-registry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAE7B,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5E,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,OAAO,GAAG,EAAE,QAAQ,EAAE,QAAiB,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;QAEvE,MAAM,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,CAAC,wBAAwB,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/F,MAAM,CACJ,gBAAgB,CAAC,mBAAmB,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CACxF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9C,EAAE,EAAE,+BAA+B;YACnC,IAAI,EAAE,+BAA+B;YACrC,QAAQ,EAAE,QAAQ;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;YACtE,+BAA+B;YAC/B,kBAAkB;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,+BAA+B,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CACpF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"model-registry.test.js","sourceRoot":"","sources":["../../src/core/model-registry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,MAAM,EACN,gBAAgB,EAChB,eAAe,EACf,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,SAAS,GAAG;IAChB,WAAW;IACX,QAAQ;IACR,QAAQ;IACR,UAAU;IACV,KAAK;IACL,SAAS;IACT,QAAQ;IACR,UAAU;IACV,YAAY;CACJ,CAAC;AACX,MAAM,eAAe,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAU,CAAC;AACpE,MAAM,UAAU,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAU,CAAC;AAEtD,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;QAE9B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,gBAAgB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnE,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAClB,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,GAAG,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,SAAS,EAAE,GAAG,KAAK,CAAC,EAAE,WAAW,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,KAAK,CAAC,EAAE,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC5E,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9F,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,KAAK,CAAC,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChF,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,0BAA0B,CAAC,CAAC,IAAI,CACzF,IAAI,CACL,CAAC;YACF,MAAM,CACJ,KAAK,CAAC,eAAe,EACrB,GAAG,KAAK,CAAC,EAAE,mCAAmC,CAC/C,CAAC,mBAAmB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC3C,MAAM,CAAC,OAAO,KAAK,CAAC,gBAAgB,EAAE,GAAG,KAAK,CAAC,EAAE,mBAAmB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtF,MAAM,CAAC,OAAO,KAAK,CAAC,cAAc,EAAE,GAAG,KAAK,CAAC,EAAE,iBAAiB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClF,MAAM,CAAC,UAAU,EAAE,GAAG,KAAK,CAAC,EAAE,WAAW,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACrE,MAAM,CAAC,eAAe,EAAE,GAAG,KAAK,CAAC,EAAE,mBAAmB,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAC1F,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;gBAC5B,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,GAAG,KAAK,CAAC,EAAE,yBAAyB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnF,CAAC;YACD,IAAI,KAAK,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBAC3C,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,KAAK,CAAC,EAAE,8BAA8B,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACjF,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,GAAG,KAAK,CAAC,EAAE,qBAAqB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;gBACtF,MAAM,CACJ,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAC1C,GAAG,KAAK,CAAC,EAAE,6BAA6B,CACzC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACb,MAAM,CACJ,KAAK,CAAC,kBAAkB,EACxB,GAAG,KAAK,CAAC,EAAE,sCAAsC,CAClD,CAAC,mBAAmB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC/C,MAAM,CAAC,YAAY,CAAC,QAAQ,EAAE,GAAG,QAAQ,mBAAmB,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7E,MAAM,CAAC,MAAM,EAAE,GAAG,QAAQ,qBAAqB,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5E,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,OAAO,GAAG,EAAE,QAAQ,EAAE,QAAiB,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;QAEvE,MAAM,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,CAAC,wBAAwB,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/F,MAAM,CACJ,gBAAgB,CAAC,mBAAmB,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CACxF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9C,EAAE,EAAE,+BAA+B;YACnC,IAAI,EAAE,+BAA+B;YACrC,QAAQ,EAAE,QAAQ;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;YACtE,+BAA+B;YAC/B,kBAAkB;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,+BAA+B,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CACpF,SAAS,CACV,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/core/oauth/gemini.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../src/core/oauth/gemini.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AA+CxE,wBAAsB,WAAW,CAAC,SAAS,EAAE,mBAAmB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CA0C3F;AAED,wBAAsB,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAaxF"}
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../src/core/oauth/gemini.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AA6ExE,wBAAsB,WAAW,CAAC,SAAS,EAAE,mBAAmB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CA2C3F;AAED,wBAAsB,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAcxF"}

package/dist/core/oauth/gemini.js

@@ -1,11 +1,14 @@
 import http from "node:http";
 import crypto from "node:crypto";
 import { generatePKCE } from "./pkce.js";
-const CLIENT_ID = "681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com";
-const CLIENT_SECRET = "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";
+const CLIENT_ID_ENV = "GGCODER_GEMINI_OAUTH_CLIENT_ID";
+const CLIENT_SECRET_ENV = "GGCODER_GEMINI_OAUTH_CLIENT_SECRET";
 const AUTHORIZE_URL = "https://accounts.google.com/o/oauth2/v2/auth";
 const TOKEN_URL = "https://oauth2.googleapis.com/token";
 const CODE_ASSIST_BASE_URL = "https://cloudcode-pa.googleapis.com";
+const CODE_ASSIST_API_VERSION = "v1internal";
+const CODE_ASSIST_POST_RETRIES = 3;
+const CODE_ASSIST_POST_RETRY_DELAY_MS = 100;
 const SCOPE = [
     "https://www.googleapis.com/auth/cloud-platform",
     "https://www.googleapis.com/auth/userinfo.email",
@@ -13,13 +16,27 @@ const SCOPE = [
 ].join(" ");
 const USER_TIER_FREE = "free-tier";
 const USER_TIER_LEGACY = "legacy-tier";
+const USER_TIER_STANDARD = "standard-tier";
+const VALIDATION_REQUIRED_REASON = "VALIDATION_REQUIRED";
+const VPC_SC_REASON = "SECURITY_POLICY_VIOLATED";
+class CodeAssistHttpError extends Error {
+    status;
+    body;
+    constructor(label, status, body) {
+        super(`Gemini Code Assist ${label} failed (${status}): ${body}`);
+        this.name = "CodeAssistHttpError";
+        this.status = status;
+        this.body = body;
+    }
+}
 export async function loginGemini(callbacks) {
+    const { clientId, clientSecret } = getGeminiOAuthClientCredentials();
     const { verifier, challenge } = await generatePKCE();
     const state = crypto.randomBytes(32).toString("hex");
     const redirectUri = await getLoopbackRedirectUri();
     const url = new URL(AUTHORIZE_URL);
     url.searchParams.set("response_type", "code");
-    url.searchParams.set("client_id", CLIENT_ID);
+    url.searchParams.set("client_id", clientId);
     url.searchParams.set("redirect_uri", redirectUri);
     url.searchParams.set("scope", SCOPE);
     url.searchParams.set("access_type", "offline");
@@ -43,20 +60,21 @@ export async function loginGemini(callbacks) {
         }
         code = parsed.code;
     }
-    const creds = await exchangeGeminiCode(code, verifier, redirectUri);
+    const creds = await exchangeGeminiCode(code, verifier, redirectUri, clientId, clientSecret);
     callbacks.onStatus("Setting up Gemini Code Assist access...");
-    const projectId = await setupCodeAssistProject(creds.accessToken);
+    const projectId = await setupCodeAssistProject(creds.accessToken, callbacks);
     return {
         ...creds,
         projectId,
     };
 }
 export async function refreshGeminiToken(refreshToken) {
+    const { clientId, clientSecret } = getGeminiOAuthClientCredentials();
     const data = await postTokenRequest({
         grant_type: "refresh_token",
         refresh_token: refreshToken,
-        client_id: CLIENT_ID,
-        client_secret: CLIENT_SECRET,
+        client_id: clientId,
+        client_secret: clientSecret,
     });
     return {
         accessToken: data.access_token,
@@ -64,6 +82,14 @@ export async function refreshGeminiToken(refreshToken) {
         expiresAt: Date.now() + data.expires_in * 1000 - 5 * 60 * 1000,
     };
 }
+function getGeminiOAuthClientCredentials() {
+    const clientId = process.env[CLIENT_ID_ENV]?.trim();
+    const clientSecret = process.env[CLIENT_SECRET_ENV]?.trim();
+    if (!clientId || !clientSecret) {
+        throw new Error(`Gemini OAuth requires ${CLIENT_ID_ENV} and ${CLIENT_SECRET_ENV} to be set.`);
+    }
+    return { clientId, clientSecret };
+}
 async function getLoopbackRedirectUri() {
     return new Promise((resolve, reject) => {
         const server = http.createServer();
@@ -122,7 +148,7 @@ async function loginWithServer(authUrl, redirectUri, expectedState, callbacks) {
                 return;
             }
             receivedCode = url.searchParams.get("code");
-            res.writeHead(200, { "Content-Type": "text/html" });
+            res.writeHead(200, { "Content-Type": "text/html", Connection: "close" });
             res.end("<html><body><h1>Login successful!</h1><p>You can close this tab.</p></body></html>");
             server.close();
         });
@@ -147,11 +173,11 @@ async function loginWithServer(authUrl, redirectUri, expectedState, callbacks) {
         });
     });
 }
-async function exchangeGeminiCode(code, verifier, redirectUri) {
+async function exchangeGeminiCode(code, verifier, redirectUri, clientId, clientSecret) {
     const data = await postTokenRequest({
         grant_type: "authorization_code",
-        client_id: CLIENT_ID,
-        client_secret: CLIENT_SECRET,
+        client_id: clientId,
+        client_secret: clientSecret,
         code,
         redirect_uri: redirectUri,
         code_verifier: verifier,
@@ -177,21 +203,33 @@ async function postTokenRequest(body) {
     }
     return (await response.json());
 }
-async function setupCodeAssistProject(accessToken) {
+async function setupCodeAssistProject(accessToken, callbacks) {
     const envProject = process.env.GOOGLE_CLOUD_PROJECT ?? process.env.GOOGLE_CLOUD_PROJECT_ID;
     if (envProject && /^\d+$/.test(envProject)) {
         throw new Error("GOOGLE_CLOUD_PROJECT must be a project ID, not a numeric project number.");
     }
-    const metadata = {
+    const coreMetadata = {
         ideType: "IDE_UNSPECIFIED",
         platform: "PLATFORM_UNSPECIFIED",
         pluginType: "GEMINI",
+    };
+    const projectMetadata = {
+        ...coreMetadata,
         ...(envProject ? { duetProject: envProject } : {}),
     };
-    const loadRes = await codeAssistPost(accessToken, "loadCodeAssist", {
-        ...(envProject ? { cloudaicompanionProject: envProject } : {}),
-        metadata,
-    });
+    let loadRes;
+    while (true) {
+        loadRes = await loadCodeAssist(accessToken, envProject, projectMetadata);
+        const validation = getValidationRequiredTier(loadRes);
+        if (!validation)
+            break;
+        callbacks.onStatus(`Gemini Code Assist requires account validation${validation.reasonMessage ? `: ${validation.reasonMessage}` : ""}`);
+        callbacks.onOpenUrl(validation.validationUrl);
+        const answer = await callbacks.onPromptCode("Complete validation in the browser, then press Enter to retry (or type cancel):");
+        if (answer.trim().toLowerCase() === "cancel") {
+            throw new Error("Gemini Code Assist account validation was cancelled.");
+        }
+    }
     if (loadRes.currentTier) {
         const project = loadRes.cloudaicompanionProject ?? envProject;
         if (!project)
@@ -203,12 +241,12 @@ async function setupCodeAssistProject(accessToken) {
         ? {
             tierId: tier.id,
             cloudaicompanionProject: undefined,
-            metadata,
+            metadata: coreMetadata,
         }
         : {
             tierId: tier.id,
             cloudaicompanionProject: envProject,
-            metadata,
+            metadata: projectMetadata,
         };
     let operation = await codeAssistPost(accessToken, "onboardUser", onboardReq);
     while (!operation.done && operation.name) {
@@ -220,6 +258,31 @@ async function setupCodeAssistProject(accessToken) {
         throwProjectError(loadRes);
     return project;
 }
+async function loadCodeAssist(accessToken, envProject, metadata) {
+    try {
+        return await codeAssistPost(accessToken, "loadCodeAssist", {
+            ...(envProject ? { cloudaicompanionProject: envProject } : {}),
+            metadata,
+        });
+    }
+    catch (err) {
+        if (err instanceof CodeAssistHttpError && isVpcScAffectedError(err)) {
+            return { currentTier: { id: USER_TIER_STANDARD } };
+        }
+        if (err instanceof CodeAssistHttpError &&
+            err.status === 403 &&
+            envProject === "cloudshell-gca") {
+            throw new Error("Access to the default Cloud Shell Gemini project was denied.\n" +
+                "Please set your own Google Cloud project by running:\n" +
+                "gcloud config set project [PROJECT_ID]\n" +
+                "or setting export GOOGLE_CLOUD_PROJECT=...", { cause: err });
+        }
+        throw err;
+    }
+}
+function getValidationRequiredTier(response) {
+    return response.ineligibleTiers?.find((tier) => tier.reasonCode === VALIDATION_REQUIRED_REASON && typeof tier.validationUrl === "string");
+}
 function getOnboardTier(response) {
     const defaultTier = response.allowedTiers?.find((tier) => tier.isDefault);
     return defaultTier ?? { id: USER_TIER_LEGACY, name: "" };
@@ -234,28 +297,73 @@ function throwProjectError(response) {
     throw new Error("Gemini requires a Google Cloud project for this account. Set GOOGLE_CLOUD_PROJECT and try again.");
 }
 async function codeAssistPost(accessToken, method, body) {
-    const response = await fetch(`${CODE_ASSIST_BASE_URL}/v1internal:${method}`, {
-        method: "POST",
-        headers: codeAssistHeaders(accessToken),
-        body: JSON.stringify(body),
-    });
-    if (!response.ok) {
-        const text = await response.text();
-        throw new Error(`Gemini Code Assist ${method} failed (${response.status}): ${text}`);
+    let lastError;
+    for (let attempt = 0; attempt <= CODE_ASSIST_POST_RETRIES; attempt++) {
+        try {
+            return await codeAssistRequest(getCodeAssistMethodUrl(method), accessToken, method, {
+                method: "POST",
+                body: JSON.stringify(body),
+            });
+        }
+        catch (err) {
+            if (!(err instanceof CodeAssistHttpError) ||
+                attempt === CODE_ASSIST_POST_RETRIES ||
+                !shouldRetryCodeAssistStatus(err.status)) {
+                throw err;
+            }
+            lastError = err;
+        }
+        await new Promise((resolve) => setTimeout(resolve, CODE_ASSIST_POST_RETRY_DELAY_MS));
     }
-    return (await response.json());
+    throw lastError ?? new Error(`Gemini Code Assist ${method} failed.`);
 }
 async function codeAssistGet(accessToken, operationName) {
-    const response = await fetch(`${CODE_ASSIST_BASE_URL}/v1internal/${operationName}`, {
+    return codeAssistRequest(getCodeAssistOperationUrl(operationName), accessToken, "operation", {
         method: "GET",
+    });
+}
+async function codeAssistRequest(url, accessToken, label, init) {
+    const response = await fetch(url, {
+        ...init,
         headers: codeAssistHeaders(accessToken),
     });
     if (!response.ok) {
         const text = await response.text();
-        throw new Error(`Gemini Code Assist operation failed (${response.status}): ${text}`);
+        throw new CodeAssistHttpError(label, response.status, text);
     }
     return (await response.json());
 }
+function getCodeAssistBaseUrl() {
+    const endpoint = process.env.CODE_ASSIST_ENDPOINT ?? CODE_ASSIST_BASE_URL;
+    const version = process.env.CODE_ASSIST_API_VERSION || CODE_ASSIST_API_VERSION;
+    return `${endpoint}/${version}`;
+}
+function getCodeAssistMethodUrl(method) {
+    return `${getCodeAssistBaseUrl()}:${method}`;
+}
+function getCodeAssistOperationUrl(operationName) {
+    return `${getCodeAssistBaseUrl()}/${operationName}`;
+}
+function shouldRetryCodeAssistStatus(status) {
+    return status === 429 || status === 499 || (status >= 500 && status <= 599);
+}
+function isVpcScAffectedError(error) {
+    try {
+        const parsed = JSON.parse(error.body);
+        if (!parsed || typeof parsed !== "object" || !("error" in parsed))
+            return false;
+        const details = parsed.error?.details;
+        return Array.isArray(details)
+            ? details.some((detail) => detail != null &&
+                typeof detail === "object" &&
+                "reason" in detail &&
+                detail.reason === VPC_SC_REASON)
+            : false;
+    }
+    catch {
+        return false;
+    }
+}
 function codeAssistHeaders(accessToken) {
     return {
         Authorization: `Bearer ${accessToken}`,