fix(prometheus): respect fallback chain when no explicit model configured
Previously, Prometheus always used params.currentModel (UI-selected model) when no explicit configuration existed. This bypassed the intended fallback chain (claude-opus-4-6 → gpt-5.4 → glm-5 → gemini-3.1-pro). Now, the UI-selected model is only used if it matches one of the models in Prometheus's fallback chain. Otherwise, the fallback chain resolution takes over and finds an available model. Fixes #2986
This commit is contained in:
@@ -0,0 +1,233 @@
|
|||||||
|
import { describe, expect, test, spyOn, afterEach, beforeEach } from "bun:test";
|
||||||
|
import { buildPrometheusAgentConfig } from "./prometheus-agent-config-builder";
|
||||||
|
import * as shared from "../shared";
|
||||||
|
import * as categoryResolver from "./category-config-resolver";
|
||||||
|
import type { CategoryConfig } from "../config/schema";
|
||||||
|
|
||||||
|
describe("buildPrometheusAgentConfig", () => {
|
||||||
|
let fetchAvailableModelsSpy: ReturnType<typeof spyOn>;
|
||||||
|
let readConnectedProvidersCacheSpy: ReturnType<typeof spyOn>;
|
||||||
|
let resolveCategoryConfigSpy: ReturnType<typeof spyOn>;
|
||||||
|
let logSpy: ReturnType<typeof spyOn>;
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
fetchAvailableModelsSpy = spyOn(shared, "fetchAvailableModels").mockResolvedValue(new Set());
|
||||||
|
readConnectedProvidersCacheSpy = spyOn(shared, "readConnectedProvidersCache").mockReturnValue(null);
|
||||||
|
resolveCategoryConfigSpy = spyOn(categoryResolver, "resolveCategoryConfig").mockImplementation(
|
||||||
|
(category) => ({ model: `${category}/default-model` } as CategoryConfig)
|
||||||
|
);
|
||||||
|
logSpy = spyOn(shared, "log").mockImplementation(() => {});
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
fetchAvailableModelsSpy.mockRestore();
|
||||||
|
readConnectedProvidersCacheSpy.mockRestore();
|
||||||
|
resolveCategoryConfigSpy.mockRestore();
|
||||||
|
logSpy.mockRestore();
|
||||||
|
});
|
||||||
|
|
||||||
|
describe("#given no explicit Prometheus model configured", () => {
|
||||||
|
describe("#when currentModel is NOT in Prometheus fallback chain", () => {
|
||||||
|
test("falls through to fallback chain instead of using currentModel as override", async () => {
|
||||||
|
// given - currentModel is a model NOT in Prometheus fallback chain
|
||||||
|
// Prometheus chain: claude-opus-4-6, gpt-5.4, glm-5, gemini-3.1-pro
|
||||||
|
const currentModel = "some-provider/gpt-5.3-codex";
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: undefined,
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then - should NOT have resolved via override (currentModel)
|
||||||
|
// The model should fall through to fallback chain
|
||||||
|
const lastLogCall = logSpy.mock.calls[logSpy.mock.calls.length - 1];
|
||||||
|
const lastLogMessage = lastLogCall?.[0] as string;
|
||||||
|
expect(lastLogMessage).not.toContain("UI selection");
|
||||||
|
expect(lastLogMessage).not.toContain("config override");
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe("#when currentModel IS in Prometheus fallback chain", () => {
|
||||||
|
test("preserves currentModel as uiSelectedModel (override)", async () => {
|
||||||
|
// given - currentModel matches a Prometheus fallback chain entry
|
||||||
|
const currentModel = "anthropic/claude-opus-4-6";
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: undefined,
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then - should have resolved via UI selection (currentModel as override)
|
||||||
|
const uiSelectionLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("UI selection")
|
||||||
|
);
|
||||||
|
expect(uiSelectionLog).toBeDefined();
|
||||||
|
expect(uiSelectionLog?.[1]).toEqual({ model: "claude-opus-4-6" });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("matches gpt-5.4 from fallback chain", async () => {
|
||||||
|
// given
|
||||||
|
const currentModel = "openai/gpt-5.4";
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: undefined,
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then
|
||||||
|
const uiSelectionLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("UI selection")
|
||||||
|
);
|
||||||
|
expect(uiSelectionLog).toBeDefined();
|
||||||
|
expect(uiSelectionLog?.[1]).toEqual({ model: "gpt-5.4" });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("matches glm-5 from fallback chain", async () => {
|
||||||
|
// given
|
||||||
|
const currentModel = "opencode-go/glm-5";
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: undefined,
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then
|
||||||
|
const uiSelectionLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("UI selection")
|
||||||
|
);
|
||||||
|
expect(uiSelectionLog).toBeDefined();
|
||||||
|
expect(uiSelectionLog?.[1]).toEqual({ model: "glm-5" });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("matches gemini-3.1-pro from fallback chain", async () => {
|
||||||
|
// given
|
||||||
|
const currentModel = "google/gemini-3.1-pro";
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: undefined,
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then
|
||||||
|
const uiSelectionLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("UI selection")
|
||||||
|
);
|
||||||
|
expect(uiSelectionLog).toBeDefined();
|
||||||
|
expect(uiSelectionLog?.[1]).toEqual({ model: "gemini-3.1-pro" });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe("#given explicit Prometheus model configured via plugin override", () => {
|
||||||
|
test("explicit config wins over currentModel and fallback chain", async () => {
|
||||||
|
// given
|
||||||
|
const currentModel = "anthropic/claude-opus-4-6";
|
||||||
|
const explicitModel = "custom-provider/custom-model";
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: { model: explicitModel },
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then - should resolve via config override, not UI selection
|
||||||
|
const configOverrideLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("config override")
|
||||||
|
);
|
||||||
|
expect(configOverrideLog).toBeDefined();
|
||||||
|
expect(configOverrideLog?.[1]).toEqual({ model: explicitModel });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe("#given category with model configured", () => {
|
||||||
|
test("category model wins when no explicit override", async () => {
|
||||||
|
// given
|
||||||
|
const currentModel = "anthropic/claude-opus-4-6";
|
||||||
|
const categoryModel = "category-provider/category-model";
|
||||||
|
|
||||||
|
resolveCategoryConfigSpy.mockReturnValue({
|
||||||
|
model: categoryModel,
|
||||||
|
} as CategoryConfig);
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: { category: "test-category" },
|
||||||
|
userCategories: { "test-category": { model: categoryModel } },
|
||||||
|
currentModel,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then - should resolve via category default
|
||||||
|
const categoryDefaultLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("category default")
|
||||||
|
);
|
||||||
|
expect(categoryDefaultLog).toBeDefined();
|
||||||
|
});
|
||||||
|
|
||||||
|
test("explicit model override wins over category model", async () => {
|
||||||
|
// given
|
||||||
|
const categoryModel = "category-provider/category-model";
|
||||||
|
const explicitModel = "explicit-provider/explicit-model";
|
||||||
|
|
||||||
|
resolveCategoryConfigSpy.mockReturnValue({
|
||||||
|
model: categoryModel,
|
||||||
|
} as CategoryConfig);
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: {
|
||||||
|
category: "test-category",
|
||||||
|
model: explicitModel,
|
||||||
|
},
|
||||||
|
userCategories: { "test-category": { model: categoryModel } },
|
||||||
|
currentModel: undefined,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then - should resolve via config override, not category default
|
||||||
|
const configOverrideLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("config override")
|
||||||
|
);
|
||||||
|
expect(configOverrideLog).toBeDefined();
|
||||||
|
expect(configOverrideLog?.[1]).toEqual({ model: explicitModel });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe("#given no currentModel and no explicit config", () => {
|
||||||
|
test("falls through to fallback chain", async () => {
|
||||||
|
// given - no currentModel, no explicit config
|
||||||
|
readConnectedProvidersCacheSpy.mockReturnValue(["anthropic"]);
|
||||||
|
|
||||||
|
// when
|
||||||
|
await buildPrometheusAgentConfig({
|
||||||
|
configAgentPlan: undefined,
|
||||||
|
pluginPrometheusOverride: undefined,
|
||||||
|
userCategories: undefined,
|
||||||
|
currentModel: undefined,
|
||||||
|
});
|
||||||
|
|
||||||
|
// then - should resolve via fallback chain
|
||||||
|
const fallbackChainLog = logSpy.mock.calls.find(
|
||||||
|
(call) => (call[0] as string).includes("fallback chain")
|
||||||
|
);
|
||||||
|
expect(fallbackChainLog).toBeDefined();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -2,6 +2,7 @@ import type { CategoryConfig } from "../config/schema";
|
|||||||
import { PROMETHEUS_PERMISSION, getPrometheusPrompt } from "../agents/prometheus";
|
import { PROMETHEUS_PERMISSION, getPrometheusPrompt } from "../agents/prometheus";
|
||||||
import { resolvePromptAppend } from "../agents/builtin-agents/resolve-file-uri";
|
import { resolvePromptAppend } from "../agents/builtin-agents/resolve-file-uri";
|
||||||
import { AGENT_MODEL_REQUIREMENTS } from "../shared/model-requirements";
|
import { AGENT_MODEL_REQUIREMENTS } from "../shared/model-requirements";
|
||||||
|
import type { FallbackEntry } from "../shared/model-requirements";
|
||||||
import {
|
import {
|
||||||
fetchAvailableModels,
|
fetchAvailableModels,
|
||||||
readConnectedProvidersCache,
|
readConnectedProvidersCache,
|
||||||
@@ -22,6 +23,20 @@ type PrometheusOverride = Record<string, unknown> & {
|
|||||||
prompt_append?: string;
|
prompt_append?: string;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
function isModelInFallbackChain(
|
||||||
|
model: string | undefined,
|
||||||
|
fallbackChain: FallbackEntry[] | undefined,
|
||||||
|
): boolean {
|
||||||
|
if (!model || !fallbackChain || fallbackChain.length === 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const modelParts = model.split("/");
|
||||||
|
const modelName = modelParts.length >= 2 ? modelParts.slice(1).join("/") : model;
|
||||||
|
|
||||||
|
return fallbackChain.some((entry) => entry.model === modelName);
|
||||||
|
}
|
||||||
|
|
||||||
export async function buildPrometheusAgentConfig(params: {
|
export async function buildPrometheusAgentConfig(params: {
|
||||||
configAgentPlan: Record<string, unknown> | undefined;
|
configAgentPlan: Record<string, unknown> | undefined;
|
||||||
pluginPrometheusOverride: PrometheusOverride | undefined;
|
pluginPrometheusOverride: PrometheusOverride | undefined;
|
||||||
@@ -42,9 +57,18 @@ export async function buildPrometheusAgentConfig(params: {
|
|||||||
const configuredPrometheusModel =
|
const configuredPrometheusModel =
|
||||||
params.pluginPrometheusOverride?.model ?? categoryConfig?.model;
|
params.pluginPrometheusOverride?.model ?? categoryConfig?.model;
|
||||||
|
|
||||||
|
const shouldUseCurrentModel = isModelInFallbackChain(
|
||||||
|
params.currentModel,
|
||||||
|
requirement?.fallbackChain,
|
||||||
|
);
|
||||||
|
|
||||||
const modelResolution = resolveModelPipeline({
|
const modelResolution = resolveModelPipeline({
|
||||||
intent: {
|
intent: {
|
||||||
uiSelectedModel: configuredPrometheusModel ? undefined : params.currentModel,
|
uiSelectedModel: configuredPrometheusModel
|
||||||
|
? undefined
|
||||||
|
: shouldUseCurrentModel
|
||||||
|
? params.currentModel
|
||||||
|
: undefined,
|
||||||
userModel: params.pluginPrometheusOverride?.model,
|
userModel: params.pluginPrometheusOverride?.model,
|
||||||
categoryDefaultModel: categoryConfig?.model,
|
categoryDefaultModel: categoryConfig?.model,
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user