Shared response buffer backend

sia/web/websockets.py

@@ -0,0 +1,28 @@
+from aiohttp import web
+
+from ..auto_approver import AutoApprover
+from ..web_agent import WebAgent
+from ..web_io_buffer import WebIOBuffer
+from ..working_memory import WorkingMemory
+from .auto_approver_websocket import AutoApproverWebSocket
+from .context_websocket import ContextWebSocket 
+from .llm_websocket import LlmWebSocket
+from .memory_websocket import MemoryWebSocket
+from .response_websocket import ResponseWebSocket
+from .stdout_websocket import StdoutWebSocket
+
+class Websockets:
+    def __init__(self, app: web.Application, agent: WebAgent, io_buffer: WebIOBuffer, auto_approver: AutoApprover, working_memory: WorkingMemory):
+        self._auto_approver_ws = AutoApproverWebSocket(auto_approver)
+        self._context_ws = ContextWebSocket(agent)
+        self._llm_ws = LlmWebSocket(agent)
+        self._memory_ws = MemoryWebSocket(working_memory)
+        self._response_ws = ResponseWebSocket(agent)
+        self._stdout_ws = StdoutWebSocket(io_buffer)
+
+        app.router.add_get("/ws/auto_approver", self._auto_approver_ws.handle_connection)
+        app.router.add_get("/ws/context", self._context_ws.handle_connection)
+        app.router.add_get("/ws/llm", self._llm_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/stdout", self._stdout_ws.handle_connection)