github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-11-12 05:40:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

style:fmt

Browse Source
This commit is contained in:
aries_ckt
2023-08-29 19:57:33 +08:00
parent 0efaffc031
commit 71b9cd14a6
36 changed files with 413 additions and 280 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
pilot/scene/chat_data/chat_excel/excel_learning/chat.py
View File

@@ -23,7 +23,14 @@ CFG = Config()
class ExcelLearning(BaseChat):
chat_scene: str = ChatScene.ExcelLearning.value()
def __init__(self, chat_session_id, user_input, parent_mode: Any=None, select_param:str=None, excel_reader:Any=None):
def __init__(
self,
chat_session_id,
user_input,
parent_mode: Any = None,
select_param: str = None,
excel_reader: Any = None,
):
chat_mode = ChatScene.ExcelLearning
""" """
self.excel_file_path = select_param
@@ -31,20 +38,19 @@ class ExcelLearning(BaseChat):
super().__init__(
chat_mode=chat_mode,
chat_session_id=chat_session_id,
current_user_input = user_input,
current_user_input=user_input,
select_param=select_param,
)
if parent_mode:
self.current_message.chat_mode = parent_mode.value()
def generate_input_values(self):
colunms, datas = self.excel_reader.get_sample_data()
datas.insert(0, colunms)
input_values = {
"data_example": json.dumps(self.excel_reader.get_sample_data(), cls=DateTimeEncoder),
"data_example": json.dumps(
self.excel_reader.get_sample_data(), cls=DateTimeEncoder
),
}
return input_values