wip

Author: jopemachine

docs/manager/graphql-reference/supergraph.graphql

@@ -1004,7 +1004,18 @@ type ArtifactStorage
   storageId: ID!
 
   """The type of the artifact storage"""
-  type: String!
+  type: ArtifactStorageType!
+}
+
+"""
+Added in 26.2.0. The type of artifact storage backend. OBJECT_STORAGE: Object storage (e.g., S3-compatible). VFS_STORAGE: Virtual folder storage. GIT_LFS: Git LFS storage.
+"""
+enum ArtifactStorageType
+  @join__type(graph: STRAWBERRY)
+{
+  OBJECT_STORAGE @join__enumValue(graph: STRAWBERRY)
+  VFS_STORAGE @join__enumValue(graph: STRAWBERRY)
+  GIT_LFS @join__enumValue(graph: STRAWBERRY)
 }
 
 enum ArtifactType

docs/manager/graphql-reference/v2-schema.graphql

@@ -715,7 +715,16 @@ type ArtifactStorage {
   storageId: ID!
 
   """The type of the artifact storage"""
-  type: String!
+  type: ArtifactStorageType!
+}
+
+"""
+Added in 26.2.0. The type of artifact storage backend. OBJECT_STORAGE: Object storage (e.g., S3-compatible). VFS_STORAGE: Virtual folder storage. GIT_LFS: Git LFS storage.
+"""
+enum ArtifactStorageType {
+  OBJECT_STORAGE
+  VFS_STORAGE
+  GIT_LFS
 }
 
 enum ArtifactType {

src/ai/backend/manager/api/gql/artifact_storage.py

@@ -1,11 +1,13 @@
 from __future__ import annotations
 
 import uuid
+from enum import StrEnum
 from typing import Self
 
 import strawberry
 from strawberry import ID, UNSET, Info
 
+from ai.backend.common.data.storage.types import ArtifactStorageType
 from ai.backend.manager.data.artifact_storages.types import (
     ArtifactStorageData,
     ArtifactStorageUpdaterSpec,
@@ -20,22 +22,60 @@
 from .types import StrawberryGQLContext
 
 
+@strawberry.enum(
+    name="ArtifactStorageType",
+    description=(
+        "Added in 26.2.0. The type of artifact storage backend. "
+        "OBJECT_STORAGE: Object storage (e.g., S3-compatible). "
+        "VFS_STORAGE: Virtual folder storage. "
+        "GIT_LFS: Git LFS storage."
+    ),
+)
+class ArtifactStorageTypeGQL(StrEnum):
+    """Artifact storage type enum."""
+
+    OBJECT_STORAGE = "object_storage"
+    VFS_STORAGE = "vfs_storage"
+    GIT_LFS = "git_lfs"
+
+    @classmethod
+    def from_internal(cls, internal_type: ArtifactStorageType) -> ArtifactStorageTypeGQL:
+        """Convert internal ArtifactStorageType to GraphQL enum."""
+        match internal_type:
+            case ArtifactStorageType.OBJECT_STORAGE:
+                return cls.OBJECT_STORAGE
+            case ArtifactStorageType.VFS_STORAGE:
+                return cls.VFS_STORAGE
+            case ArtifactStorageType.GIT_LFS:
+                return cls.GIT_LFS
+
+    def to_internal(self) -> ArtifactStorageType:
+        """Convert GraphQL enum to internal ArtifactStorageType."""
+        match self:
+            case ArtifactStorageTypeGQL.OBJECT_STORAGE:
+                return ArtifactStorageType.OBJECT_STORAGE
+            case ArtifactStorageTypeGQL.VFS_STORAGE:
+                return ArtifactStorageType.VFS_STORAGE
+            case ArtifactStorageTypeGQL.GIT_LFS:
+                return ArtifactStorageType.GIT_LFS
+
+
 @strawberry.type(description="Added in 26.2.0. Artifact storage metadata")
 class ArtifactStorage:
     id: ID = strawberry.field(description="The ID of the artifact storage")
     name: str = strawberry.field(description="The name of the artifact storage")
     storage_id: ID = strawberry.field(
         description="The ID of the underlying storage (ObjectStorage or VFSStorage)"
     )
-    type: str = strawberry.field(description="The type of the artifact storage")
+    type: ArtifactStorageTypeGQL = strawberry.field(description="The type of the artifact storage")
 
     @classmethod
     def from_dataclass(cls, data: ArtifactStorageData) -> Self:
         return cls(
             id=ID(str(data.id)),
             name=data.name,
             storage_id=ID(str(data.storage_id)),
-            type=data.type.value,
+            type=ArtifactStorageTypeGQL.from_internal(data.type),
         )