From 69f0d90d650c89f1763791c62c1de967e0e10bfc Mon Sep 17 00:00:00 2001
From: Marco Salazar <salazarm93@gmail.com>
Date: Tue, 15 Apr 2025 16:04:13 -0400
Subject: [PATCH] also gate health data

---
 .../ui-core/src/asset-data/AssetHealthDataProvider.tsx       | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/js_modules/dagster-ui/packages/ui-core/src/asset-data/AssetHealthDataProvider.tsx b/js_modules/dagster-ui/packages/ui-core/src/asset-data/AssetHealthDataProvider.tsx
index 420ef72391017..a327d8084dc72 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/asset-data/AssetHealthDataProvider.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/asset-data/AssetHealthDataProvider.tsx
@@ -1,4 +1,5 @@
 import React from 'react';
+import {FeatureFlag} from 'shared/app/FeatureFlags.oss';
 
 import {ApolloClient, gql, useApolloClient} from '../apollo-client';
 import {AssetBaseData} from './AssetBaseDataProvider';
@@ -8,6 +9,7 @@ import {liveDataFactory} from '../live-data-provider/Factory';
 import {LiveDataThreadID} from '../live-data-provider/LiveDataThread';
 import {useBlockTraceUntilTrue} from '../performance/TraceContext';
 import {AssetHealthQuery, AssetHealthQueryVariables} from './types/AssetHealthDataProvider.types';
+import {featureEnabled} from '../app/Flags';
 
 function init() {
   return liveDataFactory(
@@ -15,6 +17,9 @@ function init() {
       return useApolloClient();
     },
     async (keys, client: ApolloClient<any>) => {
+      if (!featureEnabled(FeatureFlag.flagUseNewObserveUIs)) {
+        return Object.fromEntries(keys.map((key) => [key, null]));
+      }
       const assetKeys = keys.map(tokenToAssetKey);
       const healthResponse = await client.query<AssetHealthQuery, AssetHealthQueryVariables>({
         query: ASSETS_HEALTH_INFO_QUERY,