Skip to content

Commit e59e262

Browse files
authored
fix: 修改调试页面多轮对话设置无效问题 (#945)
1 parent 9d3e0d4 commit e59e262

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

apps/application/serializers/chat_serializers.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
from application.serializers.application_serializers import ModelDatasetAssociation, DatasetSettingSerializer, \
2929
ModelSettingSerializer
3030
from application.serializers.chat_message_serializers import ChatInfo
31-
from common.config.embedding_config import ModelManage
3231
from common.constants.permission_constants import RoleConstants
3332
from common.db.search import native_search, native_page_search, page_search, get_dynamics_model
3433
from common.event import ListenerManagement
@@ -40,8 +39,6 @@
4039
from dataset.models import Document, Problem, Paragraph, ProblemParagraphMapping
4140
from dataset.serializers.common_serializers import get_embedding_model_by_dataset_id
4241
from dataset.serializers.paragraph_serializers import ParagraphSerializers
43-
from setting.models import Model
44-
from setting.models_provider import get_model
4542
from smartdoc.conf import PROJECT_DIR
4643

4744
chat_cache = caches['chat_cache']
@@ -312,7 +309,8 @@ def open(self):
312309
chat_id = str(uuid.uuid1())
313310
model_id = self.data.get('model_id')
314311
dataset_id_list = self.data.get('dataset_id_list')
315-
application = Application(id=None, dialogue_number=3, model_id=model_id,
312+
dialogue_number = 3 if self.data.get('multiple_rounds_dialogue', False) else 0
313+
application = Application(id=None, dialogue_number=dialogue_number, model_id=model_id,
316314
dataset_setting=self.data.get('dataset_setting'),
317315
model_setting=self.data.get('model_setting'),
318316
problem_optimization=self.data.get('problem_optimization'),

0 commit comments

Comments
 (0)