refactor: replace resolve global id method to it's new version

Author: seedspirit

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

@@ -17,7 +17,7 @@
     ModelDeploymentStatus as CommonDeploymentStatus,
 )
 from ai.backend.common.exception import ModelDeploymentUnavailableError
-from ai.backend.manager.api.gql.base import OrderDirection, StringFilter
+from ai.backend.manager.api.gql.base import OrderDirection, StringFilter, resolve_global_id
 from ai.backend.manager.api.gql.domain import Domain
 from ai.backend.manager.api.gql.model_deployment.access_token import (
     AccessToken,
@@ -229,7 +229,7 @@ async def access_tokens(self, info: Info[StrawberryGQLContext]) -> AccessTokenCo
                 ),
             )
 
-        _, deployment_id = AsyncNode.resolve_global_id(info, str(self._deployment_id))
+        _, deployment_id = resolve_global_id(str(self._deployment_id))
         access_token_loader = DataLoader(
             apartial(AccessToken.batch_load_by_deployment_ids, info.context)
         )
@@ -291,7 +291,7 @@ async def scaling_rule(self, info: Info[StrawberryGQLContext]) -> ScalingRule:
 
     @strawberry.field
     async def replica_state(self, info: Info[StrawberryGQLContext]) -> ReplicaState:
-        _, deployment_id = AsyncNode.resolve_global_id(info, self.id)
+        _, deployment_id = resolve_global_id(self.id)
 
         return ReplicaState(
             _deployment_id=UUID(deployment_id),
@@ -606,7 +606,7 @@ async def deployments(
 @strawberry.field(description="Added in 25.13.0")
 async def deployment(id: ID, info: Info[StrawberryGQLContext]) -> Optional[ModelDeployment]:
     """Get a specific deployment by ID."""
-    _, deployment_id = AsyncNode.resolve_global_id(info, id)
+    _, deployment_id = resolve_global_id(id)
     dataloader = DataLoader(apartial(ModelDeployment.batch_load_by_ids, info.context))
     deployment: list[ModelDeployment] = await dataloader.load(deployment_id)
 
@@ -637,7 +637,7 @@ async def update_model_deployment(
     input: UpdateModelDeploymentInput, info: Info[StrawberryGQLContext]
 ) -> UpdateModelDeploymentPayload:
     """Update an existing model deployment."""
-    _, deployment_id = AsyncNode.resolve_global_id(info, input.id)
+    _, deployment_id = resolve_global_id(input.id)
     deployment_processor = info.context.processors.deployment
     if deployment_processor is None:
         raise ModelDeploymentUnavailableError(
@@ -656,7 +656,7 @@ async def delete_model_deployment(
     input: DeleteModelDeploymentInput, info: Info[StrawberryGQLContext]
 ) -> DeleteModelDeploymentPayload:
     """Delete a model deployment."""
-    _, deployment_id = AsyncNode.resolve_global_id(info, input.id)
+    _, deployment_id = resolve_global_id(input.id)
     deployment_processor = info.context.processors.deployment
     if deployment_processor is None:
         raise ModelDeploymentUnavailableError(
@@ -695,7 +695,7 @@ class SyncReplicaPayload:
 async def sync_replicas(
     input: SyncReplicaInput, info: Info[StrawberryGQLContext]
 ) -> SyncReplicaPayload:
-    _, deployment_id = AsyncNode.resolve_global_id(info, input.model_deployment_id)
+    _, deployment_id = resolve_global_id(input.model_deployment_id)
     deployment_processor = info.context.processors.deployment
     if deployment_processor is None:
         raise ModelDeploymentUnavailableError(

src/ai/backend/manager/api/gql/model_deployment/model_replica.py

@@ -14,7 +14,7 @@
 from ai.backend.common.data.model_deployment.types import LivenessStatus as CommonLivenessStatus
 from ai.backend.common.data.model_deployment.types import ReadinessStatus as CommonReadinessStatus
 from ai.backend.common.exception import ModelDeploymentUnavailableError
-from ai.backend.manager.api.gql.base import JSONString, OrderDirection
+from ai.backend.manager.api.gql.base import JSONString, OrderDirection, resolve_global_id
 from ai.backend.manager.api.gql.session import Session
 from ai.backend.manager.api.gql.types import StrawberryGQLContext
 from ai.backend.manager.data.deployment.types import ModelReplicaData
@@ -221,7 +221,7 @@ class ReplicaStatusChangedPayload:
 @strawberry.field(description="Added in 25.13.0")
 async def replica(id: ID, info: Info[StrawberryGQLContext]) -> Optional[ModelReplica]:
     """Get a specific replica by ID."""
-    _, replica_id = AsyncNode.resolve_global_id(info, id)
+    _, replica_id = resolve_global_id(id)
     replica_loader = DataLoader(apartial(ModelReplica.batch_load_by_revision_ids, info.context))
     replicas: list[ModelReplica] = await replica_loader.load(UUID(replica_id))
     return replicas[0]

src/ai/backend/manager/api/gql/model_deployment/model_revision.py

@@ -17,7 +17,12 @@
 from ai.backend.common.types import ClusterMode as CommonClusterMode
 from ai.backend.common.types import MountPermission as CommonMountPermission
 from ai.backend.common.types import RuntimeVariant
-from ai.backend.manager.api.gql.base import JSONString, OrderDirection, StringFilter
+from ai.backend.manager.api.gql.base import (
+    JSONString,
+    OrderDirection,
+    StringFilter,
+    resolve_global_id,
+)
 from ai.backend.manager.api.gql.image import (
     Image,
 )
@@ -582,7 +587,7 @@ async def revisions(
 @strawberry.field(description="Added in 25.13.0")
 async def revision(id: ID, info: Info[StrawberryGQLContext]) -> ModelRevision:
     """Get a specific revision by ID."""
-    _, revision_id = AsyncNode.resolve_global_id(info, id)
+    _, revision_id = resolve_global_id(id)
     revision_loader = DataLoader(apartial(ModelRevision.batch_load_by_ids, info.context))
     revision: list[ModelRevision] = await revision_loader.load(revision_id)
     return revision[0]