wandb 0.19.9__py3-none-musllinux_1_2_aarch64.whl → 0.19.11__py3-none-musllinux_1_2_aarch64.whl

wandb/sdk/artifacts/_generated/project_artifact_collections.py

@@ -0,0 +1,33 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactCollectionsFragment
+
+
+class ProjectArtifactCollections(GQLBase):
+    project: Optional[ProjectArtifactCollectionsProject]
+
+
+class ProjectArtifactCollectionsProject(GQLBase):
+    artifact_type: Optional[ProjectArtifactCollectionsProjectArtifactType] = Field(
+        alias="artifactType"
+    )
+
+
+class ProjectArtifactCollectionsProjectArtifactType(GQLBase):
+    artifact_collections: Optional[ArtifactCollectionsFragment] = Field(
+        alias="artifactCollections"
+    )
+
+
+ProjectArtifactCollections.model_rebuild()
+ProjectArtifactCollectionsProject.model_rebuild()
+ProjectArtifactCollectionsProjectArtifactType.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifact_type.py

@@ -0,0 +1,24 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactTypeFragment
+
+
+class ProjectArtifactType(GQLBase):
+    project: Optional[ProjectArtifactTypeProject]
+
+
+class ProjectArtifactTypeProject(GQLBase):
+    artifact_type: Optional[ArtifactTypeFragment] = Field(alias="artifactType")
+
+
+ProjectArtifactType.model_rebuild()
+ProjectArtifactTypeProject.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifact_types.py

@@ -0,0 +1,24 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactTypesFragment
+
+
+class ProjectArtifactTypes(GQLBase):
+    project: Optional[ProjectArtifactTypesProject]
+
+
+class ProjectArtifactTypesProject(GQLBase):
+    artifact_types: ArtifactTypesFragment = Field(alias="artifactTypes")
+
+
+ProjectArtifactTypes.model_rebuild()
+ProjectArtifactTypesProject.model_rebuild()

wandb/sdk/artifacts/_generated/project_artifacts.py

@@ -0,0 +1,42 @@
+# 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, Typename
+
+from .fragments import ArtifactsFragment
+
+
+class ProjectArtifacts(GQLBase):
+    project: Optional[ProjectArtifactsProject]
+
+
+class ProjectArtifactsProject(GQLBase):
+    artifact_type: Optional[ProjectArtifactsProjectArtifactType] = Field(
+        alias="artifactType"
+    )
+
+
+class ProjectArtifactsProjectArtifactType(GQLBase):
+    artifact_collection: Optional[
+        ProjectArtifactsProjectArtifactTypeArtifactCollection
+    ] = Field(alias="artifactCollection")
+
+
+class ProjectArtifactsProjectArtifactTypeArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    name: str
+    artifacts: Optional[ArtifactsFragment]
+
+
+ProjectArtifacts.model_rebuild()
+ProjectArtifactsProject.model_rebuild()
+ProjectArtifactsProjectArtifactType.model_rebuild()
+ProjectArtifactsProjectArtifactTypeArtifactCollection.model_rebuild()

wandb/sdk/artifacts/_generated/run_input_artifacts.py

@@ -0,0 +1,51 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactFragment
+
+
+class RunInputArtifacts(GQLBase):
+    project: Optional[RunInputArtifactsProject]
+
+
+class RunInputArtifactsProject(GQLBase):
+    run: Optional[RunInputArtifactsProjectRun]
+
+
+class RunInputArtifactsProjectRun(GQLBase):
+    input_artifacts: Optional[RunInputArtifactsProjectRunInputArtifacts] = Field(
+        alias="inputArtifacts"
+    )
+
+
+class RunInputArtifactsProjectRunInputArtifacts(GQLBase):
+    total_count: int = Field(alias="totalCount")
+    edges: List[RunInputArtifactsProjectRunInputArtifactsEdges]
+    page_info: RunInputArtifactsProjectRunInputArtifactsPageInfo = Field(
+        alias="pageInfo"
+    )
+
+
+class RunInputArtifactsProjectRunInputArtifactsEdges(GQLBase):
+    node: Optional[ArtifactFragment]
+    cursor: str
+
+
+class RunInputArtifactsProjectRunInputArtifactsPageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+RunInputArtifacts.model_rebuild()
+RunInputArtifactsProject.model_rebuild()
+RunInputArtifactsProjectRun.model_rebuild()
+RunInputArtifactsProjectRunInputArtifacts.model_rebuild()
+RunInputArtifactsProjectRunInputArtifactsEdges.model_rebuild()

