digitalhub-runtime-python 0.16.0b5__py3-none-any.whl → 0.16.0b7__py3-none-any.whl

digitalhub_runtime_python/__init__.py

@@ -6,6 +6,7 @@ from digitalhub_runtime_python.entities.function.guardrail.builder import Functi
 from digitalhub_runtime_python.entities.function.hydra.builder import FunctionHydraBuilder
 from digitalhub_runtime_python.entities.function.openinference.builder import FunctionOpeninferenceBuilder
 from digitalhub_runtime_python.entities.function.python.builder import FunctionPythonBuilder
+from digitalhub_runtime_python.entities.function.ray.builder import FunctionRayBuilder
 from digitalhub_runtime_python.entities.run.guardrail_build.builder import RunGuardrailRunBuildBuilder
 from digitalhub_runtime_python.entities.run.guardrail_serve.builder import RunGuardrailRunServeBuilder
 from digitalhub_runtime_python.entities.run.hydra_build.builder import RunHydraRunBuildBuilder
@@ -16,6 +17,8 @@ from digitalhub_runtime_python.entities.run.openinference_serve.builder import R
 from digitalhub_runtime_python.entities.run.python_build.builder import RunPythonRunBuildBuilder
 from digitalhub_runtime_python.entities.run.python_job.builder import RunPythonRunJobBuilder
 from digitalhub_runtime_python.entities.run.python_serve.builder import RunPythonRunServeBuilder
+from digitalhub_runtime_python.entities.run.ray_build.builder import RunRayRunBuildBuilder
+from digitalhub_runtime_python.entities.run.ray_job.builder import RunRayRunJobBuilder
 from digitalhub_runtime_python.entities.task.guardrail_build.builder import TaskGuardrailBuildBuilder
 from digitalhub_runtime_python.entities.task.guardrail_serve.builder import TaskGuardrailServeBuilder
 from digitalhub_runtime_python.entities.task.hydra_build.builder import TaskHydraBuildBuilder
@@ -26,13 +29,15 @@ from digitalhub_runtime_python.entities.task.openinference_serve.builder import
 from digitalhub_runtime_python.entities.task.python_build.builder import TaskPythonBuildBuilder
 from digitalhub_runtime_python.entities.task.python_job.builder import TaskPythonJobBuilder
 from digitalhub_runtime_python.entities.task.python_serve.builder import TaskPythonServeBuilder
