From cf8ee9f88eabdf1b073a6d092634bd21459bac27 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Thu, 8 Jan 2026 16:39:28 +0800 Subject: [PATCH] refactor(vector_store): replace async_iter_workspace_nodes with async_list_workspace_nodes --- reme_ai/__init__.py | 2 +- reme_ai/vector_store/delete_memory_op.py | 2 +- reme_ai/vector_store/vector_store_action_op.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/reme_ai/__init__.py b/reme_ai/__init__.py index 36717b51..a1353f0e 100644 --- a/reme_ai/__init__.py +++ b/reme_ai/__init__.py @@ -30,4 +30,4 @@ __all__ = [ "vector_store", ] -__version__ = "0.2.0.6" +__version__ = "0.2.0.7" diff --git a/reme_ai/vector_store/delete_memory_op.py b/reme_ai/vector_store/delete_memory_op.py index b7828f63..6b73ea7a 100644 --- a/reme_ai/vector_store/delete_memory_op.py +++ b/reme_ai/vector_store/delete_memory_op.py @@ -44,7 +44,7 @@ class DeleteMemoryOp(BaseAsyncOp): workspace_id: str = self.context.workspace_id freq_threshold: int = self.context.freq_threshold utility_threshold: float = self.context.utility_threshold - nodes: Iterable[VectorNode] = self.vector_store.iter_workspace_nodes(workspace_id=workspace_id) + nodes: Iterable[VectorNode] = await self.vector_store.async_list_workspace_nodes(workspace_id=workspace_id) deleted_memory_ids = [] for node in nodes: diff --git a/reme_ai/vector_store/vector_store_action_op.py b/reme_ai/vector_store/vector_store_action_op.py index 670e85b0..343b92c5 100644 --- a/reme_ai/vector_store/vector_store_action_op.py +++ b/reme_ai/vector_store/vector_store_action_op.py @@ -80,7 +80,7 @@ class VectorStoreActionOp(BaseAsyncOp): def node_to_memory(node: VectorNode) -> dict: return vector_node_to_memory(node).model_dump() - result = await self.vector_store.async_iter_workspace_nodes(workspace_id=workspace_id) + result = await self.vector_store.async_list_workspace_nodes(workspace_id=workspace_id) result = [node_to_memory(node) for node in result] elif action == "load":