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

feat(agent):Fix agent bug (#1953)

Browse Source 
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
明天
2024-09-04 10:59:03 +08:00
committed by GitHub
parent d72bfb2f5f
commit b951b50689
18 changed files with 67 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
dbgpt/datasource/manages/connect_config_db.py
View File

@@ -214,10 +214,15 @@ class ConnectConfigDao(BaseDao):
def get_db_list(self, db_name: Optional[str] = None, user_id: Optional[str] = None):
"""Get db list."""
session = self.get_raw_session()
if db_name:
if db_name and user_id:
sql = f"SELECT * FROM connect_config where (user_id='{user_id}' or user_id='' or user_id IS NULL) and db_name='{db_name}'" # noqa
else:
elif user_id:
sql = f"SELECT * FROM connect_config where user_id='{user_id}' or user_id='' or user_id IS NULL" # noqa
elif db_name:
sql = f"SELECT * FROM connect_config where db_name='{db_name}'" # noqa
else:
sql = f"SELECT * FROM connect_config" # noqa
result = session.execute(text(sql))
fields = [field[0] for field in result.cursor.description] # type: ignore
data = []