Merge pull request #81 from PathOnAI/replay-cleanup

Replay cleanup
PathOnAI · Sep 25, 2024 · 4700aed · 4700aed
2 parents eece7b2 + d21283d
commit 4700aed
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/litewebagent/agents/SearchAgents/PromptSearchAgent.py b/litewebagent/agents/SearchAgents/PromptSearchAgent.py
@@ -123,6 +123,7 @@ def bfs(self):
                 "status": "pending",
                 "depth": depth,
             }
+            logger.info(trajectory_record)
             self.trajectories.append(trajectory_record)
 
             try:

diff --git a/litewebagent/utils/replay.py b/litewebagent/utils/replay.py
@@ -132,7 +132,10 @@ def take_action(step, playwright_manager, is_replay, log_folder):
     _post_extract(page)
     url = page.url
     element = find_matching_element(interactive_elements, step)
-    action = replace_number(step["action"], element['bid'])
+    if element:
+        action = replace_number(step["action"], element['bid'])
+    else:
+        action = step["action"]
     code, function_calls = action_set.to_python_code(action)
     logger.info("Executing action script")
     if is_replay: