Skip to content

Commit

Permalink
Get images clean interval parameter from instance (#98)
Browse files Browse the repository at this point in the history
* get images clean interval parameter from instance
* add env conversion
  • Loading branch information
NikolaiPetukhov authored Dec 17, 2024
1 parent 7c34871 commit 465aa94
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 3 deletions.
7 changes: 7 additions & 0 deletions agent/worker/agent_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@ class AgentOptionsJsonFields:
DOCKER_IMAGE = "dockerImage"
FORCE_CPU_ONLY = "forceCPUOnly"
LOG_LEVEL = "logLevel"
IMAGES_CLEAN_INTERVAL = "imagesCleanInterval"


def create_img_meta_str(img_size_bytes, width, height):
Expand Down Expand Up @@ -784,6 +785,12 @@ def update_env_param(name, value, default=None):

update_env_param(constants._AGENT_HOST_DIR, agent_host_dir)

update_env_param(
constants._REMOVE_IDLE_DOCKER_IMAGE_AFTER_X_DAYS,
options.get(AgentOptionsJsonFields.IMAGES_CLEAN_INTERVAL, None),
optional_defaults[constants._REMOVE_IDLE_DOCKER_IMAGE_AFTER_X_DAYS],
)

volumes = {}

def add_volume(src: str, dst: str) -> dict:
Expand Down
9 changes: 6 additions & 3 deletions agent/worker/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -189,8 +189,11 @@ def get_optional_defaults():
return _OPTIONAL_DEFAULTS.copy()


def read_optional_setting(name):
return os.getenv(name, _OPTIONAL_DEFAULTS[name])
def read_optional_setting(name, postprocess_fn = None):
value = os.getenv(name, _OPTIONAL_DEFAULTS[name])
if postprocess_fn is not None:
return postprocess_fn(value)
return value


def HOST_DIR():
Expand Down Expand Up @@ -544,7 +547,7 @@ def DISABLE_TELEMETRY():


def REMOVE_IDLE_DOCKER_IMAGE_AFTER_X_DAYS():
return read_optional_setting(_REMOVE_IDLE_DOCKER_IMAGE_AFTER_X_DAYS)
return read_optional_setting(_REMOVE_IDLE_DOCKER_IMAGE_AFTER_X_DAYS, lambda x: int(x))


def AGENT_ID():
Expand Down

0 comments on commit 465aa94

Please sign in to comment.