github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-23 01:49:58 +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

4
dbgpt/datasource/rdbms/conn_postgresql.py
View File

@@ -1,4 +1,4 @@
from typing import Any, Iterable, Optional
from typing import Any, Iterable, List, Optional, Tuple
from urllib.parse import quote
from urllib.parse import quote_plus as urlquote
@@ -85,7 +85,7 @@ class PostgreSQLDatabase(RDBMSDatabase):
print("postgresql get users error: ", e)
return []
def get_fields(self, table_name):
def get_fields(self, table_name) -> List[Tuple]:
"""Get column fields about specified table."""
session = self._db_sessions()
cursor = session.execute(