@@ -233,11 +233,25 @@ def _safe_int_env(var_name: str, default: int) -> int:
233233 # Groq
234234 "groq:openai/gpt-oss-120b" : 128_000 ,
235235 "groq:llama-3.3-70b-versatile" : 128_000 ,
236+ # Bedrock
237+ "bedrock_converse:us.anthropic.claude-sonnet-4-5-20250929-v1:0" : 200_000 ,
238+ "bedrock_converse:us.anthropic.claude-haiku-4-5-20251001-v1:0" : 200_000 ,
239+ "bedrock_converse:us.anthropic.claude-opus-4-5-20251101-v1:0" : 200_000 ,
240+ "bedrock_converse:us.moonshot.kimi-k2-thinking" : 131_072 ,
241+ "bedrock_converse:us.anthropic.claude-3-5-sonnet-20241022-v2:0" : 200_000 ,
242+ "bedrock_converse:us.anthropic.claude-3-5-haiku-20241022-v1:0" : 200_000 ,
243+ "bedrock_converse:amazon.titan-text-premier-v1:0" : 32_768 ,
244+ "bedrock_converse:us.meta.llama3-2-90b-instruct-v1:0" : 128_000 ,
245+ "bedrock_converse:us.mistral.mistral-large-2407-v1:0" : 128_000 ,
236246}
237247
238248DEFAULT_CONTEXT_WINDOW = 200_000 # Fallback for unknown models
239249
240250
241251def get_context_window (model : str ) -> int :
242252 """Get context window size for a model, with fallback."""
243- return MODEL_CONTEXT_WINDOWS .get (model , DEFAULT_CONTEXT_WINDOW )
253+ size = MODEL_CONTEXT_WINDOWS .get (model )
254+ if size is None :
255+ logger .debug (f"context_window_fallback model={ model } using default={ DEFAULT_CONTEXT_WINDOW } " )
256+ return DEFAULT_CONTEXT_WINDOW
257+ return size
0 commit comments