mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
bugfix
This commit is contained in:
parent
a535bb2d8d
commit
91706c7e44
1 changed files with 2 additions and 2 deletions
|
|
@ -48,8 +48,8 @@ class EsVectorStore(BaseVectorStore):
|
|||
}
|
||||
return self._client.indices.create(index=workspace_id, body=body)
|
||||
|
||||
def _iter_workspace_nodes(self, workspace_id: str, **kwargs) -> Iterable[VectorNode]:
|
||||
response = self._client.search(index=workspace_id, body={"query": {"match_all": {}}})
|
||||
def _iter_workspace_nodes(self, workspace_id: str, max_size: int = 10000, **kwargs) -> Iterable[VectorNode]:
|
||||
response = self._client.search(index=workspace_id, body={"query": {"match_all": {}}, "size": max_size})
|
||||
for doc in response['hits']['hits']:
|
||||
yield self.doc2node(doc, workspace_id)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue