Server side implementation for interactive entry edits

sia/web/api.py

@@ -5,11 +5,21 @@ import asyncio
 from ..auto_approver import AutoApprover
 from ..web_agent import WebAgent
 from ..web_io_buffer import WebIOBuffer
+from ..working_memory import WorkingMemory
+from ..entry.entry_factory import EntryFactory
 
 class Api:
-    def __init__(self, app: web.Application, agent: WebAgent, io_buffer: WebIOBuffer, auto_approver: AutoApprover):
+    def __init__(
+            self,
+            app: web.Application,
+            agent: WebAgent,
+            io_buffer: WebIOBuffer,
+            working_memory: WorkingMemory,
+            auto_approver: AutoApprover
+        ):
         self._app = app
         self._agent = agent
+        self._working_memory = working_memory
         self._io_buffer = io_buffer
         self._auto_approver = auto_approver
         
@@ -31,6 +41,11 @@ class Api:
         self._app.router.add_post("/api/auto_approver/context_timeout", self._set_context_timeout)
         self._app.router.add_post("/api/auto_approver/response_timeout", self._set_response_timeout)
         self._app.router.add_post("/api/auto_approver/llm", self._set_llm_name)
+        self._app.router.add_get("/api/memory", self._get_memory)
+        self._app.router.add_post("/api/memory/entry", self._create_entry)
+        self._app.router.add_put("/api/memory/entry/{id}", self._update_entry)
+        self._app.router.add_delete("/api/memory/entry/{id}", self._delete_entry)
+        self._app.router.add_post("/api/memory/entry/{id}/reset", self._reset_entry)
 
     async def _run_inference(self, request: web.Request) -> web.Response:
         """Start inference on specified LLM."""
@@ -174,4 +189,53 @@ class Api:
             self._auto_approver.llm_name = name
             return web.Response(status=200)
         except ValueError as e:
-            return web.Response(status=400, text=str(e))
+            return web.Response(status=400, text=str(e))
+
+    async def _get_memory(self, request: web.Request) -> web.Response:
+        """Get complete working memory state."""
+        entries = self._working_memory.get_entries()
+        return web.Response(
+            text=json.dumps([e.serialize() for e in entries]),
+            content_type="application/json"
+        )
+
+    async def _create_entry(self, request: web.Request) -> web.Response:
+        """Create a new entry in working memory."""
+        data = await request.json()
+        try:
+            entry = EntryFactory.create_entry(data, self._io_buffer)
+            self._working_memory.add_entry(entry)
+            return web.Response(
+                text=json.dumps({"id": entry.id}),
+                content_type="application/json"
+            )
+        except (ValueError, TypeError) as e:
+            return web.Response(status=400, text=str(e))
+
+    async def _update_entry(self, request: web.Request) -> web.Response:
+        """Update properties of an existing entry."""
+        entry_id = request.match_info["id"]
+        data = await request.json()
+        entry = self._working_memory.get_entry(entry_id)
+        if not entry:
+            return web.Response(status=404, text="Entry not found")
+        try:
+            EntryFactory.update_entry(entry, data)
+            return web.Response(status=200)
+        except ValueError as e:
+            return web.Response(status=400, text=str(e))
+
+    async def _delete_entry(self, request: web.Request) -> web.Response:
+        """Delete an entry from working memory."""
+        entry_id = request.match_info["id"]
+        self._working_memory.remove_entry(entry_id)
+        return web.Response(status=200)
+
+    async def _reset_entry(self, request: web.Request) -> web.Response:
+        """Reset an entry's state."""
+        entry_id = request.match_info["id"]
+        entry = self._working_memory.get_entry(entry_id)
+        if not entry:
+            return web.Response(status=404, text="Entry not found")
+        entry.reset()
+        return web.Response(status=200)