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

fix llm_key - render every llm_key for cua in the execute_step (#2556)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-05-31 16:26:02 -07:00
committed by GitHub
parent 48f5f0913e
commit deb38af17d
3 changed files with 6 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/forge/agent.py
View File

@@ -174,6 +174,7 @@ class ForgeAgent:
max_steps_per_run=task_block.max_steps_per_run,
error_code_mapping=task_block.error_code_mapping,
include_action_history_in_verification=task_block.include_action_history_in_verification,
model=task_block.model,
)
LOG.info(
"Created a new task for workflow run",
@@ -387,7 +388,10 @@ class ForgeAgent:
llm_caller = LLMCallerManager.get_llm_caller(task.task_id)
if not llm_caller:
# if not, create a new llm_caller
llm_caller = LLMCaller(llm_key=settings.ANTHROPIC_CUA_LLM_KEY, screenshot_scaling_enabled=True)
llm_key = task.llm_key
llm_caller = LLMCaller(
llm_key=llm_key or settings.ANTHROPIC_CUA_LLM_KEY, screenshot_scaling_enabled=True
)
# TODO: remove the code after migrating everything to llm callers
# currently, only anthropic cua tasks use llm_caller