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

use primpary llm api for user goal check (#997)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-10-17 23:07:35 -07:00
committed by GitHub
parent f132a9c435
commit ae32848275
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/forge/agent.py
View File

@@ -886,11 +886,9 @@ class ForgeAgent:
navigation_payload=task.navigation_payload, navigation_payload=task.navigation_payload,
elements=scraped_page.build_element_tree(ElementTreeFormat.HTML), elements=scraped_page.build_element_tree(ElementTreeFormat.HTML),
) )
verification_llm_api_handler = app.SECONDARY_LLM_API_HANDLER
verification_response = await verification_llm_api_handler( # this prompt is critical to our agent so let's use the primary LLM API handler
prompt=verification_prompt, step=step, screenshots=None verification_response = await app.LLM_API_HANDLER(prompt=verification_prompt, step=step, screenshots=None)
)
if "user_goal_achieved" not in verification_response or "thoughts" not in verification_response: if "user_goal_achieved" not in verification_response or "thoughts" not in verification_response:
LOG.error( LOG.error(
"Invalid LLM response for user goal success verification, skipping verification", "Invalid LLM response for user goal success verification, skipping verification",