Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

async clear memory

Browse Source
This commit is contained in:
goulustis
2026-02-02 13:37:26 +08:00
parent dd62f28be1
commit 776be9ee22
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lang_agent/base.py
View File

@@ -204,12 +204,19 @@ class GraphBase(ABC):
else: else:
self.clear_memory_device_id(device_id) self.clear_memory_device_id(device_id)
async def aclear_memory(self): async def aclear_memory(self, device_id:str=None):
# Clear the agent's (LangChain) memory if available (async version) # Clear the agent's (LangChain) memory if available (async version)
if hasattr(self, "memory") and self.memory is not None: if hasattr(self, "memory") and self.memory is not None:
if isinstance(self.memory, MemorySaver): if isinstance(self.memory, MemorySaver):
if device_id is None:
# Clear all memory
for thread_id in self.memory.storage: for thread_id in self.memory.storage:
await self.memory.adelete_thread(thread_id) await self.memory.adelete_thread(thread_id)
else:
# Clear memory for specific device_id
rm_threads = [th for th in self.memory.storage if (device_id in th)]
for thread_id in rm_threads:
await self.memory.adelete_thread(thread_id)