Merge pull request #78 from PathOnAI/check-lats

Add Config(Lats)

Author: H-Freax

visual-tree-search-app/components/ControlPanelLATS.tsx

@@ -9,6 +9,7 @@ interface SearchParams {
   goal: string;
   maxDepth: number;
   num_simulations: number;
+  iterations: number;
 }
 
 interface ControlPanelProps {
@@ -131,6 +132,19 @@ const ControlPanelLATS: React.FC<ControlPanelProps> = ({
                     className="border-slate-300 dark:border-slate-600 focus:ring-cyan-500 focus:border-cyan-500"
                   />
                 </div>
+
+                <div className="space-y-2">
+                  <Label htmlFor="iterations" className="text-slate-700 dark:text-slate-300 font-medium">Iterations</Label>
+                  <Input
+                    id="iterations"
+                    type="number"
+                    min={1}
+                    max={10}
+                    value={searchParams.iterations}
+                    onChange={(e) => handleParamChange('iterations', parseInt(e.target.value))}
+                    className="border-slate-300 dark:border-slate-600 focus:ring-cyan-500 focus:border-cyan-500"
+                  />
+                </div>
               </div>
             </div>
           )}

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

@@ -10,6 +10,7 @@ interface SearchParams {
   goal: string;
   maxDepth: number;
   num_simulations: number;
+  iterations: number;
 }
 
 interface Message {
@@ -31,7 +32,8 @@ const LATSAgent = () => {
     startingUrl: 'http://xwebarena.pathonai.org:7770/',
     goal: 'search running shoes, click on the first result',
     maxDepth: 3,
-    num_simulations: 1
+    num_simulations: 1,
+    iterations: 1
   });
 
   const [sessionId, setSessionId] = useState<string | null>(null);
@@ -103,6 +105,9 @@ const LATSAgent = () => {
             starting_url: searchParams.startingUrl,
             goal: searchParams.goal,
             search_algorithm: "lats",
+            max_depth: searchParams.maxDepth,
+            num_simulations: searchParams.num_simulations,
+            iterations: searchParams.iterations
           };
 
           wsRef.current?.send(JSON.stringify(request));

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

@@ -27,7 +27,7 @@ async def lats_search(self, websocket=None):
             for i in range(self.config.iterations):
                 await self.websocket_iteration_start(i, websocket=websocket)
 
-                print(f"Iteration {i}...")
+                print(f"Iteration {i}/{self.config.iterations} ...")
 
                 # Step 1: Node Selection
                 ## TODO: move websocket node selection into node_selection method

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

@@ -77,6 +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", 1)  # Extract iterations parameter
+        num_simulations=message.get("num_simulations", 1)
 
         # Send status update
         await websocket.send_json({
@@ -91,7 +93,9 @@ async def handle_search_request(websocket: WebSocket, message: Dict[str, Any]):
             search_algorithm=search_algorithm,
             max_depth=max_depth,
             storage_state=storage_state,
-            headless=False
+            headless=False,
+            iterations=iterations,
+            num_simulations=num_simulations
         )
 
         # Send status update