digitalhub-runtime-python 0.15.0b8__tar.gz → 0.15.0b10__tar.gz

{digitalhub_runtime_python-0.15.0b8 → digitalhub_runtime_python-0.15.0b10}/.github/workflows/release.yml

@@ -31,7 +31,7 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.12"
 
       - name: Install bumpver
         run: python -m pip install bumpver
@@ -67,7 +67,7 @@ jobs:
           ref: ${{ github.ref }}
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.12"
 
       # Build wheel to upload
       - name: Install build and build a binary wheel
@@ -91,7 +91,7 @@ jobs:
 
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.12"
 
       - name: Install cliff
         run: python -m pip install git-cliff

{digitalhub_runtime_python-0.15.0b8 → digitalhub_runtime_python-0.15.0b10}/.github/workflows/run_local_test.yml

@@ -7,7 +7,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12", "3.13"]
+        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
 
     steps:
       - uses: actions/checkout@v4

{digitalhub_runtime_python-0.15.0b8 → digitalhub_runtime_python-0.15.0b10}/.github/workflows/test_release.yml

@@ -31,7 +31,7 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.12"
 
       - name: Install bumpver
         run: python -m pip install bumpver
@@ -59,7 +59,7 @@ jobs:
 
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.12"
 
       - name: Install cliff
         run: python -m pip install git-cliff

