mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-08 03:44:14 +00:00
32 lines
757 B
Python
32 lines
757 B
Python
from typing import Dict
|
|
|
|
from dbgpt._private.config import Config
|
|
from dbgpt.app.scene import BaseChat, ChatScene
|
|
from dbgpt.util.tracer import trace
|
|
|
|
CFG = Config()
|
|
|
|
|
|
class ChatNormal(BaseChat):
|
|
chat_scene: str = ChatScene.ChatNormal.value()
|
|
|
|
keep_end_rounds: int = 10
|
|
|
|
"""Number of results to return from the query"""
|
|
|
|
def __init__(self, chat_param: Dict):
|
|
""" """
|
|
chat_param["chat_mode"] = ChatScene.ChatNormal
|
|
super().__init__(
|
|
chat_param=chat_param,
|
|
)
|
|
|
|
@trace()
|
|
async def generate_input_values(self) -> Dict:
|
|
input_values = {"input": self.current_user_input}
|
|
return input_values
|
|
|
|
@property
|
|
def chat_type(self) -> str:
|
|
return ChatScene.ChatNormal.value
|