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"""