diff --git a/src/agents/pi-embedded-runner/cache-ttl.test.ts b/src/agents/pi-embedded-runner/cache-ttl.test.ts index f5ff8be2827..066cb555a30 100644 --- a/src/agents/pi-embedded-runner/cache-ttl.test.ts +++ b/src/agents/pi-embedded-runner/cache-ttl.test.ts @@ -29,6 +29,8 @@ describe("isCacheTtlEligibleProvider", () => { it("allows moonshot and zai providers", () => { expect(isCacheTtlEligibleProvider("moonshot", "kimi-k2.5")).toBe(true); expect(isCacheTtlEligibleProvider("zai", "glm-5")).toBe(true); + expect(isCacheTtlEligibleProvider("deepinfra", "zai-org/glm-5")).toBe(true); + expect(isCacheTtlEligibleProvider("deepinfra", "moonshotai/kimi-k2.5")).toBe(true); }); it("is case-insensitive for native providers", () => { @@ -46,5 +48,6 @@ describe("isCacheTtlEligibleProvider", () => { it("rejects unsupported providers and models", () => { expect(isCacheTtlEligibleProvider("openai", "gpt-4o")).toBe(false); expect(isCacheTtlEligibleProvider("openrouter", "openai/gpt-4o")).toBe(false); + expect(isCacheTtlEligibleProvider("deepinfra", "openai/gpt-4o")).toBe(false); }); });