Function resolveUpstreamExecutionBlockIds

packages/cli/src/commands/run.ts:443–488

(
  file: DeepnoteFile,
  options: { notebook?: string; block?: string },
  pythonInterpreter: string
)

}

async function resolveUpstreamExecutionBlockIds(
  file: DeepnoteFile,
  options: { notebook?: string; block?: string },
  pythonInterpreter: string
): Promise<string[] | undefined> {
  if (!options.block) {
    return undefined
  }

  const notebooks = getNotebooksForExecutionScope(file, options)

  // If notebook is not specified, scope DAG analysis to the notebook containing the target block.
  const scopeNotebooks = selectScopeNotebooks(notebooks, options)

  const allBlocks = scopeNotebooks.flatMap(notebook => notebook.blocks)
  if (allBlocks.length === 0) {
    return undefined
  }

  const blocksToExecute = allBlocks.filter(block => block.id === options.block)
  const upstreamResult = await getUpstreamBlocks(allBlocks, blocksToExecute, {
    pythonInterpreter,
  })

  if (upstreamResult.status === 'fatal') {
    debug(`DAG analysis failed with fatal error, running single block without deps: ${upstreamResult.error.message}`)
    return undefined
  }

  if (upstreamResult.status === 'missing-deps') {
    const depsWithErrors = upstreamResult.newlyComputedBlocksContentDeps.filter(block => block.error)
    if (depsWithErrors.length > 0) {
      debug(`DAG analysis found ${depsWithErrors.length} blocks with dependency errors, using partial DAG`)
    }
  }

  const upstreamIds = upstreamResult.blocksToExecuteWithDeps
    .filter(block => block.id !== options.block)
    .map(block => block.id)
  if (upstreamIds.length === 0) {
    return undefined
  }
  const blockIds = [...new Set([...upstreamIds, options.block])]
  debug(`Block ${options.block} has ${upstreamIds.length} upstream dependencies: ${upstreamIds.join(', ')}`)
  return blockIds
}

export function createRunAction(program: Command): (path: string | undefined, options: RunOptions) => Promise<void> {
  return async (path, options) => {