-from digitalhub_runtime_python.utils.utils import handler
+from digitalhub_runtime_python.entities.task.ray_build.builder import TaskRayBuildBuilder
+from digitalhub_runtime_python.entities.task.ray_job.builder import TaskRayJobBuilder
 
 entity_builders = (
     (EntityKinds.FUNCTION_GUARDRAIL.value, FunctionGuardrailBuilder),
     (EntityKinds.FUNCTION_OPENINFERENCE.value, FunctionOpeninferenceBuilder),
     (EntityKinds.FUNCTION_PYTHON.value, FunctionPythonBuilder),
     (EntityKinds.FUNCTION_HYDRA.value, FunctionHydraBuilder),
+    (EntityKinds.FUNCTION_RAY.value, FunctionRayBuilder),
     (EntityKinds.RUN_GUARDRAIL_BUILD.value, RunGuardrailRunBuildBuilder),
     (EntityKinds.RUN_GUARDRAIL_SERVE.value, RunGuardrailRunServeBuilder),
     (EntityKinds.RUN_OPENINFERENCE_BUILD.value, RunOpeninferenceRunBuildBuilder),
@@ -43,6 +48,8 @@ entity_builders = (
     (EntityKinds.RUN_HYDRA_BUILD.value, RunHydraRunBuildBuilder),
     (EntityKinds.RUN_HYDRA_JOB.value, RunHydraRunJobBuilder),
     (EntityKinds.RUN_HYDRA_SUBTASK.value, RunHydraRunSubtaskBuilder),
+    (EntityKinds.RUN_RAY_JOB.value, RunRayRunJobBuilder),
+    (EntityKinds.RUN_RAY_BUILD.value, RunRayRunBuildBuilder),
     (EntityKinds.TASK_GUARDRAIL_BUILD.value, TaskGuardrailBuildBuilder),
     (EntityKinds.TASK_GUARDRAIL_SERVE.value, TaskGuardrailServeBuilder),
     (EntityKinds.TASK_OPENINFERENCE_BUILD.value, TaskOpeninferenceBuildBuilder),
@@ -53,6 +60,8 @@ entity_builders = (
     (EntityKinds.TASK_HYDRA_BUILD.value, TaskHydraBuildBuilder),
     (EntityKinds.TASK_HYDRA_JOB.value, TaskHydraJobBuilder),
     (EntityKinds.TASK_HYDRA_SUBTASK.value, TaskHydraSubtaskBuilder),
+    (EntityKinds.TASK_RAY_BUILD.value, TaskRayBuildBuilder),
+    (EntityKinds.TASK_RAY_JOB.value, TaskRayJobBuilder),
 )
 
 try:
@@ -64,6 +73,7 @@ try:
         RuntimeOpeninferenceBuilder,
         RuntimePythonBuilder,
         RuntimePythonJobBuilder,
+        RuntimeRayBuilder,
     )
 
     runtime_builders = (
@@ -71,6 +81,7 @@ try:
         (EntityKinds.FUNCTION_OPENINFERENCE.value, RuntimeOpeninferenceBuilder),
         (EntityKinds.FUNCTION_PYTHON.value, RuntimePythonBuilder),
         (EntityKinds.FUNCTION_HYDRA.value, RuntimeHydraBuilder),
+        (EntityKinds.FUNCTION_RAY.value, RuntimeRayBuilder),
         (EntityKinds.RUN_GUARDRAIL_BUILD.value, RuntimeGuardrailBuilder),
         (EntityKinds.RUN_GUARDRAIL_SERVE.value, RuntimeGuardrailBuilder),
         (EntityKinds.RUN_OPENINFERENCE_BUILD.value, RuntimeOpeninferenceBuilder),
@@ -81,6 +92,8 @@ try:
         (EntityKinds.RUN_HYDRA_BUILD.value, RuntimeHydraBuilder),
         (EntityKinds.RUN_HYDRA_JOB.value, RuntimeHydraJobBuilder),
         (EntityKinds.RUN_HYDRA_SUBTASK.value, RuntimeHydraSubtaskBuilder),
+        (EntityKinds.RUN_RAY_BUILD.value, RuntimeRayBuilder),
+        (EntityKinds.RUN_RAY_JOB.value, RuntimeRayBuilder),
         (EntityKinds.TASK_GUARDRAIL_BUILD.value, RuntimeGuardrailBuilder),
         (EntityKinds.TASK_GUARDRAIL_SERVE.value, RuntimeGuardrailBuilder),
         (EntityKinds.TASK_OPENINFERENCE_BUILD.value, RuntimeOpeninferenceBuilder),
@@ -91,6 +104,8 @@ try:
         (EntityKinds.TASK_HYDRA_BUILD.value, RuntimeHydraBuilder),
         (EntityKinds.TASK_HYDRA_JOB.value, RuntimeHydraJobBuilder),
         (EntityKinds.TASK_HYDRA_SUBTASK.value, RuntimeHydraSubtaskBuilder),
+        (EntityKinds.TASK_RAY_BUILD.value, RuntimeRayBuilder),
+        (EntityKinds.TASK_RAY_JOB.value, RuntimeRayBuilder),
     )
 except ImportError as e:
     from digitalhub.utils.logger.logger import get_logger

digitalhub_runtime_python/entities/_base/runtime_entity/builder.py

@@ -136,3 +136,30 @@ class RuntimeEntityBuilderHydra(RuntimeEntityBuilder):
             ),
         ]
     )
+
+class RuntimeEntityBuilderRay(RuntimeEntityBuilder):
+    EXECUTABLE_KIND = EntityKinds.FUNCTION_RAY.value
+    TASKS_KINDS = map_actions(
+        [
+            (
+                EntityKinds.TASK_RAY_JOB.value,
+                Actions.JOB.value,
+            ),
+            (
+                EntityKinds.TASK_RAY_BUILD.value,
+                Actions.BUILD.value,
+            ),
+        ]
+    )
+    RUN_KINDS = map_actions(
+        [
+            (
+                EntityKinds.RUN_RAY_JOB.value,
+                Actions.JOB.value,
+            ),
+            (
+                EntityKinds.RUN_RAY_BUILD.value,
+                Actions.BUILD.value,
+            ),
+        ]
+    )

digitalhub_runtime_python/entities/_commons/enums.py

@@ -40,6 +40,11 @@ class EntityKinds(Enum):
     RUN_HYDRA_JOB = "hydra+job:run"
     RUN_HYDRA_SUBTASK = "hydra+subtask:run"
 
+    FUNCTION_RAY = "ray"
+    TASK_RAY_BUILD = "ray+build"
+    TASK_RAY_JOB = "ray+job"
+    RUN_RAY_BUILD = "ray+build:run"
+    RUN_RAY_JOB = "ray+job:run"
 
 class Actions(Enum):
     """

digitalhub_runtime_python/entities/function/ray/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0

digitalhub_runtime_python/entities/function/ray/builder.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderRay
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.function._base.builder import FunctionBaseBuilder
+from digitalhub_runtime_python.entities.function.ray.entity import FunctionRay
+from digitalhub_runtime_python.entities.function.ray.spec import FunctionSpecRay, FunctionValidatorRay
+from digitalhub_runtime_python.entities.function.ray.status import FunctionStatusRay
+
+
+class FunctionRayBuilder(FunctionBaseBuilder, RuntimeEntityBuilderRay):
+    """
+    Ray Python function builder.
+    """
+
+    ENTITY_CLASS = FunctionRay
+    ENTITY_SPEC_CLASS = FunctionSpecRay
+    ENTITY_SPEC_VALIDATOR = FunctionValidatorRay
+    ENTITY_STATUS_CLASS = FunctionStatusRay
+    ENTITY_KIND = EntityKinds.FUNCTION_RAY.value

digitalhub_runtime_python/entities/function/ray/entity.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+import typing
+
+from digitalhub_runtime_python.entities.function._base.entity import FunctionBaseFunction
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.function.ray.spec import FunctionSpecRay
+    from digitalhub_runtime_python.entities.function.ray.status import FunctionStatusRay
+
+
+class FunctionRay(FunctionBaseFunction):
+    """
+    FunctionRay class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: FunctionSpecRay
+        self.status: FunctionStatusRay