{digitalhub_runtime_python-0.15.0b8 → digitalhub_runtime_python-0.15.0b10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalhub-runtime-python
-Version: 0.15.0b8
+Version: 0.15.0b10
 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>
@@ -213,7 +213,8 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: <3.14,>=3.10
+Classifier: Programming Language :: Python :: 3.14
+Requires-Python: <3.15,>=3.10
 Requires-Dist: digitalhub[full]<0.16,>=0.15.0b
 Requires-Dist: msgpack
 Requires-Dist: nuclio-sdk

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/__init__.py

@@ -0,0 +1,76 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.function.guardrail.builder import FunctionGuardrailBuilder
+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.run.guardrail_build.builder import RunGuardrailRunBuildBuilder
+from digitalhub_runtime_python.entities.run.guardrail_serve.builder import RunGuardrailRunServeBuilder
+from digitalhub_runtime_python.entities.run.openinference_build.builder import RunOpeninferenceRunBuildBuilder
+from digitalhub_runtime_python.entities.run.openinference_serve.builder import RunOpeninferenceRunServeBuilder
+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.task.guardrail_build.builder import TaskGuardrailBuildBuilder
+from digitalhub_runtime_python.entities.task.guardrail_serve.builder import TaskGuardrailServeBuilder
+from digitalhub_runtime_python.entities.task.openinference_build.builder import TaskOpeninferenceBuildBuilder
+from digitalhub_runtime_python.entities.task.openinference_serve.builder import TaskOpeninferenceServeBuilder
+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
+
+entity_builders = (
+    (EntityKinds.FUNCTION_GUARDRAIL.value, FunctionGuardrailBuilder),
+    (EntityKinds.FUNCTION_OPENINFERENCE.value, FunctionOpeninferenceBuilder),
+    (EntityKinds.FUNCTION_PYTHON.value, FunctionPythonBuilder),
+    (EntityKinds.RUN_GUARDRAIL_BUILD.value, RunGuardrailRunBuildBuilder),
+    (EntityKinds.RUN_GUARDRAIL_SERVE.value, RunGuardrailRunServeBuilder),
+    (EntityKinds.RUN_OPENINFERENCE_BUILD.value, RunOpeninferenceRunBuildBuilder),
+    (EntityKinds.RUN_OPENINFERENCE_SERVE.value, RunOpeninferenceRunServeBuilder),
+    (EntityKinds.RUN_PYTHON_BUILD.value, RunPythonRunBuildBuilder),
+    (EntityKinds.RUN_PYTHON_JOB.value, RunPythonRunJobBuilder),
+    (EntityKinds.RUN_PYTHON_SERVE.value, RunPythonRunServeBuilder),
+    (EntityKinds.TASK_GUARDRAIL_BUILD.value, TaskGuardrailBuildBuilder),
+    (EntityKinds.TASK_GUARDRAIL_SERVE.value, TaskGuardrailServeBuilder),
+    (EntityKinds.TASK_OPENINFERENCE_BUILD.value, TaskOpeninferenceBuildBuilder),
+    (EntityKinds.TASK_OPENINFERENCE_SERVE.value, TaskOpeninferenceServeBuilder),
+    (EntityKinds.TASK_PYTHON_BUILD.value, TaskPythonBuildBuilder),
+    (EntityKinds.TASK_PYTHON_JOB.value, TaskPythonJobBuilder),
+    (EntityKinds.TASK_PYTHON_SERVE.value, TaskPythonServeBuilder),
+)
+
+try:
+    from digitalhub_runtime_python.runtimes.builder import (
+        RuntimeGuardrailBuilder,
+        RuntimeOpeninferenceBuilder,
+        RuntimePythonBuilder,
+        RuntimePythonJobBuilder,
+    )
+
+    runtime_builders = (
+        (EntityKinds.FUNCTION_GUARDRAIL.value, RuntimeGuardrailBuilder),
+        (EntityKinds.FUNCTION_OPENINFERENCE.value, RuntimeOpeninferenceBuilder),
+        (EntityKinds.FUNCTION_PYTHON.value, RuntimePythonBuilder),
+        (EntityKinds.RUN_GUARDRAIL_BUILD.value, RuntimeGuardrailBuilder),
+        (EntityKinds.RUN_GUARDRAIL_SERVE.value, RuntimeGuardrailBuilder),
+        (EntityKinds.RUN_OPENINFERENCE_BUILD.value, RuntimeOpeninferenceBuilder),
+        (EntityKinds.RUN_OPENINFERENCE_SERVE.value, RuntimeOpeninferenceBuilder),
+        (EntityKinds.RUN_PYTHON_BUILD.value, RuntimePythonBuilder),
+        (EntityKinds.RUN_PYTHON_JOB.value, RuntimePythonJobBuilder),
+        (EntityKinds.RUN_PYTHON_SERVE.value, RuntimePythonBuilder),
+        (EntityKinds.TASK_GUARDRAIL_BUILD.value, RuntimeGuardrailBuilder),
+        (EntityKinds.TASK_GUARDRAIL_SERVE.value, RuntimeGuardrailBuilder),
+        (EntityKinds.TASK_OPENINFERENCE_BUILD.value, RuntimeOpeninferenceBuilder),
+        (EntityKinds.TASK_OPENINFERENCE_SERVE.value, RuntimeOpeninferenceBuilder),
+        (EntityKinds.TASK_PYTHON_BUILD.value, RuntimePythonBuilder),
+        (EntityKinds.TASK_PYTHON_JOB.value, RuntimePythonJobBuilder),
+        (EntityKinds.TASK_PYTHON_SERVE.value, RuntimePythonBuilder),
+    )
+except ImportError as e:
+    from digitalhub.utils.logger.logger import get_logger
+
+    logger = get_logger(__name__)
+    logger.debug(f"Error importing runtime builders: {e}")
+    runtime_builders = tuple()

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/_base/runtime_entity/builder.py

@@ -0,0 +1,102 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities._base.runtime_entity.builder import RuntimeEntityBuilder
+from digitalhub.entities._commons.utils import map_actions
+
+from digitalhub_runtime_python.entities._commons.enums import Actions, EntityKinds
+
+
+class RuntimeEntityBuilderPython(RuntimeEntityBuilder):
+    EXECUTABLE_KIND = EntityKinds.FUNCTION_PYTHON.value
+    TASKS_KINDS = map_actions(
+        [
+            (
+                EntityKinds.TASK_PYTHON_JOB.value,
+                Actions.JOB.value,
+            ),
+            (
+                EntityKinds.TASK_PYTHON_BUILD.value,
+                Actions.BUILD.value,
+            ),
+            (
+                EntityKinds.TASK_PYTHON_SERVE.value,
+                Actions.SERVE.value,
+            ),
+        ]
+    )
+    RUN_KINDS = map_actions(
+        [
+            (
+                EntityKinds.RUN_PYTHON_JOB.value,
+                Actions.JOB.value,
+            ),
+            (
+                EntityKinds.RUN_PYTHON_BUILD.value,
+                Actions.BUILD.value,
+            ),
+            (
+                EntityKinds.RUN_PYTHON_SERVE.value,
+                Actions.SERVE.value,
+            ),
+        ]
+    )
+
+
+class RuntimeEntityBuilderOpeninference(RuntimeEntityBuilder):
+    EXECUTABLE_KIND = EntityKinds.FUNCTION_OPENINFERENCE.value
+    TASKS_KINDS = map_actions(
+        [
+            (
+                EntityKinds.TASK_OPENINFERENCE_BUILD.value,
+                Actions.BUILD.value,
+            ),
+            (
+                EntityKinds.TASK_OPENINFERENCE_SERVE.value,
+                Actions.SERVE.value,
+            ),
+        ]
+    )
+    RUN_KINDS = map_actions(
+        [
+            (
+                EntityKinds.RUN_OPENINFERENCE_BUILD.value,
+                Actions.BUILD.value,
+            ),
+            (
+                EntityKinds.RUN_OPENINFERENCE_SERVE.value,
+                Actions.SERVE.value,
+            ),
+        ]
+    )
+
+
+class RuntimeEntityBuilderGuardrail(RuntimeEntityBuilder):
+    EXECUTABLE_KIND = EntityKinds.FUNCTION_GUARDRAIL.value
+    TASKS_KINDS = map_actions(
+        [
+            (
+                EntityKinds.TASK_GUARDRAIL_BUILD.value,
+                Actions.BUILD.value,
+            ),
+            (
+                EntityKinds.TASK_GUARDRAIL_SERVE.value,
+                Actions.SERVE.value,
+            ),
+        ]
+    )
+    RUN_KINDS = map_actions(
+        [
+            (
+                EntityKinds.RUN_GUARDRAIL_BUILD.value,
+                Actions.BUILD.value,
+            ),
+            (
+                EntityKinds.RUN_GUARDRAIL_SERVE.value,
+                Actions.SERVE.value,
+            ),
+        ]
+    )

{digitalhub_runtime_python-0.15.0b8 → digitalhub_runtime_python-0.15.0b10}/digitalhub_runtime_python/entities/_commons/enums.py

@@ -20,6 +20,18 @@ class EntityKinds(Enum):
     RUN_PYTHON_JOB = "python+job:run"
     RUN_PYTHON_SERVE = "python+serve:run"
 
+    FUNCTION_OPENINFERENCE = "openinference"
+    TASK_OPENINFERENCE_BUILD = "openinference+build"
+    TASK_OPENINFERENCE_SERVE = "openinference+serve"
+    RUN_OPENINFERENCE_BUILD = "openinference+build:run"
+    RUN_OPENINFERENCE_SERVE = "openinference+serve:run"
+
+    FUNCTION_GUARDRAIL = "guardrail"
+    TASK_GUARDRAIL_BUILD = "guardrail+build"
+    TASK_GUARDRAIL_SERVE = "guardrail+serve"
+    RUN_GUARDRAIL_BUILD = "guardrail+build:run"
+    RUN_GUARDRAIL_SERVE = "guardrail+serve:run"
+
 
 class Actions(Enum):
     """

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/function/guardrail/builder.py

@@ -0,0 +1,93 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.function._base.builder import FunctionBuilder
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderGuardrail
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.function.guardrail.entity import FunctionGuardrail
+from digitalhub_runtime_python.entities.function.guardrail.spec import (
+    FunctionSpecGuardrail,
+    FunctionValidatorGuardrail,
+)
+from digitalhub_runtime_python.entities.function.guardrail.status import FunctionStatusGuardrail
+from digitalhub_runtime_python.entities.function.python.utils import source_check, source_post_check
+
+
+class FunctionGuardrailBuilder(FunctionBuilder, RuntimeEntityBuilderGuardrail):
+    """
+    FunctionGuardrail builder.
+    """
+
+    ENTITY_CLASS = FunctionGuardrail
+    ENTITY_SPEC_CLASS = FunctionSpecGuardrail
+    ENTITY_SPEC_VALIDATOR = FunctionValidatorGuardrail
+    ENTITY_STATUS_CLASS = FunctionStatusGuardrail
+    ENTITY_KIND = EntityKinds.FUNCTION_GUARDRAIL.value
+
+    def build(
+        self,
+        kind: str,
+        project: str,
+        name: str,
+        uuid: str | None = None,
+        description: str | None = None,
+        labels: list[str] | None = None,
+        embedded: bool = False,
+        **kwargs,
+    ) -> FunctionGuardrail:
+        kwargs = source_check(**kwargs)
+        obj = super().build(
+            kind,
+            project,
+            name,
+            uuid,
+            description,
+            labels,
+            embedded,
+            **kwargs,
+        )
+        return source_post_check(obj)
+
+    def from_dict(self, obj: dict) -> FunctionGuardrail:
+        """
+        Create a new object from dictionary.
+
+        Parameters
+        ----------
+        obj : dict
+            Dictionary to create object from.
+
+        Returns
+        -------
+        FunctionGuardrail
+            Object instance.
+        """
+        entity = super().from_dict(obj)
+        return source_post_check(entity)
+
+    def _parse_dict(self, obj: dict) -> dict:
+        """
+        Get dictionary and parse it to a valid entity dictionary.
+
+        Parameters
+        ----------
+        obj : dict
+            Dictionary to parse.
+
+        Returns
+        -------
+        dict
+            A dictionary containing the attributes of the entity instance.
+        """
+        # Look for source in spec
+        if spec_dict := obj.get("spec", {}):
+            # Check source
+            source = spec_dict.get("source", {})
+            if source:
+                spec_dict["source"] = source_check(source=source)["source"]
+
+        return super()._parse_dict(obj)

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/function/guardrail/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.entities.function._base.entity import Function
+from digitalhub.utils.generic_utils import decode_base64_string
+from digitalhub.utils.io_utils import write_text
+from digitalhub.utils.uri_utils import has_local_scheme
+
+from digitalhub_runtime_python.entities.function.python.utils import read_installed_packages
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.function.guardrail.spec import FunctionSpecGuardrail
+    from digitalhub_runtime_python.entities.function.guardrail.status import FunctionStatusGuardrail
+
+
+class FunctionGuardrail(Function):
+    """
+    FunctionGuardrail class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: FunctionSpecGuardrail
+        self.status: FunctionStatusGuardrail
+
+    def export(self) -> str:
+        """
+        Export object as a YAML file in the context folder.
+
+        Returns
+        -------
+        str
+            Exported filepath.
+        """
+        # Strip base64 from source at following conditions:
+        # - source is local path
+        # - base64 is not None
+
+        # Check source
+        source = self.spec.source.get("source")
+        if source is not None and has_local_scheme(source):
+            # Check base64. If it is set, decode it in a local file
+            # save in variable to restore on object after export
+            base64 = self.spec.source.pop("base64", None)
+            if base64 is not None:
+                # Write local file
+                src_pth = self._context().root / source
+                write_text(src_pth, decode_base64_string(base64))
+
+                # Export and restore base64, then return
+                pth = super().export()
+                self.spec.source["base64"] = base64
+                return pth
+
+        return super().export()
+
+    def _post_create_hook_before_save(self) -> None:
+        """
+        Hook method called after the creation of the entity but before saving
+        in Core.
+        Can be overridden in subclasses to implement custom behavior.
+        """
+        self.spec.requirements = read_installed_packages(self.spec.requirements)

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/function/guardrail/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.function._base.spec import FunctionSpec, FunctionValidator
+
+from digitalhub_runtime_python.entities.function.python.models import PythonVersion, SourceValidator
+
+
+class FunctionSpecGuardrail(FunctionSpec):
+    """
+    FunctionSpecGuardrail 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,
+    ) -> None:
+        super().__init__()
+
+        self.image = image
+        self.base_image = base_image
+        self.python_version = python_version
+        self.requirements = requirements
+        self.source = source
+
+
+class FunctionValidatorGuardrail(FunctionValidator):
+    """
+    FunctionValidatorGuardrail 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] | None = None
+    "Requirements list to be installed in the image where the function will be executed"

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/function/guardrail/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 FunctionStatusGuardrail(FunctionStatus):
+    """
+    FunctionStatusGuardrail status.
+    """

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/function/openinference/builder.py

