# Generated by ariadne-codegen
# Source: tools/graphql_codegen/artifacts/

from __future__ import annotations

from typing import Literal, Optional

from pydantic import Field

from wandb._pydantic import GQLBase, GQLId, Typename


class UpdateArtifactPortfolio(GQLBase):
    update_artifact_portfolio: Optional[
        UpdateArtifactPortfolioUpdateArtifactPortfolio
    ] = Field(alias="updateArtifactPortfolio")


class UpdateArtifactPortfolioUpdateArtifactPortfolio(GQLBase):
    artifact_collection: UpdateArtifactPortfolioUpdateArtifactPortfolioArtifactCollection = Field(
        alias="artifactCollection"
    )


class UpdateArtifactPortfolioUpdateArtifactPortfolioArtifactCollection(GQLBase):
    typename__: Typename[
        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
    ]
    id: GQLId
    name: str
    description: Optional[str]


UpdateArtifactPortfolio.model_rebuild()
UpdateArtifactPortfolioUpdateArtifactPortfolio.model_rebuild()