fix: external dataset missing score_threshold_enabled (#10943)

1 year ago · 01014a6a84
parent cb0c55daa7
commit 01014a6a84
1 changed files with 1 additions and 0 deletions
--- a/api/fields/dataset_fields.py
+++ b/api/fields/dataset_fields.py
@ -41,6 +41,7 @@ dataset_retrieval_model_fields = {
 external_retrieval_model_fields = {
    "top_k": fields.Integer,
    "score_threshold": fields.Float,
+    "score_threshold_enabled": fields.Boolean,
 }

 tag_fields = {"id": fields.String, "name": fields.String, "type": fields.String}