nemo-platform-plugin 0.1.0__py3-none-any.whl

nemo_platform/types/intake/user_action_event_param.py

@@ -0,0 +1,68 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union
+from datetime import datetime
+from typing_extensions import Literal, Required, Annotated, TypedDict
+
+from ..._types import SequenceNotStr
+from ..._utils import PropertyInfo
+
+__all__ = ["UserActionEventParam"]
+
+
+class UserActionEventParam(TypedDict, total=False):
+    """Free-form user action captured by the client application.
+
+    Use this to track arbitrary user interactions with AI responses, such as copying code,
+    clicking share buttons, making purchases, or any other measurable action.
+
+    The action identifier must be ≤256 characters and should use snake-case or kebab-case
+    for consistency (e.g., ``share_clicked``, ``code_copied``, ``purchase_made``).
+    """
+
+    action: Required[str]
+    """
+    Descriptive name for the action taken by the user (e.g., 'share_clicked',
+    'code_copied', 'link_followed'). Use snake-case or kebab-case. This is a label,
+    not a unique identifier—multiple events can have the same action name.
+    """
+
+    id: str
+    """Unique identifier for the event. Populated when retrieved from database."""
+
+    created_at: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
+    """UTC timestamp when the record was created."""
+
+    created_by: Dict[str, str]
+    """Identifier of the user or system that generated the record.
+
+    Can be set of key-value pairs.
+    """
+
+    event_type: Literal["user_action"]
+
+    metadata: Dict[str, Union[str, SequenceNotStr[str], bool, float]]
+    """
+    Optional key-value pairs with additional context about the action (max 16
+    entries). Use this for details like user IDs, item IDs, timestamps, A/B test
+    variants, or any other information useful for downstream training or evaluation
+    pipelines. Example: {'user_id': '12345', 'experiment': 'variant_b',
+    'item_purchased': 'product_456'}.
+    """

nemo_platform/types/intake/user_feedback_event.py

@@ -0,0 +1,84 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Union, Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .thumb_direction import ThumbDirection
+
+__all__ = ["UserFeedbackEvent"]
+
+
+class UserFeedbackEvent(BaseModel):
+    """Structured feedback supplied by an end-user.
+
+    This event captures various forms of end-user feedback about a model's response,
+    including binary thumbs up/down ratings, numeric scores, free-text opinions,
+    suggested rewrites, and structured category ratings.
+
+    Either `thumb` or `rating` should be provided (they are mutually exclusive), but all
+    feedback fields are optional to accommodate different feedback collection patterns.
+    """
+
+    id: Optional[str] = None
+    """Unique identifier for the event. Populated when retrieved from database."""
+
+    categories: Optional[Dict[str, Union[float, str]]] = None
+    """Application-specific category ratings as key-value pairs.
+
+    Use this for custom rating dimensions (e.g., {'helpfulness': 4, 'accuracy': 5,
+    'tone': 'professional'}). Useful for radio buttons, dropdowns, or
+    multi-dimensional rating systems.
+    """
+
+    chosen_index: Optional[int] = None
+    """
+    Zero-based index of the response option the user selected when multiple
+    responses were returned. Use this when showing users multiple completion choices
+    and tracking which one they picked.
+    """
+
+    created_at: Optional[datetime] = None
+    """UTC timestamp when the record was created."""
+
+    created_by: Optional[Dict[str, str]] = None
+    """Identifier of the user or system that generated the record.
+
+    Can be set of key-value pairs.
+    """
+
+    event_type: Optional[Literal["user_feedback"]] = None
+
+    opinion: Optional[str] = None
+    """Free-text comment from the end user describing their opinion of the response."""
+
+    rating: Optional[float] = None
+    """Numeric rating (e.g., 1-5 stars) provided by the end user.
+
+    Mutually exclusive with `thumb`. Use this for star ratings or numeric scales.
+    """
+
+    rewrite: Optional[str] = None
+    """End-user's suggested text replacement for the generated response.
+
+    This is the user's idea of what the response should have been.
+    """
+
+    thumb: Optional[ThumbDirection] = None
+    """Possible thumb feedback choices."""

