github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-17 07:00:15 +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/knowledge/csv.py
View File

@@ -63,7 +63,7 @@ class CSVKnowledge(Knowledge):
)
except KeyError:
raise ValueError(
f"Source column '{self._source_column}' not found in CSV "
f"Source column '{self._source_column}' not in CSV "
f"file."
)
metadata = {"source": source, "row": i}

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)