diff --git a/pilot/scene/chat_knowledge/refine_summary/chat.py b/pilot/scene/chat_knowledge/refine_summary/chat.py index f5418bf7a..47c8167b1 100644 --- a/pilot/scene/chat_knowledge/refine_summary/chat.py +++ b/pilot/scene/chat_knowledge/refine_summary/chat.py @@ -32,14 +32,14 @@ class ExtractRefineSummary(BaseChat): } return input_values - # async def stream_call(self): - # last_output = None - # async for output in super().stream_call(): - # last_output = output - # yield output - # - # last_output = f"{last_output}" - # yield last_output + async def stream_call(self): + last_output = None + async for output in super().stream_call(): + last_output = output + yield output + + last_output = f"{last_output}" + yield last_output def stream_call_reinforce_fn(self, text): """return reference"""