use model_name instead of name (#2557)

This commit is contained in:
Shuchang Zheng
2025-05-31 18:40:36 -07:00
committed by GitHub
parent deb38af17d
commit 800a26d323
5 changed files with 17 additions and 45 deletions

View File

@@ -867,12 +867,13 @@ class ForgeAgent:
else:
if engine in CUA_ENGINES:
self.async_operation_pool.run_operation(task.task_id, AgentPhase.llm)
json_response = await app.LLM_API_HANDLER(
prompt=extract_action_prompt,
prompt_name="extract-actions",
step=step,
screenshots=scraped_page.screenshots,
llm_key_override=llm_caller.llm_key if llm_caller else None,
llm_key_override=task.llm_key,
)
try:
json_response = await self.handle_potential_verification_code(