nemo_platform/types/intake/user_feedback_event_param.py

@@ -0,0 +1,86 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union
+from datetime import datetime
+from typing_extensions import Literal, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+from .thumb_direction import ThumbDirection
+
+__all__ = ["UserFeedbackEventParam"]
+
+
+class UserFeedbackEventParam(TypedDict, total=False):
+    """Structured feedback supplied by an end-user.
+
+    This event captures various forms of end-user feedback about a model's response,
+    including binary thumbs up/down ratings, numeric scores, free-text opinions,
+    suggested rewrites, and structured category ratings.
+
+    Either `thumb` or `rating` should be provided (they are mutually exclusive), but all
+    feedback fields are optional to accommodate different feedback collection patterns.
+    """
+
+    id: str
+    """Unique identifier for the event. Populated when retrieved from database."""
+
+    categories: Dict[str, Union[float, str]]
+    """Application-specific category ratings as key-value pairs.
+
+    Use this for custom rating dimensions (e.g., {'helpfulness': 4, 'accuracy': 5,
+    'tone': 'professional'}). Useful for radio buttons, dropdowns, or
+    multi-dimensional rating systems.
+    """
+
+    chosen_index: int
+    """
+    Zero-based index of the response option the user selected when multiple
+    responses were returned. Use this when showing users multiple completion choices
+    and tracking which one they picked.
+    """
+
+    created_at: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
+    """UTC timestamp when the record was created."""
+
+    created_by: Dict[str, str]
+    """Identifier of the user or system that generated the record.
+
+    Can be set of key-value pairs.
+    """
+
+    event_type: Literal["user_feedback"]
+
+    opinion: str
+    """Free-text comment from the end user describing their opinion of the response."""
+
+    rating: float
+    """Numeric rating (e.g., 1-5 stars) provided by the end user.
+
+    Mutually exclusive with `thumb`. Use this for star ratings or numeric scales.
+    """
+
+    rewrite: str
+    """End-user's suggested text replacement for the generated response.
+
+    This is the user's idea of what the response should have been.
+    """
+
+    thumb: ThumbDirection
+    """Possible thumb feedback choices."""

nemo_platform/types/intake/user_rating.py

@@ -0,0 +1,68 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Union, Optional
+
+from ..._models import BaseModel
+from .thumb_direction import ThumbDirection
+
+__all__ = ["UserRating"]
+
+
+class UserRating(BaseModel):
+    """User's rating/evaluation of an AI response.
+
+    This captures various forms of end-user feedback about a model's response, including
+    binary thumbs up/down ratings, numeric scores, free-text opinions, suggested rewrites,
+    and structured category ratings.
+
+    Either `thumb` or `rating` should be provided (they are mutually exclusive), but all
+    fields are optional to accommodate different feedback collection patterns.
+    """
+
+    categories: Optional[Dict[str, Union[float, str]]] = None
+    """Application-specific category ratings as key-value pairs.
+
+    Use this for custom rating dimensions (e.g., {'helpfulness': 4, 'accuracy': 5,
+    'tone': 'professional'}). Useful for radio buttons, dropdowns, or
+    multi-dimensional rating systems.
+    """
+
+    chosen_index: Optional[int] = None
+    """
+    Zero-based index of the response option the user selected when multiple
+    responses were returned. Use this when showing users multiple completion choices
+    and tracking which one they picked.
+    """
+
+    opinion: Optional[str] = None
+    """Free-text comment from the end user describing their opinion of the response."""
+
+    rating: Optional[float] = None
+    """Numeric rating (e.g., 1-5 stars) provided by the end user.
+
+    Mutually exclusive with `thumb`. Use this for star ratings or numeric scales.
+    """
+
+    rewrite: Optional[str] = None
+    """End-user's suggested text replacement for the generated response.
+
+    This is the user's idea of what the response should have been.
+    """
+
+    thumb: Optional[ThumbDirection] = None
+    """Possible thumb feedback choices."""