wandb/sdk/artifacts/_generated/run_output_artifacts.py

@@ -0,0 +1,51 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactFragment
+
+
+class RunOutputArtifacts(GQLBase):
+    project: Optional[RunOutputArtifactsProject]
+
+
+class RunOutputArtifactsProject(GQLBase):
+    run: Optional[RunOutputArtifactsProjectRun]
+
+
+class RunOutputArtifactsProjectRun(GQLBase):
+    output_artifacts: Optional[RunOutputArtifactsProjectRunOutputArtifacts] = Field(
+        alias="outputArtifacts"
+    )
+
+
+class RunOutputArtifactsProjectRunOutputArtifacts(GQLBase):
+    total_count: int = Field(alias="totalCount")
+    edges: List[RunOutputArtifactsProjectRunOutputArtifactsEdges]
+    page_info: RunOutputArtifactsProjectRunOutputArtifactsPageInfo = Field(
+        alias="pageInfo"
+    )
+
+
+class RunOutputArtifactsProjectRunOutputArtifactsEdges(GQLBase):
+    node: Optional[ArtifactFragment]
+    cursor: str
+
+
+class RunOutputArtifactsProjectRunOutputArtifactsPageInfo(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+RunOutputArtifacts.model_rebuild()
+RunOutputArtifactsProject.model_rebuild()
+RunOutputArtifactsProjectRun.model_rebuild()
+RunOutputArtifactsProjectRunOutputArtifacts.model_rebuild()
+RunOutputArtifactsProjectRunOutputArtifactsEdges.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact.py

@@ -0,0 +1,26 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/artifacts/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ArtifactFragment
+
+
+class UpdateArtifact(GQLBase):
+    update_artifact: Optional[UpdateArtifactUpdateArtifact] = Field(
+        alias="updateArtifact"
+    )
+
+
+class UpdateArtifactUpdateArtifact(GQLBase):
+    artifact: ArtifactFragment
+
+
+UpdateArtifact.model_rebuild()
+UpdateArtifactUpdateArtifact.model_rebuild()

wandb/sdk/artifacts/_generated/update_artifact_portfolio.py

@@ -0,0 +1,35 @@
+# 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()

wandb/sdk/artifacts/_generated/update_artifact_sequence.py

@@ -0,0 +1,35 @@
+# 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 UpdateArtifactSequence(GQLBase):
+    update_artifact_sequence: Optional[UpdateArtifactSequenceUpdateArtifactSequence] = (
+        Field(alias="updateArtifactSequence")
+    )
+
+
+class UpdateArtifactSequenceUpdateArtifactSequence(GQLBase):
+    artifact_collection: UpdateArtifactSequenceUpdateArtifactSequenceArtifactCollection = Field(
+        alias="artifactCollection"
+    )
+
+
+class UpdateArtifactSequenceUpdateArtifactSequenceArtifactCollection(GQLBase):
+    typename__: Typename[
+        Literal["ArtifactCollection", "ArtifactPortfolio", "ArtifactSequence"]
+    ]
+    id: GQLId
+    name: str
+    description: Optional[str]
+
+
+UpdateArtifactSequence.model_rebuild()
+UpdateArtifactSequenceUpdateArtifactSequence.model_rebuild()

wandb/sdk/artifacts/_graphql_fragments.py

@@ -1,62 +1,57 @@
+from __future__ import annotations
+
+from textwrap import dedent
+
+from wandb_graphql.language.printer import print_ast
+
+from wandb.apis.public.utils import gql_compat
 from wandb.sdk.internal.internal_api import Api as InternalApi
 
-ARTIFACTS_TYPES_FRAGMENT = """
-fragment ArtifactTypesFragment on ArtifactTypeConnection {
-    edges {
-         node {
-             id
-             name
-             description
-             createdAt
-         }
-         cursor
+OMITTABLE_ARTIFACT_FIELDS = frozenset(
+    {
+        "ttlDurationSeconds",
+        "ttlIsInherited",
+        "aliases",
+        "tags",
+        "historyStep",
     }
-    pageInfo {
-        endCursor
-        hasNextPage
-    }
-}
-"""
+)
 
