Skip to content

Commit c96442d

Browse files
authored
fix: type (#67)
1 parent 3afdab8 commit c96442d

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def execute(self, message_list: List[BaseMessage],
8484
post_response_handler: PostResponseHandler,
8585
chat_model: BaseChatModel = None,
8686
paragraph_list=None,
87-
manage: = None,
87+
manage: PipelineManage = None,
8888
padding_problem_text: str = None,
8989
stream: bool = True,
9090
client_id=None, client_type=None,
@@ -125,7 +125,7 @@ def execute_stream(self, message_list: List[BaseMessage],
125125
post_response_handler: PostResponseHandler,
126126
chat_model: BaseChatModel = None,
127127
paragraph_list=None,
128-
manage: = None,
128+
manage: PipelineManage = None,
129129
padding_problem_text: str = None,
130130
client_id=None, client_type=None):
131131
# 调用模型
@@ -151,7 +151,7 @@ def execute_block(self, message_list: List[BaseMessage],
151151
post_response_handler: PostResponseHandler,
152152
chat_model: BaseChatModel = None,
153153
paragraph_list=None,
154-
manage: = None,
154+
manage: PipelineManage = None,
155155
padding_problem_text: str = None,
156156
client_id=None, client_type=None):
157157
# 调用模型

apps/application/serializers/chat_message_serializers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from langchain.chat_models.base import BaseChatModel
1616
from rest_framework import serializers
1717

18-
from application.chat_pipeline.pipeline_manage import PiplineManage
18+
from application.chat_pipeline.pipeline_manage import PipelineManage
1919
from application.chat_pipeline.step.chat_step.i_chat_step import PostResponseHandler
2020
from application.chat_pipeline.step.chat_step.impl.base_chat_step import BaseChatStep
2121
from application.chat_pipeline.step.generate_human_message_step.impl.base_generate_human_message_step import \
@@ -108,7 +108,7 @@ def handler(self,
108108
paragraph_list: List[Paragraph],
109109
problem_text: str,
110110
answer_text,
111-
manage: PiplineManage,
111+
manage: PipelineManage,
112112
step: BaseChatStep,
113113
padding_problem_text: str = None,
114114
client_id=None,
@@ -178,7 +178,7 @@ def chat(self):
178178
client_id = self.data.get('client_id')
179179
client_type = self.data.get('client_type')
180180
chat_info = self.is_valid(raise_exception=True)
181-
pipline_manage_builder = PiplineManage.builder()
181+
pipline_manage_builder = PipelineManage.builder()
182182
# 如果开启了问题优化,则添加上问题优化步骤
183183
if chat_info.application.problem_optimization:
184184
pipline_manage_builder.append_step(BaseResetProblemStep)

0 commit comments

Comments
 (0)