github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-07 23:46:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bes vector store search (#12828)

Browse Source 
**Issue:** 
fix search body in baidu cloud vectorsearch

---------

Co-authored-by: wemysschen <root@icoding-cwx.bcc-szzj.baidu.com>
This commit is contained in:
wemysschen 2023-11-04 06:39:19 +08:00 committed by GitHub
parent f04e4df7f9
commit e14aa37d59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/vectorstores/baiducloud_vector_search.py
View File

@ -95,7 +95,8 @@ class BESVectorStore(VectorStore):
connection_params: Dict[str, Any] = {} connection_params: Dict[str, Any] = {}
connection_params["hosts"] = [bes_url] connection_params["hosts"] = [bes_url]
connection_params["basic_auth"] = (username, password) if username and password:
connection_params["basic_auth"] = (username, password)
es_client = elasticsearch.Elasticsearch(**connection_params) es_client = elasticsearch.Elasticsearch(**connection_params)
try: try:
@ -236,7 +237,6 @@ class BESVectorStore(VectorStore):
if "linear" == self.index_type: if "linear" == self.index_type:
query_vector_body["linear"] = True query_vector_body["linear"] = True
query_vector_body["space_type"] = self.space_type
else: else:
query_vector_body["ef"] = search_params.get("ef", 10) query_vector_body["ef"] = search_params.get("ef", 10)
@ -279,7 +279,7 @@ class BESVectorStore(VectorStore):
logger.debug(f"Query body: {query_body}") logger.debug(f"Query body: {query_body}")
# Perform the kNN search on the BES index and return the results. # Perform the kNN search on the BES index and return the results.
response = self.client.search(index=self.index_name, **query_body) response = self.client.search(index=self.index_name, body=query_body)
logger.debug(f"response={response}") logger.debug(f"response={response}")
hits = [hit for hit in response["hits"]["hits"]] hits = [hit for hit in response["hits"]["hits"]]