window.api.notebook — Python Notebook Kernel API

import type {
  NotebookSessionState,
  NotebookRunSummary
} from '../../../../shared/notebook'

async function runAnalysis(sessionId: string, workspaceCwd: string) {
  const sessionRequest = { sessionId, workspaceCwd }

  // Subscribe to change events before querying state
  const unsubChanged = window.api.notebook.onChanged((event) => {
    console.log('Notebook changed for session:', event.sessionId)
  })

  // Query current state
  const state: NotebookSessionState = await window.api.notebook.state(sessionRequest)
  console.log('Kernel status:', state.kernelStatus)
  console.log('Existing cells:', state.cells.length)

  // Execute code directly (no persistent cell needed)
  const result: NotebookRunSummary = await window.api.notebook.execute({
    ...sessionRequest,
    code: `
import pandas as pd
df = pd.read_csv('data.csv')
print(df.describe())
    `,
    timeoutMs: 30_000
  })

  console.log('Status:', result.status)
  console.log('Output:', result.text.plain.join('\n'))

  if (result.status === 'failed') {
    console.error('Traceback:', result.text.traceback)
  }

  if (result.artifacts.length > 0) {
    console.log('Generated files:', result.artifacts.map((a) => a.name))
  }

  // Stream a multi-chunk cell write
  const { writeId, cellId } = await window.api.notebook.beginCodeCell(sessionRequest)

  await window.api.notebook.appendCodeCell({
    ...sessionRequest,
    writeId,
    cellId,
    delta: 'import matplotlib.pyplot as plt\n'
  })
  await window.api.notebook.appendCodeCell({
    ...sessionRequest,
    writeId,
    cellId,
    delta: 'plt.plot([1, 2, 3])\nplt.savefig("plot.png")\n'
  })
  await window.api.notebook.finishCodeCell({ ...sessionRequest, writeId, cellId })

  // Run the assembled cell
  const plotResult: NotebookRunSummary = await window.api.notebook.runCell({
    ...sessionRequest,
    cellId
  })
  console.log('Plot run status:', plotResult.status)

  return () => unsubChanged()
}