llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add context info in ui

Browse Source
This commit is contained in:
Niels Geens
2026-01-10 19:13:08 +01:00
parent 5821391309
commit 7014bba178
10 changed files with 141 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sia/web/websockets.py
View File

@@ -6,6 +6,7 @@ from ..web_agent import WebAgent
from ..working_memory import WorkingMemory
from .auto_approver_websocket import AutoApproverWebSocket
from .chat_websocket import ChatWebSocket
from .context_info_websocket import ContextInfoWebSocket
from .memory_websocket import MemoryWebSocket
from .response_websocket import ResponseWebSocket
from .state_websocket import StateWebSocket
@@ -14,12 +15,14 @@ class Websockets:
def __init__(self, app: web.Application, agent: WebAgent, io_buffer: ChatIOBuffer, auto_approver: AutoApprover, working_memory: WorkingMemory):
self._auto_approver_ws = AutoApproverWebSocket(auto_approver)
self._chat_ws = ChatWebSocket(io_buffer)
self._context_info_ws = ContextInfoWebSocket(agent)
self._memory_ws = MemoryWebSocket(working_memory)
self._response_ws = ResponseWebSocket(agent)
self._state_ws = StateWebSocket(agent)
app.router.add_get("/ws/auto_approver", self._auto_approver_ws.handle_connection)
app.router.add_get("/ws/chat", self._chat_ws.handle_connection)
app.router.add_get("/ws/context_info", self._context_info_ws.handle_connection)
app.router.add_get("/ws/memory", self._memory_ws.handle_connection)
app.router.add_get("/ws/response", self._response_ws.handle_connection)
app.router.add_get("/ws/state", self._state_ws.handle_connection)