refactor: parameterize image pull policy

Signed-off-by: thxCode <thxcode0824@gmail.com>

Author: thxCode

gpustack_runtime/deployer/__types__.py

@@ -685,8 +685,8 @@ class Container:
     """
     Name of the container.
     """
-    image_pull_policy: ContainerImagePullPolicyEnum = (
-        ContainerImagePullPolicyEnum.IF_NOT_PRESENT
+    image_pull_policy: ContainerImagePullPolicyEnum = field(
+        default_factory=lambda: envs.GPUSTACK_RUNTIME_DEPLOY_IMAGE_PULL_POLICY,
     )
     """
     Image pull policy of the container.

gpustack_runtime/deployer/kuberentes.py

@@ -53,7 +53,7 @@
 _LABEL_COMPONENT = f"{envs.GPUSTACK_RUNTIME_DEPLOY_LABEL_PREFIX}/component"
 
 
-class KubernetesWorkloadServiceType(str, Enum):
+class KubernetesWorkloadServiceTypeEnum(str, Enum):
     """
     Types for Kubernetes Service.
     """
@@ -84,7 +84,7 @@ class KubernetesWorkloadPlan(WorkloadPlan):
     Attributes:
         domain_suffix (str):
             Domain suffix for the cluster. Default is "cluster.local".
-        service_type (KubernetesWorkloadServiceType):
+        service_type (KubernetesWorkloadServiceTypeEnum):
             Service type for the workload. Default is CLUSTER_IP.
         resource_key_runtime_env_mapping: (dict[str, str]):
             Mapping from resource names to environment variable names for device allocation,
@@ -130,7 +130,7 @@ class KubernetesWorkloadPlan(WorkloadPlan):
     """
     Domain suffix for the cluster.
     """
-    service_type: KubernetesWorkloadServiceType = field(
+    service_type: KubernetesWorkloadServiceTypeEnum = field(
         default_factory=lambda: envs.GPUSTACK_RUNTIME_KUBERNETES_SERVICE_TYPE,
     )
     """

gpustack_runtime/envs.py

@@ -129,6 +129,10 @@
     """
     Correct the gpustack-runner image by rendering it with the host's detection.
     """
+    GPUSTACK_RUNTIME_DEPLOY_IMAGE_PULL_POLICY: str | None = None
+    """
+    Image pull policy for the deployer (e.g., Always, IfNotPresent, Never).
+    """
     GPUSTACK_RUNTIME_DEPLOY_LABEL_PREFIX: str | None = None
     """
     Label prefix for the deployer.
@@ -313,6 +317,14 @@
     "GPUSTACK_RUNTIME_DEPLOY_CORRECT_RUNNER_IMAGE": lambda: to_bool(
         getenv("GPUSTACK_RUNTIME_DEPLOY_CORRECT_RUNNER_IMAGE", "1"),
     ),
+    "GPUSTACK_RUNTIME_DEPLOY_IMAGE_PULL_POLICY": lambda: choice(
+        getenv(
+            "GPUSTACK_RUNTIME_DEPLOY_IMAGE_PULL_POLICY",
+            "IfNotPresent",
+        ),
+        options=["Always", "IfNotPresent", "Never"],
+        default="IfNotPresent",
+    ),
     "GPUSTACK_RUNTIME_DEPLOY_LABEL_PREFIX": lambda: getenv(
         "GPUSTACK_RUNTIME_DEPLOY_LABEL_PREFIX",
         "runtime.gpustack.ai",