finishing testing lats agent websocket

Author: TataKKKL

visual-tree-search-backend/app/api/lwats/agents_async/SimpleSearchAgents/lats_agent.py

@@ -121,7 +121,7 @@ async def run(self, websocket=None) -> list[LATSNode]:
                 "type": "search_complete",
                 "status": "success" if best_node.reward == 1 else "partial_success",
                 "score": best_node.reward,
-                "path": [{"id": id(node), "action": node.action} for node in best_node.get_trajectory()],
+                "path": best_node.get_trajectory(),
                 "timestamp": datetime.utcnow().isoformat()
             })
 

visual-tree-search-backend/app/api/routes/tree_search_websocket.py

@@ -127,7 +127,7 @@ async def handle_search_request(websocket: WebSocket, message: Dict[str, Any]):
             # Use the agent's built-in WebSocket-enabled DFS method
             await agent.dfs_with_websocket(websocket)
         elif search_algorithm.lower() == "lats":
-            await agent.lats_search(websocket)
+            await agent.run(websocket)
         else:
             await websocket.send_json({
                 "type": "error",