Skip to content

Commit 04a739d

Browse files
committed
finish backend refactoring
1 parent d6e9bb2 commit 04a739d

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,10 @@ async def websocket_simulation_result(self, reward, terminal_node, websocket=Non
362362
await websocket.send_json({
363363
"type": "simulation_result",
364364
"reward": reward,
365-
"terminal_node": terminal_node,
365+
"terminal_node_id": id(terminal_node),
366+
"terminal_node_parent_id": id(terminal_node.parent),
367+
"terminal_node_action": terminal_node.action,
368+
"terminal_node_description": terminal_node.natural_language_description,
366369
"timestamp": datetime.utcnow().isoformat()
367370
})
368371
else:

visual-tree-search-backend/test/test-tree-search-ws-lats.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
'tree_update_node_expansion': '\033[96m', # Cyan
4646
'tree_update_node_evaluation': '\033[96m', # Cyan
4747
'tree_update_node_children_evaluation': '\033[96m', # Cyan
48+
'tree_update_node_backpropagation': '\033[96m', # Cyan
4849
'tree_update_simulation': '\033[96m', # Cyan
4950
'trajectory_update': '\033[96m', # Cyan
5051
'removed_simulation': '\033[96m', # Cyan

0 commit comments

Comments
 (0)