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

feat: new dbgpts modules (#1910)

Browse Source 
Co-authored-by: 途杨 <tuyang.yhj@antgroup.com>
Co-authored-by: lhwan <1017484907@qq.com>
This commit is contained in:
明天
2024-08-28 21:31:42 +08:00
committed by GitHub
parent bb5d2d1f3d
commit d7a893e1a7
289 changed files with 3533 additions and 901 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/util/dbgpts/loader.py
View File

@@ -253,7 +253,7 @@ def _get_from_module(module, predicates: Optional[List[str]] = None):
return results
def _parse_package_metadata(package: InstalledPackage) -> BasePackage:
def parse_package_metadata(package: InstalledPackage) -> BasePackage:
with open(
Path(package.root) / DBGPTS_METADATA_FILE, mode="r+", encoding="utf-8"
) as f:
@@ -321,7 +321,7 @@ def _load_package_from_path(path: str):
parsed_packages = []
for package in packages:
try:
parsed_packages.append(_parse_package_metadata(package))
parsed_packages.append(parse_package_metadata(package))
except Exception as e:
logger.warning(f"Load package failed!{str(e)}", e)
@@ -343,7 +343,7 @@ def _load_flow_package_from_path(
packages = raw_packages
if not packages:
raise ValueError(f"Can't find the package {name} or {new_name}")
flow_package = _parse_package_metadata(packages[0])
flow_package = parse_package_metadata(packages[0])
if flow_package.package_type != "flow":
raise ValueError(f"Unsupported package type: {flow_package.package_type}")
return cast(FlowPackage, flow_package)