Enable extended thinking: budget_tokens=2000

This commit is contained in:
daniel
2026-05-07 23:18:48 -05:00
parent b919a13c76
commit 0a0e26ccd2

View File

@@ -253,12 +253,15 @@ async def main(payload: dict, context):
user_context += f'\nEnrolled services: {", ".join(enrolled)}' user_context += f'\nEnrolled services: {", ".join(enrolled)}'
system_prompt = build_system_prompt(user_context=user_context, actor_id=actor_id) system_prompt = build_system_prompt(user_context=user_context, actor_id=actor_id)
# Model: claude-sonnet-4-6 via cross-region inference # Model: claude-sonnet-4-6 via cross-region inference with extended thinking
from botocore.config import Config as BotoConfig from botocore.config import Config as BotoConfig
model = BedrockModel( model = BedrockModel(
model_id="us.anthropic.claude-sonnet-4-6", model_id="us.anthropic.claude-sonnet-4-6",
region_name="us-east-1", region_name="us-east-1",
boto_client_config=BotoConfig(read_timeout=600, connect_timeout=10), boto_client_config=BotoConfig(read_timeout=600, connect_timeout=10),
additional_model_request_fields={
"thinking": {"type": "enabled", "budget_tokens": 2000}
},
) )
base_tools = [web_search, web_fetch, read_workspace_file, write_workspace_file, base_tools = [web_search, web_fetch, read_workspace_file, write_workspace_file,