From 349328f608c5353adba045e15d0dd24af1d2df37 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Wed, 7 Aug 2024 15:12:32 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9=E8=B0=83=E8=AF=95?= =?UTF-8?q?=E9=A1=B5=E9=9D=A2=E5=A4=9A=E8=BD=AE=E5=AF=B9=E8=AF=9D=E8=AE=BE?= =?UTF-8?q?=E7=BD=AE=E6=97=A0=E6=95=88=E9=97=AE=E9=A2=98=20(#945)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit (cherry picked from commit e59e26208b8d75dfd6a9956b1385326da89bb4e2) --- apps/application/serializers/chat_serializers.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/apps/application/serializers/chat_serializers.py b/apps/application/serializers/chat_serializers.py index 8692dc328..0255f1c6c 100644 --- a/apps/application/serializers/chat_serializers.py +++ b/apps/application/serializers/chat_serializers.py @@ -28,7 +28,6 @@ from application.models.api_key_model import ApplicationAccessToken from application.serializers.application_serializers import ModelDatasetAssociation, DatasetSettingSerializer, \ ModelSettingSerializer from application.serializers.chat_message_serializers import ChatInfo -from common.config.embedding_config import ModelManage from common.constants.permission_constants import RoleConstants from common.db.search import native_search, native_page_search, page_search, get_dynamics_model from common.event import ListenerManagement @@ -40,8 +39,6 @@ from common.util.lock import try_lock, un_lock from dataset.models import Document, Problem, Paragraph, ProblemParagraphMapping from dataset.serializers.common_serializers import get_embedding_model_by_dataset_id from dataset.serializers.paragraph_serializers import ParagraphSerializers -from setting.models import Model -from setting.models_provider import get_model from smartdoc.conf import PROJECT_DIR chat_cache = caches['chat_cache'] @@ -312,7 +309,8 @@ class ChatSerializers(serializers.Serializer): chat_id = str(uuid.uuid1()) model_id = self.data.get('model_id') dataset_id_list = self.data.get('dataset_id_list') - application = Application(id=None, dialogue_number=3, model_id=model_id, + dialogue_number = 3 if self.data.get('multiple_rounds_dialogue', False) else 0 + application = Application(id=None, dialogue_number=dialogue_number, model_id=model_id, dataset_setting=self.data.get('dataset_setting'), model_setting=self.data.get('model_setting'), problem_optimization=self.data.get('problem_optimization'),