Respect per-agent fallback chains in runtime fallback

2026-03-04 20:25:25 +01:00
parent 4300f60aaf
commit 38c925697b
2 changed files with 68 additions and 14 deletions
@@ -0,0 +1,66 @@
+import { afterEach, describe, expect, test } from "bun:test"
+
+import { getFallbackModelsForSession } from "./fallback-models"
+import { SessionCategoryRegistry } from "../../shared/session-category-registry"
+
+describe("runtime-fallback fallback-models", () => {
+  afterEach(() => {
+    SessionCategoryRegistry.clear()
+  })
+
+  test("uses category fallback_models when session category is registered", () => {
+    //#given
+    const sessionID = "ses_runtime_fallback_category"
+    SessionCategoryRegistry.register(sessionID, "quick")
+    const pluginConfig = {
+      categories: {
+        quick: {
+          fallback_models: ["openai/gpt-5.2", "anthropic/claude-opus-4-6"],
+        },
+      },
+    } as any
+
+    //#when
+    const result = getFallbackModelsForSession(sessionID, undefined, pluginConfig)
+
+    //#then
+    expect(result).toEqual(["openai/gpt-5.2", "anthropic/claude-opus-4-6"])
+  })
+
+  test("uses agent-specific fallback_models when agent is resolved", () => {
+    //#given
+    const pluginConfig = {
+      agents: {
+        oracle: {
+          fallback_models: ["openai/gpt-5.2", "anthropic/claude-opus-4-6"],
+        },
+      },
+    } as any
+
+    //#when
+    const result = getFallbackModelsForSession("ses_runtime_fallback_agent", "oracle", pluginConfig)
+
+    //#then
+    expect(result).toEqual(["openai/gpt-5.2", "anthropic/claude-opus-4-6"])
+  })
+
+  test("does not fall back to another agent chain when agent cannot be resolved", () => {
+    //#given
+    const pluginConfig = {
+      agents: {
+        sisyphus: {
+          fallback_models: ["quotio/gpt-5.2", "quotio/glm-5", "quotio/kimi-k2.5"],
+        },
+        oracle: {
+          fallback_models: ["openai/gpt-5.2", "anthropic/claude-opus-4-6"],
+        },
+      },
+    } as any
+
+    //#when
+    const result = getFallbackModelsForSession("ses_runtime_fallback_unknown", undefined, pluginConfig)
+
+    //#then
+    expect(result).toEqual([])
+  })
+})