feat: 新增use_remote_resource参数

This commit is contained in:
Xuwznln
2025-08-11 16:09:27 +08:00
parent 50e93cb1af
commit 573bcf1a6c

View File

@@ -95,6 +95,11 @@ def parse_args():
action="store_true", action="store_true",
help="启动unilab时同时报送注册表信息", help="启动unilab时同时报送注册表信息",
) )
parser.add_argument(
"--use_remote_resource",
action="store_true",
help="启动unilab时使用远程资源启动",
)
parser.add_argument( parser.add_argument(
"--config", "--config",
type=str, type=str,
@@ -195,6 +200,16 @@ def main():
print_status(f"当前工作目录为 {working_dir}", "info") print_status(f"当前工作目录为 {working_dir}", "info")
load_config_from_file(config_path, args_dict["labid"]) load_config_from_file(config_path, args_dict["labid"])
if args_dict["use_remote_resource"]:
print_status("使用远程资源启动", "info")
from unilabos.app.web import http_client
res = http_client.resource_get("host_node", False)
if str(res.get("code", 0)) == "0" and len(res.get("data", [])) > 0:
print_status("远程资源已存在,使用云端物料!", "info")
args_dict["graph"] = None
else:
print_status("远程资源不存在,本地将进行首次上报!", "info")
# 设置BasicConfig参数 # 设置BasicConfig参数
BasicConfig.working_dir = working_dir BasicConfig.working_dir = working_dir
BasicConfig.is_host_mode = not args_dict.get("without_host", False) BasicConfig.is_host_mode = not args_dict.get("without_host", False)