-ARTIFACT_FILES_FRAGMENT = """fragment FilesFragment on FileConnection {
-    edges {
-        node {
-            id
-            name: displayName
-            url
-            sizeBytes
-            storagePath
-            mimetype
-            updatedAt
-            digest
-            md5
-            directUrl
-        }
-        cursor
-    }
-    pageInfo {
-        endCursor
-        hasNextPage
-    }
-}"""
+
+def omit_artifact_fields(api: InternalApi) -> set[str]:
+    """Return names of Artifact fields to remove from GraphQL requests (for server compatibility)."""
+    allowed_fields = set(api.server_artifact_introspection())
+    return set(OMITTABLE_ARTIFACT_FIELDS - allowed_fields)
 
 
 def _gql_artifact_fragment(include_aliases: bool = True) -> str:
     """Return a GraphQL query fragment with all parseable Artifact attributes."""
-    allowed_fields = set(InternalApi().server_artifact_introspection())
+    omit_fields = omit_artifact_fields(api=InternalApi())
 
-    supports_ttl = "ttlIsInherited" in allowed_fields
-    supports_tags = "tags" in allowed_fields
+    # Respect the `include_aliases` flag
+    if not include_aliases:
+        omit_fields.add("aliases")
 
-    ttl_duration_seconds = "ttlDurationSeconds" if supports_ttl else ""
-    ttl_is_inherited = "ttlIsInherited" if supports_ttl else ""
-
-    tags = "tags {name}" if supports_tags else ""
-
-    # The goal is to move all artifact aliases fetches to the membership level in the future
-    # but this is a quick fix to unblock the registry work
-    aliases = (
-        """aliases {
+    artifact_fragment_str = dedent(
+        """\
+        fragment ArtifactFragment on Artifact {
+            id
+            artifactSequence {
+                project {
+                    entityName
+                    name
+                }
+                name
+            }
+            versionIndex
+            artifactType {
+                name
+            }
+            description
+            metadata
+            ttlDurationSeconds
+            ttlIsInherited
+            aliases {
                 artifactCollection {
                     project {
                         entityName
@@ -65,43 +60,25 @@ def _gql_artifact_fragment(include_aliases: bool = True) -> str:
                     name
                 }
                 alias
-            }"""
-        if include_aliases
-        else ""
-    )
-
-    return f"""
-        fragment ArtifactFragment on Artifact {{
-            id
-            artifactSequence {{
-                project {{
-                    entityName
-                    name
-                }}
-                name
-            }}
-            versionIndex
-            artifactType {{
+            }
+            tags {
                 name
-            }}
-            description
-            metadata
-            {ttl_duration_seconds}
-            {ttl_is_inherited}
-            {aliases}
-            {tags}
+            }
+            historyStep
             state
-            currentManifest {{
-                file {{
+            currentManifest {
+                file {
                     directUrl
-                }}
-            }}
+                }
+            }
             commitHash
             fileCount
             createdAt
             updatedAt
-        }}
-    """
+        }"""
+    )
+    compat_doc = gql_compat(artifact_fragment_str, omit_fields=omit_fields)
+    return print_ast(compat_doc)
 
 
 def _gql_registry_fragment() -> str:
@@ -120,5 +97,6 @@ def _gql_registry_fragment() -> str:
             description
             createdAt
             updatedAt
+            access
         }
     """

wandb/sdk/artifacts/_validators.py

@@ -3,9 +3,14 @@
 from __future__ import annotations
 
 import re
+from dataclasses import dataclass, field
 from functools import wraps
-from typing import TYPE_CHECKING, Any, Callable, TypeVar, cast, overload
+from typing import TYPE_CHECKING, Any, Callable, Literal, TypeVar, cast, overload
 
+from wandb.sdk.artifacts._generated.fragments import (
+    ArtifactPortfolioTypeFields,
+    ArtifactSequenceTypeFields,
+)
 from wandb.sdk.artifacts.exceptions import (
     ArtifactFinalizedError,
     ArtifactNotLoggedError,
@@ -24,6 +29,41 @@ if TYPE_CHECKING:
 REGISTRY_PREFIX: Final[str] = "wandb-registry-"
 MAX_ARTIFACT_METADATA_KEYS: Final[int] = 100
 
+ARTIFACT_NAME_MAXLEN: Final[int] = 128
+ARTIFACT_NAME_INVALID_CHARS: Final[frozenset[str]] = frozenset({"/"})
+
+LINKED_ARTIFACT_COLLECTION_TYPE: Final[str] = ArtifactPortfolioTypeFields.model_fields[
+    "typename__"
+].default
+SOURCE_ARTIFACT_COLLECTION_TYPE: Final[str] = ArtifactSequenceTypeFields.model_fields[
+    "typename__"
+].default
+
+
+@dataclass
+class _LinkArtifactFields:
+    """Keep this list updated with fields where the linked artifact and the source artifact differ."""
+
+    entity_name: str
+    project_name: str
+    name: str
+    version: str
+    aliases: list[str]
+
+    # These fields shouldn't be set as they should always be
+    # these values for a linked artifact
+    # These fields shouldn't be set by the user as they should always be these values for a linked artifact
+    _is_link: Literal[True] = field(init=False, default=True)
+    _linked_artifacts: list[Artifact] = field(init=False, default_factory=list)
+
+    @property
+    def is_link(self) -> bool:
+        return self._is_link
+
+    @property
+    def linked_artifacts(self) -> list[Artifact]:
+        return self._linked_artifacts
+
 
 # For mypy checks
 @overload
@@ -43,6 +83,69 @@ def always_list(obj: Any, base_type: Any = (str, bytes)) -> list[T]:
     return [obj] if isinstance(obj, base_type) else list(obj)
 
 
+def validate_artifact_name(name: str) -> str:
+    """Validate the artifact name, returning it if successful.
+
+    Raises:
+        ValueError: If the artifact name is invalid.
+    """
+    if len(name) > ARTIFACT_NAME_MAXLEN:
+        short_name = f"{name[:ARTIFACT_NAME_MAXLEN]} ..."
+        raise ValueError(
+            f"Artifact name is longer than {ARTIFACT_NAME_MAXLEN} characters: {short_name!r}"
+        )
+
+    if ARTIFACT_NAME_INVALID_CHARS.intersection(name):
+        raise ValueError(
+            "Artifact names must not contain any of the following characters: "
+            f"{', '.join(sorted(ARTIFACT_NAME_INVALID_CHARS))}.  Got: {name!r}"
+        )
+
+    return name
+
+
+INVALID_URL_CHARACTERS = ("/", "\\", "#", "?", "%", ":", "\r", "\n")
+
+
+def validate_project_name(name: str) -> None:
+    """Validates a project name according to W&B rules.
+
+    Args:
+        name: The project name string.
+
+    Raises:
+        ValueError: If the name is invalid (too long or contains invalid characters).
+    """
+    max_len = 128
+
+    if len(name) == 0:
+        raise ValueError("Project name cannot be empty")
+
+    registry_name = ""
+    if name.startswith(REGISTRY_PREFIX):
+        registry_name = name[len(REGISTRY_PREFIX) :]
+        if len(registry_name) == 0:
+            raise ValueError("Registry name cannot be empty")
+
+    if len(name) > max_len:
+        if registry_name:
+            raise ValueError(
+                f"Invalid registry name {registry_name!r}, must be {max_len - len(REGISTRY_PREFIX)} characters or less"
+            )
+        else:
+            raise ValueError(
+                f"Invalid project name {name!r}, must be {max_len} characters or less"
+            )
+
+    # Find the first occurrence of any invalid character
+    if invalid_chars := set(INVALID_URL_CHARACTERS).intersection(name):
+        error_name = registry_name or name
+        invalid_chars_repr = ", ".join(sorted(map(repr, invalid_chars)))
+        raise ValueError(
+            f"Invalid project/registry name {error_name!r}, cannot contain characters: {invalid_chars_repr!s}"
+        )
+
+
 def validate_aliases(aliases: Collection[str] | str) -> list[str]:
     """Validate the artifact aliases and return them as a list.
 
@@ -59,6 +162,22 @@ def validate_aliases(aliases: Collection[str] | str) -> list[str]:
     return aliases_list
 
 
+def validate_artifact_types_list(artifact_types: list[str]) -> list[str]:
+    """Return True if the artifact types list is valid, False otherwise."""
+    artifact_types = always_list(artifact_types)
+    invalid_chars = ("/", ":")
+    if any(
+        char in type or len(type) > 128
+        for type in artifact_types
+        for char in invalid_chars
+    ):
+        raise ValueError(
+            f"""Artifact types must not contain any of the following characters: {", ".join(invalid_chars)}
+              and must be less than equal to 128 characters"""
+        )
+    return artifact_types
+
+
 _VALID_TAG_PATTERN: re.Pattern[str] = re.compile(r"^[-\w]+( +[-\w]+)*$")