wip

Author: jopemachine

changes/10300.feature.md

@@ -0,0 +1 @@
+Add `admin_update_deployment_policy` GQL mutation

docs/manager/graphql-reference/supergraph.graphql

@@ -7718,6 +7718,9 @@ type Mutation
   """Added in 25.16.0"""
   addModelRevision(input: AddRevisionInput!): AddRevisionPayload! @join__field(graph: STRAWBERRY)
 
+  """Added in 26.4.0"""
+  adminUpdateDeploymentPolicy(input: UpdateDeploymentPolicyInput!): UpdateDeploymentPolicyPayload! @join__field(graph: STRAWBERRY)
+
   """Create a new notification channel (admin only)"""
   adminCreateNotificationChannel(input: CreateNotificationChannelInput!): CreateNotificationChannelPayload! @join__field(graph: STRAWBERRY)
 
@@ -12942,6 +12945,27 @@ type UpdateDeploymentPayload
   deployment: ModelDeployment!
 }
 
+"""
+Added in 26.3.0. Input for updating a deployment policy. Internally upserts the policy.
+"""
+input UpdateDeploymentPolicyInput
+  @join__type(graph: STRAWBERRY)
+{
+  deploymentId: ID!
+  strategy: DeploymentStrategyType!
+  rollbackOnFailure: Boolean! = false
+  rollingUpdate: RollingUpdateConfigInput = null
+  blueGreen: BlueGreenConfigInput = null
+}
+
+"""Added in 26.3.0. Result of updating a deployment policy."""
+type UpdateDeploymentPolicyPayload
+  @join__type(graph: STRAWBERRY)
+{
+  deploymentPolicy: DeploymentPolicy!
+  created: Boolean!
+}
+
 """Added in 25.14.0"""
 input UpdateHuggingFaceRegistryInput
   @join__type(graph: STRAWBERRY)

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

@@ -4074,6 +4074,9 @@ type Mutation {
   """Added in 25.16.0"""
   addModelRevision(input: AddRevisionInput!): AddRevisionPayload!
 
+  """Added in 26.4.0"""
+  adminUpdateDeploymentPolicy(input: UpdateDeploymentPolicyInput!): UpdateDeploymentPolicyPayload!
+
   """Create a new notification channel (admin only)"""
   adminCreateNotificationChannel(input: CreateNotificationChannelInput!): CreateNotificationChannelPayload!
 
@@ -7885,6 +7888,23 @@ type UpdateDeploymentPayload {
   deployment: ModelDeployment!
 }
 
+"""
+Added in 26.3.0. Input for updating a deployment policy. Internally upserts the policy.
+"""
+input UpdateDeploymentPolicyInput {
+  deploymentId: ID!
+  strategy: DeploymentStrategyType!
+  rollbackOnFailure: Boolean! = false
+  rollingUpdate: RollingUpdateConfigInput = null
+  blueGreen: BlueGreenConfigInput = null
+}
+
+"""Added in 26.3.0. Result of updating a deployment policy."""
+type UpdateDeploymentPolicyPayload {
+  deploymentPolicy: DeploymentPolicy!
+  created: Boolean!
+}
+
 """Added in 25.14.0"""
 input UpdateHuggingFaceRegistryInput {
   id: ID!

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

@@ -16,6 +16,8 @@
     # Revision
     activate_deployment_revision,
     add_model_revision,
+    # Policy
+    admin_update_deployment_policy,
     # Access Token
     create_access_token,
     # Auto Scaling
@@ -40,8 +42,6 @@
     routes,
     sync_replicas,
     update_auto_scaling_rule,
-    # Policy
-    update_deployment_policy,
     update_model_deployment,
     update_route_traffic_status,
 )
@@ -275,7 +275,7 @@
     "sync_replicas",
     "update_model_deployment",
     # Resolvers - Policy
-    "update_deployment_policy",
+    "admin_update_deployment_policy",
     # Resolvers - Replica
     "replica",
     "replica_status_changed",

src/ai/backend/manager/api/gql/deployment/resolver/__init__.py

@@ -22,7 +22,7 @@
     update_model_deployment,
 )
 from .policy import (
-    update_deployment_policy,
+    admin_update_deployment_policy,
 )
 from .replica import (
     replica,

src/ai/backend/manager/api/gql/deployment/resolver/policy.py

@@ -13,17 +13,19 @@
     UpdateDeploymentPolicyPayloadGQL,
 )
 from ai.backend.manager.api.gql.types import StrawberryGQLContext
+from ai.backend.manager.api.gql.utils import check_admin_only
 from ai.backend.manager.services.deployment.actions.deployment_policy.upsert_deployment_policy import (
     UpsertDeploymentPolicyAction,
 )
 
 
 @strawberry.mutation(description="Added in 26.4.0")  # type: ignore[misc]
-async def update_deployment_policy(
+async def admin_update_deployment_policy(
     input: UpdateDeploymentPolicyInputGQL,
     info: Info[StrawberryGQLContext],
 ) -> UpdateDeploymentPolicyPayloadGQL:
     """Update (upsert) a deployment policy for a deployment."""
+    check_admin_only()
     deployment_uuid = UUID(str(input.deployment_id))
     upserter = input.to_upserter(deployment_uuid)
 

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

@@ -138,7 +138,7 @@ def to_spec(self) -> BlueGreenSpec:
 
 @strawberry.input(
     name="UpdateDeploymentPolicyInput",
-    description="Added in 26.3.0. Input for updating a deployment policy. Internally upserts the policy.",
+    description="Added in 26.4.0. Input for updating a deployment policy. Internally upserts the policy.",
 )
 class UpdateDeploymentPolicyInputGQL:
     deployment_id: ID
@@ -177,7 +177,7 @@ def to_upserter(self, deployment_uuid: UUID) -> DeploymentPolicyUpserter:
 
 @strawberry.type(
     name="UpdateDeploymentPolicyPayload",
-    description="Added in 26.3.0. Result of updating a deployment policy.",
+    description="Added in 26.4.0. Result of updating a deployment policy.",
 )
 class UpdateDeploymentPolicyPayloadGQL:
     deployment_policy: DeploymentPolicyGQL

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

@@ -52,6 +52,7 @@
     # Revision
     activate_deployment_revision,
     add_model_revision,
+    admin_update_deployment_policy,
     # Access Token
     create_access_token,
     # Auto Scaling
@@ -76,7 +77,6 @@
     routes,
     sync_replicas,
     update_auto_scaling_rule,
-    update_deployment_policy,
     update_model_deployment,
     update_route_traffic_status,
 )
@@ -430,7 +430,7 @@ class Mutation:
     delete_model_deployment = delete_model_deployment
     sync_replicas = sync_replicas
     add_model_revision = add_model_revision
-    update_deployment_policy = update_deployment_policy
+    admin_update_deployment_policy = admin_update_deployment_policy
     # Notification - Admin APIs
     admin_create_notification_channel = admin_create_notification_channel
     admin_update_notification_channel = admin_update_notification_channel