.

dagster-io · Nov 17, 2024 · 0bdb899 · 0bdb899
1 parent 9ea4930
commit 0bdb899
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/js_modules/dagster-ui/packages/ui-core/src/asset-selection/useAssetSelectionFiltering.tsx b/js_modules/dagster-ui/packages/ui-core/src/asset-selection/useAssetSelectionFiltering.tsx
@@ -46,5 +46,10 @@ export const useAssetSelectionFiltering = <
     );
   }, [graphAssetKeys, assetsByKey]);
 
-  return {filtered, fetchResult, graphQueryItems};
+  const filteredByKey = useMemo(
+    () => Object.fromEntries(filtered.map((asset) => [tokenForAssetKey(asset.key), asset])),
+    [filtered],
+  );
+
+  return {filtered, filteredByKey, fetchResult, graphAssetKeys, graphQueryItems};
 };
diff --git a/js_modules/dagster-ui/packages/ui-core/src/search/types.ts b/js_modules/dagster-ui/packages/ui-core/src/search/types.ts
@@ -9,7 +9,7 @@ import {
   SearchScheduleFragment,
   SearchSensorFragment,
 } from './types/useGlobalSearch.types';
-import {DefinitionTag} from '../graphql/types';
+import {AssetKey, DefinitionTag} from '../graphql/types';
 
 export enum SearchResultType {
   AssetGroup,
@@ -54,6 +54,7 @@ export function isAssetFilterSearchResultType(
 }
 
 export type SearchResult = {
+  key?: AssetKey;
   label: string;
   description: string;
   href: string;

diff --git a/js_modules/dagster-ui/packages/ui-core/src/search/useGlobalSearch.tsx b/js_modules/dagster-ui/packages/ui-core/src/search/useGlobalSearch.tsx
@@ -208,6 +208,7 @@ const secondaryDataToSearchResults = (
   const assets: SearchResult[] = nodes
     .filter(({definition}) => definition !== null)
     .map((node) => ({
+      key: node.key,
       label: displayNameForAssetKey(node.key),
       description: `Asset in ${buildRepoPathForHuman(
         node.definition!.repository.name,