fix(memory): harden context window cache collisions

This commit is contained in:
Vignesh Natarajan
2026-02-15 19:31:41 -08:00
parent 559c8d9930
commit cbf58d2e1c
3 changed files with 52 additions and 11 deletions

View File

@@ -1,7 +1,22 @@
import { describe, expect, it } from "vitest";
import { applyConfiguredContextWindows } from "./context.js";
import { applyConfiguredContextWindows, applyDiscoveredContextWindows } from "./context.js";
import { createSessionManagerRuntimeRegistry } from "./pi-extensions/session-manager-runtime-registry.js";
describe("applyDiscoveredContextWindows", () => {
it("keeps the smallest context window when duplicate model ids are discovered", () => {
const cache = new Map<string, number>();
applyDiscoveredContextWindows({
cache,
models: [
{ id: "claude-sonnet-4-5", contextWindow: 1_000_000 },
{ id: "claude-sonnet-4-5", contextWindow: 200_000 },
],
});
expect(cache.get("claude-sonnet-4-5")).toBe(200_000);
});
});
describe("applyConfiguredContextWindows", () => {
it("overrides discovered cache values with explicit models.providers contextWindow", () => {
const cache = new Map<string, number>([["anthropic/claude-opus-4-6", 1_000_000]]);