modify websocket message for both lats agent and simple search agents for the new websocket endpoint

[TataKKKL]

https://github.com/PathOnAI/VisualTreeSearch-Demo/tree/main/visual-tree-search-backend/app/api/lwats/agents_async/SearchAgents

[TataKKKL]

#53

[TataKKKL]

Websocket Message Types

Type 1: Iteration (LATS specific)

await websocket.send_json({
    "type": "iteration_start",
    "iteration": i + 1,
    "timestamp": datetime.utcnow().isoformat()
})

Type 2: Step Start

await websocket.send_json({
    "type": "step_start",
    "step": 5,
    "step_name": "backpropagation",
    "timestamp": datetime.utcnow().isoformat()
})

Type 3: Search Complete

LATS version:

await websocket.send_json({
    "type": "search_complete",
    "status": "success" if best_node.reward == 1 else "partial_success",
    "score": best_node.reward,
    "path": best_node.get_trajectory(),
    "timestamp": datetime.utcnow().isoformat()
})

Simple search version:

await websocket.send_json({
    "type": "search_complete",
    "status": "success",
    "score": score,
    "path": [{"id": id(node), "action": node.action} for node in path[1:]],
    "timestamp": datetime.utcnow().isoformat()
})

Type 4: Node Operations

Node expanding:

await websocket.send_json({
    "type": "node_expanding", 
    "node_id": 4744117904, 
    "timestamp": datetime.utcnow().isoformat()
})

Node created:

await websocket.send_json({
    "type": "node_created", 
    "node_id": 4491780752, 
    "parent_id": 4744117904, 
    "action": "fill('223', 'running shoes')", 
    "description": "enter 'running shoes' in the search box", 
    "timestamp": datetime.utcnow().isoformat()
})

Type 5: Setup Operations

• account_reset
• browser_setup

Type 6: Tree and Trajectory Updates

# Tree update
await websocket.send_json({
    "type": "tree_update",
    "tree": tree_data,
    "timestamp": datetime.utcnow().isoformat()
})

# Trajectory update
await websocket.send_json({
    "type": "trajectory_update",
    "trajectory": trajectory_data,
    "timestamp": datetime.utcnow().isoformat()
})

Type 7: Node Selected

• Implemented for LATS
• May be optional for BFS/DFS