feat: Add compute session node ref in model replica node

Author: seedspirit

docs/manager/graphql-reference/schema.graphql

@@ -128,7 +128,7 @@ type Query {
     """Added in 24.03.1"""
     id: String
     reference: String
-    architecture: String = "x86_64"
+    architecture: String = "aarch64"
   ): Image
   images(
     """
@@ -1541,7 +1541,7 @@ type VirtualFolderPermission implements Item {
 }
 
 """Added in 24.09.0."""
-type ComputeSessionNode implements Node {
+type ComputeSessionNode implements Node @key(fields: "id") {
   """The ID of the object"""
   id: ID!
 
@@ -2243,7 +2243,7 @@ type ServiceConfigEdge {
   cursor: String!
 }
 
-union _Entity = ImageNode | DomainNode | ScalingGroupNode | GroupNode | UserNode | VirtualFolderNode | EndpointToken | EndpointAutoScalingRuleNode
+union _Entity = ImageNode | DomainNode | ScalingGroupNode | GroupNode | UserNode | VirtualFolderNode | ComputeSessionNode | EndpointToken | EndpointAutoScalingRuleNode
 
 scalar _Any
 
@@ -2331,7 +2331,7 @@ type Mutation {
   ): RescanImages
   preload_image(references: [String]!, target_agents: [String]!): PreloadImage
   unload_image(references: [String]!, target_agents: [String]!): UnloadImage
-  modify_image(architecture: String = "x86_64", props: ModifyImageInput!, target: String!): ModifyImage
+  modify_image(architecture: String = "aarch64", props: ModifyImageInput!, target: String!): ModifyImage
 
   """Added in 25.6.0"""
   clear_image_custom_resource_limit(key: ClearImageCustomResourceLimitKey!): ClearImageCustomResourceLimitPayload
@@ -2340,7 +2340,7 @@ type Mutation {
   forget_image_by_id(image_id: String!): ForgetImageById
 
   """Deprecated since 25.4.0. Use `forget_image_by_id` instead."""
-  forget_image(architecture: String = "x86_64", reference: String!): ForgetImage @deprecated(reason: "Deprecated since 25.4.0. Use `forget_image_by_id` instead.")
+  forget_image(architecture: String = "aarch64", reference: String!): ForgetImage @deprecated(reason: "Deprecated since 25.4.0. Use `forget_image_by_id` instead.")
 
   """Added in 25.4.0"""
   purge_image_by_id(
@@ -2352,7 +2352,7 @@ type Mutation {
 
   """Added in 24.03.1"""
   untag_image_from_registry(image_id: String!): UntagImageFromRegistry
-  alias_image(alias: String!, architecture: String = "x86_64", target: String!): AliasImage
+  alias_image(alias: String!, architecture: String = "aarch64", target: String!): AliasImage
   dealias_image(alias: String!): DealiasImage
   clear_images(registry: String): ClearImages
 
@@ -2924,7 +2924,7 @@ type ClearImageCustomResourceLimitPayload {
 """Added in 25.6.0."""
 input ClearImageCustomResourceLimitKey {
   image_canonical: String!
-  architecture: String! = "x86_64"
+  architecture: String! = "aarch64"
 }
 
 """Added in 24.03.0."""

docs/manager/graphql-reference/supergraph.graphql

@@ -761,7 +761,7 @@ input ClearImageCustomResourceLimitKey
   @join__type(graph: GRAPHENE)
 {
   image_canonical: String!
-  architecture: String! = "x86_64"
+  architecture: String! = "aarch64"
 }
 
 """Added in 25.6.0."""
@@ -952,75 +952,76 @@ type ComputeSessionList implements PaginatedList
 """Added in 24.09.0."""
 type ComputeSessionNode implements Node
   @join__implements(graph: GRAPHENE, interface: "Node")
-  @join__type(graph: GRAPHENE)
+  @join__type(graph: GRAPHENE, key: "id")
+  @join__type(graph: STRAWBERRY, key: "id", extension: true)
 {
   """The ID of the object"""
   id: ID!
 
   """ID of session."""
-  row_id: UUID
-  tag: String
-  name: String
-  type: String
+  row_id: UUID @join__field(graph: GRAPHENE)
+  tag: String @join__field(graph: GRAPHENE)
+  name: String @join__field(graph: GRAPHENE)
+  type: String @join__field(graph: GRAPHENE)
 
   """Added in 24.09.0."""
-  priority: Int
-  cluster_template: String
-  cluster_mode: String
-  cluster_size: Int
-  domain_name: String
-  project_id: UUID
-  user_id: UUID
+  priority: Int @join__field(graph: GRAPHENE)
+  cluster_template: String @join__field(graph: GRAPHENE)
+  cluster_mode: String @join__field(graph: GRAPHENE)
+  cluster_size: Int @join__field(graph: GRAPHENE)
+  domain_name: String @join__field(graph: GRAPHENE)
+  project_id: UUID @join__field(graph: GRAPHENE)
+  user_id: UUID @join__field(graph: GRAPHENE)
 
   """Added in 25.13.0."""
-  owner: UserNode
-  access_key: String
+  owner: UserNode @join__field(graph: GRAPHENE)
+  access_key: String @join__field(graph: GRAPHENE)
 
   """
   One of ['read_attribute', 'update_attribute', 'delete_session', 'start_app', 'execute', 'convert_to_image'].
   """
-  permissions: [SessionPermissionValueField]
-  status: String
-  status_info: String
-  status_data: JSONString
-  status_history: JSONString
-  created_at: DateTime
-  terminated_at: DateTime
-  starts_at: DateTime
-  scheduled_at: DateTime
+  permissions: [SessionPermissionValueField] @join__field(graph: GRAPHENE)
+  status: String @join__field(graph: GRAPHENE)
+  status_info: String @join__field(graph: GRAPHENE)
+  status_data: JSONString @join__field(graph: GRAPHENE)
+  status_history: JSONString @join__field(graph: GRAPHENE)
+  created_at: DateTime @join__field(graph: GRAPHENE)
+  terminated_at: DateTime @join__field(graph: GRAPHENE)
+  starts_at: DateTime @join__field(graph: GRAPHENE)
+  scheduled_at: DateTime @join__field(graph: GRAPHENE)
 
   """Added in 25.13.0."""
-  queue_position: Int
-  startup_command: String
-  result: String
-  commit_status: String
-  abusing_reports: [JSONString]
-  idle_checks: JSONString
-  agent_ids: [String]
-  resource_opts: JSONString
-  scaling_group: String
-  service_ports: JSONString
-  vfolder_mounts: [String]
-  occupied_slots: JSONString
-  requested_slots: JSONString
+  queue_position: Int @join__field(graph: GRAPHENE)
+  startup_command: String @join__field(graph: GRAPHENE)
+  result: String @join__field(graph: GRAPHENE)
+  commit_status: String @join__field(graph: GRAPHENE)
+  abusing_reports: [JSONString] @join__field(graph: GRAPHENE)
+  idle_checks: JSONString @join__field(graph: GRAPHENE)
+  agent_ids: [String] @join__field(graph: GRAPHENE)
+  resource_opts: JSONString @join__field(graph: GRAPHENE)
+  scaling_group: String @join__field(graph: GRAPHENE)
+  service_ports: JSONString @join__field(graph: GRAPHENE)
+  vfolder_mounts: [String] @join__field(graph: GRAPHENE)
+  occupied_slots: JSONString @join__field(graph: GRAPHENE)
+  requested_slots: JSONString @join__field(graph: GRAPHENE)
 
   """Added in 25.4.0."""
-  image_references: [String]
+  image_references: [String] @join__field(graph: GRAPHENE)
 
   """Added in 25.4.0."""
-  vfolder_nodes(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): VirtualFolderConnection
-  num_queries: BigInt
-  inference_metrics: JSONString
-  kernel_nodes(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): KernelConnection
+  vfolder_nodes(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): VirtualFolderConnection @join__field(graph: GRAPHENE)
+  num_queries: BigInt @join__field(graph: GRAPHENE)
+  inference_metrics: JSONString @join__field(graph: GRAPHENE)
+  kernel_nodes(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): KernelConnection @join__field(graph: GRAPHENE)
 
   """Added in 24.09.0."""
-  dependents(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): ComputeSessionConnection
+  dependents(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): ComputeSessionConnection @join__field(graph: GRAPHENE)
 
   """Added in 24.09.0."""
-  dependees(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): ComputeSessionConnection
+  dependees(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): ComputeSessionConnection @join__field(graph: GRAPHENE)
 
   """Added in 24.09.0."""
-  graph(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): ComputeSessionConnection
+  graph(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): ComputeSessionConnection @join__field(graph: GRAPHENE)
 }
 
 """Deprecated since 24.09.0. use `ContainerRegistryNode` instead"""
@@ -1583,34 +1584,10 @@ type DealiasImage
   msg: String
 }
 
-<<<<<<< Updated upstream
-=======
 """Decimal (fixed-point)"""
 scalar Decimal
   @join__type(graph: STRAWBERRY)
 
-"""Added in 25.13.0"""
-input DeleteArtifactRevisionsInput
-  @join__type(graph: STRAWBERRY)
-{
-  targets: [DeleteArtifactRevisionTarget!]!
-}
-
-"""Added in 25.13.0"""
-type DeleteArtifactRevisionsPayload
-  @join__type(graph: STRAWBERRY)
-{
-  artifactRevisionIds: [ID!]!
-}
-
-"""Added in 25.13.0"""
-input DeleteArtifactRevisionTarget
-  @join__type(graph: STRAWBERRY)
-{
-  artifactRevisionId: ID!
-  storageNamespaceId: ID!
-}
-
 input DeleteAutoScalingRuleInput
   @join__type(graph: STRAWBERRY)
 {
@@ -1623,7 +1600,6 @@ type DeleteAutoScalingRulePayload
   id: ID!
 }
 
->>>>>>> Stashed changes
 """Deprecated since 24.09.0. use `DeleteContainerRegistryNode` instead"""
 type DeleteContainerRegistry
   @join__type(graph: GRAPHENE)
@@ -3113,11 +3089,6 @@ type ModelReplica implements Node
   id: ID!
   revision: ModelRevision!
 
-  """
-  The session ID associated with the replica. This can be null right after replica creation.
-  """
-  sessionId: ID
-
   """
   This represents whether the replica has been checked and its health state.
   """
@@ -3144,6 +3115,11 @@ type ModelReplica implements Node
   live statistics of the routing node. e.g. "live_stat": "{\"cpu_util\": {\"current\": \"7.472\", \"capacity\": \"1000\", \"pct\": \"0.75\", \"unit_hint\": \"percent\"}}"
   """
   liveStat: JSONString!
+
+  """
+  The session ID associated with the replica. This can be null right after replica creation.
+  """
+  session: ComputeSessionNode!
 }
 
 """Added in 25.13.0"""
