clean code & setup iterations 1

Author: H-Freax

visual-tree-search-app/pages/LATSAgent.tsx

@@ -33,7 +33,7 @@ const LATSAgent = () => {
     goal: 'search running shoes, click on the first result',
     maxDepth: 3,
     num_simulations: 1,
-    iterations: 3
+    iterations: 1
   });
 
   const [sessionId, setSessionId] = useState<string | null>(null);
@@ -107,7 +107,7 @@ const LATSAgent = () => {
             search_algorithm: "lats",
             max_depth: searchParams.maxDepth,
             num_simulations: searchParams.num_simulations,
-            iterations: 3
+            iterations: searchParams.iterations
           };
 
           wsRef.current?.send(JSON.stringify(request));

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

@@ -77,7 +77,8 @@ async def handle_search_request(websocket: WebSocket, message: Dict[str, Any]):
         search_algorithm = message.get("search_algorithm", "bfs")
         max_depth = message.get("max_depth", 3)
         storage_state = message.get("storage_state", "app/api/shopping.json")
-        iterations = message.get("iterations", 3)  # Extract iterations parameter
+        iterations = message.get("iterations", 1)  # Extract iterations parameter
+        num_simulations=message.get("num_simulations", 1)
 
         # Send status update
         await websocket.send_json({
@@ -94,7 +95,7 @@ async def handle_search_request(websocket: WebSocket, message: Dict[str, Any]):
             storage_state=storage_state,
             headless=False,
             iterations=iterations,
-            num_simulations=message.get("num_simulations", 1)
+            num_simulations=num_simulations
         )
 
         # Send status update