mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-10 03:30:56 +00:00
refactor(vector_store): replace async_iter_workspace_nodes with async_list_workspace_nodes
This commit is contained in:
parent
59aa253e85
commit
cf8ee9f88e
3 changed files with 3 additions and 3 deletions
|
|
@ -30,4 +30,4 @@ __all__ = [
|
|||
"vector_store",
|
||||
]
|
||||
|
||||
__version__ = "0.2.0.6"
|
||||
__version__ = "0.2.0.7"
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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":
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue