From 5bccbc20d9dad1059d1c613b3d88aa263d2940ce Mon Sep 17 00:00:00 2001
From: Nikolai Petukhov <petukhov.niko@gmail.com>
Date: Wed, 11 Dec 2024 15:09:30 -0300
Subject: [PATCH] get images clean interval parameter from instance

---
 agent/worker/agent_utils.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/agent/worker/agent_utils.py b/agent/worker/agent_utils.py
index bc3f474..7eab498 100644
--- a/agent/worker/agent_utils.py
+++ b/agent/worker/agent_utils.py
@@ -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):
@@ -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: