Merge branch 'feat/external-knowledge-api' of github.com:langgenius/dify into feat/external-knowledge-api

feat/external-knowledge-api
Yi 2 years ago
commit 918df23f64

@ -319,11 +319,12 @@ class ExternalDatasetService:
headers = {"Content-Type": "application/json"} headers = {"Content-Type": "application/json"}
if settings.get("api_key"): if settings.get("api_key"):
headers["Authorization"] = f"Bearer {settings.get('api_key')}" headers["Authorization"] = f"Bearer {settings.get('api_key')}"
score_threshold_enabled = external_retrieval_parameters.get("score_threshold_enabled") or False
score_threshold = external_retrieval_parameters.get("score_threshold", 0.0) if score_threshold_enabled else 0.0
request_params = { request_params = {
"retrieval_setting": { "retrieval_setting": {
"top_k": external_retrieval_parameters.get("top_k"), "top_k": external_retrieval_parameters.get("top_k"),
"score_threshold": external_retrieval_parameters.get("score_threshold"), "score_threshold": score_threshold,
}, },
"query": query, "query": query,
"knowledge_id": external_knowledge_binding.external_knowledge_id, "knowledge_id": external_knowledge_binding.external_knowledge_id,

Loading…
Cancel
Save