digitalhub_runtime_python/entities/function/ray/spec.py

@@ -0,0 +1,57 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.function._base.spec import FunctionSpec, FunctionValidator
+
+from digitalhub_runtime_python.entities.function.python.models import PythonVersion, SourceValidator
+
+
+class FunctionSpecRay(FunctionSpec):
+    """
+    FunctionSpecRay specifications.
+    """
+
+    def __init__(
+        self,
+        source: dict | None = None,
+        image: str | None = None,
+        base_image: str | None = None,
+        python_version: str | None = None,
+        requirements: list | None = None,
+        ray_version: str | None = None,
+    ) -> None:
+        super().__init__()
+
+        self.image = image
+        self.base_image = base_image
+        self.python_version = python_version
+        self.requirements = requirements
+        self.ray_version = ray_version
+        self.source = source
+
+
+class FunctionValidatorRay(FunctionValidator):
+    """
+    FunctionValidatorRay validator.
+    """
+
+    source: SourceValidator
+    """Source code validator"""
+
+    python_version: PythonVersion
+    "Python version"
+
+    image: str | None = None
+    "Image where the function will be executed"
+
+    base_image: str | None = None
+    "Base image used to build the image where the function will be executed"
+
+    requirements: list[str] | str | None = None
+    "Requirements list to be installed in the image where the function will be executed"
+
+    ray_version: str | None = None
+    "Ray version"

digitalhub_runtime_python/entities/function/ray/status.py

@@ -0,0 +1,13 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.function._base.status import FunctionStatus
+
+
+class FunctionStatusRay(FunctionStatus):
+    """
+    FunctionStatusRay status.
+    """

digitalhub_runtime_python/entities/run/ray_build/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0

digitalhub_runtime_python/entities/run/ray_build/builder.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.run._base.builder import RunBuilder
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderRay
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.run.ray_build.entity import RunRayRunBuild
+from digitalhub_runtime_python.entities.run.ray_build.spec import RunSpecRayRunBuild, RunValidatorRayRunBuild
+from digitalhub_runtime_python.entities.run.ray_build.status import RunStatusRayRunBuild
+
+
+class RunRayRunBuildBuilder(RunBuilder, RuntimeEntityBuilderRay):
+    """
+    RunRayRunBuildBuilder runner.
+    """
+
+    ENTITY_CLASS = RunRayRunBuild
+    ENTITY_SPEC_CLASS = RunSpecRayRunBuild
+    ENTITY_SPEC_VALIDATOR = RunValidatorRayRunBuild
+    ENTITY_STATUS_CLASS = RunStatusRayRunBuild
+    ENTITY_KIND = EntityKinds.RUN_RAY_BUILD.value

