fix: use keyword argument for IndicesClient.refresh() for opensearch-py 3.x (#21248)
In opensearch-py >= 3.0.0, IndicesClient.refresh() no longer accepts the index name as a positional argument. This causes a TypeError when uploading documents to knowledge bases with OpenSearch backend. Changes positional arguments to keyword arguments (index=...) in all three refresh() calls in the OpenSearch vector DB client. Fixes #20649
This commit is contained in:
@@ -211,7 +211,7 @@ class OpenSearchClient(VectorDBBase):
|
|||||||
for item in batch
|
for item in batch
|
||||||
]
|
]
|
||||||
bulk(self.client, actions)
|
bulk(self.client, actions)
|
||||||
self.client.indices.refresh(self._get_index_name(collection_name))
|
self.client.indices.refresh(index=self._get_index_name(collection_name))
|
||||||
|
|
||||||
def upsert(self, collection_name: str, items: list[VectorItem]):
|
def upsert(self, collection_name: str, items: list[VectorItem]):
|
||||||
self._create_index_if_not_exists(
|
self._create_index_if_not_exists(
|
||||||
@@ -234,7 +234,7 @@ class OpenSearchClient(VectorDBBase):
|
|||||||
for item in batch
|
for item in batch
|
||||||
]
|
]
|
||||||
bulk(self.client, actions)
|
bulk(self.client, actions)
|
||||||
self.client.indices.refresh(self._get_index_name(collection_name))
|
self.client.indices.refresh(index=self._get_index_name(collection_name))
|
||||||
|
|
||||||
def delete(
|
def delete(
|
||||||
self,
|
self,
|
||||||
@@ -263,7 +263,7 @@ class OpenSearchClient(VectorDBBase):
|
|||||||
self.client.delete_by_query(
|
self.client.delete_by_query(
|
||||||
index=self._get_index_name(collection_name), body=query_body
|
index=self._get_index_name(collection_name), body=query_body
|
||||||
)
|
)
|
||||||
self.client.indices.refresh(self._get_index_name(collection_name))
|
self.client.indices.refresh(index=self._get_index_name(collection_name))
|
||||||
|
|
||||||
def reset(self):
|
def reset(self):
|
||||||
indices = self.client.indices.get(index=f"{self.index_prefix}_*")
|
indices = self.client.indices.get(index=f"{self.index_prefix}_*")
|
||||||
|
|||||||
Reference in New Issue
Block a user