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

from __future__ import annotations
from .dedicateddatasetmodality import (
    DedicatedDatasetModality,
    DedicatedDatasetModalityTypedDict,
)
from friendli.core.types import BaseModel
import pydantic
from typing_extensions import Annotated, TypedDict


class CreateDatasetRequestTypedDict(TypedDict):
    """Create dataset request."""

    name: str
    "Name of the dataset."
    project_id: str
    "ID of the project."
    modality: DedicatedDatasetModalityTypedDict
    "Dataset modality."


class CreateDatasetRequest(BaseModel):
    """Create dataset request."""

    name: str
    "Name of the dataset."
    project_id: Annotated[str, pydantic.Field(alias="projectId")]
    "ID of the project."
    modality: DedicatedDatasetModality
    "Dataset modality."
