From c31e168f57345db6fe2c6c1c327a1fcd25b82f97 Mon Sep 17 00:00:00 2001
From: JamieDeMaria <jamie@dagsterlabs.com>
Date: Tue, 14 Jan 2025 10:22:51 -0500
Subject: [PATCH] gql schema changes

---
 .../ui-core/src/graphql/schema.graphql        |  2 ++
 .../packages/ui-core/src/graphql/types.ts     | 10 +++++++++
 .../dagster_graphql/schema/instigation.py     | 21 +++++++++++++++++++
 3 files changed, 33 insertions(+)

diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql b/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
index 17eb08bdcaa32..2628e3cbe2320 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
+++ b/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
@@ -2627,7 +2627,9 @@ type InstigationTick {
   endTimestamp: Float
   requestedAssetKeys: [AssetKey!]!
   requestedAssetMaterializationCount: Int!
+  submittedAssetMaterializationCount: Int!
   requestedMaterializationsForAssets: [RequestedMaterializationsForAsset!]!
+  submittedMaterializationsForAssets: [RequestedMaterializationsForAsset!]!
   autoMaterializeAssetEvaluationId: ID
   instigationType: InstigationType!
 }
diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts b/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
index 3834fb8da919b..2fc2d45391868 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
+++ b/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
@@ -2011,6 +2011,8 @@ export type InstigationTick = {
   runs: Array<Run>;
   skipReason: Maybe<Scalars['String']['output']>;
   status: InstigationTickStatus;
+  submittedAssetMaterializationCount: Scalars['Int']['output'];
+  submittedMaterializationsForAssets: Array<RequestedMaterializationsForAsset>;
   tickId: Scalars['ID']['output'];
   timestamp: Scalars['Float']['output'];
 };
@@ -9196,6 +9198,14 @@ export const buildInstigationTick = (
       overrides && overrides.hasOwnProperty('status')
         ? overrides.status!
         : InstigationTickStatus.FAILURE,
+    submittedAssetMaterializationCount:
+      overrides && overrides.hasOwnProperty('submittedAssetMaterializationCount')
+        ? overrides.submittedAssetMaterializationCount!
+        : 5903,
+    submittedMaterializationsForAssets:
+      overrides && overrides.hasOwnProperty('submittedMaterializationsForAssets')
+        ? overrides.submittedMaterializationsForAssets!
+        : [],
     tickId:
       overrides && overrides.hasOwnProperty('tickId')
         ? overrides.tickId!
diff --git a/python_modules/dagster-graphql/dagster_graphql/schema/instigation.py b/python_modules/dagster-graphql/dagster_graphql/schema/instigation.py
index 622f0ea1792b3..b4ebbcbfb339a 100644
--- a/python_modules/dagster-graphql/dagster_graphql/schema/instigation.py
+++ b/python_modules/dagster-graphql/dagster_graphql/schema/instigation.py
@@ -250,7 +250,9 @@ class GrapheneInstigationTick(graphene.ObjectType):
     endTimestamp = graphene.Field(graphene.Float)
     requestedAssetKeys = non_null_list(GrapheneAssetKey)
     requestedAssetMaterializationCount = graphene.NonNull(graphene.Int)
+    submittedAssetMaterializationCount = graphene.NonNull(graphene.Int)
     requestedMaterializationsForAssets = non_null_list(GrapheneRequestedMaterializationsForAsset)
+    submittedMaterializationsForAssets = non_null_list(GrapheneRequestedMaterializationsForAsset)
     autoMaterializeAssetEvaluationId = graphene.Field(graphene.ID)
     instigationType = graphene.NonNull(GrapheneInstigationType)
 
@@ -322,9 +324,28 @@ def resolve_requestedMaterializationsForAssets(self, _):
             for asset_key, partition_keys in self._tick.requested_assets_and_partitions.items()
         ]
 
+    def resolve_submittedMaterializationsForAssets(self, _):
+        """The asset materializations that were successfully submitted for materialization
+        in this tick. May not be the same as requestedMaterializationsForAssets if the tick was terminated
+        mid-iteration.
+        """
+        return [
+            GrapheneRequestedMaterializationsForAsset(
+                assetKey=GrapheneAssetKey(path=asset_key.path), partitionKeys=list(partition_keys)
+            )
+            for asset_key, partition_keys in self._tick.submitted_assets_and_partitions.items()
+        ]
+
     def resolve_requestedAssetMaterializationCount(self, _):
         return self._tick.requested_asset_materialization_count
 
+    def resolve_submittedAssetMaterializationCount(self, _):
+        """The number of asset materializations that were successfully submitted for materialization
+        in this tick. May not be the same as requestedAssetMaterializationCount if the tick was terminated
+        mid-iteration.
+        """
+        return self._tick.submitted_asset_materialization_count
+
 
 class GrapheneDryRunInstigationTick(graphene.ObjectType):
     timestamp = graphene.Float()