run graphql

dagster-io · Jan 16, 2025 · a81756e · a81756e
1 parent eb04b1b
commit a81756e
Show file tree

Hide file tree

Showing 3 changed files with 13 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/pipelines/pipeline.py b/python_modules/dagster-graphql/dagster_graphql/schema/pipelines/pipeline.py
@@ -389,6 +389,7 @@ class GrapheneRun(graphene.ObjectType):
     hasDeletePermission = graphene.NonNull(graphene.Boolean)
     hasConcurrencyKeySlots = graphene.NonNull(graphene.Boolean)
     rootConcurrencyKeys = graphene.List(graphene.NonNull(graphene.String))
+    allConcurrencyKeys = graphene.List(graphene.NonNull(graphene.String))
     hasUnconstrainedRootNodes = graphene.NonNull(graphene.Boolean)
     hasRunMetricsEnabled = graphene.NonNull(graphene.Boolean)
 
@@ -622,6 +623,12 @@ def resolve_hasUnconstrainedRootNodes(self, graphene_info: ResolveInfo):
 
         return False
 
+    def resolve_allConcurrencyKeys(self, graphene_info: ResolveInfo):
+        if not self.dagster_run.run_op_concurrency:
+            return None
+
+        return list(self.dagster_run.run_op_concurrency.all_pools)
+
     def resolve_rootConcurrencyKeys(self, graphene_info: ResolveInfo):
         if not self.dagster_run.run_op_concurrency:
             return None