fix: correct Ollama context metadata for GPT-5.5 and GPT-5.4 models (#420)

Author: TutuchanXD

src/ollama/bridge.ts

@@ -53,8 +53,11 @@ class OllamaBridgeError extends Error {
 }
 
 const CONTEXT_WINDOW_OVERRIDES = new Map<string, number>([
-  ["gpt-5.4", 272000],
-  ["gpt-5.4-mini", 272000],
+  ["gpt-5.5", 400000],
+  ["gpt-5.4", 400000],
+  ["gpt-5.4-pro", 400000],
+  ["gpt-5.4-mini", 400000],
+  ["gpt-5.4-nano", 400000],
   ["gpt-5.3-codex", 272000],
   ["gpt-5.3-codex-spark", 272000],
   ["gpt-5.2", 272000],
@@ -107,6 +110,7 @@ function responseHeaders(init: HeadersInit, request?: Request): Headers {
 
 function inferFamily(modelId: string): string {
   const normalized = modelId.toLowerCase();
+  if (normalized.startsWith("gpt-5.5")) return "gpt-5.5";
   if (normalized.startsWith("gpt-5.4")) return "gpt-5.4";
   if (normalized.startsWith("gpt-5.3")) return "gpt-5.3";
   if (normalized.startsWith("gpt-5.2")) return "gpt-5.2";

tests/unit/ollama/bridge.test.ts

@@ -105,36 +105,40 @@ describe("Ollama bridge routes", () => {
     expect(body.models[0].digest).toMatch(/^[a-f0-9]{64}$/);
   });
 
-  it("returns model show metadata and can suppress vision capability", async () => {
-    fetchMock.mockResolvedValueOnce(json({
-      id: "gpt-5.4-mini",
-      displayName: "GPT 5.4 mini",
-      inputModalities: ["text", "image"],
-      supportedReasoningEfforts: [{ reasoningEffort: "medium" }],
-      defaultReasoningEffort: "medium",
-    }));
-    const app = createApp(true);
-
-    const res = await app.request("/api/show", {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ model: "gpt-5.4-mini" }),
-    });
-
-    expect(res.status).toBe(200);
-    expect(fetchMock).toHaveBeenCalledWith(
-      "http://upstream.test/v1/models/gpt-5.4-mini/info",
-      expect.any(Object),
-    );
-    const body = await res.json() as Record<string, unknown>;
-    expect(body.capabilities).toEqual(["completion", "tools", "thinking"]);
-    expect(body.parameters).toBe("num_ctx 272000\nreasoning medium");
-    expect(body.model_info).toMatchObject({
-      "gpt-5.4.context_length": 272000,
-      upstream_id: "gpt-5.4-mini",
-      input_modalities: ["text", "image"],
+  it.each(["gpt-5.5", "gpt-5.4-mini"])(
+    "returns 400k context metadata for %s and can suppress vision capability",
+    async (model) => {
+      fetchMock.mockResolvedValueOnce(json({
+        id: model,
+        displayName: model,
+        inputModalities: ["text", "image"],
+        supportedReasoningEfforts: [{ reasoningEffort: "medium" }],
+        defaultReasoningEffort: "medium",
+      }));
+      const app = createApp(true);
+
+      const res = await app.request("/api/show", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ model }),
+      });
+
+      expect(res.status).toBe(200);
+      expect(fetchMock).toHaveBeenCalledWith(
+        `http://upstream.test/v1/models/${model}/info`,
+        expect.any(Object),
+      );
+      const body = await res.json() as Record<string, unknown>;
+      expect(body.capabilities).toEqual(["completion", "tools", "thinking"]);
+      expect(body.parameters).toBe("num_ctx 400000\nreasoning medium");
+      const architecture = model.startsWith("gpt-5.4") ? "gpt-5.4" : model;
+      expect(body.model_info).toMatchObject({
+        [`${architecture}.context_length`]: 400000,
+        context_length: 400000,
+        upstream_id: model,
+        input_modalities: ["text", "image"],
+      });
     });
-  });
 
   it("converts non-streaming Ollama chat requests and responses", async () => {
     fetchMock.mockResolvedValueOnce(json({