mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-26 03:54:10 +00:00
refactor: The first refactored version for sdk release (#907)
Co-authored-by: chengfangyin2 <chengfangyin3@jd.com>
This commit is contained in:
33
dbgpt/app/scene/chat_knowledge/refine_summary/chat.py
Normal file
33
dbgpt/app/scene/chat_knowledge/refine_summary/chat.py
Normal file
@@ -0,0 +1,33 @@
|
||||
from typing import Dict
|
||||
|
||||
from dbgpt.app.scene import BaseChat, ChatScene
|
||||
|
||||
|
||||
class ExtractRefineSummary(BaseChat):
|
||||
chat_scene: str = ChatScene.ExtractRefineSummary.value()
|
||||
|
||||
"""extract final summary by llm"""
|
||||
|
||||
def __init__(self, chat_param: Dict):
|
||||
""" """
|
||||
chat_param["chat_mode"] = ChatScene.ExtractRefineSummary
|
||||
super().__init__(
|
||||
chat_param=chat_param,
|
||||
)
|
||||
|
||||
self.existing_answer = chat_param["select_param"]
|
||||
|
||||
async def generate_input_values(self):
|
||||
input_values = {
|
||||
# "context": self.user_input,
|
||||
"existing_answer": self.existing_answer,
|
||||
}
|
||||
return input_values
|
||||
|
||||
def stream_plugin_call(self, text):
|
||||
"""return summary label"""
|
||||
return f"<summary>{text}</summary>"
|
||||
|
||||
@property
|
||||
def chat_type(self) -> str:
|
||||
return ChatScene.ExtractRefineSummary.value
|
Reference in New Issue
Block a user