Function handleSnapshotLoad

packages/mcp/src/tools/snapshots.ts:265–378

(args: Record<string, unknown>)

}

async function handleSnapshotLoad(args: Record<string, unknown>) {
  if (args.path === undefined) {
    return snapshotError('path is required')
  }
  const parsedArgs = snapshotLoadArgsSchema.safeParse(args)
  if (!parsedArgs.success) {
    return snapshotError(`Invalid arguments for deepnote_snapshot_load: ${formatFirstIssue(parsedArgs.error)}`)
  }
  const { path: filePath, snapshotDir } = parsedArgs.data

  try {
    const absolutePath = path.resolve(filePath)

    // Check if it's a snapshot file directly
    if (filePath.endsWith('.snapshot.deepnote')) {
      const snapshot = await loadSnapshotFile(absolutePath)

      // Count outputs
      let totalOutputs = 0
      for (const notebook of snapshot.project.notebooks) {
        for (const block of notebook.blocks) {
          const execBlock = block as { outputs?: unknown[] }
          if (execBlock.outputs && execBlock.outputs.length > 0) {
            totalOutputs++
          }
        }
      }

      return {
        content: [
          {
            type: 'text',
            text: JSON.stringify(
              {
                path: absolutePath,
                projectName: snapshot.project.name,
                projectId: snapshot.project.id,
                version: snapshot.version,
                notebooks: snapshot.project.notebooks.length,
                blocksWithOutputs: totalOutputs,
                execution: snapshot.execution,
                environment: snapshot.environment,
              },
              null,
              2
            ),
          },
        ],
      }
    }

    // Otherwise, find and load latest snapshot for the source file
    const content = await fs.readFile(absolutePath, 'utf-8')
    const file = deserializeDeepnoteFile(content)
    const projectId = file.project.id
    const notebookId = resolveSnapshotNotebookId(file)
    const snapshotOptions = { ...(snapshotDir ? { snapshotDir } : {}), ...(notebookId ? { notebookId } : {}) }
    const snapshot = await loadLatestSnapshot(absolutePath, projectId, snapshotOptions)