feat:add summary label

longerking · Nov 14, 2023 · 6281910 · 6281910
1 parent 85da219
commit 6281910
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/pilot/scene/chat_knowledge/refine_summary/chat.py b/pilot/scene/chat_knowledge/refine_summary/chat.py
@@ -32,14 +32,14 @@ def generate_input_values(self):
         }
         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"""