neuro-simulator 0.4.0__tar.gz → 0.4.1__tar.gz

{neuro_simulator-0.4.0 → neuro_simulator-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuro_simulator
-Version: 0.4.0
+Version: 0.4.1
 Summary: Neuro Simulator Server
 Author-email: Moha-Master <hongkongreporter@outlook.com>
 License-Expression: MIT

{neuro_simulator-0.4.0 → neuro_simulator-0.4.1}/neuro_simulator/agent/memory/manager.py

@@ -69,10 +69,22 @@ class MemoryManager:
     async def _save_init_memory(self):
         with open(self.init_memory_file, 'w', encoding='utf-8') as f:
             json.dump(self.init_memory, f, ensure_ascii=False, indent=2)
-            
-    async def update_init_memory(self, new_memory: Dict[str, Any]):
-        self.init_memory.update(new_memory)
+
+    async def replace_init_memory(self, new_memory: Dict[str, Any]):
+        """Replaces the entire init memory with a new object."""
+        self.init_memory = new_memory
+        await self._save_init_memory()
+
+    async def update_init_memory_item(self, key: str, value: Any):
+        """Updates or adds a single key-value pair in init memory."""
+        self.init_memory[key] = value
         await self._save_init_memory()
+
+    async def delete_init_memory_key(self, key: str):
+        """Deletes a key from init memory."""
+        if key in self.init_memory:
+            del self.init_memory[key]
+            await self._save_init_memory()
             
     async def _save_core_memory(self):
         with open(self.core_memory_file, 'w', encoding='utf-8') as f:
@@ -93,6 +105,13 @@ class MemoryManager:
         if len(self.temp_memory) > 20:
             self.temp_memory = self.temp_memory[-20:]
         await self._save_temp_memory()
+
+    async def delete_temp_memory_item(self, item_id: str):
+        """Deletes an item from temp memory by its ID."""
+        initial_len = len(self.temp_memory)
+        self.temp_memory = [item for item in self.temp_memory if item.get("id") != item_id]
+        if len(self.temp_memory) < initial_len:
+            await self._save_temp_memory()
         
     async def get_core_memory_blocks(self) -> Dict[str, Any]:
         return self.core_memory.get("blocks", {})

{neuro_simulator-0.4.0 → neuro_simulator-0.4.1}/neuro_simulator/core/application.py

@@ -341,6 +341,10 @@ async def handle_admin_ws_message(websocket: WebSocket, data: dict):
             blocks = await agent.get_memory_blocks()
             response["payload"] = blocks
         
+        elif action == "get_core_memory_block":
+            block = await agent.get_memory_block(**payload)
+            response["payload"] = block
+
         elif action == "create_core_memory_block":
             block_id = await agent.create_memory_block(**payload)
             response["payload"] = {"status": "success", "block_id": block_id}
@@ -373,6 +377,12 @@ async def handle_admin_ws_message(websocket: WebSocket, data: dict):
             updated_temp_mem = await agent.get_temp_memory()
             await connection_manager.broadcast_to_admins({"type": "temp_memory_updated", "payload": updated_temp_mem})
 
+        elif action == "delete_temp_memory_item":
+            await agent.delete_temp_memory_item(**payload)
+            response["payload"] = {"status": "success"}
+            updated_temp_mem = await agent.get_temp_memory()
+            await connection_manager.broadcast_to_admins({"type": "temp_memory_updated", "payload": updated_temp_mem})
+
         elif action == "clear_temp_memory":
             await agent.clear_temp_memory()
             response["payload"] = {"status": "success"}
@@ -390,6 +400,18 @@ async def handle_admin_ws_message(websocket: WebSocket, data: dict):
             updated_init_mem = await agent.get_init_memory()
             await connection_manager.broadcast_to_admins({"type": "init_memory_updated", "payload": updated_init_mem})
 
+        elif action == "update_init_memory_item":
+            await agent.update_init_memory_item(**payload)
+            response["payload"] = {"status": "success"}
+            updated_init_mem = await agent.get_init_memory()
+            await connection_manager.broadcast_to_admins({"type": "init_memory_updated", "payload": updated_init_mem})
+
+        elif action == "delete_init_memory_key":
+            await agent.delete_init_memory_key(**payload)
+            response["payload"] = {"status": "success"}
+            updated_init_mem = await agent.get_init_memory()
+            await connection_manager.broadcast_to_admins({"type": "init_memory_updated", "payload": updated_init_mem})
+
         # Tool Actions
         elif action == "get_all_tools":
             agent_instance = getattr(agent, 'agent_instance', agent)

{neuro_simulator-0.4.0 → neuro_simulator-0.4.1}/neuro_simulator/services/builtin.py

@@ -66,7 +66,17 @@ class BuiltinAgentWrapper(BaseAgent):
         return self.agent_instance.memory_manager.init_memory
 
     async def update_init_memory(self, memory: Dict[str, Any]):
-        await self.agent_instance.memory_manager.update_init_memory(memory)
+        await self.agent_instance.memory_manager.replace_init_memory(memory)
+        updated_init_mem = await self.get_init_memory()
+        await connection_manager.broadcast_to_admins({"type": "init_memory_updated", "payload": updated_init_mem})
+
+    async def update_init_memory_item(self, key: str, value: Any):
+        await self.agent_instance.memory_manager.update_init_memory_item(key, value)
+        updated_init_mem = await self.get_init_memory()
+        await connection_manager.broadcast_to_admins({"type": "init_memory_updated", "payload": updated_init_mem})
+
+    async def delete_init_memory_key(self, key: str):
+        await self.agent_instance.memory_manager.delete_init_memory_key(key)
         updated_init_mem = await self.get_init_memory()
         await connection_manager.broadcast_to_admins({"type": "init_memory_updated", "payload": updated_init_mem})
 
@@ -79,6 +89,11 @@ class BuiltinAgentWrapper(BaseAgent):
         updated_temp_mem = await self.get_temp_memory()
         await connection_manager.broadcast_to_admins({"type": "temp_memory_updated", "payload": updated_temp_mem})
 
+    async def delete_temp_memory_item(self, item_id: str):
+        await self.agent_instance.memory_manager.delete_temp_memory_item(item_id)
+        updated_temp_mem = await self.get_temp_memory()
+        await connection_manager.broadcast_to_admins({"type": "temp_memory_updated", "payload": updated_temp_mem})
+
     async def clear_temp_memory(self):
         await self.agent_instance.memory_manager.reset_temp_memory()
         updated_temp_mem = await self.get_temp_memory()

{neuro_simulator-0.4.0 → neuro_simulator-0.4.1}/neuro_simulator.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuro_simulator
-Version: 0.4.0
+Version: 0.4.1
 Summary: Neuro Simulator Server
 Author-email: Moha-Master <hongkongreporter@outlook.com>
 License-Expression: MIT

{neuro_simulator-0.4.0 → neuro_simulator-0.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "neuro_simulator"
-version = "0.4.0"
+version = "0.4.1"
 description = "Neuro Simulator Server"
 readme = "README.md"
 authors = [{ name = "Moha-Master", email = "hongkongreporter@outlook.com" }]