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

feat(core): Dynamically loading dbgpts (#1211)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-02-29 15:57:49 +08:00
committed by GitHub
parent 673ddaab5b
commit 1d90711952
15 changed files with 504 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
dbgpt/core/awel/dag/loader.py
View File

@@ -63,19 +63,23 @@ def _process_file(filepath) -> List[DAG]:
return results
def _load_modules_from_file(filepath: str):
def _load_modules_from_file(
filepath: str, mod_name: str | None = None, show_log: bool = True
):
import importlib
import importlib.machinery
import importlib.util
logger.info(f"Importing {filepath}")
if show_log:
logger.info(f"Importing {filepath}")
org_mod_name, _ = os.path.splitext(os.path.split(filepath)[-1])
path_hash = hashlib.sha1(filepath.encode("utf-8")).hexdigest()
mod_name = f"unusual_prefix_{path_hash}_{org_mod_name}"
if mod_name is None:
mod_name = f"unusual_prefix_{path_hash}_{org_mod_name}"
if mod_name in sys.modules:
del sys.modules[mod_name]
if mod_name in sys.modules:
del sys.modules[mod_name]
def parse(mod_name, filepath):
try: