@@ -22,6 +22,7 @@ import {
2222 BEDROCK_DEFAULT_CONTEXT ,
2323 AWS_INFERENCE_PROFILE_MAPPING ,
2424 BEDROCK_CLAUDE_SONNET_4_MODEL_ID ,
25+ BEDROCK_1M_CONTEXT_MODEL_IDS ,
2526} from "@roo-code/types"
2627
2728import { ApiStream } from "../transform/stream"
@@ -378,10 +379,11 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
378379 inferenceConfig . topP = 0.1
379380 }
380381
381- // Check if 1M context is enabled for Claude Sonnet 4
382+ // Check if 1M context is enabled for Claude Sonnet 4 / 4.5
382383 // Use parseBaseModelId to handle cross-region inference prefixes
383384 const baseModelId = this . parseBaseModelId ( modelConfig . id )
384- const is1MContextEnabled = baseModelId === BEDROCK_CLAUDE_SONNET_4_MODEL_ID && this . options . awsBedrock1MContext
385+ const is1MContextEnabled =
386+ BEDROCK_1M_CONTEXT_MODEL_IDS . includes ( baseModelId as any ) && this . options . awsBedrock1MContext
385387
386388 // Add anthropic_beta for 1M context to additionalModelRequestFields
387389 if ( is1MContextEnabled ) {
@@ -976,10 +978,10 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
976978 }
977979 }
978980
979- // Check if 1M context is enabled for Claude Sonnet 4
981+ // Check if 1M context is enabled for Claude Sonnet 4 / 4.5
980982 // Use parseBaseModelId to handle cross-region inference prefixes
981983 const baseModelId = this . parseBaseModelId ( modelConfig . id )
982- if ( baseModelId === BEDROCK_CLAUDE_SONNET_4_MODEL_ID && this . options . awsBedrock1MContext ) {
984+ if ( BEDROCK_1M_CONTEXT_MODEL_IDS . includes ( baseModelId as any ) && this . options . awsBedrock1MContext ) {
983985 // Update context window to 1M tokens when 1M context beta is enabled
984986 modelConfig . info = {
985987 ...modelConfig . info ,
0 commit comments