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