Feature/session memory

agent/nodes/brain.py

@@ -198,6 +198,13 @@ async def run(
                 query=query,
             )
 
+            await context.websocket.send_json({
+                "type": "runtime_action",
+                "action": "web_search",
+                "id": tool_call_id,
+                "status": "completed",
+            })
+
             context.runtime_search_result = search_result
             context.runtime_search_result_id = tool_call_id
 

app_settings.py

@@ -9,6 +9,7 @@
 class AppSettings:
 
     CHAT_ENDPOINT: str
+    MODELS_ENDPOINT: str
 
     USE_SERVICE_AS_BRAIN: bool
 
@@ -39,6 +40,7 @@ class AppSettings:
 settings = AppSettings(
 
     CHAT_ENDPOINT=config.CHAT_ENDPOINT,
+    MODELS_ENDPOINT=config.MODELS_ENDPOINT,
 
     USE_SERVICE_AS_BRAIN=config.USE_SERVICE_AS_BRAIN,