digitalhub_runtime_python/entities/run/ray_build/entity.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+import typing
+
+from digitalhub_runtime_python.entities.run._base.entity import RunBaseBuildRun
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.run.ray_build.spec import RunSpecRayRunBuild
+    from digitalhub_runtime_python.entities.run.ray_build.status import RunStatusRayRunBuild
+
+
+class RunRayRunBuild(RunBaseBuildRun):
+    """
+    RunRayRunBuild class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: RunSpecRayRunBuild
+        self.status: RunStatusRayRunBuild

digitalhub_runtime_python/entities/run/ray_build/spec.py

@@ -0,0 +1,15 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub_runtime_python.entities.run._base.spec import RunSpecBaseRun, RunValidatorBaseRun
+
+
+class RunSpecRayRunBuild(RunSpecBaseRun):
+    """RunSpecRayRunBuild specifications."""
+
+
+class RunValidatorRayRunBuild(RunValidatorBaseRun):
+    """RunValidatorRayRunBuild validator."""

digitalhub_runtime_python/entities/run/ray_build/status.py

@@ -0,0 +1,13 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub_runtime_python.entities.run._base.status import RunStatusBaseRun
+
+
+class RunStatusRayRunBuild(RunStatusBaseRun):
+    """
+    RunStatusRayRunBuild status.
+    """

digitalhub_runtime_python/entities/run/ray_job/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0

digitalhub_runtime_python/entities/run/ray_job/builder.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.run._base.builder import RunBuilder
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderPython
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.run.ray_job.entity import RunRayRunJob
+from digitalhub_runtime_python.entities.run.ray_job.spec import RunSpecRayRunJob, RunValidatorRayRunJob
+from digitalhub_runtime_python.entities.run.ray_job.status import RunStatusRayRunJob
+
+
+class RunRayRunJobBuilder(RunBuilder, RuntimeEntityBuilderPython):
+    """
+    RunRayRunJobBuilder runner.
+    """
+
+    ENTITY_CLASS = RunRayRunJob
+    ENTITY_SPEC_CLASS = RunSpecRayRunJob
+    ENTITY_SPEC_VALIDATOR = RunValidatorRayRunJob
+    ENTITY_STATUS_CLASS = RunStatusRayRunJob
+    ENTITY_KIND = EntityKinds.RUN_RAY_JOB.value

digitalhub_runtime_python/entities/run/ray_job/entity.py

@@ -0,0 +1,69 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+import typing
+
+from digitalhub_runtime_python.entities.run._base.entity import RunBaseRun
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.run.ray_job.spec import RunSpecRayRunJob
+    from digitalhub_runtime_python.entities.run.ray_job.status import RunStatusRayRunJob
+
+
+class RunRayRunJob(RunBaseRun):
+    """
+    RunRayRunJob class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: RunSpecRayRunJob
+        self.status: RunStatusRayRunJob
+
+    def local_execution(self) -> bool:
+        """
+        Check if run has local execution.
+
+        Returns
+        -------
+        bool
+            True if run has local execution, False otherwise.
+        """
+        return self.spec.local_execution
+
+    def add_output(self, name: str, value: str) -> None:
+        """
+        Add an output to the run job.
+
+        Parameters
+        ----------
+        name : str
+            The name of the output.
+        value : str
+            The value of the output.
+        """
+        if self.status.outputs is None:
+            self.status.outputs = {}
+        self.status.outputs[name] = value
+
+    def set_status(self, status: dict) -> None:
+        """
+        Patch to merge outputs when updating status.
+
+        Parameters
+        ----------
+        status : dict
+            The status dictionary to update.
+        """
+        if self.status.outputs is None:
+            self.status.outputs = {}
+        if "outputs" in status:
+            status["outputs"] = {
+                **self.status.outputs,
+                **status["outputs"],
+            }
+        return super().set_status(status)

digitalhub_runtime_python/entities/run/ray_job/spec.py

@@ -0,0 +1,68 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub_runtime_python.entities.run._base.spec import RunSpecBaseRun, RunValidatorBaseRun
+
+
+class RunSpecRayRunJob(RunSpecBaseRun):
+    """RunSpecRayRunJob specifications."""
+
+    def __init__(
+        self,
+        task: str,
+        function: str | None = None,
+        workflow: str | None = None,
+        volumes: list[dict] | None = None,
+        resources: dict | None = None,
+        envs: list[dict] | None = None,
+        secrets: list[str] | None = None,
+        profile: str | None = None,
+        source: dict | None = None,
+        image: str | None = None,
+        base_image: str | None = None,
+        python_version: str | None = None,
+        requirements: list | None = None,
+        service_type: str | None = None,
+        service_name: str | None = None,
+        replicas: int | None = None,
+        instructions: dict | None = None,
+        inputs: dict | None = None,
+        parameters: dict | None = None,
+        init_parameters: dict | None = None,
+        local_execution: bool = False,
+        **kwargs,
+    ) -> None:
+        super().__init__(
+            task,
+            function,
+            workflow,
+            volumes,
+            resources,
+            envs,
+            secrets,
+            profile,
+            source,
+            image,
+            base_image,
+            python_version,
+            requirements,
+            service_type,
+            service_name,
+            replicas,
+            instructions,
+            inputs,
+            parameters,
+            init_parameters,
+            **kwargs,
+        )
+        self.local_execution = local_execution
+
+
+class RunValidatorRayRunJob(RunValidatorBaseRun):
+    """RunValidatorRayRunJob validator."""
+
+    local_execution: bool = False
+    """Whether to execute the run locally instead of in the cluster."""

digitalhub_runtime_python/entities/run/ray_job/status.py

@@ -0,0 +1,13 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub_runtime_python.entities.run._base.status import RunStatusBaseRun
+
+
+class RunStatusRayRunJob(RunStatusBaseRun):
+    """
+    RunStatusRayRunJob status.
+    """

digitalhub_runtime_python/entities/task/ray_build/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0

digitalhub_runtime_python/entities/task/ray_build/builder.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.task._base.builder import TaskBuilder
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderRay
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.task.ray_build.entity import TaskRayBuild
+from digitalhub_runtime_python.entities.task.ray_build.spec import TaskSpecRayBuild, TaskValidatorRayBuild
+from digitalhub_runtime_python.entities.task.ray_build.status import TaskStatusRayBuild
+
+
+class TaskRayBuildBuilder(TaskBuilder, RuntimeEntityBuilderRay):
+    """
+    TaskRayBuild builder.
+    """
+
+    ENTITY_CLASS = TaskRayBuild
+    ENTITY_SPEC_CLASS = TaskSpecRayBuild
+    ENTITY_SPEC_VALIDATOR = TaskValidatorRayBuild
+    ENTITY_STATUS_CLASS = TaskStatusRayBuild
+    ENTITY_KIND = EntityKinds.TASK_RAY_BUILD.value

