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

fix(ChatKnowledge): summary support zhipu message_converter (#1085)

Browse Source
This commit is contained in:
Aries-ckt
2024-01-18 15:09:09 +08:00
committed by GitHub
parent 0936856c3a
commit 0162dee474
3 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/rag/extractor/summary.py
View File

@@ -18,7 +18,7 @@ Write a quick summary of the following context:
the summary should be as concise as possible and not overly lengthy.Please keep the answer within approximately 200 characters.
"""
REFINE_SUMMARY_TEMPLATE_ZH = """我们已经提供了一个到某一点的现有总结:{context}\n 请根据你之前推理的内容进行最终的总结,总结回答的时候最好按照1.2.3.进行. 注意:请用<中文>来进行总结。"""
REFINE_SUMMARY_TEMPLATE_ZH = """我们已经提供了一个到某一点的现有总结:{context}\n 请根据你之前推理的内容进行总结,总结回答的时候最好按照1.2.3.进行. 注意:请用<中文>来进行总结。"""
REFINE_SUMMARY_TEMPLATE_EN = """
We have provided an existing summary up to a certain point: {context}, We have the opportunity to refine the existing summary (only if needed) with some more context below.
@@ -144,7 +144,7 @@ class SummaryExtractor(Extractor):
from dbgpt.core import ModelMessage
prompt = prompt_template.format(context=chunk_text)
messages = [ModelMessage(role=ModelMessageRoleType.SYSTEM, content=prompt)]
messages = [ModelMessage(role=ModelMessageRoleType.HUMAN, content=prompt)]
request = ModelRequest(model=self._model_name, messages=messages)
tasks.append(self._llm_client.generate(request))
summary_results = await run_async_tasks(