Enable extended thinking: budget_tokens=2000
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user