Fix Trinity main-session compatibility mismatch

clawsweeper[bot] · clawsweeper[bot] · commit 4c669d66cb15 · 2026-05-02T23:02:41.000Z
diff --git a/extensions/arcee/index.test.ts b/extensions/arcee/index.test.ts
@@ -150,6 +150,10 @@ describe("arcee provider plugin", () => {
       } as never),
     ).toMatchObject({
       id: "arcee/trinity-large-thinking",
+      compat: {
+        supportsReasoningEffort: false,
+        supportsTools: false,
+      },
     });
 
     expect(
@@ -196,6 +200,10 @@ describe("arcee provider plugin", () => {
     ).toMatchObject({
       id: "arcee/trinity-large-thinking",
       baseUrl: "https://openrouter.ai/api/v1",
+      compat: {
+        supportsReasoningEffort: false,
+        supportsTools: false,
+      },
     });
 
     expect(
diff --git a/extensions/arcee/index.ts b/extensions/arcee/index.ts
@@ -19,6 +19,7 @@ import {
 } from "./provider-catalog.js";
 import {
   ARCEE_TRINITY_LARGE_THINKING_COMPAT,
+  applyArceeTrinityLargeThinkingCompat,
   normalizeArceeProviderConfig,
   shouldContributeArceeTrinityLargeThinkingCompat,
 } from "./provider-policy.js";
@@ -100,7 +101,7 @@ function normalizeArceeResolvedModel<T extends { baseUrl?: string; id: string }>
     return undefined;
   }
   return {
-    ...model,
+    ...applyArceeTrinityLargeThinkingCompat(model),
     id: normalizedId,
     baseUrl: normalizedBaseUrl,
   };
diff --git a/extensions/arcee/provider-policy.ts b/extensions/arcee/provider-policy.ts
@@ -75,6 +75,32 @@ export function shouldContributeArceeTrinityLargeThinkingCompat(params: {
   return normalizeBaseUrl(params.model.baseUrl) === normalizeBaseUrl(ARCEE_BASE_URL);
 }
 
+export function applyArceeTrinityLargeThinkingCompat<T extends { id: string; compat?: unknown }>(
+  model: T,
+): T {
+  if (!isArceeTrinityLargeThinkingModelId(model.id)) {
+    return model;
+  }
+  const compat =
+    model.compat && typeof model.compat === "object"
+      ? (model.compat as Record<string, unknown>)
+      : undefined;
+  if (
+    compat?.supportsReasoningEffort ===
+      ARCEE_TRINITY_LARGE_THINKING_COMPAT.supportsReasoningEffort &&
+    compat?.supportsTools === ARCEE_TRINITY_LARGE_THINKING_COMPAT.supportsTools
+  ) {
+    return model;
+  }
+  return {
+    ...model,
+    compat: {
+      ...compat,
+      ...ARCEE_TRINITY_LARGE_THINKING_COMPAT,
+    } as T extends { compat?: infer TCompat } ? TCompat : never,
+  } as T;
+}
+
 export function normalizeArceeProviderConfig(
   providerConfig: ModelProviderConfig,
 ): ModelProviderConfig {
@@ -91,24 +117,12 @@ export function normalizeArceeProviderConfig(
   const hasModels = Array.isArray(providerConfig.models);
   const models = hasModels
     ? providerConfig.models.map((model) => {
-        if (!isArceeTrinityLargeThinkingModelId(model.id)) {
-          return model;
-        }
-        if (
-          model.compat?.supportsReasoningEffort ===
-            ARCEE_TRINITY_LARGE_THINKING_COMPAT.supportsReasoningEffort &&
-          model.compat?.supportsTools === ARCEE_TRINITY_LARGE_THINKING_COMPAT.supportsTools
-        ) {
+        const normalizedModel = applyArceeTrinityLargeThinkingCompat(model);
+        if (normalizedModel === model) {
           return model;
         }
         changed = true;
-        return {
-          ...model,
-          compat: {
-            ...model.compat,
-            ...ARCEE_TRINITY_LARGE_THINKING_COMPAT,
-          },
-        };
+        return normalizedModel;
       })
     : providerConfig.models;
 