@@ -3850,7 +3826,7 @@ type Mutation
   ): RescanImages @join__field(graph: GRAPHENE)
   preload_image(references: [String]!, target_agents: [String]!): PreloadImage @join__field(graph: GRAPHENE)
   unload_image(references: [String]!, target_agents: [String]!): UnloadImage @join__field(graph: GRAPHENE)
-  modify_image(architecture: String = "x86_64", props: ModifyImageInput!, target: String!): ModifyImage @join__field(graph: GRAPHENE)
+  modify_image(architecture: String = "aarch64", props: ModifyImageInput!, target: String!): ModifyImage @join__field(graph: GRAPHENE)
 
   """Added in 25.6.0"""
   clear_image_custom_resource_limit(key: ClearImageCustomResourceLimitKey!): ClearImageCustomResourceLimitPayload @join__field(graph: GRAPHENE)
@@ -3859,7 +3835,7 @@ type Mutation
   forget_image_by_id(image_id: String!): ForgetImageById @join__field(graph: GRAPHENE)
 
   """Deprecated since 25.4.0. Use `forget_image_by_id` instead."""
-  forget_image(architecture: String = "x86_64", reference: String!): ForgetImage @join__field(graph: GRAPHENE) @deprecated(reason: "Deprecated since 25.4.0. Use `forget_image_by_id` instead.")
+  forget_image(architecture: String = "aarch64", reference: String!): ForgetImage @join__field(graph: GRAPHENE) @deprecated(reason: "Deprecated since 25.4.0. Use `forget_image_by_id` instead.")
 
   """Added in 25.4.0"""
   purge_image_by_id(
@@ -3871,7 +3847,7 @@ type Mutation
 
   """Added in 24.03.1"""
   untag_image_from_registry(image_id: String!): UntagImageFromRegistry @join__field(graph: GRAPHENE)
-  alias_image(alias: String!, architecture: String = "x86_64", target: String!): AliasImage @join__field(graph: GRAPHENE)
+  alias_image(alias: String!, architecture: String = "aarch64", target: String!): AliasImage @join__field(graph: GRAPHENE)
   dealias_image(alias: String!): DealiasImage @join__field(graph: GRAPHENE)
   clear_images(registry: String): ClearImages @join__field(graph: GRAPHENE)
 
@@ -4117,9 +4093,6 @@ type Mutation
   """Added in 25.14.0"""
   updateObjectStorage(input: UpdateObjectStorageInput!): UpdateObjectStoragePayload! @join__field(graph: STRAWBERRY)
 
-<<<<<<< Updated upstream
-  """Added in 25.14.0"""
-=======
   """Added in 25.13.0"""
   createAutoScalingRule(input: CreateAutoScalingRuleInput!): CreateAutoScalingRulePayload! @join__field(graph: STRAWBERRY)
 
@@ -4129,8 +4102,7 @@ type Mutation
   """Added in 25.13.0"""
   deleteAutoScalingRule(input: DeleteAutoScalingRuleInput!): DeleteAutoScalingRulePayload! @join__field(graph: STRAWBERRY)
 
-  """Added in 25.13.0"""
->>>>>>> Stashed changes
+  """Added in 25.14.0"""
   deleteObjectStorage(input: DeleteObjectStorageInput!): DeleteObjectStoragePayload! @join__field(graph: STRAWBERRY)
 
   """Added in 25.14.0"""
@@ -4590,7 +4562,7 @@ type Query
     """Added in 24.03.1"""
     id: String
     reference: String
-    architecture: String = "x86_64"
+    architecture: String = "aarch64"
   ): Image @join__field(graph: GRAPHENE)
   images(
     """
@@ -4886,21 +4858,19 @@ type Query
   """Added in 25.14.0"""
   huggingfaceRegistries(before: String = null, after: String = null, first: Int = null, last: Int = null, offset: Int = null, limit: Int = null): HuggingFaceRegistryConnection! @join__field(graph: STRAWBERRY)
 
-<<<<<<< Updated upstream
   """Added in 25.14.0"""
   reservoirRegistry(id: ID!): ReservoirRegistry @join__field(graph: STRAWBERRY)
 
   """Added in 25.14.0"""
   reservoirRegistries(before: String = null, after: String = null, first: Int = null, last: Int = null, offset: Int = null, limit: Int = null): ReservoirRegistryConnection! @join__field(graph: STRAWBERRY)
-=======
+
   """
   Added in 25.13.0 Get configuration JSON Schemas for all inference runtimes
   """
   inferenceRuntimeConfigs: JSON! @join__field(graph: STRAWBERRY)
 
   """Added in 25.13.0. Get JSON Schema for inference runtime configuration"""
   inferenceRuntimeConfig(name: String!): JSON! @join__field(graph: STRAWBERRY)
->>>>>>> Stashed changes
 }
 
 type QuotaDetails
@@ -5033,7 +5003,6 @@ type RescanImages
   task_id: UUID
 }
 
-<<<<<<< Updated upstream
 """Added in 25.14.0"""
 type ReservoirRegistry implements Node
   @join__implements(graph: STRAWBERRY, interface: "Node")
@@ -5071,9 +5040,7 @@ type ReservoirRegistryEdge
   node: ReservoirRegistry!
 }
 
-=======
 """Added in 25.13.0"""
->>>>>>> Stashed changes
 type ResourceConfig
   @join__type(graph: STRAWBERRY)
 {