Merge pull request #100 from PathOnAI/localmode

bugfix localmode

Author: pentium3

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

@@ -443,7 +443,7 @@ async def node_expansion(self, node: LATSNode, websocket = None) -> None:
 
      # node evaluation
      # change the node evaluation to use the new prompt
-    async def node_children_evaluation(self, node: LATSNode) -> None:
+    async def node_children_evaluation(self, node: LATSNode, websocket = None) -> None:
         if websocket:
             await websocket.send_json({
                 "type": "evaluation_start",
@@ -480,7 +480,7 @@ async def node_children_evaluation(self, node: LATSNode) -> None:
             child.value = score
             # child.reward = score
 
-    async def node_evaluation(self, node: LATSNode) -> None:
+    async def node_evaluation(self, node: LATSNode, websocket = None) -> None:
         """Evaluate the current node and assign its score."""
         if websocket:
             await websocket.send_json({