From 62819107a5e339cc0d5f119caf0a5c83f93f2b86 Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Tue, 14 Nov 2023 22:35:11 +0800
Subject: [PATCH] feat:add summary label

---
 .../scene/chat_knowledge/refine_summary/chat.py  | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

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"<summary>{last_output}</summary>"
-    #     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"<summary>{last_output}</summary>"
+        yield last_output
 
     def stream_call_reinforce_fn(self, text):
         """return reference"""