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

chore: Add pylint for DB-GPT rag lib (#1267)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-07 23:27:43 +08:00
committed by GitHub
parent aaaf34db17
commit 7446817340
70 changed files with 1135 additions and 587 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,3 +1,5 @@
"""Summary for rdbms database."""
from typing import List
from dbgpt._private.config import Config
@@ -9,21 +11,28 @@ CFG = Config()
class RdbmsSummary(DBSummary):
"""Get rdbms db table summary template.
summary example:
table_name(column1(column1 comment),column2(column2 comment),column3(column3 comment) and index keys, and table comment is {table_comment})
Summary example:
table_name(column1(column1 comment),column2(column2 comment),
column3(column3 comment) and index keys, and table comment is {table_comment})
"""
def __init__(self, name, type):
def __init__(self, name: str, type: str):
"""Create a new RdbmsSummary."""
self.name = name
self.type = type
self.summary_template = "{table_name}({columns})"
self.tables = {}
self.tables_info = []
self.vector_tables_info = []
# self.tables_info = []
# self.vector_tables_info = []
if not CFG.LOCAL_DB_MANAGE:
raise ValueError("Local db manage is not initialized.")
# TODO: Don't use the global variable.
self.db = CFG.LOCAL_DB_MANAGE.get_connect(name)
self.metadata = """user info :{users}, grant info:{grant}, charset:{charset}, collation:{collation}""".format(
self.metadata = """user info :{users}, grant info:{grant}, charset:{charset},
collation:{collation}""".format(
users=self.db.get_users(),
grant=self.db.get_grants(),
charset=self.db.get_charset(),
@@ -36,8 +45,10 @@ class RdbmsSummary(DBSummary):
def get_table_summary(self, table_name):
"""Get table summary for table.
example:
table_name(column1(column1 comment),column2(column2 comment),column3(column3 comment) and index keys, and table comment: {table_comment})
table_name(column1(column1 comment),column2(column2 comment),
column3(column3 comment) and index keys, and table comment: {table_comment})
"""
return _parse_table_summary(self.db, self.summary_template, table_name)
@@ -74,7 +85,8 @@ def _parse_table_summary(
table_name (str): table name
Examples:
table_name(column1(column1 comment),column2(column2 comment),column3(column3 comment) and index keys, and table comment: {table_comment})
table_name(column1(column1 comment),column2(column2 comment),
column3(column3 comment) and index keys, and table comment: {table_comment})
"""
columns = []
for column in conn.get_columns(table_name):