nemo_platform/types/intake/user_rating_param.py

@@ -0,0 +1,70 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union
+from typing_extensions import TypedDict
+
+from .thumb_direction import ThumbDirection
+
+__all__ = ["UserRatingParam"]
+
+
+class UserRatingParam(TypedDict, total=False):
+    """User's rating/evaluation of an AI response.
+
+    This captures various forms of end-user feedback about a model's response, including
+    binary thumbs up/down ratings, numeric scores, free-text opinions, suggested rewrites,
+    and structured category ratings.
+
+    Either `thumb` or `rating` should be provided (they are mutually exclusive), but all
+    fields are optional to accommodate different feedback collection patterns.
+    """
+
+    categories: Dict[str, Union[float, str]]
+    """Application-specific category ratings as key-value pairs.
+
+    Use this for custom rating dimensions (e.g., {'helpfulness': 4, 'accuracy': 5,
+    'tone': 'professional'}). Useful for radio buttons, dropdowns, or
+    multi-dimensional rating systems.
+    """
+
+    chosen_index: int
+    """
+    Zero-based index of the response option the user selected when multiple
+    responses were returned. Use this when showing users multiple completion choices
+    and tracking which one they picked.
+    """
+
+    opinion: str
+    """Free-text comment from the end user describing their opinion of the response."""
+
+    rating: float
+    """Numeric rating (e.g., 1-5 stars) provided by the end user.
+
+    Mutually exclusive with `thumb`. Use this for star ratings or numeric scales.
+    """
+
+    rewrite: str
+    """End-user's suggested text replacement for the generated response.
+
+    This is the user's idea of what the response should have been.
+    """
+
+    thumb: ThumbDirection
+    """Possible thumb feedback choices."""

nemo_platform/types/jobs/__init__.py

