vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

enable thinking in anthropic CUA (#2240)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-29 00:16:22 +08:00
committed by GitHub
parent 8e6819c987
commit 1449720038
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/sdk/api/llm/api_handler_factory.py
View File

@@ -693,6 +693,7 @@ class LLMCaller:
max_tokens = active_parameters.get("max_completion_tokens") or active_parameters.get("max_tokens") or 4096
model_name = self.llm_config.model_name.replace("bedrock/", "").replace("anthropic/", "")
betas = active_parameters.get("betas", NOT_GIVEN)
thinking = active_parameters.get("thinking", NOT_GIVEN)
LOG.info("Anthropic request", betas=betas, tools=tools, timeout=timeout)
response = await app.ANTHROPIC_CLIENT.beta.messages.create(
max_tokens=max_tokens,
@@ -701,6 +702,7 @@ class LLMCaller:
tools=tools or NOT_GIVEN,
timeout=timeout,
betas=betas,
thinking=thinking,
)
LOG.info("Anthropic response", response=response, betas=betas, tools=tools, timeout=timeout)
return response