gql schema changes

dagster-io · Jan 15, 2025 · f8d1214 · f8d1214
1 parent e8fbe15
commit f8d1214
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
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
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
diff --git a/python_modules/dagster-graphql/dagster_graphql/schema/instigation.py b/python_modules/dagster-graphql/dagster_graphql/schema/instigation.py
@@ -249,7 +249,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)
 
@@ -317,9 +319,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()