mirror of
https://github.com/csunny/DB-GPT.git
synced 2026-07-27 00:11:25 +00:00
31 lines
747 B
Python
31 lines
747 B
Python
from typing import Dict
|
|
|
|
from pilot.scene.base_chat import BaseChat
|
|
from pilot.scene.base import ChatScene
|
|
from pilot.configs.config import Config
|
|
|
|
from pilot.scene.chat_normal.prompt import prompt
|
|
|
|
CFG = Config()
|
|
|
|
|
|
class ChatNormal(BaseChat):
|
|
chat_scene: str = ChatScene.ChatNormal.value()
|
|
|
|
"""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,
|
|
)
|
|
|
|
def generate_input_values(self):
|
|
input_values = {"input": self.current_user_input}
|
|
return input_values
|
|
|
|
@property
|
|
def chat_type(self) -> str:
|
|
return ChatScene.ChatNormal.value
|