mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-20 06:04:15 +00:00
Prepare for alternative Codex harness certification
This commit is contained in:
parent
11dd2f476b
commit
6dcb64565a
11 changed files with 337 additions and 375 deletions
|
|
@ -1,17 +1,9 @@
|
|||
import { mkdtempSync, readFileSync, writeFileSync } from "node:fs";
|
||||
import { tmpdir } from "node:os";
|
||||
import { join } from "node:path";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { getCodexInstructions } from "../src/providers/openai-codex/prompts/codex.js";
|
||||
import { describe, expect, it } from "vitest";
|
||||
import { type RequestBody, transformRequestBody } from "../src/providers/openai-codex/request-transformer.js";
|
||||
import { parseCodexError } from "../src/providers/openai-codex/response-handler.js";
|
||||
|
||||
const DEFAULT_PROMPT_PREFIX =
|
||||
"You are an expert coding assistant. You help users with coding tasks by reading files, executing commands";
|
||||
const FALLBACK_PROMPT = readFileSync(
|
||||
new URL("../src/providers/openai-codex/prompts/codex-instructions.md", import.meta.url),
|
||||
"utf8",
|
||||
);
|
||||
|
||||
describe("openai-codex request transformer", () => {
|
||||
it("filters item_reference and strips ids", async () => {
|
||||
|
|
@ -95,79 +87,3 @@ describe("openai-codex error parsing", () => {
|
|||
expect(info.rateLimits?.primary?.used_percent).toBe(99);
|
||||
});
|
||||
});
|
||||
|
||||
describe("openai-codex prompt caching", () => {
|
||||
const originalFetch = global.fetch;
|
||||
const originalAgentDir = process.env.PI_CODING_AGENT_DIR;
|
||||
|
||||
beforeEach(() => {
|
||||
vi.restoreAllMocks();
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
global.fetch = originalFetch;
|
||||
if (originalAgentDir === undefined) {
|
||||
delete process.env.PI_CODING_AGENT_DIR;
|
||||
} else {
|
||||
process.env.PI_CODING_AGENT_DIR = originalAgentDir;
|
||||
}
|
||||
});
|
||||
|
||||
it("caches prompts with etag and reuses cache", async () => {
|
||||
const tempDir = mkdtempSync(join(tmpdir(), "pi-codex-"));
|
||||
process.env.PI_CODING_AGENT_DIR = tempDir;
|
||||
|
||||
const tag = "rust-v0.0.0";
|
||||
const promptText = "PROMPT_CONTENT";
|
||||
const etag = '"etag-123"';
|
||||
|
||||
const fetchMock = vi.fn(async (input: string | URL, init?: RequestInit) => {
|
||||
const url = typeof input === "string" ? input : input.toString();
|
||||
if (url === "https://api.github.com/repos/openai/codex/releases/latest") {
|
||||
return new Response(JSON.stringify({ tag_name: tag }), { status: 200 });
|
||||
}
|
||||
if (url.startsWith("https://raw.githubusercontent.com/openai/codex/")) {
|
||||
const headerValue =
|
||||
init?.headers && typeof init.headers === "object" && "If-None-Match" in init.headers
|
||||
? String((init.headers as Record<string, string>)["If-None-Match"])
|
||||
: undefined;
|
||||
if (headerValue === etag) {
|
||||
return new Response("", { status: 304, headers: { etag } });
|
||||
}
|
||||
return new Response(promptText, { status: 200, headers: { etag } });
|
||||
}
|
||||
return new Response("not found", { status: 404 });
|
||||
});
|
||||
|
||||
global.fetch = fetchMock as typeof fetch;
|
||||
|
||||
const first = await getCodexInstructions("gpt-5.1-codex");
|
||||
expect(first).toBe(promptText);
|
||||
|
||||
const metaPath = join(tempDir, "cache", "openai-codex", "codex-instructions-meta.json");
|
||||
const meta = JSON.parse(readFileSync(metaPath, "utf8")) as { etag: string; tag: string; lastChecked: number };
|
||||
writeFileSync(metaPath, JSON.stringify({ ...meta, lastChecked: 0 }), "utf8");
|
||||
|
||||
const second = await getCodexInstructions("gpt-5.1-codex");
|
||||
expect(second).toBe(promptText);
|
||||
expect(fetchMock).toHaveBeenCalled();
|
||||
const rawCalls = fetchMock.mock.calls.filter((call) =>
|
||||
String(call[0]).startsWith("https://raw.githubusercontent.com/openai/codex/"),
|
||||
);
|
||||
expect(rawCalls.length).toBeGreaterThan(0);
|
||||
});
|
||||
|
||||
it("falls back to bundled instructions when cache and network are unavailable", async () => {
|
||||
const tempDir = mkdtempSync(join(tmpdir(), "pi-codex-"));
|
||||
process.env.PI_CODING_AGENT_DIR = tempDir;
|
||||
|
||||
const fetchMock = vi.fn(async () => {
|
||||
throw new Error("network down");
|
||||
});
|
||||
|
||||
global.fetch = fetchMock as typeof fetch;
|
||||
|
||||
const instructions = await getCodexInstructions("gpt-5.1-codex");
|
||||
expect(instructions).toBe(FALLBACK_PROMPT);
|
||||
});
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue