diff --git a/agent/worker/task_update.py b/agent/worker/task_update.py index c40833b..2c3b117 100644 --- a/agent/worker/task_update.py +++ b/agent/worker/task_update.py @@ -136,10 +136,13 @@ def check_and_pull_sly_net_if_needed( if sly_net_client_image_name is None: sly_net_client_image_name = cur_container.attrs["Config"]["Image"] - docker_hub_image_info = ic.get_registry_data(sly_net_client_image_name) + registry = docker_utils.resolve_registry(sly_net_client_image_name) + auth = docker_utils._registry_auth_from_env(registry) + + docker_registry_image_info = ic.get_registry_data(sly_net_client_image_name, auth_config=auth) name_with_digest: str = cur_container.image.attrs.get("RepoDigests", [""])[0] - if name_with_digest.endswith(docker_hub_image_info.id): + if name_with_digest.endswith(docker_registry_image_info.id): logger.info("sly-net-client is already updated") return False else: