fix(ai,coding-agent): support anthropic-style cache control for openai compatibles closes #3392
This commit is contained in:
152
packages/ai/test/openai-completions-cache-control-format.test.ts
Normal file
152
packages/ai/test/openai-completions-cache-control-format.test.ts
Normal file
@@ -0,0 +1,152 @@
|
||||
import { Type } from "@sinclair/typebox";
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { getModel } from "../src/models.js";
|
||||
import { streamOpenAICompletions } from "../src/providers/openai-completions.js";
|
||||
import type { Model } from "../src/types.js";
|
||||
|
||||
interface CacheControl {
|
||||
type: "ephemeral";
|
||||
ttl?: string;
|
||||
}
|
||||
|
||||
interface TextPart {
|
||||
type: "text";
|
||||
text: string;
|
||||
cache_control?: CacheControl;
|
||||
}
|
||||
|
||||
interface ToolWithCacheControl {
|
||||
type: string;
|
||||
cache_control?: CacheControl;
|
||||
}
|
||||
|
||||
interface CapturedParams {
|
||||
messages: Array<{
|
||||
role: string;
|
||||
content: string | TextPart[] | null;
|
||||
}>;
|
||||
tools?: ToolWithCacheControl[];
|
||||
}
|
||||
|
||||
const mockState = vi.hoisted(() => ({
|
||||
lastParams: undefined as CapturedParams | undefined,
|
||||
}));
|
||||
|
||||
vi.mock("openai", () => {
|
||||
class FakeOpenAI {
|
||||
chat = {
|
||||
completions: {
|
||||
create: (params: CapturedParams) => {
|
||||
mockState.lastParams = params;
|
||||
const stream = {
|
||||
async *[Symbol.asyncIterator]() {
|
||||
yield {
|
||||
id: "chatcmpl-test",
|
||||
choices: [{ delta: {}, finish_reason: "stop" }],
|
||||
usage: {
|
||||
prompt_tokens: 1,
|
||||
completion_tokens: 1,
|
||||
prompt_tokens_details: { cached_tokens: 0 },
|
||||
completion_tokens_details: { reasoning_tokens: 0 },
|
||||
},
|
||||
};
|
||||
},
|
||||
};
|
||||
const promise = Promise.resolve(stream) as Promise<typeof stream> & {
|
||||
withResponse: () => Promise<{
|
||||
data: typeof stream;
|
||||
response: { status: number; headers: Headers };
|
||||
}>;
|
||||
};
|
||||
promise.withResponse = async () => ({
|
||||
data: stream,
|
||||
response: { status: 200, headers: new Headers() },
|
||||
});
|
||||
return promise;
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
return { default: FakeOpenAI };
|
||||
});
|
||||
|
||||
async function capturePayload(
|
||||
model: Model<"openai-completions">,
|
||||
options?: { cacheRetention?: "none" | "short" | "long" },
|
||||
): Promise<CapturedParams> {
|
||||
const timestamp = Date.now();
|
||||
|
||||
await streamOpenAICompletions(
|
||||
model,
|
||||
{
|
||||
systemPrompt: "System prompt",
|
||||
messages: [{ role: "user", content: "Hello", timestamp }],
|
||||
tools: [
|
||||
{
|
||||
name: "read",
|
||||
description: "Read a file",
|
||||
parameters: Type.Object({
|
||||
path: Type.String(),
|
||||
}),
|
||||
},
|
||||
],
|
||||
},
|
||||
{ apiKey: "test-key", ...options },
|
||||
).result();
|
||||
|
||||
if (!mockState.lastParams) {
|
||||
throw new Error("Expected payload to be captured");
|
||||
}
|
||||
|
||||
return mockState.lastParams;
|
||||
}
|
||||
|
||||
function getInstructionMessage(params: CapturedParams) {
|
||||
return params.messages.find((message) => message.role === "system" || message.role === "developer");
|
||||
}
|
||||
|
||||
function expectAnthropicCacheMarkers(params: CapturedParams): void {
|
||||
const instructionMessage = getInstructionMessage(params);
|
||||
expect(instructionMessage).toBeDefined();
|
||||
expect(Array.isArray(instructionMessage?.content)).toBe(true);
|
||||
expect((instructionMessage?.content as TextPart[])[0]?.cache_control).toEqual({ type: "ephemeral" });
|
||||
|
||||
expect(params.tools).toHaveLength(1);
|
||||
expect(params.tools?.[0]?.cache_control).toEqual({ type: "ephemeral" });
|
||||
|
||||
const lastMessage = params.messages[params.messages.length - 1];
|
||||
expect(lastMessage.role).toBe("user");
|
||||
expect(Array.isArray(lastMessage.content)).toBe(true);
|
||||
expect((lastMessage.content as TextPart[])[0]?.cache_control).toEqual({ type: "ephemeral" });
|
||||
}
|
||||
|
||||
describe("openai-completions cacheControlFormat", () => {
|
||||
beforeEach(() => {
|
||||
mockState.lastParams = undefined;
|
||||
});
|
||||
|
||||
it("applies Anthropic-style cache markers for built-in opencode-go Qwen models", async () => {
|
||||
const model = getModel("opencode-go", "qwen3.5-plus");
|
||||
expect(model.compat?.cacheControlFormat).toBe("anthropic");
|
||||
|
||||
const params = await capturePayload(model);
|
||||
expectAnthropicCacheMarkers(params);
|
||||
});
|
||||
|
||||
it("preserves Anthropic-style cache markers for OpenRouter Anthropic models", async () => {
|
||||
const model = getModel("openrouter", "anthropic/claude-sonnet-4");
|
||||
const params = await capturePayload(model);
|
||||
expectAnthropicCacheMarkers(params);
|
||||
});
|
||||
|
||||
it("omits Anthropic-style cache markers when cacheRetention is none", async () => {
|
||||
const model = getModel("opencode-go", "qwen3.5-plus");
|
||||
const params = await capturePayload(model, { cacheRetention: "none" });
|
||||
const instructionMessage = getInstructionMessage(params);
|
||||
|
||||
expect(Array.isArray(instructionMessage?.content)).toBe(false);
|
||||
expect(params.tools?.[0]?.cache_control).toBeUndefined();
|
||||
expect(typeof params.messages[params.messages.length - 1]?.content).toBe("string");
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user