@@ -0,0 +1,93 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from digitalhub.entities.function._base.builder import FunctionBuilder
+
+from digitalhub_runtime_python.entities._base.runtime_entity.builder import RuntimeEntityBuilderOpeninference
+from digitalhub_runtime_python.entities._commons.enums import EntityKinds
+from digitalhub_runtime_python.entities.function.openinference.entity import FunctionOpeninference
+from digitalhub_runtime_python.entities.function.openinference.spec import (
+    FunctionSpecOpeninference,
+    FunctionValidatorOpeninference,
+)
+from digitalhub_runtime_python.entities.function.openinference.status import FunctionStatusOpeninference
+from digitalhub_runtime_python.entities.function.python.utils import source_check, source_post_check
+
+
+class FunctionOpeninferenceBuilder(FunctionBuilder, RuntimeEntityBuilderOpeninference):
+    """
+    FunctionOpeninference builder.
+    """
+
+    ENTITY_CLASS = FunctionOpeninference
+    ENTITY_SPEC_CLASS = FunctionSpecOpeninference
+    ENTITY_SPEC_VALIDATOR = FunctionValidatorOpeninference
+    ENTITY_STATUS_CLASS = FunctionStatusOpeninference
+    ENTITY_KIND = EntityKinds.FUNCTION_OPENINFERENCE.value
+
+    def build(
+        self,
+        kind: str,
+        project: str,
+        name: str,
+        uuid: str | None = None,
+        description: str | None = None,
+        labels: list[str] | None = None,
+        embedded: bool = False,
+        **kwargs,
+    ) -> FunctionOpeninference:
+        kwargs = source_check(**kwargs)
+        obj = super().build(
+            kind,
+            project,
+            name,
+            uuid,
+            description,
+            labels,
+            embedded,
+            **kwargs,
+        )
+        return source_post_check(obj)
+
+    def from_dict(self, obj: dict) -> FunctionOpeninference:
+        """
+        Create a new object from dictionary.
+
+        Parameters
+        ----------
+        obj : dict
+            Dictionary to create object from.
+
+        Returns
+        -------
+        FunctionOpeninference
+            Object instance.
+        """
+        entity = super().from_dict(obj)
+        return source_post_check(entity)
+
+    def _parse_dict(self, obj: dict) -> dict:
+        """
+        Get dictionary and parse it to a valid entity dictionary.
+
+        Parameters
+        ----------
+        obj : dict
+            Dictionary to parse.
+
+        Returns
+        -------
+        dict
+            A dictionary containing the attributes of the entity instance.
+        """
+        # Look for source in spec
+        if spec_dict := obj.get("spec", {}):
+            # Check source
+            source = spec_dict.get("source", {})
+            if source:
+                spec_dict["source"] = source_check(source=source)["source"]
+
+        return super()._parse_dict(obj)

