15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:02:46 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: type

This commit is contained in:
shaohuzhang1 2024-04-15 13:44:52 +08:00
parent 3afdab8acf
commit 05972ea017
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py
View File

@ -84,7 +84,7 @@ class BaseChatStep(IChatStep):
post_response_handler: PostResponseHandler,
chat_model: BaseChatModel = None,
paragraph_list=None,
manage: = None,
manage: PipelineManage = None,
padding_problem_text: str = None,
stream: bool = True,
client_id=None, client_type=None,
@ -125,7 +125,7 @@ class BaseChatStep(IChatStep):
post_response_handler: PostResponseHandler,
chat_model: BaseChatModel = None,
paragraph_list=None,
manage: = None,
manage: PipelineManage = None,
padding_problem_text: str = None,
client_id=None, client_type=None):
# 调用模型
@ -151,7 +151,7 @@ class BaseChatStep(IChatStep):
post_response_handler: PostResponseHandler,
chat_model: BaseChatModel = None,
paragraph_list=None,
manage: = None,
manage: PipelineManage = None,
padding_problem_text: str = None,
client_id=None, client_type=None):
# 调用模型

6
apps/application/serializers/chat_message_serializers.py
View File

@ -15,7 +15,7 @@ from django.db.models import QuerySet
from langchain.chat_models.base import BaseChatModel
from rest_framework import serializers
from application.chat_pipeline.pipeline_manage import PiplineManage
from application.chat_pipeline.pipeline_manage import PipelineManage
from application.chat_pipeline.step.chat_step.i_chat_step import PostResponseHandler
from application.chat_pipeline.step.chat_step.impl.base_chat_step import BaseChatStep
from application.chat_pipeline.step.generate_human_message_step.impl.base_generate_human_message_step import \
@ -108,7 +108,7 @@ def get_post_handler(chat_info: ChatInfo):
paragraph_list: List[Paragraph],
problem_text: str,
answer_text,
manage: PiplineManage,
manage: PipelineManage,
step: BaseChatStep,
padding_problem_text: str = None,
client_id=None,
@ -178,7 +178,7 @@ class ChatMessageSerializer(serializers.Serializer):
client_id = self.data.get('client_id')
client_type = self.data.get('client_type')
chat_info = self.is_valid(raise_exception=True)
pipline_manage_builder = PiplineManage.builder()
pipline_manage_builder = PipelineManage.builder()
# 如果开启了问题优化,则添加上问题优化步骤
if chat_info.application.problem_optimization:
pipline_manage_builder.append_step(BaseResetProblemStep)