From 01014a6a8481add41deb1312c4e5f50bdab5b0e6 Mon Sep 17 00:00:00 2001 From: "cooper.wu" Date: Thu, 21 Nov 2024 18:01:47 +0800 Subject: [PATCH] fix: external dataset missing score_threshold_enabled (#10943) --- api/fields/dataset_fields.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/fields/dataset_fields.py b/api/fields/dataset_fields.py index b32423f10c..533e3a0837 100644 --- 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}