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

feat: (0.6)New UI (#1855)

Browse Source 
Co-authored-by: 夏姜 <wenfengjiang.jwf@digital-engine.com>
Co-authored-by: aries_ckt <916701291@qq.com>
Co-authored-by: wb-lh513319 <wb-lh513319@alibaba-inc.com>
Co-authored-by: csunny <cfqsunny@163.com>
This commit is contained in:
明天
2024-08-21 17:37:45 +08:00
committed by GitHub
parent 3fc82693ba
commit b124ecc10b
824 changed files with 93371 additions and 2515 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
dbgpt/serve/agent/resource/datasource.py
View File

@@ -27,10 +27,15 @@ class DatasourceDBParameters(DBParameters):
cls,
parameters: Type["DatasourceDBParameters"],
version: Optional[str] = None,
**kwargs,
) -> Any:
"""Convert the parameters to configurations."""
conf: List[ParameterDescription] = cast(
List[ParameterDescription], super().to_configurations(parameters)
List[ParameterDescription],
super().to_configurations(
parameters,
**kwargs,
),
)
version = version or cls._resource_version()
if version != "v1":
@@ -58,9 +63,16 @@ class DatasourceResource(RDBMSConnectorResource):
super().__init__(name, connector=conn, db_name=db_name, **kwargs)
@classmethod
def resource_parameters_class(cls) -> Type[DatasourceDBParameters]:
dbs = CFG.local_db_manager.get_db_list()
results = [db["db_name"] for db in dbs]
def resource_parameters_class(cls, **kwargs) -> Type[DatasourceDBParameters]:
dbs = CFG.local_db_manager.get_db_list(user_id=kwargs.get("user_id", None))
results = [
{
"label": "[" + db["db_type"] + "]" + db["db_name"],
"key": db["db_name"],
"description": db["comment"],
}
for db in dbs
]
@dataclasses.dataclass
class _DynDBParameters(DatasourceDBParameters):