This commit is contained in:
takatost 2024-07-20 00:57:41 +08:00
parent dad1a967ee
commit beaac5033a

View File

@ -85,7 +85,10 @@ class GraphEngine:
# run graph
generator = self._run(start_node_id=self.graph.root_node_id)
if self.init_params.workflow_type == WorkflowType.CHAT:
answer_stream_processor = AnswerStreamProcessor(self.graph)
answer_stream_processor = AnswerStreamProcessor(
graph=self.graph,
variable_pool=self.graph_runtime_state.variable_pool
)
generator = answer_stream_processor.process(generator)
for item in generator: