diff --git a/js_modules/dagster-ui/packages/ui-core/src/app/ExecutionSessionStorage.tsx b/js_modules/dagster-ui/packages/ui-core/src/app/ExecutionSessionStorage.tsx
index a2ed8f2367f13..32d449cfaec4c 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/app/ExecutionSessionStorage.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/app/ExecutionSessionStorage.tsx
@@ -31,8 +31,22 @@ export interface PipelineRunTag {
 }
 
 export type SessionBase =
-  | {presetName: string; tags: PipelineRunTag[] | null}
-  | {partitionsSetName: string; partitionName: string | null; tags: PipelineRunTag[] | null};
+  | {
+      type: 'preset';
+      presetName: string;
+      tags: PipelineRunTag[] | null;
+    }
+  | {
+      type: 'asset-job-partition';
+      partitionName: string | null;
+      tags: PipelineRunTag[] | null;
+    }
+  | {
+      type: 'op-job-partition-set';
+      partitionsSetName: string;
+      partitionName: string | null;
+      tags: PipelineRunTag[] | null;
+    };
 
 export interface IExecutionSession {
   key: string;
@@ -230,7 +244,7 @@ export const useInitialDataForMode = (
   partitionSets: LaunchpadSessionPartitionSetsFragment,
   rootDefaultYaml: string | undefined,
   shouldPopulateWithDefaults: boolean,
-) => {
+): {base?: SessionBase; runConfigYaml?: string} => {
   const {isJob, isAssetJob, presets} = pipeline;
   const partitionSetsForMode = partitionSets.results;
 
@@ -242,14 +256,23 @@ export const useInitialDataForMode = (
     // `default` preset
     if (presetsForMode.length === 1 && (isAssetJob || partitionSetsForMode.length === 0)) {
       return {
-        base: {presetName: presetsForMode[0]!.name, tags: null},
+        base: {
+          type: 'preset',
+          presetName: presetsForMode[0]!.name,
+          tags: null,
+        },
         runConfigYaml: presetsForMode[0]!.runConfigYaml,
       };
     }
 
     if (!presetsForMode.length && partitionSetsForMode.length === 1) {
       return {
-        base: {partitionsSetName: partitionSetsForMode[0]!.name, partitionName: null, tags: null},
+        base: {
+          type: 'op-job-partition-set',
+          partitionsSetName: partitionSetsForMode[0]!.name,
+          partitionName: null,
+          tags: null,
+        },
         runConfigYaml: rootDefaultYaml,
       };
     }
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetChoosePartitionsDialog.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetChoosePartitionsDialog.tsx
index 1a0c043276cfb..e4131e39df565 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetChoosePartitionsDialog.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetChoosePartitionsDialog.tsx
@@ -47,7 +47,6 @@ import {PartitionDimensionSelection, usePartitionHealthData} from './usePartitio
 import {showCustomAlert} from '../app/CustomAlertProvider';
 import {PipelineRunTag} from '../app/ExecutionSessionStorage';
 import {usePermissionsForLocation} from '../app/Permissions';
-import {PythonErrorInfo} from '../app/PythonErrorInfo';
 import {
   displayNameForAssetKey,
   isHiddenAssetGroupJob,
@@ -60,13 +59,9 @@ import {
   LaunchPartitionBackfillMutation,
   LaunchPartitionBackfillMutationVariables,
 } from '../instance/backfill/types/BackfillUtils.types';
-import {CONFIG_PARTITION_SELECTION_QUERY} from '../launchpad/ConfigEditorConfigPicker';
+import {fetchTagsAndConfigForAssetJob} from '../launchpad/ConfigFetch';
 import {useLaunchPadHooks} from '../launchpad/LaunchpadHooksContext';
 import {TagContainer, TagEditor} from '../launchpad/TagEditor';
-import {
-  ConfigPartitionSelectionQuery,
-  ConfigPartitionSelectionQueryVariables,
-} from '../launchpad/types/ConfigEditorConfigPicker.types';
 import {
   DAEMON_NOT_RUNNING_ALERT_INSTANCE_FRAGMENT,
   DaemonNotRunningAlert,
@@ -86,7 +81,7 @@ import {RepoAddress} from '../workspace/types';
 
 const MISSING_FAILED_STATUSES = [AssetPartitionStatus.MISSING, AssetPartitionStatus.FAILED];
 
-interface Props {
+export interface LaunchAssetChoosePartitionsDialogProps {
   open: boolean;
   setOpen: (open: boolean) => void;
   repoAddress: RepoAddress;
@@ -99,7 +94,9 @@ interface Props {
   refetch?: () => Promise<void>;
 }
 
-export const LaunchAssetChoosePartitionsDialog = (props: Props) => {
+export const LaunchAssetChoosePartitionsDialog = (
+  props: LaunchAssetChoosePartitionsDialogProps,
+) => {
   const displayName =
     props.assets.length > 1
       ? `${props.assets.length} assets`
@@ -135,7 +132,7 @@ const LaunchAssetChoosePartitionsDialogBody = ({
   target,
   upstreamAssetKeys,
   refetch: _refetch,
-}: Props) => {
+}: LaunchAssetChoosePartitionsDialogProps) => {
   const partitionedAssets = assets.filter((a) => !!a.partitionDefinition);
 
   const {
@@ -269,50 +266,19 @@ const LaunchAssetChoosePartitionsDialogBody = ({
       });
     }
 
-    const {data: tagAndConfigData} = await client.query<
-      ConfigPartitionSelectionQuery,
-      ConfigPartitionSelectionQueryVariables
-    >({
-      query: CONFIG_PARTITION_SELECTION_QUERY,
-      fetchPolicy: 'network-only',
-      variables: {
-        repositorySelector: {
-          repositoryLocationName: repoAddress.location,
-          repositoryName: repoAddress.name,
-        },
-        partitionSetName: target.partitionSetName,
-        partitionName: keysFiltered[0]!,
-      },
+    const config = await fetchTagsAndConfigForAssetJob(client, {
+      partitionName: keysFiltered[0]!,
+      repositoryLocationName: repoAddress.location,
+      repositoryName: repoAddress.name,
+      assetKeys: target.assetKeys,
+      jobName: target.jobName,
     });
-
-    if (
-      !tagAndConfigData ||
-      !tagAndConfigData.partitionSetOrError ||
-      tagAndConfigData.partitionSetOrError.__typename !== 'PartitionSet' ||
-      !tagAndConfigData.partitionSetOrError.partition
-    ) {
-      return;
-    }
-
-    const {partition} = tagAndConfigData.partitionSetOrError;
-
-    if (partition.tagsOrError.__typename === 'PythonError') {
-      showCustomAlert({
-        title: 'Unable to load tags',
-        body: <PythonErrorInfo error={partition.tagsOrError} />,
-      });
-      return;
-    }
-    if (partition.runConfigOrError.__typename === 'PythonError') {
-      showCustomAlert({
-        title: 'Unable to load tags',
-        body: <PythonErrorInfo error={partition.runConfigOrError} />,
-      });
+    if (!config) {
       return;
     }
 
-    const runConfigData = partition.runConfigOrError.yaml || '';
-    let allTags = [...partition.tagsOrError.results, ...tags];
+    const runConfigData = config.yaml || '';
+    let allTags = [...config.tags, ...tags];
 
     if (launchWithRangesAsTags) {
       allTags = allTags.filter((t) => !t.key.startsWith(DagsterTag.Partition));
@@ -331,7 +297,6 @@ const LaunchAssetChoosePartitionsDialogBody = ({
         executionParams: {
           ...executionParamsForAssetJob(repoAddress, target.jobName, assets, allTags),
           runConfigData,
-          mode: partition.mode,
         },
       },
       'toast',
@@ -351,7 +316,8 @@ const LaunchAssetChoosePartitionsDialogBody = ({
             partitionNames: keysFiltered,
             fromFailure: false,
             selector: {
-              partitionSetName: target.partitionSetName,
+              // Todo: Fix after PR #23720 merges
+              partitionSetName: `${target.jobName}_partition_set`,
               repositorySelector: {
                 repositoryLocationName: repoAddress.location,
                 repositoryName: repoAddress.name,
@@ -618,9 +584,8 @@ const LaunchAssetChoosePartitionsDialogBody = ({
       <DialogFooter
         topBorder={!previewNotice}
         left={
-          'partitionSetName' in target && (
-            <RunningBackfillsNotice partitionSetName={target.partitionSetName} />
-          )
+          'assetKeys' in target &&
+          target.assetKeys && <RunningBackfillsNotice assetSelection={target.assetKeys} />
         }
       >
         <Button intent="none" onClick={() => setOpen(false)}>
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetExecutionButton.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetExecutionButton.tsx
index 61b1b7f351dd3..9518163e1d496 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetExecutionButton.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/LaunchAssetExecutionButton.tsx
@@ -57,7 +57,7 @@ import {repoAddressAsHumanString} from '../workspace/repoAddressAsString';
 import {RepoAddress} from '../workspace/types';
 
 export type LaunchAssetsChoosePartitionsTarget =
-  | {type: 'job'; jobName: string; partitionSetName: string}
+  | {type: 'job'; jobName: string; assetKeys: AssetKey[]}
   | {type: 'pureWithAnchorAsset'; anchorAssetKey: AssetKey}
   | {type: 'pureAll'};
 
@@ -525,12 +525,6 @@ async function stateForLaunchingAssets(
   if (pipeline.__typename !== 'Pipeline') {
     return {type: 'error', error: pipeline.message};
   }
-  const partitionSets = resourceResult.data.partitionSetsOrError;
-  if (partitionSets.__typename !== 'PartitionSets') {
-    return {type: 'error', error: partitionSets.message};
-  }
-
-  const partitionSetName = partitionSets.results[0]?.name;
   const requiredResourceKeys = assets.flatMap((a) => a.requiredResources.map((r) => r.resourceKey));
   const resources = pipeline.modes[0]!.resources.filter((r) =>
     requiredResourceKeys.includes(r.name),
@@ -565,17 +559,17 @@ async function stateForLaunchingAssets(
         ),
         includeSeparatelyExecutableChecks: true,
         solidSelectionQuery: assetOpNames.map((name) => `"${name}"`).join(', '),
-        base: partitionSetName
-          ? {partitionsSetName: partitionSetName, partitionName: null, tags: []}
+        base: partitionDefinition
+          ? {type: 'asset-job-partition', partitionName: null, tags: []}
           : undefined,
       },
     };
   }
-  if (partitionDefinition && partitionSetName) {
+  if (partitionDefinition) {
     return {
       type: 'partitions',
       assets,
-      target: {type: 'job', jobName, partitionSetName},
+      target: {type: 'job', jobName, assetKeys: assets.map(asAssetKeyInput)},
       upstreamAssetKeys: getUpstreamAssetKeys(assets),
       repoAddress,
     };
@@ -866,27 +860,6 @@ export const LAUNCH_ASSET_LOADER_RESOURCE_QUERY = gql`
     $repositoryLocationName: String!
     $repositoryName: String!
   ) {
-    partitionSetsOrError(
-      pipelineName: $pipelineName
-      repositorySelector: {
-        repositoryName: $repositoryName
-        repositoryLocationName: $repositoryLocationName
-      }
-    ) {
-      ... on PythonError {
-        message
-      }
-      ... on PipelineNotFoundError {
-        message
-      }
-      ... on PartitionSets {
-        results {
-          id
-          name
-        }
-      }
-    }
-
     pipelineOrError(
       params: {
         pipelineName: $pipelineName
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/RunningBackfillsNotice.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/RunningBackfillsNotice.tsx
index 5697400afb19b..ac12c9b07a82c 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/RunningBackfillsNotice.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/RunningBackfillsNotice.tsx
@@ -1,14 +1,17 @@
 import {gql, useQuery} from '@apollo/client';
 import {Box, Colors, Icon} from '@dagster-io/ui-components';
+import {useMemo} from 'react';
 import {Link} from 'react-router-dom';
 
 import {
   RunningBackfillsNoticeQuery,
   RunningBackfillsNoticeQueryVariables,
 } from './types/RunningBackfillsNotice.types';
+import {tokenForAssetKey} from '../asset-graph/Utils';
+import {AssetKeyInput} from '../graphql/types';
 import {useBlockTraceOnQueryResult} from '../performance/TraceContext';
 
-export const RunningBackfillsNotice = ({partitionSetName}: {partitionSetName: string}) => {
+export const RunningBackfillsNotice = ({assetSelection}: {assetSelection: AssetKeyInput[]}) => {
   const queryResult = useQuery<RunningBackfillsNoticeQuery, RunningBackfillsNoticeQueryVariables>(
     RUNNING_BACKFILLS_NOTICE_QUERY,
   );
@@ -20,18 +23,24 @@ export const RunningBackfillsNotice = ({partitionSetName}: {partitionSetName: st
       ? data.partitionBackfillsOrError.results
       : [];
 
+  const assetSelectionTokens = useMemo(
+    () => new Set(assetSelection.map(tokenForAssetKey)),
+    [assetSelection],
+  );
+
   const runningBackfillCount = runningBackfills.filter(
-    (r) => r.partitionSetName === partitionSetName,
+    (r) => r.assetSelection?.some((a) => assetSelectionTokens.has(tokenForAssetKey(a))),
   ).length;
 
   if (runningBackfillCount === 0) {
     return <span />;
   }
+
   return (
     <div style={{color: Colors.textLight(), maxWidth: 350}}>
       {runningBackfillCount === 1
-        ? 'Note: A backfill has been requested for this job and may be refreshing displayed assets. '
-        : `Note: ${runningBackfillCount} backfills have been requested for this job and may be refreshing displayed assets. `}
+        ? `Note: A backfill has been requested and may be refreshing displayed assets. `
+        : `Note: ${runningBackfillCount} backfills have been requested and may be refreshing displayed assets. `}
       <Link to="/overview/backfills" target="_blank">
         <Box flex={{gap: 4, display: 'inline-flex', alignItems: 'center'}}>
           View <Icon name="open_in_new" color={Colors.linkDefault()} />
@@ -48,6 +57,9 @@ export const RUNNING_BACKFILLS_NOTICE_QUERY = gql`
         results {
           id
           partitionSetName
+          assetSelection {
+            path
+          }
         }
       }
     }
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetChoosePartitionsDialog.fixtures.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetChoosePartitionsDialog.fixtures.tsx
index e7d88d0b361fd..7ad98b797597d 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetChoosePartitionsDialog.fixtures.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetChoosePartitionsDialog.fixtures.tsx
@@ -1,6 +1,7 @@
 import {assetNodes} from './LaunchAssetLoaderQuery.fixtures';
+import {LaunchAssetChoosePartitionsDialogProps} from '../LaunchAssetChoosePartitionsDialog';
 
-export const ReleasesJobProps = {
+export const ReleasesJobProps: Omit<LaunchAssetChoosePartitionsDialogProps, 'open' | 'setOpen'> = {
   assets: assetNodes,
   upstreamAssetKeys: [],
   repoAddress: {
@@ -10,7 +11,6 @@ export const ReleasesJobProps = {
   target: {
     type: 'job' as const,
     jobName: '__ASSET_JOB_0',
-    partitionSetName: '__ASSET_JOB_0_partition_set',
+    assetKeys: [{path: ['asset_key_1']}],
   },
-  open: true,
 };
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetExecutionButton.fixtures.ts b/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetExecutionButton.fixtures.ts
index 52516f437bc3d..4a70909d6782d 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetExecutionButton.fixtures.ts
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/__fixtures__/LaunchAssetExecutionButton.fixtures.ts
@@ -21,8 +21,10 @@ import {
   buildMaterializationEvent,
   buildMode,
   buildPartitionDefinition,
-  buildPartitionSet,
-  buildPartitionSets,
+  buildPartitionRunConfig,
+  buildPartitionTags,
+  buildPartitionTagsAndConfig,
+  buildPipeline,
   buildRegularConfigType,
   buildRepository,
   buildRepositoryLocation,
@@ -32,8 +34,8 @@ import {
 } from '../../graphql/types';
 import {LAUNCH_PARTITION_BACKFILL_MUTATION} from '../../instance/backfill/BackfillUtils';
 import {LaunchPartitionBackfillMutation} from '../../instance/backfill/types/BackfillUtils.types';
-import {CONFIG_PARTITION_SELECTION_QUERY} from '../../launchpad/ConfigEditorConfigPicker';
-import {ConfigPartitionSelectionQuery} from '../../launchpad/types/ConfigEditorConfigPicker.types';
+import {CONFIG_PARTITION_FOR_ASSET_JOB_QUERY} from '../../launchpad/ConfigFetch';
+import {ConfigPartitionForAssetJobQuery} from '../../launchpad/types/ConfigFetch.types';
 import {LAUNCH_PIPELINE_EXECUTION_MUTATION} from '../../runs/RunUtils';
 import {
   LaunchPipelineExecutionMutation,
@@ -94,7 +96,7 @@ export const UNPARTITIONED_ASSET = buildAssetNode({
   dependencyKeys: [],
   dependedByKeys: [],
   graphName: null,
-  jobNames: ['__ASSET_JOB_7', 'my_asset_job'],
+  jobNames: ['__ASSET_JOB', 'my_asset_job'],
   opNames: ['unpartitioned_asset'],
   opVersion: null,
   description: null,
@@ -114,7 +116,7 @@ export const UNPARTITIONED_ASSET = buildAssetNode({
 export const CHECKED_ASSET = buildAssetNode({
   ...UNPARTITIONED_ASSET,
   id: 'test.py.repo.["checked_asset"]',
-  jobNames: ['__ASSET_JOB_7', 'checks_included_job', 'checks_excluded_job'],
+  jobNames: ['__ASSET_JOB', 'checks_included_job', 'checks_excluded_job'],
   assetKey: buildAssetKey({path: ['checked_asset']}),
   configField: BASE_CONFIG_TYPE_FIELD,
   assetChecksOrError: buildAssetChecks({
@@ -165,14 +167,14 @@ export const UNPARTITIONED_ASSET_WITH_REQUIRED_CONFIG = buildAssetNode({
 export const MULTI_ASSET_OUT_1 = buildAssetNode({
   ...UNPARTITIONED_ASSET,
   id: 'test.py.repo.["multi_asset_out_1"]',
-  jobNames: ['__ASSET_JOB_7'],
+  jobNames: ['__ASSET_JOB'],
   assetKey: buildAssetKey({path: ['multi_asset_out_1']}),
 });
 
 export const MULTI_ASSET_OUT_2 = buildAssetNode({
   ...UNPARTITIONED_ASSET,
   id: 'test.py.repo.["multi_asset_out_2"]',
-  jobNames: ['__ASSET_JOB_7'],
+  jobNames: ['__ASSET_JOB'],
   assetKey: buildAssetKey({path: ['multi_asset_out_2']}),
 });
 
@@ -189,7 +191,7 @@ export const ASSET_DAILY = buildAssetNode({
   dependencyKeys: [],
   dependedByKeys: [{__typename: 'AssetKey', path: ['asset_weekly']}],
   graphName: null,
-  jobNames: ['__ASSET_JOB_7', 'my_asset_job'],
+  jobNames: ['__ASSET_JOB', 'my_asset_job'],
   opNames: ['asset_daily'],
   opVersion: null,
   description: null,
@@ -442,7 +444,6 @@ export const buildLaunchAssetLoaderGenericJobMock = (jobName: string) => {
     result: {
       data: {
         __typename: 'Query',
-        partitionSetsOrError: buildPartitionSets({results: []}),
         pipelineOrError: {
           id: '8e2d3f9597c4a45bb52fe9ab5656419f4329d4fb',
           modes: [
@@ -463,7 +464,7 @@ export const LaunchAssetLoaderResourceJob7Mock: MockedResponse<LaunchAssetLoader
   request: {
     query: LAUNCH_ASSET_LOADER_RESOURCE_QUERY,
     variables: {
-      pipelineName: '__ASSET_JOB_7',
+      pipelineName: '__ASSET_JOB',
       repositoryLocationName: 'test.py',
       repositoryName: 'repo',
     },
@@ -471,15 +472,6 @@ export const LaunchAssetLoaderResourceJob7Mock: MockedResponse<LaunchAssetLoader
   result: {
     data: {
       __typename: 'Query',
-      partitionSetsOrError: {
-        results: [
-          buildPartitionSet({
-            id: '5b10aae97b738c48a4262b1eca530f89b13e9afc',
-            name: '__ASSET_JOB_7_partition_set',
-          }),
-        ],
-        __typename: 'PartitionSets',
-      },
       pipelineOrError: {
         id: '8e2d3f9597c4a45bb52fe9ab5656419f4329d4fb',
         modes: [
@@ -575,15 +567,6 @@ export const LaunchAssetLoaderResourceJob8Mock: MockedResponse<LaunchAssetLoader
   result: {
     data: {
       __typename: 'Query',
-      partitionSetsOrError: {
-        results: [
-          buildPartitionSet({
-            id: '129179973a9144278c2429d3ba680bf0f809a59b',
-            name: '__ASSET_JOB_8_partition_set',
-          }),
-        ],
-        __typename: 'PartitionSets',
-      },
       pipelineOrError: {
         id: '8689a9dcd052f769b73d73dfe57e89065dac369d',
         modes: [
@@ -680,15 +663,6 @@ export const LaunchAssetLoaderResourceMyAssetJobMock: MockedResponse<LaunchAsset
     result: {
       data: {
         __typename: 'Query',
-        partitionSetsOrError: {
-          results: [
-            buildPartitionSet({
-              id: '129179973a9144278c2429d3ba680bf0f809a59b',
-              name: 'my_asset_job_partition_set',
-            }),
-          ],
-          __typename: 'PartitionSets',
-        },
         pipelineOrError: {
           id: '8689a9dcd052f769b73d73dfe57e89065dac369d',
           modes: [
@@ -738,35 +712,30 @@ export const LaunchAssetCheckUpstreamWeeklyRootMock: MockedResponse<LaunchAssetC
 
 export function buildConfigPartitionSelectionLatestPartitionMock(
   partitionName: string,
-  partitionSetName: string,
-): MockedResponse<ConfigPartitionSelectionQuery> {
+  jobName: string,
+): MockedResponse<ConfigPartitionForAssetJobQuery> {
   return {
     request: {
-      query: CONFIG_PARTITION_SELECTION_QUERY,
+      query: CONFIG_PARTITION_FOR_ASSET_JOB_QUERY,
       variables: {
+        jobName,
         partitionName,
-        partitionSetName,
-        repositorySelector: {
-          repositoryLocationName: 'test.py',
-          repositoryName: 'repo',
-        },
+        repositoryLocationName: 'test.py',
+        repositoryName: 'repo',
+        assetKeys: [{path: ['asset_daily']}],
       },
     },
     result: {
       data: {
         __typename: 'Query',
-        partitionSetOrError: {
-          __typename: 'PartitionSet',
-          id: '5b10aae97b738c48a4262b1eca530f89b13e9afc',
-          partition: {
+        pipelineOrError: buildPipeline({
+          partition: buildPartitionTagsAndConfig({
             name: '2023-03-14',
-            solidSelection: null,
-            runConfigOrError: {
+            runConfigOrError: buildPartitionRunConfig({
               yaml: '{}\n',
               __typename: 'PartitionRunConfig',
-            },
-            mode: 'default',
-            tagsOrError: {
+            }),
+            tagsOrError: buildPartitionTags({
               results: [
                 {
                   key: 'dagster/partition',
@@ -775,15 +744,13 @@ export function buildConfigPartitionSelectionLatestPartitionMock(
                 },
                 {
                   key: 'dagster/partition_set',
-                  value: partitionSetName,
+                  value: `${jobName}_partition_set`,
                   __typename: 'PipelineTag',
                 },
               ],
-              __typename: 'PartitionTags',
-            },
-            __typename: 'Partition',
-          },
-        },
+            }),
+          }),
+        }),
       },
     },
   };
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetChoosePartitionsDialog.test.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetChoosePartitionsDialog.test.tsx
index d790f9317e470..96e67a6c16464 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetChoosePartitionsDialog.test.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetChoosePartitionsDialog.test.tsx
@@ -102,8 +102,8 @@ describe('launchAssetChoosePartitionsDialog', () => {
               setOpen={(_open: boolean) => {}}
               repoAddress={buildRepoAddress('test', 'test')}
               target={{
-                jobName: '__ASSET_JOB_0',
-                partitionSetName: '__ASSET_JOB_0_partition_set',
+                jobName: '__ASSET_JOB',
+                assetKeys: [assetA.assetKey, assetB.assetKey],
                 type: 'job',
               }}
               assets={[assetA, assetB]}
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetExecutionButton.test.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetExecutionButton.test.tsx
index 7a2a14d345fc0..7961f4c209ad4 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetExecutionButton.test.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/LaunchAssetExecutionButton.test.tsx
@@ -240,7 +240,7 @@ describe('LaunchAssetExecutionButton', () => {
         selector: {
           repositoryLocationName: 'test.py',
           repositoryName: 'repo',
-          pipelineName: '__ASSET_JOB_7',
+          pipelineName: '__ASSET_JOB',
           assetSelection: [{path: ['unpartitioned_asset']}],
           assetCheckSelection: [],
         },
@@ -375,7 +375,7 @@ describe('LaunchAssetExecutionButton', () => {
         executionMetadata: {
           tags: [
             {key: 'dagster/partition', value: '2023-02-22'},
-            {key: 'dagster/partition_set', value: '__ASSET_JOB_7_partition_set'},
+            {key: 'dagster/partition_set', value: '__ASSET_JOB_partition_set'},
           ],
         },
         mode: 'default',
@@ -383,7 +383,7 @@ describe('LaunchAssetExecutionButton', () => {
         selector: {
           assetSelection: [{path: ['asset_daily']}],
           assetCheckSelection: [],
-          pipelineName: '__ASSET_JOB_7',
+          pipelineName: '__ASSET_JOB',
           repositoryLocationName: 'test.py',
           repositoryName: 'repo',
         },
@@ -593,7 +593,7 @@ describe('LaunchAssetExecutionButton', () => {
         selector: {
           repositoryLocationName: 'test.py',
           repositoryName: 'repo',
-          pipelineName: '__ASSET_JOB_7',
+          pipelineName: '__ASSET_JOB',
           assetSelection: [
             asAssetKeyInput(MULTI_ASSET_OUT_1.assetKey),
             asAssetKeyInput(MULTI_ASSET_OUT_2.assetKey),
@@ -644,9 +644,9 @@ function renderButton({
     LaunchAssetCheckUpstreamWeeklyRootMock,
     ...PartitionHealthAssetMocks,
     buildLaunchAssetWarningsMock([]),
-    buildConfigPartitionSelectionLatestPartitionMock('2020-01-02', 'my_asset_job_partition_set'),
-    buildConfigPartitionSelectionLatestPartitionMock('2023-02-22', 'my_asset_job_partition_set'),
-    buildConfigPartitionSelectionLatestPartitionMock('2023-02-22', '__ASSET_JOB_7_partition_set'),
+    buildConfigPartitionSelectionLatestPartitionMock('2020-01-02', 'my_asset_job'),
+    buildConfigPartitionSelectionLatestPartitionMock('2023-02-22', 'my_asset_job'),
+    buildConfigPartitionSelectionLatestPartitionMock('2023-02-22', '__ASSET_JOB'),
     buildLaunchAssetLoaderMock([MULTI_ASSET_OUT_1.assetKey], {
       assetNodeAdditionalRequiredKeys: [MULTI_ASSET_OUT_2.assetKey],
     }),
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/types/LaunchAssetExecutionButton.types.ts b/js_modules/dagster-ui/packages/ui-core/src/assets/types/LaunchAssetExecutionButton.types.ts
index 5c19aa1207e4d..eb2a9cf449bf7 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/types/LaunchAssetExecutionButton.types.ts
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/types/LaunchAssetExecutionButton.types.ts
@@ -1871,13 +1871,6 @@ export type LaunchAssetLoaderResourceQueryVariables = Types.Exact<{
 
 export type LaunchAssetLoaderResourceQuery = {
   __typename: 'Query';
-  partitionSetsOrError:
-    | {
-        __typename: 'PartitionSets';
-        results: Array<{__typename: 'PartitionSet'; id: string; name: string}>;
-      }
-    | {__typename: 'PipelineNotFoundError'; message: string}
-    | {__typename: 'PythonError'; message: string};
   pipelineOrError:
     | {__typename: 'InvalidSubsetError'; message: string}
     | {
diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/types/RunningBackfillsNotice.types.ts b/js_modules/dagster-ui/packages/ui-core/src/assets/types/RunningBackfillsNotice.types.ts
index c0f01d715a94c..937d80662e604 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/types/RunningBackfillsNotice.types.ts
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/types/RunningBackfillsNotice.types.ts
@@ -13,6 +13,7 @@ export type RunningBackfillsNoticeQuery = {
           __typename: 'PartitionBackfill';
           id: string;
           partitionSetName: string | null;
+          assetSelection: Array<{__typename: 'AssetKey'; path: Array<string>}> | null;
         }>;
       }
     | {__typename: 'PythonError'};
diff --git a/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigEditorConfigPicker.tsx b/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigEditorConfigPicker.tsx
index b570ad9b84927..3f95d51b641ea 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigEditorConfigPicker.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigEditorConfigPicker.tsx
@@ -20,7 +20,8 @@ import styled from 'styled-components';
 import {
   ConfigEditorGeneratorPipelineFragment,
   ConfigEditorPipelinePresetFragment,
-  ConfigPartitionResultFragment,
+  ConfigPartitionsAssetsQuery,
+  ConfigPartitionsAssetsQueryVariables,
   ConfigPartitionsQuery,
   ConfigPartitionsQueryVariables,
   PartitionSetForConfigEditorFragment,
@@ -42,7 +43,6 @@ import {RepoAddress} from '../workspace/types';
 type Pipeline = ConfigEditorGeneratorPipelineFragment;
 type Preset = ConfigEditorPipelinePresetFragment;
 type PartitionSet = PartitionSetForConfigEditorFragment;
-type Partition = ConfigPartitionResultFragment;
 type ConfigGenerator = Preset | PartitionSet;
 
 interface ConfigEditorConfigPickerProps {
@@ -94,8 +94,10 @@ export const ConfigEditorConfigPicker = (props: ConfigEditorConfigPickerProps) =
     if ('presetName' in base) {
       return `Preset: ${base.presetName}`;
     }
-
-    return `Partition Set: ${base.partitionsSetName}`;
+    if ('partitionsSetName' in base) {
+      return `Partition Set: ${base.partitionsSetName}`;
+    }
+    return `Partitions`;
   };
 
   const onSelect = (item: ConfigGenerator) => {
@@ -103,6 +105,7 @@ export const ConfigEditorConfigPicker = (props: ConfigEditorConfigPickerProps) =
       onSaveSession({
         mode: item.mode,
         base: {
+          type: 'op-job-partition-set',
           partitionsSetName: item.name,
           partitionName: null,
           tags: base ? base.tags : null,
@@ -122,10 +125,10 @@ export const ConfigEditorConfigPicker = (props: ConfigEditorConfigPickerProps) =
           onSelect={onSelect}
         />
       )}
-      {base && 'partitionsSetName' in base ? (
+      {base && 'partitionName' in base ? (
         <ConfigEditorPartitionPicker
           pipeline={pipeline}
-          partitionSetName={base.partitionsSetName}
+          partitionSetName={'partitionsSetName' in base ? base.partitionsSetName : ''}
           value={base.partitionName}
           onSelect={onSelectPartition}
           repoAddress={repoAddress}
@@ -157,19 +160,33 @@ const ConfigEditorPartitionPicker = React.memo((props: ConfigEditorPartitionPick
   const {basePath} = React.useContext(AppContext);
   const repositorySelector = repoAddressToSelector(repoAddress);
 
-  const queryResult = useQuery<ConfigPartitionsQuery, ConfigPartitionsQueryVariables>(
+  const queryResultNoAssets = useQuery<ConfigPartitionsQuery, ConfigPartitionsQueryVariables>(
     CONFIG_PARTITIONS_QUERY,
     {
-      variables: {
-        repositorySelector,
-        partitionSetName,
-        assetKeys: assetSelection
-          ? assetSelection.map((selection) => ({path: selection.assetKey.path}))
-          : [],
-      },
+      skip: !!assetSelection,
+      variables: {repositorySelector, partitionSetName},
       fetchPolicy: 'network-only',
     },
   );
+
+  const queryResultAssets = useQuery<
+    ConfigPartitionsAssetsQuery,
+    ConfigPartitionsAssetsQueryVariables
+  >(CONFIG_PARTITIONS_ASSETS_QUERY, {
+    skip: !assetSelection,
+    variables: {
+      params: {
+        ...repositorySelector,
+        pipelineName: props.pipeline.name,
+      },
+      assetKeys: assetSelection
+        ? assetSelection.map((selection) => ({path: selection.assetKey.path}))
+        : [],
+    },
+    fetchPolicy: 'network-only',
+  });
+
+  const queryResult = assetSelection ? queryResultAssets : queryResultNoAssets;
   const {data, refetch, loading} = queryResult;
   useBlockTraceOnQueryResult(queryResult, 'ConfigPartitionsQuery');
 
@@ -182,26 +199,18 @@ const ConfigEditorPartitionPicker = React.memo((props: ConfigEditorPartitionPick
   );
 
   const doesAnyAssetHavePartitions = React.useMemo(
-    () => data?.assetNodes?.some((node) => !!node.partitionDefinition),
+    () =>
+      data && 'assetNodes' in data && data.assetNodes?.some((node) => !!node.partitionDefinition),
     [data],
   );
 
-  const partitions: Partition[] = React.useMemo(() => {
-    const retrieved =
-      data?.partitionSetOrError.__typename === 'PartitionSet' &&
-      data?.partitionSetOrError.partitionsOrError.__typename === 'Partitions'
-        ? data.partitionSetOrError.partitionsOrError.results
-        : [];
+  const partitions: string[] = React.useMemo(() => {
+    const retrieved = partitionKeysFromData(data);
     return sortOrder === 'asc' ? retrieved : [...retrieved].reverse();
   }, [data, sortOrder]);
 
-  const error =
-    data?.partitionSetOrError.__typename === 'PartitionSet' &&
-    data?.partitionSetOrError.partitionsOrError.__typename !== 'Partitions'
-      ? data.partitionSetOrError.partitionsOrError
-      : null;
-
-  const selected = partitions.find((p) => p.name === value);
+  const error = errorFromData(data);
+  const selected = value && partitions.includes(value) ? value : undefined;
 
   const onClickSort = React.useCallback(
     (event: React.MouseEvent<HTMLButtonElement>) => {
@@ -225,7 +234,7 @@ const ConfigEditorPartitionPicker = React.memo((props: ConfigEditorPartitionPick
   };
 
   const {isDynamicPartition, dynamicPartitionsDefinitionName} = React.useMemo(() => {
-    const assetNodes = data?.assetNodes;
+    const assetNodes = data && 'assetNodes' in data ? data?.assetNodes : undefined;
     const definition = assetNodes?.find((a) => !!a.partitionDefinition)?.partitionDefinition;
     if (
       !definition ||
@@ -240,7 +249,7 @@ const ConfigEditorPartitionPicker = React.memo((props: ConfigEditorPartitionPick
       isDynamicPartition: definition.type === PartitionDefinitionType.DYNAMIC,
       dynamicPartitionsDefinitionName: definition.name,
     };
-  }, [data?.assetNodes]);
+  }, [data]);
 
   const [showCreatePartition, setShowCreatePartition] = React.useState(false);
 
@@ -289,24 +298,24 @@ const ConfigEditorPartitionPicker = React.memo((props: ConfigEditorPartitionPick
   // and ensure it is re-applied to the internal state when it changes (via `key` below).
   return (
     <>
-      <Suggest<Partition>
-        key={selected ? selected.name : 'none'}
+      <Suggest<string>
+        key={selected ? selected : 'none'}
         defaultSelectedItem={selected}
         items={partitions}
         inputProps={inputProps}
-        inputValueRenderer={(partition) => partition.name}
-        itemPredicate={(query, partition) => query.length === 0 || partition.name.includes(query)}
+        inputValueRenderer={(partition) => partition}
+        itemPredicate={(query, partition) => query.length === 0 || partition.includes(query)}
         itemRenderer={(partition, props) => (
           <MenuItem
             active={props.modifiers.active}
             onClick={props.handleClick}
-            key={partition.name}
-            text={partition.name}
+            key={partition}
+            text={partition}
           />
         )}
         noResults={<MenuItem disabled={true} text="No presets." />}
         onItemSelect={(item) => {
-          onSelect(repositorySelector, partitionSetName, item.name);
+          onSelect(repositorySelector, partitionSetName, item);
         }}
       />
       {isDynamicPartition ? (
@@ -457,7 +466,6 @@ const CONFIG_PARTITIONS_QUERY = gql`
   query ConfigPartitionsQuery(
     $repositorySelector: RepositorySelector!
     $partitionSetName: String!
-    $assetKeys: [AssetKeyInput!]
   ) {
     partitionSetOrError(
       repositorySelector: $repositorySelector
@@ -475,13 +483,6 @@ const CONFIG_PARTITIONS_QUERY = gql`
         }
       }
     }
-    assetNodes(assetKeys: $assetKeys) {
-      id
-      partitionDefinition {
-        name
-        type
-      }
-    }
   }
 
   fragment ConfigPartitionResult on Partition {
@@ -491,40 +492,32 @@ const CONFIG_PARTITIONS_QUERY = gql`
   ${PYTHON_ERROR_FRAGMENT}
 `;
 
-export const CONFIG_PARTITION_SELECTION_QUERY = gql`
-  query ConfigPartitionSelectionQuery(
-    $repositorySelector: RepositorySelector!
-    $partitionSetName: String!
-    $partitionName: String!
-  ) {
-    partitionSetOrError(
-      repositorySelector: $repositorySelector
-      partitionSetName: $partitionSetName
-    ) {
-      ... on PartitionSet {
+const CONFIG_PARTITIONS_ASSETS_QUERY = gql`
+  query ConfigPartitionsAssetsQuery($params: PipelineSelector!, $assetKeys: [AssetKeyInput!]) {
+    pipelineOrError(params: $params) {
+      ...PythonErrorFragment
+      ... on PipelineNotFoundError {
+        message
+      }
+      ... on InvalidSubsetError {
+        message
+      }
+      ... on Pipeline {
         id
-        partition(partitionName: $partitionName) {
-          name
-          solidSelection
-          runConfigOrError {
-            ... on PartitionRunConfig {
-              yaml
-            }
-            ...PythonErrorFragment
-          }
-          mode
-          tagsOrError {
-            ... on PartitionTags {
-              results {
-                key
-                value
-              }
-            }
-            ...PythonErrorFragment
+        partitionKeysOrError(selectedAssetKeys: $assetKeys) {
+          ... on PartitionKeys {
+            partitionKeys
           }
         }
       }
     }
+    assetNodes(assetKeys: $assetKeys) {
+      id
+      partitionDefinition {
+        name
+        type
+      }
+    }
   }
 
   ${PYTHON_ERROR_FRAGMENT}
@@ -571,3 +564,44 @@ export const CONFIG_EDITOR_GENERATOR_PARTITION_SETS_FRAGMENT = gql`
     solidSelection
   }
 `;
+
+function partitionKeysFromData(
+  data: ConfigPartitionsAssetsQuery | ConfigPartitionsQuery | undefined,
+) {
+  if (!data) {
+    return [];
+  }
+
+  if (
+    'partitionSetOrError' in data &&
+    data.partitionSetOrError.__typename === 'PartitionSet' &&
+    data.partitionSetOrError.partitionsOrError.__typename === 'Partitions'
+  ) {
+    return data.partitionSetOrError.partitionsOrError.results.map((a) => a.name);
+  }
+  if ('pipelineOrError' in data && data.pipelineOrError.__typename === 'Pipeline') {
+    return data.pipelineOrError.partitionKeysOrError.partitionKeys;
+  }
+  return [];
+}
+
+function errorFromData(data: ConfigPartitionsAssetsQuery | ConfigPartitionsQuery | undefined) {
+  if (!data) {
+    return null;
+  }
+  if (
+    'partitionSetOrError' in data &&
+    data.partitionSetOrError.__typename === 'PartitionSet' &&
+    data.partitionSetOrError.partitionsOrError.__typename !== 'Partitions'
+  ) {
+    return data.partitionSetOrError.partitionsOrError;
+  }
+  if (
+    'pipelineOrError' in data &&
+    data.pipelineOrError &&
+    data.pipelineOrError.__typename !== 'Pipeline'
+  ) {
+    return data.pipelineOrError;
+  }
+  return null;
+}
diff --git a/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigFetch.tsx b/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigFetch.tsx
new file mode 100644
index 0000000000000..5dee4a82ff419
--- /dev/null
+++ b/js_modules/dagster-ui/packages/ui-core/src/launchpad/ConfigFetch.tsx
@@ -0,0 +1,185 @@
+import {ApolloClient, gql} from '@apollo/client';
+
+import {
+  ConfigPartitionForAssetJobQuery,
+  ConfigPartitionForAssetJobQueryVariables,
+  ConfigPartitionSelectionQuery,
+  ConfigPartitionSelectionQueryVariables,
+} from './types/ConfigFetch.types';
+import {showCustomAlert} from '../app/CustomAlertProvider';
+import {PYTHON_ERROR_FRAGMENT} from '../app/PythonErrorFragment';
+import {PythonErrorInfo} from '../app/PythonErrorInfo';
+
+export async function fetchTagsAndConfigForJob(
+  client: ApolloClient<any>,
+  variables: ConfigPartitionSelectionQueryVariables,
+) {
+  const {data} = await client.query<
+    ConfigPartitionSelectionQuery,
+    ConfigPartitionSelectionQueryVariables
+  >({
+    query: CONFIG_PARTITION_SELECTION_QUERY,
+    variables,
+  });
+
+  if (
+    !data ||
+    !data.partitionSetOrError ||
+    data.partitionSetOrError.__typename !== 'PartitionSet' ||
+    !data.partitionSetOrError.partition
+  ) {
+    return;
+  }
+
+  const {partition} = data.partitionSetOrError;
+
+  if (partition.tagsOrError.__typename === 'PythonError') {
+    showCustomAlert({
+      title: 'Unable to load tags',
+      body: <PythonErrorInfo error={partition.tagsOrError} />,
+    });
+    return;
+  }
+  if (partition.runConfigOrError.__typename === 'PythonError') {
+    showCustomAlert({
+      title: 'Unable to load tags',
+      body: <PythonErrorInfo error={partition.runConfigOrError} />,
+    });
+    return;
+  }
+
+  return {
+    yaml: partition.runConfigOrError.yaml,
+    tags: partition.tagsOrError.results.map((t) => ({key: t.key, value: t.value})),
+    solidSelection: partition.solidSelection,
+    mode: partition.mode,
+  };
+}
+
+export async function fetchTagsAndConfigForAssetJob(
+  client: ApolloClient<any>,
+  variables: ConfigPartitionForAssetJobQueryVariables,
+) {
+  const {data: tagAndConfigData} = await client.query<
+    ConfigPartitionForAssetJobQuery,
+    ConfigPartitionForAssetJobQueryVariables
+  >({
+    query: CONFIG_PARTITION_FOR_ASSET_JOB_QUERY,
+    fetchPolicy: 'network-only',
+    variables,
+  });
+
+  if (
+    !tagAndConfigData ||
+    !tagAndConfigData.pipelineOrError ||
+    tagAndConfigData.pipelineOrError.__typename !== 'Pipeline' ||
+    !tagAndConfigData.pipelineOrError.partition
+  ) {
+    return;
+  }
+
+  const {partition} = tagAndConfigData.pipelineOrError;
+
+  if (partition.tagsOrError.__typename === 'PythonError') {
+    showCustomAlert({
+      title: 'Unable to load tags',
+      body: <PythonErrorInfo error={partition.tagsOrError} />,
+    });
+    return;
+  }
+  if (partition.runConfigOrError.__typename === 'PythonError') {
+    showCustomAlert({
+      title: 'Unable to load tags',
+      body: <PythonErrorInfo error={partition.runConfigOrError} />,
+    });
+    return;
+  }
+
+  return {
+    yaml: partition.runConfigOrError.yaml,
+    tags: partition.tagsOrError.results.map((t) => ({key: t.key, value: t.value})),
+    solidSelection: null,
+    mode: null,
+  };
+}
+
+export const CONFIG_PARTITION_FOR_ASSET_JOB_QUERY = gql`
+  query ConfigPartitionForAssetJobQuery(
+    $repositoryName: String!
+    $repositoryLocationName: String!
+    $jobName: String!
+    $partitionName: String!
+    $assetKeys: [AssetKeyInput!]!
+  ) {
+    pipelineOrError(
+      params: {
+        pipelineName: $jobName
+        repositoryName: $repositoryName
+        repositoryLocationName: $repositoryLocationName
+      }
+    ) {
+      ... on Pipeline {
+        id
+        partition(partitionName: $partitionName, selectedAssetKeys: $assetKeys) {
+          name
+          runConfigOrError {
+            ... on PartitionRunConfig {
+              yaml
+            }
+            ...PythonErrorFragment
+          }
+          tagsOrError {
+            ... on PartitionTags {
+              results {
+                key
+                value
+              }
+            }
+            ...PythonErrorFragment
+          }
+        }
+      }
+    }
+  }
+
+  ${PYTHON_ERROR_FRAGMENT}
+`;
+
+export const CONFIG_PARTITION_SELECTION_QUERY = gql`
+  query ConfigPartitionSelectionQuery(
+    $repositorySelector: RepositorySelector!
+    $partitionSetName: String!
+    $partitionName: String!
+  ) {
+    partitionSetOrError(
+      repositorySelector: $repositorySelector
+      partitionSetName: $partitionSetName
+    ) {
+      ... on PartitionSet {
+        id
+        partition(partitionName: $partitionName) {
+          name
+          solidSelection
+          runConfigOrError {
+            ... on PartitionRunConfig {
+              yaml
+            }
+            ...PythonErrorFragment
+          }
+          mode
+          tagsOrError {
+            ... on PartitionTags {
+              results {
+                key
+                value
+              }
+            }
+            ...PythonErrorFragment
+          }
+        }
+      }
+    }
+  }
+
+  ${PYTHON_ERROR_FRAGMENT}
+`;
diff --git a/js_modules/dagster-ui/packages/ui-core/src/launchpad/LaunchpadSession.tsx b/js_modules/dagster-ui/packages/ui-core/src/launchpad/LaunchpadSession.tsx
index 1b6ce5e88cb68..341b6c5a81bbc 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/launchpad/LaunchpadSession.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/launchpad/LaunchpadSession.tsx
@@ -24,11 +24,9 @@ import * as React from 'react';
 import styled from 'styled-components';
 import * as yaml from 'yaml';
 
-import {
-  CONFIG_PARTITION_SELECTION_QUERY,
-  ConfigEditorConfigPicker,
-} from './ConfigEditorConfigPicker';
+import {ConfigEditorConfigPicker} from './ConfigEditorConfigPicker';
 import {ConfigEditorModePicker} from './ConfigEditorModePicker';
+import {fetchTagsAndConfigForAssetJob, fetchTagsAndConfigForJob} from './ConfigFetch';
 import {LaunchpadConfigExpansionButton} from './LaunchpadConfigExpansionButton';
 import {useLaunchPadHooks} from './LaunchpadHooksContext';
 import {LoadingOverlay} from './LoadingOverlay';
@@ -38,11 +36,7 @@ import {SessionSettingsBar} from './SessionSettingsBar';
 import {TagContainer, TagEditor} from './TagEditor';
 import {scaffoldPipelineConfig} from './scaffoldType';
 import {LaunchpadType} from './types';
-import {
-  ConfigEditorPipelinePresetFragment,
-  ConfigPartitionSelectionQuery,
-  ConfigPartitionSelectionQueryVariables,
-} from './types/ConfigEditorConfigPicker.types';
+import {ConfigEditorPipelinePresetFragment} from './types/ConfigEditorConfigPicker.types';
 import {
   LaunchpadSessionPartitionSetsFragment,
   LaunchpadSessionPipelineFragment,
@@ -62,7 +56,6 @@ import {
   SessionBase,
 } from '../app/ExecutionSessionStorage';
 import {usePermissionsForLocation} from '../app/Permissions';
-import {PythonErrorInfo} from '../app/PythonErrorInfo';
 import {ShortcutHandler} from '../app/ShortcutHandler';
 import {displayNameForAssetKey, tokenForAssetKey} from '../asset-graph/Utils';
 import {asAssetCheckHandleInput, asAssetKeyInput} from '../assets/asInput';
@@ -454,7 +447,7 @@ const LaunchpadSession = (props: LaunchpadSessionProps) => {
     const newBaseTags = preset.tags.map(onlyKeyAndValue);
 
     onSaveSession({
-      base: {presetName: preset.name, tags: newBaseTags},
+      base: {type: 'preset', presetName: preset.name, tags: newBaseTags},
       name: preset.name,
       runConfigYaml: preset.runConfigYaml || '',
       solidSelection: preset.solidSelection,
@@ -474,55 +467,35 @@ const LaunchpadSession = (props: LaunchpadSessionProps) => {
     onConfigLoading();
     try {
       const {base} = currentSession;
-      const {data} = await client.query<
-        ConfigPartitionSelectionQuery,
-        ConfigPartitionSelectionQueryVariables
-      >({
-        query: CONFIG_PARTITION_SELECTION_QUERY,
-        variables: {repositorySelector, partitionSetName, partitionName},
-      });
 
-      if (
-        !data ||
-        !data.partitionSetOrError ||
-        data.partitionSetOrError.__typename !== 'PartitionSet' ||
-        !data.partitionSetOrError.partition
-      ) {
+      const resp = props.session.assetSelection
+        ? await fetchTagsAndConfigForAssetJob(client, {
+            ...repositorySelector,
+            jobName: pipeline.name,
+            assetKeys: props.session.assetSelection.map(asAssetKeyInput),
+            partitionName,
+          })
+        : await fetchTagsAndConfigForJob(client, {
+            repositorySelector,
+            partitionSetName,
+            partitionName,
+          });
+
+      if (!resp) {
         onConfigLoaded();
         return;
       }
 
-      const {partition} = data.partitionSetOrError;
-
-      let newBaseTags: {key: string; value: string}[] = [];
-      if (partition.tagsOrError.__typename === 'PythonError') {
-        showCustomAlert({
-          body: <PythonErrorInfo error={partition.tagsOrError} />,
-        });
-      } else {
-        newBaseTags = partition.tagsOrError.results.map(onlyKeyAndValue);
-      }
-
-      let runConfigYaml;
-      if (partition.runConfigOrError.__typename === 'PythonError') {
-        runConfigYaml = '';
-        showCustomAlert({
-          body: <PythonErrorInfo error={partition.runConfigOrError} />,
-        });
-      } else {
-        runConfigYaml = mergeYaml(currentSession.runConfigYaml, partition.runConfigOrError.yaml);
-      }
-
-      const solidSelection = sessionSolidSelection || partition.solidSelection;
+      const solidSelection = sessionSolidSelection || resp.solidSelection;
 
       onSaveSession({
-        name: partition.name,
-        base: Object.assign({}, base, {partitionName: partition.name, tags: newBaseTags}),
-        runConfigYaml,
+        name: partitionName,
+        base: Object.assign({}, base, {partitionName, tags: resp.tags}),
+        runConfigYaml: mergeYaml(currentSession.runConfigYaml, resp.yaml),
         solidSelection,
         solidSelectionQuery: solidSelection === null ? '*' : solidSelection.join(','),
-        mode: partition.mode,
-        tags: tagsApplyingNewBaseTags(newBaseTags),
+        mode: resp.mode,
+        tags: tagsApplyingNewBaseTags(resp.tags),
         needsRefresh: false,
       });
     } catch {}
@@ -544,17 +517,16 @@ const LaunchpadSession = (props: LaunchpadSessionProps) => {
     }
 
     // Otherwise, handle partition-based configuration.
-    const {partitionName, partitionsSetName} = base;
     const repositorySelector = repoAddressToSelector(repoAddress);
 
     // It is expected that `partitionName` is set here, since we shouldn't be showing the
     // button at all otherwise.
-    if (partitionName) {
+    if (base.partitionName) {
       onConfigLoading();
       await onSelectPartition(
         repositorySelector,
-        partitionsSetName,
-        partitionName,
+        'partitionsSetName' in base ? base.partitionsSetName : '',
+        base.partitionName,
         currentSession.solidSelection,
       );
       onConfigLoaded();
@@ -587,7 +559,7 @@ const LaunchpadSession = (props: LaunchpadSessionProps) => {
     if (
       base &&
       needsRefresh &&
-      ('presetName' in base || (base.partitionsSetName && base.partitionName))
+      ('presetName' in base || ('partitionName' in base && base.partitionName))
     ) {
       return base;
     }
diff --git a/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigEditorConfigPicker.types.ts b/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigEditorConfigPicker.types.ts
index 89e9311c0e42a..69f031d6d7bba 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigEditorConfigPicker.types.ts
+++ b/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigEditorConfigPicker.types.ts
@@ -5,7 +5,6 @@ import * as Types from '../../graphql/types';
 export type ConfigPartitionsQueryVariables = Types.Exact<{
   repositorySelector: Types.RepositorySelector;
   partitionSetName: Types.Scalars['String']['input'];
-  assetKeys?: Types.InputMaybe<Array<Types.AssetKeyInput> | Types.AssetKeyInput>;
 }>;
 
 export type ConfigPartitionsQuery = {
@@ -29,67 +28,44 @@ export type ConfigPartitionsQuery = {
       }
     | {__typename: 'PartitionSetNotFoundError'}
     | {__typename: 'PythonError'};
-  assetNodes: Array<{
-    __typename: 'AssetNode';
-    id: string;
-    partitionDefinition: {
-      __typename: 'PartitionDefinition';
-      name: string | null;
-      type: Types.PartitionDefinitionType;
-    } | null;
-  }>;
 };
 
 export type ConfigPartitionResultFragment = {__typename: 'Partition'; name: string};
 
-export type ConfigPartitionSelectionQueryVariables = Types.Exact<{
-  repositorySelector: Types.RepositorySelector;
-  partitionSetName: Types.Scalars['String']['input'];
-  partitionName: Types.Scalars['String']['input'];
+export type ConfigPartitionsAssetsQueryVariables = Types.Exact<{
+  params: Types.PipelineSelector;
+  assetKeys?: Types.InputMaybe<Array<Types.AssetKeyInput> | Types.AssetKeyInput>;
 }>;
 
-export type ConfigPartitionSelectionQuery = {
+export type ConfigPartitionsAssetsQuery = {
   __typename: 'Query';
-  partitionSetOrError:
+  pipelineOrError:
+    | {__typename: 'InvalidSubsetError'; message: string}
     | {
-        __typename: 'PartitionSet';
+        __typename: 'Pipeline';
         id: string;
-        partition: {
-          __typename: 'Partition';
-          name: string;
-          solidSelection: Array<string> | null;
-          mode: string;
-          runConfigOrError:
-            | {__typename: 'PartitionRunConfig'; yaml: string}
-            | {
-                __typename: 'PythonError';
-                message: string;
-                stack: Array<string>;
-                errorChain: Array<{
-                  __typename: 'ErrorChainLink';
-                  isExplicitLink: boolean;
-                  error: {__typename: 'PythonError'; message: string; stack: Array<string>};
-                }>;
-              };
-          tagsOrError:
-            | {
-                __typename: 'PartitionTags';
-                results: Array<{__typename: 'PipelineTag'; key: string; value: string}>;
-              }
-            | {
-                __typename: 'PythonError';
-                message: string;
-                stack: Array<string>;
-                errorChain: Array<{
-                  __typename: 'ErrorChainLink';
-                  isExplicitLink: boolean;
-                  error: {__typename: 'PythonError'; message: string; stack: Array<string>};
-                }>;
-              };
-        } | null;
+        partitionKeysOrError: {__typename: 'PartitionKeys'; partitionKeys: Array<string>};
       }
-    | {__typename: 'PartitionSetNotFoundError'}
-    | {__typename: 'PythonError'};
+    | {__typename: 'PipelineNotFoundError'; message: string}
+    | {
+        __typename: 'PythonError';
+        message: string;
+        stack: Array<string>;
+        errorChain: Array<{
+          __typename: 'ErrorChainLink';
+          isExplicitLink: boolean;
+          error: {__typename: 'PythonError'; message: string; stack: Array<string>};
+        }>;
+      };
+  assetNodes: Array<{
+    __typename: 'AssetNode';
+    id: string;
+    partitionDefinition: {
+      __typename: 'PartitionDefinition';
+      name: string | null;
+      type: Types.PartitionDefinitionType;
+    } | null;
+  }>;
 };
 
 export type ConfigEditorGeneratorPipelineFragment = {
diff --git a/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigFetch.types.ts b/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigFetch.types.ts
new file mode 100644
index 0000000000000..3843307184121
--- /dev/null
+++ b/js_modules/dagster-ui/packages/ui-core/src/launchpad/types/ConfigFetch.types.ts
@@ -0,0 +1,104 @@
+// Generated GraphQL types, do not edit manually.
+
+import * as Types from '../../graphql/types';
+
+export type ConfigPartitionForAssetJobQueryVariables = Types.Exact<{
+  repositoryName: Types.Scalars['String']['input'];
+  repositoryLocationName: Types.Scalars['String']['input'];
+  jobName: Types.Scalars['String']['input'];
+  partitionName: Types.Scalars['String']['input'];
+  assetKeys: Array<Types.AssetKeyInput> | Types.AssetKeyInput;
+}>;
+
+export type ConfigPartitionForAssetJobQuery = {
+  __typename: 'Query';
+  pipelineOrError:
+    | {__typename: 'InvalidSubsetError'}
+    | {
+        __typename: 'Pipeline';
+        id: string;
+        partition: {
+          __typename: 'PartitionTagsAndConfig';
+          name: string;
+          runConfigOrError:
+            | {__typename: 'PartitionRunConfig'; yaml: string}
+            | {
+                __typename: 'PythonError';
+                message: string;
+                stack: Array<string>;
+                errorChain: Array<{
+                  __typename: 'ErrorChainLink';
+                  isExplicitLink: boolean;
+                  error: {__typename: 'PythonError'; message: string; stack: Array<string>};
+                }>;
+              };
+          tagsOrError:
+            | {
+                __typename: 'PartitionTags';
+                results: Array<{__typename: 'PipelineTag'; key: string; value: string}>;
+              }
+            | {
+                __typename: 'PythonError';
+                message: string;
+                stack: Array<string>;
+                errorChain: Array<{
+                  __typename: 'ErrorChainLink';
+                  isExplicitLink: boolean;
+                  error: {__typename: 'PythonError'; message: string; stack: Array<string>};
+                }>;
+              };
+        } | null;
+      }
+    | {__typename: 'PipelineNotFoundError'}
+    | {__typename: 'PythonError'};
+};
+
+export type ConfigPartitionSelectionQueryVariables = Types.Exact<{
+  repositorySelector: Types.RepositorySelector;
+  partitionSetName: Types.Scalars['String']['input'];
+  partitionName: Types.Scalars['String']['input'];
+}>;
+
+export type ConfigPartitionSelectionQuery = {
+  __typename: 'Query';
+  partitionSetOrError:
+    | {
+        __typename: 'PartitionSet';
+        id: string;
+        partition: {
+          __typename: 'Partition';
+          name: string;
+          solidSelection: Array<string> | null;
+          mode: string;
+          runConfigOrError:
+            | {__typename: 'PartitionRunConfig'; yaml: string}
+            | {
+                __typename: 'PythonError';
+                message: string;
+                stack: Array<string>;
+                errorChain: Array<{
+                  __typename: 'ErrorChainLink';
+                  isExplicitLink: boolean;
+                  error: {__typename: 'PythonError'; message: string; stack: Array<string>};
+                }>;
+              };
+          tagsOrError:
+            | {
+                __typename: 'PartitionTags';
+                results: Array<{__typename: 'PipelineTag'; key: string; value: string}>;
+              }
+            | {
+                __typename: 'PythonError';
+                message: string;
+                stack: Array<string>;
+                errorChain: Array<{
+                  __typename: 'ErrorChainLink';
+                  isExplicitLink: boolean;
+                  error: {__typename: 'PythonError'; message: string; stack: Array<string>};
+                }>;
+              };
+        } | null;
+      }
+    | {__typename: 'PartitionSetNotFoundError'}
+    | {__typename: 'PythonError'};
+};