@@ -0,0 +1,95 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .container_spec import ContainerSpec as ContainerSpec
+from .step_lifecycle import StepLifecycle as StepLifecycle
+from .job_list_params import JobListParams as JobListParams
+from .step_list_params import StepListParams as StepListParams
+from .compute_resources import ComputeResources as ComputeResources
+from .image_pull_secret import ImagePullSecret as ImagePullSecret
+from .job_create_params import JobCreateParams as JobCreateParams
+from .kubernetes_volume import KubernetesVolume as KubernetesVolume
+from .platform_job_spec import PlatformJobSpec as PlatformJobSpec
+from .platform_job_step import PlatformJobStep as PlatformJobStep
+from .platform_job_task import PlatformJobTask as PlatformJobTask
+from .result_list_params import ResultListParams as ResultListParams
+from .docker_volume_mount import DockerVolumeMount as DockerVolumeMount
+from .job_get_logs_params import JobGetLogsParams as JobGetLogsParams
+from .container_spec_param import ContainerSpecParam as ContainerSpecParam
+from .result_create_params import ResultCreateParams as ResultCreateParams
+from .step_lifecycle_param import StepLifecycleParam as StepLifecycleParam
+from .compute_resource_spec import ComputeResourceSpec as ComputeResourceSpec
+from .platform_job_response import PlatformJobResponse as PlatformJobResponse
+from .cpu_execution_provider import CPUExecutionProvider as CPUExecutionProvider
+from .gpu_execution_provider import GPUExecutionProvider as GPUExecutionProvider
+from .platform_job_step_spec import PlatformJobStepSpec as PlatformJobStepSpec
+from .compute_resources_param import ComputeResourcesParam as ComputeResourcesParam
+from .kubernetes_volume_mount import KubernetesVolumeMount as KubernetesVolumeMount
+from .platform_job_sort_field import PlatformJobSortField as PlatformJobSortField
+from .platform_job_spec_param import PlatformJobSpecParam as PlatformJobSpecParam
+from .docker_job_network_config import DockerJobNetworkConfig as DockerJobNetworkConfig
+from .docker_job_storage_config import DockerJobStorageConfig as DockerJobStorageConfig
+from .e2e_job_execution_profile import E2EJobExecutionProfile as E2EJobExecutionProfile
+from .step_update_status_params import StepUpdateStatusParams as StepUpdateStatusParams
+from .kubernetes_object_metadata import KubernetesObjectMetadata as KubernetesObjectMetadata
+from .compute_resource_spec_param import ComputeResourceSpecParam as ComputeResourceSpecParam
+from .kubernetes_empty_dir_volume import KubernetesEmptyDirVolume as KubernetesEmptyDirVolume
+from .platform_job_responses_page import PlatformJobResponsesPage as PlatformJobResponsesPage
+from .cpu_execution_provider_param import CPUExecutionProviderParam as CPUExecutionProviderParam
+from .docker_job_execution_profile import DockerJobExecutionProfile as DockerJobExecutionProfile
+from .gpu_execution_provider_param import GPUExecutionProviderParam as GPUExecutionProviderParam
+from .job_execution_profile_config import JobExecutionProfileConfig as JobExecutionProfileConfig
+from .platform_job_step_spec_param import PlatformJobStepSpecParam as PlatformJobStepSpecParam
+from .task_create_or_update_params import TaskCreateOrUpdateParams as TaskCreateOrUpdateParams
+from .kubernetes_job_storage_config import KubernetesJobStorageConfig as KubernetesJobStorageConfig
+from .subprocess_execution_provider import SubprocessExecutionProvider as SubprocessExecutionProvider
+from .volcano_job_execution_profile import VolcanoJobExecutionProfile as VolcanoJobExecutionProfile
+from .platform_job_step_with_context import PlatformJobStepWithContext as PlatformJobStepWithContext
+from .platform_job_list_task_response import PlatformJobListTaskResponse as PlatformJobListTaskResponse
+from .platform_jobs_list_filter_param import PlatformJobsListFilterParam as PlatformJobsListFilterParam
+from .job_update_status_details_params import JobUpdateStatusDetailsParams as JobUpdateStatusDetailsParams
+from .kubernetes_job_execution_profile import KubernetesJobExecutionProfile as KubernetesJobExecutionProfile
+from .subprocess_job_execution_profile import SubprocessJobExecutionProfile as SubprocessJobExecutionProfile
+from .platform_job_environment_variable import PlatformJobEnvironmentVariable as PlatformJobEnvironmentVariable
+from .distributed_gpu_execution_provider import DistributedGPUExecutionProvider as DistributedGPUExecutionProvider
+from .kubernetes_persistent_volume_claim import KubernetesPersistentVolumeClaim as KubernetesPersistentVolumeClaim
+from .docker_job_execution_profile_config import DockerJobExecutionProfileConfig as DockerJobExecutionProfileConfig
+from .subprocess_execution_provider_param import SubprocessExecutionProviderParam as SubprocessExecutionProviderParam
+from .job_list_execution_profiles_response import JobListExecutionProfilesResponse as JobListExecutionProfilesResponse
+from .platform_job_step_with_contexts_page import PlatformJobStepWithContextsPage as PlatformJobStepWithContextsPage
+from .platform_job_steps_list_filter_param import PlatformJobStepsListFilterParam as PlatformJobStepsListFilterParam
+from .volcano_job_execution_profile_config import VolcanoJobExecutionProfileConfig as VolcanoJobExecutionProfileConfig
+from .kubernetes_job_execution_profile_config import (
+    KubernetesJobExecutionProfileConfig as KubernetesJobExecutionProfileConfig,
+)
+from .platform_job_environment_variable_param import (
+    PlatformJobEnvironmentVariableParam as PlatformJobEnvironmentVariableParam,
+)
+from .subprocess_job_execution_profile_config import (
+    SubprocessJobExecutionProfileConfig as SubprocessJobExecutionProfileConfig,
+)
+from .distributed_gpu_execution_provider_param import (
+    DistributedGPUExecutionProviderParam as DistributedGPUExecutionProviderParam,
+)
+from .platform_job_secret_environment_variable_ref import (
+    PlatformJobSecretEnvironmentVariableRef as PlatformJobSecretEnvironmentVariableRef,
+)
+from .platform_job_secret_environment_variable_ref_param import (
+    PlatformJobSecretEnvironmentVariableRefParam as PlatformJobSecretEnvironmentVariableRefParam,
+)

