feat(editor): ChatExcel

🔥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

View File

@ -27,6 +27,14 @@ class ConnectManager:
subclasses += self.get_all_subclasses(subclass)
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):
chat_classes = self.get_all_subclasses(BaseConnect)
result = None

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])
async def db_support_types():
support_types = [
DBType.Mysql,
DBType.MSSQL,
DBType.DuckDb,
DBType.SQLite,
DBType.Clickhouse,
]
support_types = CFG.LOCAL_DB_MANAGE.get_all_completed_types()
db_type_infos = []
for type in support_types:
db_type_infos.append(