feat(RHOAIENG-29391): Store entrypoint scripts in configMaps

Signed-off-by: Pat O'Connor <[email protected]>

Author: chipspeak

helper.py

@@ -0,0 +1,58 @@
+"""
+Helper script for testing multiple script mounting in RayJobs
+"""
+import ray
+from datetime import datetime
+
+
+def get_cluster_info():
+    """Get information about the Ray cluster"""
+    try:
+        cluster_info = {
+            "ray_version": ray.__version__,
+            "cluster_resources": ray.cluster_resources(),
+            "available_resources": ray.available_resources(),
+            "nodes": len(ray.nodes()),
+            "timestamp": datetime.now().isoformat(),
+        }
+        return cluster_info
+    except Exception as e:
+        return {"error": str(e), "timestamp": datetime.now().isoformat()}
+
+
+def run_simple_ray_task():
+    """Run a simple Ray task to test cluster functionality"""
+
+    @ray.remote
+    def hello_task(name):
+        import time
+
+        time.sleep(1)  # Simulate some work
+        return f"Hello {name} from Ray worker!"
+
+    # Submit multiple tasks
+    tasks = [hello_task.remote(f"Task-{i}") for i in range(3)]
+    results = ray.get(tasks)
+
+    return results
+
+
+def print_mounted_files():
+    """Check what files are available in the scripts directory"""
+    import os
+
+    script_dir = "/home/ray/scripts"
+
+    print(f"📁 Contents of {script_dir}:")
+    try:
+        if os.path.exists(script_dir):
+            files = os.listdir(script_dir)
+            for file in sorted(files):
+                file_path = os.path.join(script_dir, file)
+                if os.path.isfile(file_path):
+                    size = os.path.getsize(file_path)
+                    print(f"  📄 {file} ({size} bytes)")
+        else:
+            print(f"  ❌ Directory {script_dir} does not exist")
+    except Exception as e:
+        print(f"  ❌ Error listing files: {e}")

src/codeflare_sdk/ray/rayjobs/config.py

@@ -20,7 +20,7 @@
 
 import pathlib
 from dataclasses import dataclass, field, fields
-from typing import Dict, List, Optional, Union, get_args, get_origin, Any
+from typing import Dict, List, Optional, Union, get_args, get_origin, Any, Tuple
 from kubernetes.client import (
     V1ConfigMapVolumeSource,
     V1KeyToPath,
@@ -515,3 +515,88 @@ def _build_gcs_ft_options(self) -> Dict[str, Any]:
             }
 
         return gcs_ft_options
+
+    def add_script_volumes(
+        self, configmap_name: str, mount_path: str = "/home/ray/scripts"
+    ):
+        """
+        Add script volume and mount references to cluster configuration.
+
+        Args:
+            configmap_name: Name of the ConfigMap containing scripts
+            mount_path: Where to mount scripts in containers (default: /home/ray/scripts)
+        """
+        # Check if script volume already exists
+        volume_name = "ray-job-scripts"
+        existing_volume = next(
+            (v for v in self.volumes if getattr(v, "name", None) == volume_name), None
+        )
+        if existing_volume:
+            logger.debug(f"Script volume '{volume_name}' already exists, skipping...")
+            return
+
+        # Check if script mount already exists
+        existing_mount = next(
+            (m for m in self.volume_mounts if getattr(m, "name", None) == volume_name),
+            None,
+        )
+        if existing_mount:
+            logger.debug(
+                f"Script volume mount '{volume_name}' already exists, skipping..."
+            )
+            return
+
+        # Add script volume to cluster configuration
+        script_volume = V1Volume(
+            name=volume_name, config_map=V1ConfigMapVolumeSource(name=configmap_name)
+        )
+        self.volumes.append(script_volume)
+
+        # Add script volume mount to cluster configuration
+        script_mount = V1VolumeMount(name=volume_name, mount_path=mount_path)
+        self.volume_mounts.append(script_mount)
+
+        logger.info(
+            f"Added script volume '{configmap_name}' to cluster config: mount_path={mount_path}"
+        )
+
+    def build_script_configmap_spec(
+        self, job_name: str, namespace: str, scripts: Dict[str, str]
+    ) -> Dict[str, Any]:
+        """
+        Build ConfigMap specification for scripts (no API calls).
+
+        Args:
+            job_name: Name of the RayJob (used for ConfigMap naming)
+            namespace: Kubernetes namespace
+            scripts: Dictionary of script_name -> script_content
+
+        Returns:
+            Dict: ConfigMap specification ready for Kubernetes API
+        """
+        configmap_name = f"{job_name}-scripts"
+        return {
+            "apiVersion": "v1",
+            "kind": "ConfigMap",
+            "metadata": {"name": configmap_name, "namespace": namespace},
+            "data": scripts,
+        }
+
+    def build_script_volume_specs(
+        self, configmap_name: str, mount_path: str = "/home/ray/scripts"
+    ) -> Tuple[Dict[str, Any], Dict[str, Any]]:
+        """
+        Build volume and mount specifications for scripts (no API calls).
+
+        Args:
+            configmap_name: Name of the ConfigMap containing scripts
+            mount_path: Where to mount scripts in containers
+
+        Returns:
+            Tuple of (volume_spec, mount_spec) as dictionaries
+        """
+        volume_spec = {"name": "ray-job-scripts", "configMap": {"name": configmap_name}}
+
+        mount_spec = {"name": "ray-job-scripts", "mountPath": mount_path}
+
+        return volume_spec, mount_spec