digitalhub_runtime_python/entities/task/ray_build/entity.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+import typing
+
+from digitalhub.entities.task._base.entity import Task
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.task.ray_build.spec import TaskSpecRayBuild
+    from digitalhub_runtime_python.entities.task.ray_build.status import TaskStatusRayBuild
+
+
+class TaskRayBuild(Task):
+    """
+    TaskRayBuild class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: TaskSpecRayBuild
+        self.status: TaskStatusRayBuild

digitalhub_runtime_python/entities/task/ray_build/spec.py

@@ -0,0 +1,42 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.task._base.spec import TaskSpecFunction, TaskValidatorFunction
+
+
+class TaskSpecRayBuild(TaskSpecFunction):
+    """TaskSpecRayBuild specifications."""
+
+    def __init__(
+        self,
+        function: str,
+        volumes: list[dict] | None = None,
+        resources: dict | None = None,
+        envs: list[dict] | None = None,
+        secrets: list[str] | None = None,
+        profile: str | None = None,
+        instructions: list | None = None,
+        **kwargs,
+    ) -> None:
+        super().__init__(
+            function=function,
+            volumes=volumes,
+            resources=resources,
+            envs=envs,
+            secrets=secrets,
+            profile=profile,
+            **kwargs,
+        )
+        self.instructions = instructions
+
+
+class TaskValidatorRayBuild(TaskValidatorFunction):
+    """
+    TaskValidatorRayBuild validator.
+    """
+
+    instructions: list[str] | None = None
+    """Build instructions."""

digitalhub_runtime_python/entities/task/ray_build/status.py

@@ -0,0 +1,13 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.task._base.status import TaskStatus
+
+
+class TaskStatusRayBuild(TaskStatus):
+    """
+    TaskStatusRayBuild status.
+    """

digitalhub_runtime_python/entities/task/ray_job/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0

digitalhub_runtime_python/entities/task/ray_job/builder.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.task._base.builder import TaskBuilder
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderRay
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.task.ray_job.entity import TaskRayJob
+from digitalhub_runtime_python.entities.task.ray_job.spec import TaskSpecRayJob, TaskValidatorRayJob
+from digitalhub_runtime_python.entities.task.ray_job.status import TaskStatusRayJob
+
+
+class TaskRayJobBuilder(TaskBuilder, RuntimeEntityBuilderRay):
+    """
+    TaskRayJobBuilder jober.
+    """
+
+    ENTITY_CLASS = TaskRayJob
+    ENTITY_SPEC_CLASS = TaskSpecRayJob
+    ENTITY_SPEC_VALIDATOR = TaskValidatorRayJob
+    ENTITY_STATUS_CLASS = TaskStatusRayJob
+    ENTITY_KIND = EntityKinds.TASK_RAY_JOB.value

digitalhub_runtime_python/entities/task/ray_job/entity.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+import typing
+
+from digitalhub.entities.task._base.entity import Task
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.task.ray_job.spec import TaskSpecRayJob
+    from digitalhub_runtime_python.entities.task.ray_job.status import TaskStatusRayJob
+
+
+class TaskRayJob(Task):
+    """
+    TaskRayJob class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: TaskSpecRayJob
+        self.status: TaskStatusRayJob

digitalhub_runtime_python/entities/task/ray_job/spec.py

@@ -0,0 +1,52 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.task._base.spec import TaskSpecFunction, TaskValidatorFunction
+
+
+class TaskSpecRayJob(TaskSpecFunction):
+    """
+    TaskSpecRayJob specifications.
+    """
+    def __init__(
+        self,
+        function: str,
+        volumes: list[dict] | None = None,
+        resources: dict | None = None,
+        envs: list[dict] | None = None,
+        secrets: list[str] | None = None,
+        profile: str | None = None,
+        replicas: int | None = None,
+        min_replicas: int | None = None,
+        max_replicas: int | None = None,
+        **kwargs,
+    ) -> None:
+        super().__init__(
+            function=function,
+            volumes=volumes,
+            resources=resources,
+            envs=envs,
+            secrets=secrets,
+            profile=profile,
+            **kwargs,
+        )
+        self.replicas = replicas
+        self.min_replicas = min_replicas
+        self.max_replicas = max_replicas
+
+class TaskValidatorRayJob(TaskValidatorFunction):
+    """
+    TaskValidatorRayJob validator.
+    """
+
+    replicas: int | None = None
+    """Number of replicas."""
+
+    min_replicas: int | None = None
+    """Minimum number of replicas."""
+
+    max_replicas: int | None = None
+    """Maximum number of replicas."""

digitalhub_runtime_python/entities/task/ray_job/status.py

@@ -0,0 +1,13 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.task._base.status import TaskStatus
+
+
+class TaskStatusRayJob(TaskStatus):
+    """
+    TaskStatusRayJob status.
+    """

digitalhub_runtime_python/runtimes/builder.py

@@ -16,6 +16,7 @@ from digitalhub_runtime_python.runtimes.runtime import (
     RuntimeOpeninference,
     RuntimePython,
     RuntimePythonJob,
+    RuntimeRay,
 )
 
 
