From ff9ead0859c4b91115d534d724df3f92a498e5d2 Mon Sep 17 00:00:00 2001
From: yhjun1026 <460342015@qq.com>
Date: Thu, 31 Aug 2023 20:30:44 +0800
Subject: [PATCH] feat(editor): ChatExcel
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🔥ChatExcel Mode Operation Manual
---
 pilot/connections/manages/connection_manager.py       | 11 +++++++----
 pilot/connections/rdbms/base.py                       |  1 -
 .../chat_data/chat_excel/excel_learning/out_parser.py |  2 +-
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/pilot/connections/manages/connection_manager.py b/pilot/connections/manages/connection_manager.py
index 711f1d6db..a933205c2 100644
--- a/pilot/connections/manages/connection_manager.py
+++ b/pilot/connections/manages/connection_manager.py
@@ -153,12 +153,15 @@ class ConnectManager:
                 db_user = db_info.db_user
                 db_pwd = db_info.db_pwd
                 return connect_instance.from_uri_db(
-                    host=db_host, port=db_port, user=db_user, pwd=db_pwd, db_name=db_name
+                    host=db_host,
+                    port=db_port,
+                    user=db_user,
+                    pwd=db_pwd,
+                    db_name=db_name,
                 )
         except Exception as e:
-            print(f'{db_info.db_name} Test connect Failure!{str(e)}')
-            raise ValueError(f'{db_info.db_name} Test connect Failure!{str(e)}')
-
+            print(f"{db_info.db_name} Test connect Failure!{str(e)}")
+            raise ValueError(f"{db_info.db_name} Test connect Failure!{str(e)}")
 
     def get_db_list(self):
         return self.storage.get_db_list()
diff --git a/pilot/connections/rdbms/base.py b/pilot/connections/rdbms/base.py
index dbd8de5f3..97d90e344 100644
--- a/pilot/connections/rdbms/base.py
+++ b/pilot/connections/rdbms/base.py
@@ -501,7 +501,6 @@ class RDBMSDatabase(BaseConnect):
         except Exception as e:
             return []
 
-
     def get_table_comments(self, db_name):
         cursor = self.session.execute(
             text(
diff --git a/pilot/scene/chat_data/chat_excel/excel_learning/out_parser.py b/pilot/scene/chat_data/chat_excel/excel_learning/out_parser.py
index c583e958d..c990b042c 100644
--- a/pilot/scene/chat_data/chat_excel/excel_learning/out_parser.py
+++ b/pilot/scene/chat_data/chat_excel/excel_learning/out_parser.py
@@ -62,4 +62,4 @@ class LearningExcelOutputParser(BaseOutputParser):
             html = f"""{html_title}\n{html_colunms}\n{html_plans}"""
             return html
         else:
-            return speak
\ No newline at end of file
+            return speak