From d0e3686cafd9ac483bc55724ab187bab21bd8f81 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9D=92=E8=BD=A9?= Date: Wed, 10 Jul 2024 16:50:21 +0800 Subject: [PATCH] remove async call interface --- memory_scope/memory/worker/read/retrieve_memory_worker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/memory_scope/memory/worker/read/retrieve_memory_worker.py b/memory_scope/memory/worker/read/retrieve_memory_worker.py index 42f7fe03..aedd40da 100644 --- a/memory_scope/memory/worker/read/retrieve_memory_worker.py +++ b/memory_scope/memory/worker/read/retrieve_memory_worker.py @@ -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)