Update opensearch.py

This commit is contained in:
Vineeth B V 2025-02-06 13:36:11 +05:30 committed by GitHub
parent fd6b039859
commit 7c78facfd9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -114,9 +114,9 @@ class OpenSearchClient:
return self._result_to_search_result(result)
def query(
self, index_name: str, filter: dict, limit: Optional[int] = None
self, collection_name: str, filter: dict, limit: Optional[int] = None
) -> Optional[GetResult]:
if not self.has_collection(index_name):
if not self.has_collection(collection_name):
return None
query_body = {
@ -137,7 +137,7 @@ class OpenSearchClient:
try:
result = self.client.search(
index=f"{self.index_prefix}_{index_name}",
index=f"{self.index_prefix}_{collection_name}",
body=query_body,
size=size
)