github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-05 10:29:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(editor): ChatExcel

Browse Source 
🔥ChatExcel Mode Complete
This commit is contained in:
yhjun1026 2023-08-25 12:00:44 +08:00
parent 5ca38154e1
commit ca5ed11623
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pilot/connections/manages/connection_manager.py
View File

@ -27,6 +27,14 @@ class ConnectManager:
subclasses += self.get_all_subclasses(subclass) subclasses += self.get_all_subclasses(subclass)
return subclasses return subclasses
def get_all_completed_types(self):
chat_classes = self.get_all_subclasses(BaseConnect)
support_types = []
for cls in chat_classes:
if cls.db_type:
support_types.append(DBType.of_db_type(cls.db_type))
return support_types
def get_cls_by_dbtype(self, db_type): def get_cls_by_dbtype(self, db_type):
chat_classes = self.get_all_subclasses(BaseConnect) chat_classes = self.get_all_subclasses(BaseConnect)
result = None result = None

9
pilot/openapi/api_v1/api_v1.py
View File

@ -110,13 +110,8 @@ async def db_connect_delete(db_name: str = None):
@router.get("/v1/chat/db/support/type", response_model=Result[DbTypeInfo]) @router.get("/v1/chat/db/support/type", response_model=Result[DbTypeInfo])
async def db_support_types(): async def db_support_types():
support_types = [
DBType.Mysql, support_types = CFG.LOCAL_DB_MANAGE.get_all_completed_types()
DBType.MSSQL,
DBType.DuckDb,
DBType.SQLite,
DBType.Clickhouse,
]
db_type_infos = [] db_type_infos = []
for type in support_types: for type in support_types:
db_type_infos.append( db_type_infos.append(