Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions open-sse/services/autoCombo/__tests__/autoCombo.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -649,4 +649,11 @@ describe("Task Fitness DB Resolution Chain", () => {
const lowerScore = getTaskFitness("claude-sonnet", "coding");
expect(upperScore).toBe(lowerScore);
});

it("normalizes -free model suffix when looking up fitness table and wildcard boosts", () => {
const baseResult = getTaskFitnessWithSource("claude-sonnet", "coding");
const freeResult = getTaskFitnessWithSource("claude-sonnet-free", "coding");
expect(freeResult.source).toBe(baseResult.source);
expect(freeResult.score).toBe(baseResult.score);
});
});
19 changes: 16 additions & 3 deletions open-sse/services/autoCombo/taskFitness.ts
Original file line number Diff line number Diff line change
Expand Up @@ -277,10 +277,19 @@ export function getModelsDevTierFitness(model: string, taskType: string): number
const dbScore = queryModelIntelligence(normalizedModel, normalizedTask, "models_dev_tier");
if (dbScore !== null) return dbScore;

const baseModel = normalizedModel.endsWith(FREE_SUFFIX)
? normalizedModel.slice(0, -FREE_SUFFIX.length)
: normalizedModel;

if (baseModel !== normalizedModel) {
const baseDbScore = queryModelIntelligence(baseModel, normalizedTask, "models_dev_tier");
if (baseDbScore !== null) return baseDbScore;
}

const caps = loadModelCapabilities();
if (!caps) return null;

const capRow = caps[normalizedModel];
const capRow = caps[normalizedModel] || caps[baseModel];
if (!capRow) return null;

const tier = deriveTierFromCapabilities(capRow);
Expand Down Expand Up @@ -347,12 +356,16 @@ export function getTaskFitnessWithSource(
return { score: tierScore, source: "models_dev_tier" };
}

const staticScore = lookupStaticFitnessTable(normalizedModel, normalizedTask);
const baseModel = normalizedModel.endsWith(FREE_SUFFIX)
? normalizedModel.slice(0, -FREE_SUFFIX.length)
: normalizedModel;

const staticScore = lookupStaticFitnessTable(baseModel, normalizedTask);
if (staticScore !== null) {
return { score: staticScore, source: "fitness_table" };
}

return { score: lookupWildcardBoosts(normalizedModel, normalizedTask), source: "wildcard_boost" };
return { score: lookupWildcardBoosts(baseModel, normalizedTask), source: "wildcard_boost" };
}

/** Suffix used to mark free-tier model variants (e.g. "mimo-v2.5-free"). */
Expand Down
Loading