Merge pull request #81 from PathOnAI/replay-cleanup

Replay cleanup

Author: IBMCIBMC265

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:

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: