fix: clickhouse connect error fix

dbgpt/app/scene/chat_db/professional_qa/chat.py

@@ -28,7 +28,6 @@ def __init__(self, chat_param: Dict):
 
         if self.db_name:
             self.database = CFG.LOCAL_DB_MANAGE.get_connect(self.db_name)
-            self.db_connect = self.database.session
             self.tables = self.database.get_table_names()
 
         self.top_k = (

dbgpt/app/static/_next/static/01sV4-dnn1dUTis5H2TCy/_ssgManifest.js

@@ -0,0 +1 @@
+self.__SSG_MANIFEST=new Set([]);self.__SSG_MANIFEST_CB&&self.__SSG_MANIFEST_CB()

dbgpt/datasource/base.py

@@ -58,6 +58,14 @@ def get_table_comments(self, db_name):
         """
         pass
 
+    def get_column_comments(self, db_name, table_name):
+        """Get column comments.
+
+        Args:
+            table_name (_type_): _description_
+        """
+        pass
+
     def run(self, command: str, fetch: str = "all") -> List:
         """Execute sql command.
 

dbgpt/datasource/manages/connect_config_db.py

@@ -197,6 +197,7 @@ def get_db_config(self, db_name):
         else:
             raise ValueError("Cannot get database by name" + db_name)
 
+        print(result)
         fields = [field[0] for field in result.cursor.description]
         row_dict = {}
         row_1 = list(result.cursor.fetchall()[0])

dbgpt/datasource/rdbms/base.py

@@ -292,20 +292,21 @@ def _query(self, query: str, fetch: str = "all"):
             query (str): SQL query to run
             fetch (str): fetch type
         """
+        result = []
+
         print(f"Query[{query}]")
         if not query:
-            return []
+            return result
         cursor = self.session.execute(text(query))
         if cursor.returns_rows:
             if fetch == "all":
                 result = cursor.fetchall()
             elif fetch == "one":
-                result = cursor.fetchone()[0]  # type: ignore
+                result = [cursor.fetchone()]
             else:
                 raise ValueError("Fetch parameter must be either 'one' or 'all'")
             field_names = tuple(i[0:] for i in cursor.keys())
 
-            result = list(result)
             result.insert(0, field_names)
             return result
 
@@ -538,7 +539,8 @@ def get_users(self):
     def get_table_comments(self, db_name):
         cursor = self.session.execute(
             text(
-                f"""SELECT table_name, table_comment    FROM information_schema.tables   WHERE table_schema = '{db_name}'""".format(
+                f"""SELECT table_name, table_comment    FROM information_schema.tables  
+                    WHERE table_schema = '{db_name}'""".format(
                     db_name
                 )
             )
@@ -548,6 +550,21 @@ def get_table_comments(self, db_name):
             (table_comment[0], table_comment[1]) for table_comment in table_comments
         ]
 
+    def get_column_comments(self, db_name, table_name):
+        cursor = self.session.execute(
+            text(
+                f"""SELECT column_name, column_comment FROM information_schema.columns 
+                    WHERE table_schema = '{db_name}' and table_name = '{table_name}'
+                """.format(
+                    db_name, table_name
+                )
+            )
+        )
+        column_comments = cursor.fetchall()
+        return [
+            (column_comment[0], column_comment[1]) for column_comment in column_comments
+        ]
+
     def get_database_list(self):
         session = self._db_sessions()
         cursor = session.execute(text(" show databases;"))