From 2157f84c4f66f398a8b367679f81b62eb76d93ee Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Tue, 4 Mar 2025 18:39:49 +0800 Subject: [PATCH] fix: Reference segmentation limit (#2487) --- apps/application/serializers/application_serializers.py | 4 ++-- apps/dataset/serializers/dataset_serializers.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/apps/application/serializers/application_serializers.py b/apps/application/serializers/application_serializers.py index e047a1868..543f53ae2 100644 --- a/apps/application/serializers/application_serializers.py +++ b/apps/application/serializers/application_serializers.py @@ -117,7 +117,7 @@ def valid_model_params_setting(model_id, model_params_setting): class DatasetSettingSerializer(serializers.Serializer): - top_n = serializers.FloatField(required=True, max_value=100, min_value=1, + top_n = serializers.FloatField(required=True, max_value=10000, min_value=1, error_messages=ErrMessage.float(_("Reference segment number"))) similarity = serializers.FloatField(required=True, max_value=1, min_value=0, error_messages=ErrMessage.float(_("Acquaintance"))) @@ -581,7 +581,7 @@ class ApplicationSerializer(serializers.Serializer): id = serializers.CharField(required=True, error_messages=ErrMessage.uuid(_("Application ID"))) user_id = serializers.UUIDField(required=False, error_messages=ErrMessage.uuid(_("User ID"))) query_text = serializers.CharField(required=True, error_messages=ErrMessage.char(_("Query text"))) - top_number = serializers.IntegerField(required=True, max_value=100, min_value=1, + top_number = serializers.IntegerField(required=True, max_value=10000, min_value=1, error_messages=ErrMessage.integer(_("topN"))) similarity = serializers.FloatField(required=True, max_value=2, min_value=0, error_messages=ErrMessage.float(_("Relevance"))) diff --git a/apps/dataset/serializers/dataset_serializers.py b/apps/dataset/serializers/dataset_serializers.py index 47f321bb5..f9e9a7132 100644 --- a/apps/dataset/serializers/dataset_serializers.py +++ b/apps/dataset/serializers/dataset_serializers.py @@ -567,7 +567,7 @@ class DataSetSerializers(serializers.ModelSerializer): id = serializers.CharField(required=True, error_messages=ErrMessage.char("id")) user_id = serializers.UUIDField(required=False, error_messages=ErrMessage.char(_('user id'))) query_text = serializers.CharField(required=True, error_messages=ErrMessage.char(_('query text'))) - top_number = serializers.IntegerField(required=True, max_value=100, min_value=1, + top_number = serializers.IntegerField(required=True, max_value=10000, min_value=1, error_messages=ErrMessage.char("top number")) similarity = serializers.FloatField(required=True, max_value=2, min_value=0, error_messages=ErrMessage.char(_('similarity')))