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

clear memory by device_id

Browse Source
This commit is contained in:
goulustis
2026-02-02 09:05:16 +08:00
parent 40ee9b7420
commit f51c408cae
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lang_agent/base.py
View File

@@ -185,13 +185,25 @@ class GraphBase(ABC):
else: else:
return img return img
def clear_memory(self): def clear_all_memory(self):
# Clear the agent's (LangChain) memory if available # NOTE: self.memory = MemorySaver
if hasattr(self, "memory") and self.memory is not None:
if isinstance(self.memory, MemorySaver):
for thread_id in self.memory.storage: for thread_id in self.memory.storage:
self.memory.delete_thread(thread_id) self.memory.delete_thread(thread_id)
def clear_memory_device_id(self, device_id:str):
# NOTE: self.memory = MemorySaver
rm_threads = [th for th in self.memory.storage if (device_id in th)]
for thread_id in rm_threads:
self.memory.delete_thread(thread_id)
def clear_memory(self, device_id:str=None):
# Clear the agent's (LangChain) memory if available
if hasattr(self, "memory") and self.memory is not None:
if isinstance(self.memory, MemorySaver) and (device_id is None):
self.clear_all_memory()
else:
self.clear_memory_device_id(device_id)
async def aclear_memory(self): async def aclear_memory(self):
# 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: