github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-19 08:26:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:add summary label

Browse Source
This commit is contained in:
aries_ckt
2023-11-14 22:35:11 +08:00
parent 85da219a0a
commit 62819107a5
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pilot/scene/chat_knowledge/refine_summary/chat.py
View File

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