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

fix: clickhouse connect error fix (#958)

Browse Source 
Co-authored-by: Magic <magic@B-4TMH9N3X-2120.local>
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
magic.chen
2023-12-22 11:44:26 +08:00
committed by GitHub
parent d9065227bd
commit 681a8e2ed5
26 changed files with 630 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/rag/summary/rdbms_db_summary.py
View File

@@ -76,7 +76,7 @@ def _parse_table_summary(
table_name(column1(column1 comment),column2(column2 comment),column3(column3 comment) and index keys, and table comment: {table_comment})
"""
columns = []
for column in conn._inspector.get_columns(table_name):
for column in conn.get_columns(table_name):
if column.get("comment"):
columns.append(f"{column['name']} ({column.get('comment')})")
else:
@@ -84,7 +84,7 @@ def _parse_table_summary(
column_str = ", ".join(columns)
index_keys = []
for index_key in conn._inspector.get_indexes(table_name):
for index_key in conn.get_indexes(table_name):
key_str = ", ".join(index_key["column_names"])
index_keys.append(f"{index_key['name']}(`{key_str}`) ")
table_str = summary_template.format(table_name=table_name, columns=column_str)
@@ -92,7 +92,7 @@ def _parse_table_summary(
index_key_str = ", ".join(index_keys)
table_str += f", and index keys: {index_key_str}"
try:
comment = conn._inspector.get_table_comment(table_name)
comment = conn.get_table_comment(table_name)
except Exception:
comment = dict(text=None)
if comment.get("text"):