fix: enable claude-cli 1m context override

This commit is contained in:
Arthur Kazemi
2026-04-24 12:48:50 +10:00
committed by Ayaan Zaidi
parent bb389a37d0
commit 8abbae0101
2 changed files with 3 additions and 26 deletions

View File

@@ -240,13 +240,12 @@ describe("resolveContextTokensForModel", () => {
expect(result).toBe(ANTHROPIC_CONTEXT_1M_TOKENS); expect(result).toBe(ANTHROPIC_CONTEXT_1M_TOKENS);
}); });
it("does not force 1M context when claude-cli context1m is enabled", () => { it("returns 1M context when claude-cli context1m is enabled for opus/sonnet", () => {
const result = resolveContextTokensForModel({ const result = resolveContextTokensForModel({
cfg: { cfg: {
models: { models: {
providers: { providers: {
"claude-cli": { "claude-cli": {
baseUrl: "https://api.anthropic.com",
models: [testModelContextWindow("claude-opus-4-7", 200_000)], models: [testModelContextWindow("claude-opus-4-7", 200_000)],
}, },
}, },
@@ -267,29 +266,7 @@ describe("resolveContextTokensForModel", () => {
allowAsyncLoad: false, allowAsyncLoad: false,
}); });
expect(result).toBe(200_000); expect(result).toBe(ANTHROPIC_CONTEXT_1M_TOKENS);
});
it("does not force 1M context for claude-cli aliases when context1m is enabled", () => {
const result = resolveContextTokensForModel({
cfg: {
agents: {
defaults: {
models: {
"claude-cli/sonnet": {
params: { context1m: true },
},
},
},
},
},
provider: "claude-cli",
model: "sonnet",
fallbackContextTokens: 200_000,
allowAsyncLoad: false,
});
expect(result).toBe(200_000);
}); });
it("does not force 1M context when context1m is not enabled", () => { it("does not force 1M context when context1m is not enabled", () => {

View File

@@ -375,7 +375,7 @@ function resolveConfiguredProviderContextTokens(
} }
function isAnthropic1MModel(provider: string, model: string): boolean { function isAnthropic1MModel(provider: string, model: string): boolean {
if (provider !== "anthropic") { if (provider !== "anthropic" && provider !== "claude-cli") {
return false; return false;
} }
const modelId = resolveModelFamilyId(model); const modelId = resolveModelFamilyId(model);