neuro-simulator 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

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/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/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.dist-info → neuro_simulator-0.4.1.dist-info}/METADATA

@@ -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.dist-info → neuro_simulator-0.4.1.dist-info}/RECORD

@@ -6,7 +6,7 @@ neuro_simulator/agent/llm.py,sha256=xPBEXpZ19WOt7YkERNaY9rscNI-ePASTjIt-_sZV7UI,
 neuro_simulator/agent/memory_prompt.txt,sha256=wdpdnbOYhMKgPJnnGlcSejGdt2uItrXzDgz9_8cKnqw,824
 neuro_simulator/agent/neuro_prompt.txt,sha256=WSN5Fa6AwVnJaSFhz98fQTb2EtQ35U6w2qga_C-s1Go,1438
 neuro_simulator/agent/memory/__init__.py,sha256=YJ7cynQJI6kD7vjyv3rKc-CZqmoYSuGQtRZl_XdGEps,39
-neuro_simulator/agent/memory/manager.py,sha256=tnwxxAlpS0wEd0jfX2MG9hnyAcuhUiZOIP7mz9LCPh8,5388
+neuro_simulator/agent/memory/manager.py,sha256=R9OsPTt8lXsc2Z95lYRJCmKAOghOtHARzpOtqOnU_Wg,6214
 neuro_simulator/agent/tools/__init__.py,sha256=1WZy6PADfi6o1avyy1y-ThWBFAPJ_bBqtkobyYpf5ao,38
 neuro_simulator/agent/tools/add_temp_memory.py,sha256=R6K-iVMD6ykd7pS3uI811dZIkuuAf7nPn9c6xCzqiHc,2127
 neuro_simulator/agent/tools/add_to_core_memory_block.py,sha256=dt7y3w-qr379gS79mdata9w3hHG45wxdVIDXPnROO1Y,2306
@@ -24,13 +24,13 @@ neuro_simulator/api/system.py,sha256=OJT6m6HIYATMCAKrgeBRhficaiUjIDl9f-WyUT-RoBw
 neuro_simulator/core/__init__.py,sha256=-ojq25c8XA0CU25b0OxcGjH4IWFEDHR-HXSRSZIuKe8,57
 neuro_simulator/core/agent_factory.py,sha256=p3IKT6sNzSpUojQjvbZJu0pbmyyb258sGn_YNSJlqwI,1717
 neuro_simulator/core/agent_interface.py,sha256=ZXUCtkQUvsBQ5iCb0gTILJaShn5KmSrEgKhd7PK18HE,2794
-neuro_simulator/core/application.py,sha256=oJUuKW3U8Uadr1IlzZThef8CgNoW1zncW_U94OFOUPQ,25518
+neuro_simulator/core/application.py,sha256=mW5SlqonSn5oQvb5xuGR8FTZ_-sEx3OUTXwfC_N_-2c,26703
 neuro_simulator/core/config.py,sha256=QRbkm0h_FBTtYd-z5RRauSa1B0cp0LkNHVLqGBOBTYY,15324
 neuro_simulator/core/path_manager.py,sha256=hfjI4s8-WXlgwvPWDSLYMQ2d0OaXPOMYWWlP5xe5NLg,2954
 neuro_simulator/services/__init__.py,sha256=s3ZrAHg5TpJakadAAGY1h0wDw_xqN4Je4aJwJyRBmk4,61
 neuro_simulator/services/audience.py,sha256=sAmvkz1ip1MNqaw7t-9abqfnmp0yh8EdG5bS5KYR-Us,4999
 neuro_simulator/services/audio.py,sha256=EElBue80njZY8_CKQhVtZEcGchJUdmS4EDOvJEEu_LY,2909
-neuro_simulator/services/builtin.py,sha256=NrdQhf4BuB8_evxnz8m6wiY9DgLwhY3l7Yp5TnmHGYo,5156
+neuro_simulator/services/builtin.py,sha256=7b5F2StGX2ZT1jMiAAuqov0mx5OMHvUYheo8E1c9fjQ,6097
 neuro_simulator/services/letta.py,sha256=OEM4qYeRRPoj8qY645YZwCHmMfg1McLp0eTbSAu3KL0,11976
 neuro_simulator/services/stream.py,sha256=eueAaxhAwX_n0mT1W0W4YOVCe7fH5uPoMdRf2o0wQMI,5937
 neuro_simulator/utils/__init__.py,sha256=xSEFzjT827W81mNyQ_DLtr00TgFlttqfFgpz9pSxFXQ,58
@@ -39,8 +39,8 @@ neuro_simulator/utils/process.py,sha256=9OYWx8fzaJZqmFUcjQX37AnBhl7YWvrLxDWBa30v
 neuro_simulator/utils/queue.py,sha256=bg-mIFF8ycClwmmfcKSPjAXtvjImzTKf6z7xZc2VX20,2196
 neuro_simulator/utils/state.py,sha256=DdBqSAYfjOFtJfB1hEGhYPh32r1ZvFuVlN_-29_-luA,664
 neuro_simulator/utils/websocket.py,sha256=fjC-qipzjgM_e7XImP12DFmLhvxkMOSr2GnUWws7esE,2058
-neuro_simulator-0.4.0.dist-info/METADATA,sha256=n3zf_b6P5veJQ-VPxlqlCsxCfZXTd2-loTHn6XB4N3w,8643
-neuro_simulator-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuro_simulator-0.4.0.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
-neuro_simulator-0.4.0.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
-neuro_simulator-0.4.0.dist-info/RECORD,,
+neuro_simulator-0.4.1.dist-info/METADATA,sha256=CGD8xcQwXRwCPdWX7jsgI0jVNeUNev2S0uQZv7D8Qwk,8643
+neuro_simulator-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuro_simulator-0.4.1.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
+neuro_simulator-0.4.1.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
+neuro_simulator-0.4.1.dist-info/RECORD,,