add score threshold

1df8f5a0 · jyong · 642842d6 · 1df8f5a0
Commit 1df8f5a0 authored Sep 11, 2023 by jyong
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

dataset_retriever_tool.py api/core/tool/dataset_retriever_tool.py +2 -1

No files found.
--- a/api/core/tool/dataset_retriever_tool.py
+++ b/api/core/tool/dataset_retriever_tool.py
@@ -94,7 +94,8 @@ class DatasetRetrieverTool(BaseTool):
                    query,
                    search_type='similarity_score_threshold',
                    search_kwargs={
-                        'k': self.k
+                        'k': self.k,
+                        'score_threshold': 0.8
                    }
                )
            else: