fix(BA-6036): expose resource_group on ModelDeploymentMetadata (#11598)

Author: jopemachine

changes/11598.fix.md

@@ -0,0 +1 @@
+Expose `resource_group` on `ModelDeploymentMetadata` so the resource group selected at deployment creation is visible in detail responses (GraphQL and REST v2).

docs/manager/graphql-reference/supergraph.graphql

@@ -9350,6 +9350,11 @@ Added in 25.19.0. Contains metadata information for a model deployment including
 type ModelDeploymentMetadata
   @join__type(graph: STRAWBERRY)
 {
+  """
+  Added in UNRELEASED. The resource group this deployment runs in, resolved via DataLoader.
+  """
+  resourceGroup: ResourceGroup
+
   """The project of this entity."""
   project: GroupNode @deprecated(reason: "Use project_v2 instead.")
 
@@ -9370,6 +9375,7 @@ type ModelDeploymentMetadata
   name: String!
   status: DeploymentStatus!
   tags: [String!]!
+  resourceGroupName: String!
   createdAt: DateTime!
   updatedAt: DateTime!
 }

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

@@ -6147,6 +6147,11 @@ type ModelDeploymentEdge {
 Added in 25.19.0. Contains metadata information for a model deployment including its name, status, tags, and timestamps. Also provides access to the associated project and domain.
 """
 type ModelDeploymentMetadata {
+  """
+  Added in UNRELEASED. The resource group this deployment runs in, resolved via DataLoader.
+  """
+  resourceGroup: ResourceGroup
+
   """The project of this entity."""
   project: GroupNode @deprecated(reason: "Use project_v2 instead.")
 
@@ -6167,6 +6172,7 @@ type ModelDeploymentMetadata {
   name: String!
   status: DeploymentStatus!
   tags: [String!]!
+  resourceGroupName: String!
   createdAt: DateTime!
   updatedAt: DateTime!
 }

src/ai/backend/common/dto/manager/v2/deployment/types.py

@@ -434,6 +434,7 @@ class DeploymentMetadataInfoDTO(BaseResponseModel):
     name: str
     status: ModelDeploymentStatus
     tags: list[str]
+    resource_group_name: str
     created_at: datetime
     updated_at: datetime
 

src/ai/backend/manager/api/adapters/deployment/adapter.py

@@ -2178,6 +2178,7 @@ def _deployment_data_to_dto(data: ModelDeploymentData) -> DeploymentNode:
                 name=data.metadata.name,
                 status=data.metadata.status,
                 tags=data.metadata.tags,
+                resource_group_name=data.metadata.resource_group_name,
                 created_at=data.metadata.created_at,
                 updated_at=data.metadata.updated_at,
             ),

src/ai/backend/manager/api/gql/deployment/types/deployment.py

@@ -167,6 +167,7 @@
 if TYPE_CHECKING:
     from ai.backend.manager.api.gql.domain_v2.types.node import DomainV2GQL
     from ai.backend.manager.api.gql.project_v2.types.node import ProjectV2GQL
+    from ai.backend.manager.api.gql.resource_group.types import ResourceGroupGQL
     from ai.backend.manager.api.gql.user.types.node import UserV2GQL
 
 DeploymentStatusGQL: type[ModelDeploymentStatus] = gql_enum(
@@ -231,6 +232,29 @@ class ModelDeploymentMetadata:
     tags: list[str]
     created_at: datetime
     updated_at: datetime
+    resource_group_name: str = gql_added_field(
+        BackendAIGQLMeta(
+            added_version=NEXT_RELEASE_VERSION,
+            description="Name of the resource group (scaling group) this deployment runs in.",
+        )
+    )
+
+    @gql_added_field(
+        BackendAIGQLMeta(
+            added_version=NEXT_RELEASE_VERSION,
+            description="The resource group this deployment runs in, resolved via DataLoader.",
+        )
+    )  # type: ignore[misc]
+    async def resource_group(
+        self, info: Info[StrawberryGQLContext]
+    ) -> (
+        Annotated[
+            ResourceGroupGQL,
+            strawberry.lazy("ai.backend.manager.api.gql.resource_group.types"),
+        ]
+        | None
+    ):
+        return await info.context.data_loaders.resource_group_loader.load(self.resource_group_name)
 
     @gql_field(
         description="The project of this entity.",

src/ai/backend/manager/data/deployment/types.py

@@ -1002,6 +1002,7 @@ class ModelDeploymentMetadataInfo:
     tags: list[str]
     project_id: UUID
     domain_name: str
+    resource_group_name: str
     created_at: datetime
     updated_at: datetime
 

src/ai/backend/manager/services/deployment/service.py

@@ -252,6 +252,7 @@ def _convert_deployment_info_to_data(info: DeploymentInfo) -> ModelDeploymentDat
             tags=[info.metadata.tag] if info.metadata.tag else [],
             project_id=info.metadata.project,
             domain_name=info.metadata.domain,
+            resource_group_name=info.metadata.resource_group,
             created_at=info.metadata.created_at or datetime.now(UTC),
             updated_at=info.metadata.created_at or datetime.now(UTC),
         ),

tests/unit/common/dto/manager/v2/deployment/test_response.py

@@ -105,6 +105,7 @@ def _make_deployment_metadata(**kwargs: object) -> DeploymentMetadataInfoDTO:
         "name": "test-deployment",
         "status": ModelDeploymentStatus.READY,
         "tags": [],
+        "resource_group_name": "default",
         "created_at": now,
         "updated_at": now,
     }