From 5caee79f43d6750e9b371d7065e6e9676a303e4b Mon Sep 17 00:00:00 2001
From: Automation <automation@omniroute>
Date: Sun, 10 May 2026 23:47:31 +0200
Subject: [PATCH] fix(catalog): ensure individual models get context_length via
 getTokenLimit fallback
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

When the /v1/models catalog builds entries for individual provider
chat models, context_length was previously only set when the
REGISTRY provider entry carried defaultContextLength. For providers
without that field (or when alias resolution fails to map to a
REGISTRY key), models shipped without any context_length, causing
OpenCode and other clients to fall back to a ~4000 token limit.

Now getDefaultContextFallback calls getTokenLimit() as the ultimate
fallback, which resolves through env overrides, models.dev DB,
name heuristics, and hardcoded defaults — always returning a value.

Fixes the same class of bug as 3dc7542e (combo context_length)
but for individual (non-combo) models.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
---
 src/app/api/v1/models/catalog.ts        |  8 +++-
 tests/unit/models-catalog-route.test.ts | 57 +++++++++++++++++++++++++
 2 files changed, 64 insertions(+), 1 deletion(-)

diff --git a/src/app/api/v1/models/catalog.ts b/src/app/api/v1/models/catalog.ts
index 747253635f..49e9d7c944 100644
--- a/src/app/api/v1/models/catalog.ts
+++ b/src/app/api/v1/models/catalog.ts
@@ -839,7 +839,13 @@ export async function getUnifiedModelsResponse(
       const provider = typeof model.owned_by === "string" ? model.owned_by : null;
       if (!provider) return undefined;
       const canonicalId = aliasToProviderId[provider] || provider;
-      return REGISTRY[canonicalId]?.defaultContextLength;
+
+      const registryFallback = REGISTRY[canonicalId]?.defaultContextLength;
+      if (registryFallback) return registryFallback;
+
+      const modelId =
+        model.root || (typeof model.id === "string" ? model.id.split("/").pop() : undefined);
+      return modelId ? getTokenLimit(canonicalId, modelId) : getTokenLimit(canonicalId);
     };
 
     const enrichedModels = finalModels.map((model) => {
diff --git a/tests/unit/models-catalog-route.test.ts b/tests/unit/models-catalog-route.test.ts
index d0036c91b9..8ee3ed4704 100644
--- a/tests/unit/models-catalog-route.test.ts
+++ b/tests/unit/models-catalog-route.test.ts
@@ -926,6 +926,63 @@ test("v1 models catalog auto-calculates combo context_length from targets when n
   );
 });
 
+test("v1 models catalog includes context_length for individual chat models", async () => {
+  await seedConnection("openai", { name: "openai-context" });
+  await seedConnection("claude", {
+    authType: "oauth",
+    name: "claude-context",
+    apiKey: null,
+    accessToken: "claude-access",
+  });
+
+  const response = await v1ModelsCatalog.getUnifiedModelsResponse(
+    new Request("http://localhost/api/v1/models")
+  );
+  const body = (await response.json()) as any;
+  const chatModels = body.data.filter((item) => !item.type || item.type === "chat");
+
+  assert.equal(response.status, 200);
+  assert.ok(chatModels.length > 0, "should have at least one chat model");
+
+  for (const model of chatModels) {
+    assert.ok(
+      typeof model.context_length === "number" && model.context_length > 0,
+      `chat model ${model.id} should have a positive context_length, got ${model.context_length}`
+    );
+  }
+});
+
+test("v1 models catalog falls back to getTokenLimit for models without registry defaultContextLength", async () => {
+  // opencode-go has defaultContextLength in REGISTRY, but we test the fallback
+  // path by verifying models from the synced path still get context_length
+  const connection = await seedConnection("opencode-go", {
+    name: "opencode-go-context-fallback",
+    apiKey: "go-key",
+  });
+
+  await modelsDb.replaceSyncedAvailableModelsForConnection("opencode-go", (connection as any).id, [
+    {
+      id: "test-model-no-context",
+      name: "Test Model No Context",
+      source: "imported",
+      supportedEndpoints: ["chat"],
+    },
+  ]);
+
+  const response = await v1ModelsCatalog.getUnifiedModelsResponse(
+    new Request("http://localhost/api/v1/models")
+  );
+  const body = (await response.json()) as any;
+  const model = body.data.find((item) => item.id === "opencode-go/test-model-no-context");
+
+  assert.equal(response.status, 200);
+  assert.ok(model, "synced model should appear");
+  assert.ok(
+    typeof model.context_length === "number" && model.context_length > 0,
+    `synced model without inputTokenLimit should get context_length via getTokenLimit fallback, got ${model.context_length}`
+  );
+});
+
 test("v1 models catalog prefers manual combo context_length over auto-calculated", async () => {
   await seedConnection("openai", { name: "openai-manual-context" });