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

feat(editor): ChatExcel

Browse Source 
ChatExcel devlop part 4
This commit is contained in:
yhjun1026
2023-08-23 09:54:19 +08:00
parent 20fc743cd3
commit c67e6d1a65
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,5 +1,6 @@
import json
import os
from typing import Any
from pilot.scene.base_message import (
HumanMessage,
@@ -22,18 +23,19 @@ CFG = Config()
class ExcelLearning(BaseChat):
chat_scene: str = ChatScene.ExcelLearning.value()
def __init__(self, chat_session_id, user_input, select_param:str=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
self.excel_reader = ExcelReader(select_param)
self.excel_reader = excel_reader
super().__init__(
chat_mode=chat_mode,
chat_session_id=chat_session_id,
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):