diff --git a/src/components/FlowRunGraph.vue b/src/components/FlowRunGraph.vue index ddd35f2d0..c403fc217 100644 --- a/src/components/FlowRunGraph.vue +++ b/src/components/FlowRunGraph.vue @@ -23,7 +23,6 @@ import { useColorTheme } from '@prefecthq/prefect-design' import { computed, ref } from 'vue' import FlowRunGraphConfirmation from '@/components/FlowRunGraphConfirmation.vue' - import { useCan } from '@/compositions' import { useTaskRunsCount } from '@/compositions/useTaskRunsCount' import { useWorkspaceApi } from '@/compositions/useWorkspaceApi' import { FlowRun } from '@/models/FlowRun' @@ -46,7 +45,6 @@ }>() const api = useWorkspaceApi() - const can = useCan() const { value: colorThemeValue } = useColorTheme() const load = ref(true) @@ -108,7 +106,7 @@ const config = computed(() => ({ runId: props.flowRun.id, fetch: (id) => { - return api.flowRuns.getFlowRunsGraph(id, Boolean(can.access.nestedTaskRunGraphs)) + return api.flowRuns.getFlowRunsGraph(id, { nestedTaskRunGraphs: true }) }, fetchEvents: props.fetchEvents, styles: { diff --git a/src/services/WorkspaceFlowRunsApi.ts b/src/services/WorkspaceFlowRunsApi.ts index ab06cbef0..c63e857ae 100644 --- a/src/services/WorkspaceFlowRunsApi.ts +++ b/src/services/WorkspaceFlowRunsApi.ts @@ -17,6 +17,10 @@ import { WorkspaceApi } from '@/services/WorkspaceApi' import { Schema, SchemaValues } from '@/types/schemas' import { toMap } from '@/utilities' +type GetFlowRunGraphOptions = { + nestedTaskRunGraphs: boolean, +} + export class WorkspaceFlowRunsApi extends WorkspaceApi { protected override routePrefix = '/flow_runs' @@ -77,7 +81,7 @@ export class WorkspaceFlowRunsApi extends WorkspaceApi { return data } - public async getFlowRunsGraph(id: string, nestedTaskRunGraphs: boolean): Promise { + public async getFlowRunsGraph(id: string, { nestedTaskRunGraphs }: GetFlowRunGraphOptions): Promise { const { data: graph } = await this.get(`/${id}/graph-v2`) return mapper.map('RunGraphDataResponse', { graph, nestedTaskRunGraphs }, 'RunGraphData') diff --git a/src/services/can.ts b/src/services/can.ts index 5ef208791..7ea057fc1 100644 --- a/src/services/can.ts +++ b/src/services/can.ts @@ -1,10 +1,7 @@ import { InjectionKey, ref } from 'vue' import { MaybeRef } from '@/types/reactivity' -export const workspaceFeatureFlags = [ - 'access:deploymentScheduleFlowRunInfraOverrides', - 'access:nestedTaskRunGraphs', -] as const satisfies Readonly<`access:${string}`[]> +export const workspaceFeatureFlags = ['access:deploymentScheduleFlowRunInfraOverrides'] as const satisfies Readonly<`access:${string}`[]> export type WorkspaceFeatureFlag = typeof workspaceFeatureFlags[number]