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

from __future__ import annotations

from typing import Annotated

import pydantic
from typing_extensions import TypedDict

from friendli.core.types import BaseModel

from .dedicateddatasetmodality import (
    DedicatedDatasetModality,
    DedicatedDatasetModalityTypedDict,
)


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

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


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

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