Browse Source

fix: external dataset missing score_threshold_enabled (#10943)

cooper.wu 5 months ago
parent
commit
01014a6a84
1 changed files with 1 additions and 0 deletions
  1. 1 0
      api/fields/dataset_fields.py

+ 1 - 0
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}