digitalhub_runtime_python-0.15.0b10/digitalhub_runtime_python/entities/function/openinference/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.entities.function._base.entity import Function
+from digitalhub.utils.generic_utils import decode_base64_string
+from digitalhub.utils.io_utils import write_text
+from digitalhub.utils.uri_utils import has_local_scheme
+
+from digitalhub_runtime_python.entities.function.python.utils import read_installed_packages
+
+if typing.TYPE_CHECKING:
+    from digitalhub_runtime_python.entities.function.openinference.spec import FunctionSpecOpeninference
+    from digitalhub_runtime_python.entities.function.openinference.status import FunctionStatusOpeninference
+
+
+class FunctionOpeninference(Function):
+    """
+    FunctionOpeninference class.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.spec: FunctionSpecOpeninference
+        self.status: FunctionStatusOpeninference
+
+    def export(self) -> str:
+        """
+        Export object as a YAML file in the context folder.
+
+        Returns
+        -------
+        str
+            Exported filepath.
+        """
+        # Strip base64 from source at following conditions:
+        # - source is local path
+        # - base64 is not None
+
+        # Check source
+        source = self.spec.source.get("source")
+        if source is not None and has_local_scheme(source):
+            # Check base64. If it is set, decode it in a local file
+            # save in variable to restore on object after export
+            base64 = self.spec.source.pop("base64", None)
+            if base64 is not None:
+                # Write local file
+                src_pth = self._context().root / source
+                write_text(src_pth, decode_base64_string(base64))
+
+                # Export and restore base64, then return
+                pth = super().export()
+                self.spec.source["base64"] = base64
+                return pth
+
+        return super().export()
+
+    def _post_create_hook_before_save(self) -> None:
+        """
+        Hook method called after the creation of the entity but before saving
+        in Core.
+        Can be overridden in subclasses to implement custom behavior.
+        """
+        self.spec.requirements = read_installed_packages(self.spec.requirements)