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

feat(model): Support yi proxy LLM (#1303)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-15 22:15:37 +08:00
committed by GitHub
parent 6974329116
commit 130ffb08c9
144 changed files with 1286 additions and 1088 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -99,7 +99,7 @@ def _parse_table_summary(
index_keys = []
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}`) ")
index_keys.append(f"{index_key['name']}(`{key_str}`) ") # noqa
table_str = summary_template.format(table_name=table_name, columns=column_str)
if len(index_keys) > 0:
index_key_str = ", ".join(index_keys)