github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-13 13:36:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch], langchain[minor]: Add retriever self_query and score_threshold in DingoDB (#18106)

Browse Source
This commit is contained in:
Hech
2024-03-06 07:47:29 +08:00
committed by GitHub
parent d039dcb6ba
commit 6a08134661
6 changed files with 656 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/community/langchain_community/vectorstores/dingo.py
View File

@@ -145,7 +145,7 @@ class Dingo(VectorStore):
List of Documents most similar to the query and score for each
"""
docs_and_scores = self.similarity_search_with_score(
query, k=k, search_params=search_params
query, k=k, search_params=search_params, **kwargs
)
return [doc for doc, _ in docs_and_scores]
@@ -177,9 +177,15 @@ class Dingo(VectorStore):
return []
for res in results[0]["vectorWithDistances"]:
score = res["distance"]
if (
"score_threshold" in kwargs
and kwargs.get("score_threshold") is not None
):
if score > kwargs.get("score_threshold"):
continue
metadatas = res["scalarData"]
id = res["id"]
score = res["distance"]
text = metadatas[self._text_key]["fields"][0]["data"]
metadata = {"id": id, "text": text, "score": score}
for meta_key in metadatas.keys():