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

Handle LLM returning element_id instead of id, and handle cases where it may not be returned at all (#301)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-05-12 20:18:34 -07:00
committed by GitHub
parent ab88f24df6
commit 6a83f367ba
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
skyvern/webeye/actions/actions.py
View File

@@ -138,7 +138,13 @@ class CompleteAction(DecisiveAction):
def parse_actions(task: Task, json_response: List[Dict[str, Any]]) -> List[Action]: def parse_actions(task: Task, json_response: List[Dict[str, Any]]) -> List[Action]:
actions = [] actions = []
for action in json_response: for action in json_response:
element_id = action["id"] if "id" in action:
element_id = action["id"]
elif "element_id" in action:
element_id = action["element_id"]
else:
element_id = None
reasoning = action["reasoning"] if "reasoning" in action else None reasoning = action["reasoning"] if "reasoning" in action else None
if "action_type" not in action or action["action_type"] is None: if "action_type" not in action or action["action_type"] is None:
actions.append(NullAction(reasoning=reasoning)) actions.append(NullAction(reasoning=reasoning))