"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from .dedicateddatasetmodalitytype import DedicatedDatasetModalityType
from friendli.core.types import BaseModel
from typing import List, Optional
from typing_extensions import NotRequired, TypedDict


class DedicatedDatasetModalityTypedDict(TypedDict):
    """Dataset modality."""

    input_modals: NotRequired[List[DedicatedDatasetModalityType]]
    output_modals: NotRequired[List[DedicatedDatasetModalityType]]


class DedicatedDatasetModality(BaseModel):
    """Dataset modality."""

    input_modals: Optional[List[DedicatedDatasetModalityType]] = None
    output_modals: Optional[List[DedicatedDatasetModalityType]] = None
