Jelajahi Sumber

fix: external dataset missing score_threshold_enabled (#10943)

cooper.wu 1 tahun lalu
induk
melakukan
01014a6a84
1 mengubah file dengan 1 tambahan dan 0 penghapusan
  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}