nemo_platform/types/jobs/compute_resource_spec.py

@@ -0,0 +1,32 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["ComputeResourceSpec"]
+
+
+class ComputeResourceSpec(BaseModel):
+    """Resource specification."""
+
+    cpu: Optional[str] = None
+    """CPU specification (e.g., '250m', '1', '2.5')."""
+
+    memory: Optional[str] = None
+    """Memory specification (e.g., '128Mi', '1Gi', '512M')."""

nemo_platform/types/jobs/compute_resource_spec_param.py

@@ -0,0 +1,32 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["ComputeResourceSpecParam"]
+
+
+class ComputeResourceSpecParam(TypedDict, total=False):
+    """Resource specification."""
+
+    cpu: str
+    """CPU specification (e.g., '250m', '1', '2.5')."""
+
+    memory: str
+    """Memory specification (e.g., '128Mi', '1Gi', '512M')."""

nemo_platform/types/jobs/compute_resources.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+from .compute_resource_spec import ComputeResourceSpec
+
+__all__ = ["ComputeResources"]
+
+
+class ComputeResources(BaseModel):
+    """Resource requirements matching k8s ResourceRequirements format."""
+
+    limits: Optional[ComputeResourceSpec] = None
+    """Resource specification."""
+
+    num_gpus: Optional[int] = None
+    """Step requesting number of GPUs."""
+
+    num_nodes: Optional[int] = None
+    """Number of nodes to use."""
+
+    requests: Optional[ComputeResourceSpec] = None
+    """Resource specification."""
+
+    shm_size: Optional[str] = None
+    """Shared memory (/dev/shm) size as a Kubernetes quantity (e.g.
+
+    '1Gi', '4Gi'). Used for GPU and distributed-GPU job executors. When unset,
+    defaults to 1Gi per allocated GPU.
+    """

nemo_platform/types/jobs/compute_resources_param.py

@@ -0,0 +1,47 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+from .compute_resource_spec_param import ComputeResourceSpecParam
+
+__all__ = ["ComputeResourcesParam"]
+
+
+class ComputeResourcesParam(TypedDict, total=False):
+    """Resource requirements matching k8s ResourceRequirements format."""
+
+    limits: ComputeResourceSpecParam
+    """Resource specification."""
+
+    num_gpus: int
+    """Step requesting number of GPUs."""
+
+    num_nodes: int
+    """Number of nodes to use."""
+
+    requests: ComputeResourceSpecParam
+    """Resource specification."""
+
+    shm_size: str
+    """Shared memory (/dev/shm) size as a Kubernetes quantity (e.g.
+
+    '1Gi', '4Gi'). Used for GPU and distributed-GPU job executors. When unset,
+    defaults to 1Gi per allocated GPU.
+    """

nemo_platform/types/jobs/container_spec.py

@@ -0,0 +1,35 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from ..._models import BaseModel
+
+__all__ = ["ContainerSpec"]
+
+
+class ContainerSpec(BaseModel):
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    image: str
+
+    command: Optional[List[str]] = None
+
+    entrypoint: Optional[List[str]] = None