瀏覽代碼

Update opensearch.py

Vineeth B V 2 月之前
父節點
當前提交
7c78facfd9
共有 1 個文件被更改,包括 3 次插入3 次删除
  1. 3 3
      backend/open_webui/retrieval/vector/dbs/opensearch.py

+ 3 - 3
backend/open_webui/retrieval/vector/dbs/opensearch.py

@@ -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
             )