scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-10 22:41:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

remove async call interface

Browse source
This commit is contained in:
青轩 2024-07-10 16:50:21 +08:00
parent 2d6713376d
commit d0e3686caf
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
memory_scope/memory/worker/read/retrieve_memory_worker.py
View file

@ -21,7 +21,7 @@ class RetrieveMemoryWorker(MemoryBaseWorker):
"status": MemoryNodeStatus.ACTIVE.value,
"memory_type": [MemoryTypeEnum.OBSERVATION.value, MemoryTypeEnum.OBS_CUSTOMIZED.value],
}
return await self.memory_store.a_retrieve_memories(query=query,
return self.memory_store.retrieve_memories(query=query,
top_k=self.retrieve_obs_top_k,
filter_dict=filter_dict)
@ -36,7 +36,7 @@ class RetrieveMemoryWorker(MemoryBaseWorker):
"status": MemoryNodeStatus.ACTIVE.value,
"memory_type": MemoryTypeEnum.INSIGHT.value,
}
return await self.memory_store.a_retrieve_memories(query=query,
return self.memory_store.retrieve_memories(query=query,
top_k=self.retrieve_ins_pf_top_k,
filter_dict=filter_dict)
@ -51,7 +51,7 @@ class RetrieveMemoryWorker(MemoryBaseWorker):
"status": MemoryNodeStatus.EXPIRED.value,
"memory_type": [MemoryTypeEnum.OBSERVATION.value, MemoryTypeEnum.OBS_CUSTOMIZED.value],
}
return await self.memory_store.a_retrieve_memories(query=query,
return self.memory_store.retrieve_memories(query=query,
top_k=self.retrieve_expired_top_k,
filter_dict=filter_dict)