@@ -59,3 +60,8 @@ class RuntimeHydraSubtaskBuilder(RuntimeBuilder):
     """RuntimeHydraSubtaskBuilder class."""
 
     RUNTIME_CLASS = RuntimeHydraSubtask
+
+class RuntimeRayBuilder(RuntimeBuilder):
+    """RuntimeRayBuilder class."""
+
+    RUNTIME_CLASS = RuntimeRay

digitalhub_runtime_python/runtimes/hydra_runtime.py

@@ -132,7 +132,6 @@ class RuntimeHydraJob(RuntimePythonJob):
         if "parameters" in spec:
             sys.argv += [f"{key}={value}" for key, value in spec["parameters"].items()]
 
-
         # import from source
         fnc = import_function_from_source(
             source_path,
@@ -149,6 +148,10 @@ class RuntimeHydraSubtask(RuntimePythonJob):
     """
 
     def _configure_execution(self, spec: dict) -> tuple[Callable, bool]:
+        from hydra.core.utils import setup_globals
+
+        setup_globals()
+
         source_spec = spec.get("source", {})
         path = self.runtime_dir
 
@@ -158,13 +161,13 @@ class RuntimeHydraSubtask(RuntimePythonJob):
         return fnc, False
 
     def _compose_args(self, func, spec, project):
-        # expect to be wrapped with hydra.main, 'cfg' is in the parameters, and Omecaconf is present
+        # expect to be wrapped with hydra.main, 'cfg_passthrough' is in the parameters, and Omecaconf is present
         from omegaconf import OmegaConf
 
-        args = super()._compose_args(func, spec, project)
+        cfg = spec.get("parameters", {}).get("cfg_passthrough", {})
         try:
-            args["cfg_passthrough"] = OmegaConf.create(spec.get("parameters", {}).get("cfg_passthrough", {}))
+            args = {"cfg_passthrough": OmegaConf.create(cfg)}
         except Exception as e:  # noqa: BLE001
             print(f"Failed to convert cfg to container. Exception: {e.__class__}. Error: {e.args}")
-            args["cfg_passthrough"] = {}
+            args = {"cfg_passthrough": {}}
         return args

digitalhub_runtime_python/runtimes/runtime.py

@@ -151,3 +151,8 @@ class RuntimeGuardrail(RuntimePython):
     """
     Runtime Guardrail class.
     """
+
+class RuntimeRay(RuntimePython):
+    """
+    Runtime Ray class.
+    """

{digitalhub_runtime_python-0.16.0b5.dist-info → digitalhub_runtime_python-0.16.0b7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: digitalhub-runtime-python
-Version: 0.16.0b5
+Version: 0.16.0b7
 Summary: Python runtime for DHCore
 Project-URL: Homepage, https://github.com/scc-digitalhub/digitalhub-sdk-runtime-python
 Author-email: Fondazione Bruno Kessler <digitalhub@fbk.eu>, Matteo Martini <mmartini@fbk.eu>

{digitalhub_runtime_python-0.16.0b5.dist-info → digitalhub_runtime_python-0.16.0b7.dist-info}/RECORD

@@ -1,10 +1,10 @@
-digitalhub_runtime_python/__init__.py,sha256=TsiuxnaUlDDpwXUfcjDh0AIZyWsn_vIqh8UAjMbGSr8,6610
+digitalhub_runtime_python/__init__.py,sha256=qy17THET_iivXjtbaV0UfuN8tAWtEqND2Y_BXWt8NQ8,7623
 digitalhub_runtime_python/entities/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/_base/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/_base/runtime_entity/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
-digitalhub_runtime_python/entities/_base/runtime_entity/builder.py,sha256=THkwWt_iul8fler4KJrHUMKhHtuqYyuk-pRoADqHH0U,3688
+digitalhub_runtime_python/entities/_base/runtime_entity/builder.py,sha256=VaARLEaABav4-xCe6GDS6HhkPI4akmHLVHyobDi3sPk,4361
 digitalhub_runtime_python/entities/_commons/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
-digitalhub_runtime_python/entities/_commons/enums.py,sha256=G24c6GRO5wkZ4hEvk5tq4AG0Q6Mg-vQq6tk0VsMa5T4,1381
+digitalhub_runtime_python/entities/_commons/enums.py,sha256=nnCvH9EOzf8V7GQsr91FPV8U2eB7ITzPQGb6C2lY6j8,1536
 digitalhub_runtime_python/entities/_commons/utils.py,sha256=nmwB-4KoruimUK0NhGvzMeo2P4ulFnr13zfuxgE9ZMs,5714
 digitalhub_runtime_python/entities/_commons/requirement_parser/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/_commons/requirement_parser/enums.py,sha256=plZQOcr83YVYxfStdIFFw8U3BG9u8716H0B8KZUYzBc,540
@@ -44,6 +44,11 @@ digitalhub_runtime_python/entities/function/python/entity.py,sha256=P62w_UKVfFzC
 digitalhub_runtime_python/entities/function/python/models.py,sha256=sJ0Gd_EV66zRmKbYPjrHgLt3cgAmz5JhkFOJkSeG0yo,1031
 digitalhub_runtime_python/entities/function/python/spec.py,sha256=6TkwV0kGVryvZmvMY12RzfsTzw4D2Pm43dRNi3IvGy0,1428
 digitalhub_runtime_python/entities/function/python/status.py,sha256=XK3KPKovCRiMfEFzFeCzaVkoQyuwfci9nI2JzYxCIec,308
+digitalhub_runtime_python/entities/function/ray/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
+digitalhub_runtime_python/entities/function/ray/builder.py,sha256=JdA-w_7hdNeJv0EAQNEqp5Mo_UjB5KTHBCABKBjBJwI,1012
+digitalhub_runtime_python/entities/function/ray/entity.py,sha256=lw5rgO-FLb7fgiYEOVMotwDexSQjwaNJVdoNcy-l1PM,698
+digitalhub_runtime_python/entities/function/ray/spec.py,sha256=ZVL6e8W6bmW5QSS4Vy3L32ixXkZHEQjBpW6xD8c6BZM,1548
+digitalhub_runtime_python/entities/function/ray/status.py,sha256=oaZ4E7sgOlmAqVNtWP3XSbERki0NaimyX_jN4DBSi00,302
 digitalhub_runtime_python/entities/run/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/run/_base/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/run/_base/entity.py,sha256=BusdxsgFU6pTwxawOodRGqAr_26h0RFBl6_8A2UmVDQ,6130
@@ -100,6 +105,16 @@ digitalhub_runtime_python/entities/run/python_serve/builder.py,sha256=bMpSF05YdE
 digitalhub_runtime_python/entities/run/python_serve/entity.py,sha256=AqDYz7vg2D_pTkdpUaJBZhrpt8ciN_tWsyxM0GJdwb0,2438
 digitalhub_runtime_python/entities/run/python_serve/spec.py,sha256=JWgbuBn4a0HtwPNQkCUNzRHAmn4KlSnqTarcykf76ak,442
 digitalhub_runtime_python/entities/run/python_serve/status.py,sha256=eYrDLQy2rTxdhr1UfmwlhEPn2ZqYf7nr11OiCO3tziY,328
+digitalhub_runtime_python/entities/run/ray_build/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
+digitalhub_runtime_python/entities/run/ray_build/builder.py,sha256=nXrMlwImOdmCiWNOr0cJ31Dcqu8GxD40vpxVGUCgR0M,1007
+digitalhub_runtime_python/entities/run/ray_build/entity.py,sha256=RJ9HfBVVeITGt3UxtdbYVxGcQYUHfYrX9IlWiTAafYk,703
+digitalhub_runtime_python/entities/run/ray_build/spec.py,sha256=kKAmp-c1_-YF77pRYTNl57GOdahua2a7EiHaLyh4leU,430
+digitalhub_runtime_python/entities/run/ray_build/status.py,sha256=GB40PoeN9_qyagmpfLELn4QzCBsdbT6Y4Wfw3hGn2uM,322
+digitalhub_runtime_python/entities/run/ray_job/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
+digitalhub_runtime_python/entities/run/ray_job/builder.py,sha256=X5e-qQFqokdnR9JQJoSpyshy4DRzTUuWG8vhwUm6b6k,985
+digitalhub_runtime_python/entities/run/ray_job/entity.py,sha256=C8xYddOROdYoBuAaJg9JdJfgr3akJWKodK2LlbrGFIM,1829
+digitalhub_runtime_python/entities/run/ray_job/spec.py,sha256=yWa3RjW03K8x7esa8ZZNqMTu7ODMnZ296phV0LysxtA,1964
+digitalhub_runtime_python/entities/run/ray_job/status.py,sha256=fNvKxkyYGh-SuNOZk_dL4vqhzrXdaNOi4hlyhY_jGu8,318
 digitalhub_runtime_python/entities/task/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/task/guardrail_build/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
 digitalhub_runtime_python/entities/task/guardrail_build/builder.py,sha256=TSPhUAvKv_bu7SOMNSRMR72KenDsHr8ynkR3iB-eJDo,1097
@@ -151,16 +166,26 @@ digitalhub_runtime_python/entities/task/python_serve/builder.py,sha256=brciGDBI0
 digitalhub_runtime_python/entities/task/python_serve/entity.py,sha256=UP9iLJJ5bwRymGdvmEq9RCSryfPaVLUl8N3iPBqhW2I,681
 digitalhub_runtime_python/entities/task/python_serve/spec.py,sha256=VDkEvgoxYv5O7v5n_xR2PQnRJW4fxPPNeEHZpV1kvBc,1504
 digitalhub_runtime_python/entities/task/python_serve/status.py,sha256=OqKVBAhiQDXJb_qogxV8qqh6eyPjudT4puXx8bg_6XU,298
+digitalhub_runtime_python/entities/task/ray_build/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
+digitalhub_runtime_python/entities/task/ray_build/builder.py,sha256=lmcVrWgrqE1bhYtfifBSbbcKdjVpVwx5A-yfBWHRhRE,988
+digitalhub_runtime_python/entities/task/ray_build/entity.py,sha256=ccalJFjpgtX3JkMvS9tM4OL_kMTkyh8dVnp6skGb59A,657
+digitalhub_runtime_python/entities/task/ray_build/spec.py,sha256=VCDMBdJiafe4z20XZbfZrRV5xPxeu9EJSPfw1GgZoNM,1103
+digitalhub_runtime_python/entities/task/ray_build/status.py,sha256=Ipmuz093ODy0gOXs-ZG-9qO8Bqtw6bmspbVc-Czt5Tk,292
+digitalhub_runtime_python/entities/task/ray_job/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
+digitalhub_runtime_python/entities/task/ray_job/builder.py,sha256=3kwmbaofzNdBySAHGg3AAlkMkXHngxKu0M3fkmQgEMQ,965
+digitalhub_runtime_python/entities/task/ray_job/entity.py,sha256=JklCYqnbseC7Ds0Zd76av7jM901WWd7QjDHDMI5diCo,641
+digitalhub_runtime_python/entities/task/ray_job/spec.py,sha256=fe1qDy4uiquidErcLEanUDWJ6AExgwBP8tJi81f9EN8,1394
+digitalhub_runtime_python/entities/task/ray_job/status.py,sha256=Ouc9v4pKQ93ECp5CjnWeedFoDLe3FxuhaPRhQiF1JT4,288
 digitalhub_runtime_python/runtimes/__init__.py,sha256=IRY2i9U97wptE3OcC_NO-EBmcAH6-Q36gMCspKetQ0k,107
-digitalhub_runtime_python/runtimes/builder.py,sha256=vfSyxh5iEkSmTngTWM2BMpcnyvtAagBJpbN0M7ICjKw,1370
-digitalhub_runtime_python/runtimes/hydra_runtime.py,sha256=lE2n19FoYwK4hg4DOmJ3AGm4WMDL3TvR-f-gC6UaH9Y,6334
-digitalhub_runtime_python/runtimes/runtime.py,sha256=kJgDfXk_UbXXryRw7LNAIETJJH5nB7IwjnJgYVBrBRI,3965
+digitalhub_runtime_python/runtimes/builder.py,sha256=dw6cR459hHqyaf4veC9MohOGBvpGBWJzSTmGR6CHNc0,1494
+digitalhub_runtime_python/runtimes/hydra_runtime.py,sha256=kC_C-OX6jlpwozheH0YVEQ6gNGBgkO0cm70AQNob28A,6386
+digitalhub_runtime_python/runtimes/runtime.py,sha256=IlcHcI_75-xMFIlFaly-QyVnjdHAiZLqJbh1axqaxyA,4038
 digitalhub_runtime_python/utils/configuration.py,sha256=1GyCOtl-mQ4lgv6UplpaB-wj1E-VoWChAC_G6HbYXzE,9177
 digitalhub_runtime_python/utils/inputs.py,sha256=c-dU61jHXrTd-5XJt4LacjfDzi_vWZFKwuJCbqAGdJQ,5806
 digitalhub_runtime_python/utils/outputs.py,sha256=irGUqy43F26NC1_ruH831NDyDTIVQHwIf15oXpR-q2M,9571
 digitalhub_runtime_python/utils/utils.py,sha256=qO-wh-9qJFvoJnWIWNeajtFQuBSYw3xhCI3JqV9gVHE,1936
-digitalhub_runtime_python-0.16.0b5.dist-info/METADATA,sha256=O1Qyswy5ugEHNL0LjxZG_DmDjkXocynFS0wOyU0NQNQ,17080
-digitalhub_runtime_python-0.16.0b5.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-digitalhub_runtime_python-0.16.0b5.dist-info/licenses/AUTHORS,sha256=iDleK_2EAMmh0o8Te_E9mdXQCP4rBfFr9dW9d0-pCno,301
-digitalhub_runtime_python-0.16.0b5.dist-info/licenses/LICENSE,sha256=z3xQCHfGmTnigfoUe3uidW_GUSVeFBdos30w9VNTRec,11361
-digitalhub_runtime_python-0.16.0b5.dist-info/RECORD,,
+digitalhub_runtime_python-0.16.0b7.dist-info/METADATA,sha256=AmUyYfESy88itgSrzQrrZpBJ2-oGHgCFyCKBYSIVfLI,17080
+digitalhub_runtime_python-0.16.0b7.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+digitalhub_runtime_python-0.16.0b7.dist-info/licenses/AUTHORS,sha256=iDleK_2EAMmh0o8Te_E9mdXQCP4rBfFr9dW9d0-pCno,301
+digitalhub_runtime_python-0.16.0b7.dist-info/licenses/LICENSE,sha256=z3xQCHfGmTnigfoUe3uidW_GUSVeFBdos30w9VNTRec,11361
+digitalhub_runtime_python-0.16.0b7.dist-info/RECORD,,