fix: type

Author: shaohuzhang1

apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py

@@ -84,7 +84,7 @@ def execute(self, message_list: List[BaseMessage],
                 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 @@ def execute_stream(self, message_list: List[BaseMessage],
                        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 @@ def execute_block(self, message_list: List[BaseMessage],
                       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):
         # 调用模型

apps/application/serializers/chat_message_serializers.py

@@ -15,7 +15,7 @@
 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 handler(self,
                     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 @@ def chat(self):
         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)