digitalhub-runtime-python 0.16.0b6__tar.gz → 0.16.0b7__tar.gz

{digitalhub_runtime_python-0.16.0b6 → digitalhub_runtime_python-0.16.0b7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: digitalhub-runtime-python
-Version: 0.16.0b6
+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.0b6 → digitalhub_runtime_python-0.16.0b7}/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-0.16.0b6 → digitalhub_runtime_python-0.16.0b7}/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-0.16.0b6 → digitalhub_runtime_python-0.16.0b7}/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/digitalhub_runtime_python/entities/task/python_job/__init__.py

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

digitalhub_runtime_python-0.16.0b7/digitalhub_runtime_python/entities/task/python_serve/__init__.py

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

digitalhub_runtime_python-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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-0.16.0b7/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