mathieu/ai-suite-rocm-local
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

qol impro.

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-29 12:10:20 +02:00
parent 157e6efe16
commit 06d9e39f9d
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/utils.py
View File

@ -63,6 +63,8 @@ def run_command(command: str, exit_on_error: bool = True):
if process.returncode != 0: if process.returncode != 0:
logger.fatal(f"Failed to run command: {command}") logger.fatal(f"Failed to run command: {command}")
if exit_on_error:
raise Exception(f"Failed to run command: {command}") raise Exception(f"Failed to run command: {command}")
return out, err, process.returncode return out, err, process.returncode
@ -71,13 +73,9 @@ def run_command(command: str, exit_on_error: bool = True):
def load_service_from_string(service: str) -> Stack: def load_service_from_string(service: str) -> Stack:
logger.debug(f"Loading service from string: {service}") logger.debug(f"Loading service from string: {service}")
try:
service_name = service.replace("_", " ").title().replace(" ", "") service_name = service.replace("_", " ").title().replace(" ", "")
module = importlib.import_module(f"services.{service}") module = importlib.import_module(f"services.{service}")
met = getattr(module, service_name) met = getattr(module, service_name)
return met() return met()
except ModuleNotFoundError as e:
logger.error(f"Failed to load service: {e}")
return None