@oceanum/oceanumlab 4.6.0

package/lib/kernelHandoff.js

@@ -0,0 +1,236 @@
+import { NotebookActions } from '@jupyterlab/notebook';
+import { CodeCell } from '@jupyterlab/cells';
+import { harvestOutputs } from './notebookRun';
+/** Longest wait for a starting kernel before the cell is reported as not run. */
+const KERNEL_START_WAIT_MS = 10000;
+export class KernelHandoff {
+    /**
+     * @param _target The conversation's notebook, brought to the front -- and
+     *   opened again first, if it was closed. Asked only when there is
+     *   something to place, so an answer with no blocks leaves the tabs alone.
+     * @param _readFrom The panel the request read its selected cell from, if
+     *   any. A code answer replaces the selected cell only in that same panel:
+     *   one opened since has some other cell selected, and replacing it would
+     *   overwrite the user's work.
+     */
+    constructor(_target, _readFrom) {
+        this._target = _target;
+        this._readFrom = _readFrom;
+    }
+    /**
+     * Place every block the response carries, in order, and return the message
+     * for the chat window.
+     *
+     * This used to switch on `response.type`, because a response was exactly one
+     * of text, code or markdown. It can now carry any number of blocks in any
+     * combination (OCE-173) — a markdown table describing a dataset followed by
+     * the query that produced it is one response, not two — so the branch became
+     * a loop. No blocks is the commonest case and still touches no notebook.
+     */
+    async inject(response, options = {}) {
+        var _a;
+        const runs = [];
+        let halted = false;
+        if (response.blocks.length === 0) {
+            // Nothing to place — just the message for the chat window.
+            return { message: response.message, runs };
+        }
+        // Every answer goes into the conversation's notebook, whichever tab is in
+        // front, and that notebook is brought to the front so the user sees it.
+        const notebookPanel = await this._target();
+        if (!notebookPanel) {
+            return {
+                message: response.message + '\n\n(Could not open a notebook)',
+                runs
+            };
+        }
+        const notebook = notebookPanel.content;
+        const { sessionContext } = notebookPanel;
+        // The user selected ONE cell, so only the first code block can replace it;
+        // everything after is inserted below in order. Captured before the loop
+        // because inserting moves `activeCell`.
+        const selected = notebook.activeCell;
+        let replaceTarget = options.replaceCodeCell &&
+            notebookPanel === this._readFrom() &&
+            selected instanceof CodeCell
+            ? selected
+            : null;
+        for (const block of response.blocks) {
+            if ((_a = options.signal) === null || _a === void 0 ? void 0 : _a.aborted) {
+                break;
+            }
+            if (block.type === 'code') {
+                let cell = null;
+                // The target can be gone by now: a markdown block placed before this
+                // one yielded while it rendered, and the user deleted the cell.
+                if (replaceTarget && !replaceTarget.isDisposed) {
+                    replaceTarget.model.sharedModel.setSource(block.content);
+                    cell = replaceTarget;
+                    replaceTarget = null;
+                }
+                else {
+                    NotebookActions.insertBelow(notebook);
+                    // `insertBelow` makes whatever the user's default cell type is;
+                    // code goes in a code cell regardless.
+                    if (notebook.activeCell &&
+                        !(notebook.activeCell instanceof CodeCell)) {
+                        NotebookActions.changeCellType(notebook, 'code');
+                    }
+                    const newCell = notebook.activeCell;
+                    if (!newCell) {
+                        continue;
+                    }
+                    newCell.model.sharedModel.setSource(block.content);
+                    cell = newCell instanceof CodeCell ? newCell : null;
+                }
+                // Run THIS cell, not the selection: the cell just written need not be
+                // the active one (a markdown block placed before it moved the
+                // selection), and the user may have several cells selected. Reading
+                // the outputs afterwards is what makes the iterate workflow possible
+                // -- it is the only place the kernel's answer can be seen. A blank
+                // block is placed but not run: `CodeCell.execute` would send nothing
+                // and the empty result would read as a kernel failure.
+                if (options.autoRun && cell && block.content.trim()) {
+                    const { outcome, executed } = await this._run(notebookPanel, cell, options.signal);
+                    runs.push({
+                        code: block.content,
+                        message: response.message,
+                        ...outcome
+                    });
+                    if (!executed) {
+                        halted = true;
+                    }
+                }
+            }
+            else {
+                NotebookActions.insertBelow(notebook);
+                if (!notebook.activeCell) {
+                    continue;
+                }
+                // Convert to markdown cell and populate
+                NotebookActions.changeCellType(notebook, 'markdown');
+                const mdCell = notebook.activeCell;
+                if (mdCell) {
+                    mdCell.model.sharedModel.setSource(block.content);
+                    // Render the markdown cell
+                    await NotebookActions.run(notebook, sessionContext);
+                }
+            }
+        }
+        return { message: response.message, runs, halted };
+    }
+    /**
+     * Execute one cell and report what happened, and whether it happened at all.
+     *
+     * `runCells` resolves false for a kernel error (the traceback is in the
+     * outputs) and for some of the ways a cell can fail to execute at all
+     * (pending input, interrupted); it resolves TRUE with nothing run when there
+     * is no kernel, when the kernel is terminating, or when the kernel is still
+     * starting (`CodeCell.execute` bails silently on a session without a
+     * kernel); and it rejects when the kernel is dead. None of the did-not-run
+     * paths touch the outputs, so a replaced cell would still show its previous
+     * run; they are cleared first so that whatever is there afterwards came from
+     * this run. The execution count is cleared with them and only a kernel
+     * reply sets it again, so a null count afterwards means nothing ran,
+     * whatever `runCells` resolved to; that is reported as an error rather than
+     * as a clean, silent run.
+     *
+     * Pending input is refused BEFORE clearing: clearing the outputs of a cell
+     * that is waiting on `input()` disposes the prompt widget, and the kernel
+     * then waits forever for an answer nobody can give.
+     */
+    async _run(notebookPanel, cell, signal) {
+        const { content: notebook, sessionContext } = notebookPanel;
+        if (sessionContext.kernelDisplayStatus === 'initializing') {
+            // A kernel is starting (for a notebook this call just created, say).
+            // Bounded: `ready` never resolves if the kernel ends up needing to be
+            // selected by hand.
+            await waitForReady(sessionContext, signal);
+            if (signal === null || signal === void 0 ? void 0 : signal.aborted) {
+                return didNotRun('stopped before the kernel was ready');
+            }
+        }
+        // Closing the notebook, or deleting the cell, disposes it and nulls its
+        // model; every access below would throw and take the whole reply with it.
+        if (cell.isDisposed) {
+            return didNotRun('the cell was deleted or its notebook closed');
+        }
+        if (sessionContext.pendingInput) {
+            return didNotRun('the kernel is waiting for input that must be answered first');
+        }
+        const interrupt = () => {
+            var _a, _b;
+            void ((_b = (_a = sessionContext.session) === null || _a === void 0 ? void 0 : _a.kernel) === null || _b === void 0 ? void 0 : _b.interrupt().catch(() => {
+                // Already dead or gone; there is nothing left to interrupt.
+            }));
+        };
+        cell.model.sharedModel.transact(() => {
+            cell.model.clearExecution();
+        }, false);
+        signal === null || signal === void 0 ? void 0 : signal.addEventListener('abort', interrupt, { once: true });
+        let ran;
+        let failure = null;
+        try {
+            ran = await NotebookActions.runCells(notebook, [cell], sessionContext);
+        }
+        catch (err) {
+            // A dead kernel rejects rather than resolving false. Report it like any
+            // other did-not-run so the rounds already placed keep their messages,
+            // and log it: a rejection can also be a fault inside JupyterLab's
+            // executor, and the chat line alone would hide that.
+            console.error('Oceanum AI: cell execution failed', err);
+            ran = false;
+            failure = err instanceof Error ? err.message : String(err);
+        }
+        finally {
+            signal === null || signal === void 0 ? void 0 : signal.removeEventListener('abort', interrupt);
+        }
+        if (cell.isDisposed) {
+            return didNotRun('the cell was deleted or its notebook closed');
+        }
+        // Only stream and error outputs are wanted; serialising a rendered plot
+        // or DataFrame just to discard it is not.
+        const outputs = [];
+        const model = cell.model.outputs;
+        for (let i = 0; i < model.length; i++) {
+            const item = model.get(i);
+            if (item.type === 'stream' || item.type === 'error') {
+                outputs.push(item.toJSON());
+            }
+        }
+        const outcome = harvestOutputs(outputs);
+        if ((!ran || cell.model.executionCount === null) &&
+            outcome.status === 'ok') {
+            return didNotRun(failure !== null && failure !== void 0 ? failure : 'no kernel available yet, or the execution was interrupted');
+        }
+        return { outcome, executed: true };
+    }
+}
+/**
+ * Resolve when the session context is ready, when Stop is pressed, or after
+ * `KERNEL_START_WAIT_MS`, whichever comes first. Stop is raced too so the
+ * button answers at once rather than after the kernel, or the timeout.
+ */
+function waitForReady(sessionContext, signal) {
+    return new Promise(resolve => {
+        const done = () => {
+            clearTimeout(timer);
+            signal === null || signal === void 0 ? void 0 : signal.removeEventListener('abort', done);
+            resolve();
+        };
+        const timer = setTimeout(done, KERNEL_START_WAIT_MS);
+        void sessionContext.ready.then(done);
+        signal === null || signal === void 0 ? void 0 : signal.addEventListener('abort', done, { once: true });
+    });
+}
+function didNotRun(reason) {
+    return {
+        outcome: {
+            status: 'error',
+            stdout: '',
+            error: `The cell did not execute (${reason}).`
+        },
+        executed: false
+    };
+}
+//# sourceMappingURL=kernelHandoff.js.map

package/lib/kernelHandoff.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kernelHandoff.js","sourceRoot":"","sources":["../src/kernelHandoff.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAiB,MAAM,sBAAsB,CAAC;AACtE,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAI7C,OAAO,EAAE,cAAc,EAA2B,MAAM,eAAe,CAAC;AAgBxE,iFAAiF;AACjF,MAAM,oBAAoB,GAAG,KAAK,CAAC;AAEnC,MAAM,OAAO,aAAa;IACxB;;;;;;;;OAQG;IACH,YACU,OAA4C,EAC5C,SAAqC;QADrC,YAAO,GAAP,OAAO,CAAqC;QAC5C,cAAS,GAAT,SAAS,CAA4B;IAC5C,CAAC;IAEJ;;;;;;;;;OASG;IACH,KAAK,CAAC,MAAM,CACV,QAAyB,EACzB,UAAyB,EAAE;;QAE3B,MAAM,IAAI,GAAkB,EAAE,CAAC;QAC/B,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,2DAA2D;YAC3D,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;QAC7C,CAAC;QAED,0EAA0E;QAC1E,wEAAwE;QACxE,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO,GAAG,iCAAiC;gBAC7D,IAAI;aACL,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC;QACvC,MAAM,EAAE,cAAc,EAAE,GAAG,aAAa,CAAC;QAEzC,2EAA2E;QAC3E,wEAAwE;QACxE,wCAAwC;QACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;QACrC,IAAI,aAAa,GACf,OAAO,CAAC,eAAe;YACvB,aAAa,KAAK,IAAI,CAAC,SAAS,EAAE;YAClC,QAAQ,YAAY,QAAQ;YAC1B,CAAC,CAAC,QAAQ;YACV,CAAC,CAAC,IAAI,CAAC;QAEX,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpC,IAAI,MAAA,OAAO,CAAC,MAAM,0CAAE,OAAO,EAAE,CAAC;gBAC5B,MAAM;YACR,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,IAAI,IAAI,GAAoB,IAAI,CAAC;gBACjC,qEAAqE;gBACrE,gEAAgE;gBAChE,IAAI,aAAa,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;oBAC/C,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACzD,IAAI,GAAG,aAAa,CAAC;oBACrB,aAAa,GAAG,IAAI,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACN,eAAe,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;oBACtC,gEAAgE;oBAChE,uCAAuC;oBACvC,IACE,QAAQ,CAAC,UAAU;wBACnB,CAAC,CAAC,QAAQ,CAAC,UAAU,YAAY,QAAQ,CAAC,EAC1C,CAAC;wBACD,eAAe,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACnD,CAAC;oBACD,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC;oBACpC,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,SAAS;oBACX,CAAC;oBACD,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACnD,IAAI,GAAG,OAAO,YAAY,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;gBACtD,CAAC;gBAED,sEAAsE;gBACtE,8DAA8D;gBAC9D,oEAAoE;gBACpE,qEAAqE;gBACrE,mEAAmE;gBACnE,qEAAqE;gBACrE,uDAAuD;gBACvD,IAAI,OAAO,CAAC,OAAO,IAAI,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;oBACpD,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAC3C,aAAa,EACb,IAAI,EACJ,OAAO,CAAC,MAAM,CACf,CAAC;oBACF,IAAI,CAAC,IAAI,CAAC;wBACR,IAAI,EAAE,KAAK,CAAC,OAAO;wBACnB,OAAO,EAAE,QAAQ,CAAC,OAAO;wBACzB,GAAG,OAAO;qBACX,CAAC,CAAC;oBACH,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,MAAM,GAAG,IAAI,CAAC;oBAChB,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,eAAe,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;gBACtC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;oBACzB,SAAS;gBACX,CAAC;gBAED,wCAAwC;gBACxC,eAAe,CAAC,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;gBACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACnC,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBAClD,2BAA2B;oBAC3B,MAAM,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IACrD,CAAC;IAED;;;;;;;;;;;;;;;;;;;OAmBG;IACK,KAAK,CAAC,IAAI,CAChB,aAA4B,EAC5B,IAAc,EACd,MAAoB;QAEpB,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,aAAa,CAAC;QAC5D,IAAI,cAAc,CAAC,mBAAmB,KAAK,cAAc,EAAE,CAAC;YAC1D,qEAAqE;YACrE,sEAAsE;YACtE,oBAAoB;YACpB,MAAM,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,EAAE,CAAC;gBACpB,OAAO,SAAS,CAAC,qCAAqC,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QACD,wEAAwE;QACxE,0EAA0E;QAC1E,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC,6CAA6C,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,cAAc,CAAC,YAAY,EAAE,CAAC;YAChC,OAAO,SAAS,CACd,6DAA6D,CAC9D,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GAAG,GAAG,EAAE;;YACrB,KAAK,CAAA,MAAA,MAAA,cAAc,CAAC,OAAO,0CAAE,MAAM,0CAAE,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;gBAC1D,4DAA4D;YAC9D,CAAC,CAAC,CAAA,CAAC;QACL,CAAC,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,GAAG,EAAE;YACnC,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;QAC9B,CAAC,EAAE,KAAK,CAAC,CAAC;QACV,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,gBAAgB,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,IAAI,GAAY,CAAC;QACjB,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,EAAE,cAAc,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,wEAAwE;YACxE,sEAAsE;YACtE,kEAAkE;YAClE,qDAAqD;YACrD,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,GAAG,CAAC,CAAC;YACxD,GAAG,GAAG,KAAK,CAAC;YACZ,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,CAAC;gBAAS,CAAC;YACT,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,mBAAmB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAClD,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC,6CAA6C,CAAC,CAAC;QAClE,CAAC;QAED,wEAAwE;QACxE,0CAA0C;QAC1C,MAAM,OAAO,GAAuB,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACpD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QAExC,IACE,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,cAAc,KAAK,IAAI,CAAC;YAC5C,OAAO,CAAC,MAAM,KAAK,IAAI,EACvB,CAAC;YACD,OAAO,SAAS,CACd,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,2DAA2D,CACvE,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACrC,CAAC;CACF;AAED;;;;GAIG;AACH,SAAS,YAAY,CACnB,cAA+B,EAC/B,MAAoB;IAEpB,OAAO,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;QACjC,MAAM,IAAI,GAAG,GAAG,EAAE;YAChB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC3C,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;QACrD,KAAK,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,MAAc;IAC/B,OAAO;QACL,OAAO,EAAE;YACP,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,6BAA6B,MAAM,IAAI;SAC/C;QACD,QAAQ,EAAE,KAAK;KAChB,CAAC;AACJ,CAAC"}

package/lib/notebookContext.d.ts

@@ -0,0 +1,48 @@
+import type { Notebook } from '@jupyterlab/notebook';
+/**
+ * A notebook cell reduced to what the chat context needs.
+ */
+export interface IContextCell {
+    kind: 'code' | 'markdown';
+    source: string;
+}
+/** What a request carries from the conversation's notebook. */
+export interface INotebookSnapshot {
+    cells: IContextCell[];
+    /** The selected cell, which a code answer may replace. None when closed. */
+    selected: {
+        source: string;
+        isCode: boolean;
+    } | null;
+}
+export declare const MAX_NOTEBOOK_CELLS = 500;
+export declare const MAX_CELL_CHARS = 100000;
+export declare const MAX_CELLS_BYTES: number;
+/** An open notebook's code and markdown cells, and its selected cell. */
+export declare function snapshotOf(notebook: Notebook): INotebookSnapshot;
+/**
+ * A closed notebook's cells, read from its file.
+ *
+ * Read rather than opened again: re-opening the tab is for an answer with
+ * cells to place, not for asking a question. A closed notebook has no
+ * selected cell. Anything unexpected in the file reads as no cells rather
+ * than an error -- context is best-effort.
+ */
+export declare function snapshotFromIpynb(content: unknown): INotebookSnapshot;
+/**
+ * The notebook as `notebookCells`: every non-empty code and markdown cell, in
+ * order.
+ *
+ * The server renders all of them inside ONE ```python block, each headed
+ * `# Cell N`, so markdown sent raw would read there as broken code. It goes as
+ * comment lines under a jupytext-style `# %% [markdown]` marker instead: still
+ * valid Python, and unambiguous to the model.
+ *
+ * Markdown is also what can push a request past the server's limits, and those
+ * reject the whole request -- costing the user the chat for the sake of some
+ * prose. So markdown gives way first: inline images are dropped from it, an
+ * oversized markdown cell is left out, and when the markdown would take the
+ * cells past the server's cell count or past MAX_CELLS_BYTES, only the code is
+ * sent -- which is what was sent before markdown was included.
+ */
+export declare function formatNotebookCells(cells: readonly IContextCell[]): string[];

package/lib/notebookContext.js

@@ -0,0 +1,118 @@
+// Mirrors oceanum-ai's app/config.py. Past either limit the server rejects the
+// whole request, so the chat fails outright rather than losing some context.
+export const MAX_NOTEBOOK_CELLS = 500;
+export const MAX_CELL_CHARS = 100000;
+// The server also caps the WHOLE request body at 2 MiB (MAX_REQUEST_BYTES),
+// and that body carries chat history and, on follow-up rounds, run outputs as
+// well as the cells. Markdown may take the cells up to half of it.
+export const MAX_CELLS_BYTES = 1024 * 1024;
+// An inline image: the one thing in markdown that is routinely huge. Any media
+// type, any `;param` or `;param=value` (charset, base64, utf8), then the data
+// up to the first space, bracket or quote.
+const DATA_URI = /data:[\w.+-]+\/[\w.+-]+(?:;[\w.+-]+(?:=[\w.+-]+)?)*,[^\s)"']+/g;
+/** An open notebook's code and markdown cells, and its selected cell. */
+export function snapshotOf(notebook) {
+    const cells = [];
+    for (const cell of notebook.widgets) {
+        const type = cell.model.type;
+        if (type === 'code' || type === 'markdown') {
+            cells.push({ kind: type, source: cell.model.sharedModel.getSource() });
+        }
+    }
+    const active = notebook.activeCell;
+    return {
+        cells,
+        selected: active
+            ? {
+                source: active.model.sharedModel.getSource(),
+                isCode: active.model.type === 'code'
+            }
+            : null
+    };
+}
+/**
+ * A closed notebook's cells, read from its file.
+ *
+ * Read rather than opened again: re-opening the tab is for an answer with
+ * cells to place, not for asking a question. A closed notebook has no
+ * selected cell. Anything unexpected in the file reads as no cells rather
+ * than an error -- context is best-effort.
+ */
+export function snapshotFromIpynb(content) {
+    const raw = content === null || content === void 0 ? void 0 : content.cells;
+    const cells = [];
+    if (Array.isArray(raw)) {
+        for (const cell of raw) {
+            const { cell_type: type, source } = (cell !== null && cell !== void 0 ? cell : {});
+            if (type !== 'code' && type !== 'markdown') {
+                continue;
+            }
+            // nbformat allows the source as one string or as a list of lines.
+            const text = Array.isArray(source)
+                ? source.join('')
+                : typeof source === 'string'
+                    ? source
+                    : '';
+            cells.push({ kind: type, source: text });
+        }
+    }
+    return { cells, selected: null };
+}
+/**
+ * The notebook as `notebookCells`: every non-empty code and markdown cell, in
+ * order.
+ *
+ * The server renders all of them inside ONE ```python block, each headed
+ * `# Cell N`, so markdown sent raw would read there as broken code. It goes as
+ * comment lines under a jupytext-style `# %% [markdown]` marker instead: still
+ * valid Python, and unambiguous to the model.
+ *
+ * Markdown is also what can push a request past the server's limits, and those
+ * reject the whole request -- costing the user the chat for the sake of some
+ * prose. So markdown gives way first: inline images are dropped from it, an
+ * oversized markdown cell is left out, and when the markdown would take the
+ * cells past the server's cell count or past MAX_CELLS_BYTES, only the code is
+ * sent -- which is what was sent before markdown was included.
+ */
+export function formatNotebookCells(cells) {
+    const code = [];
+    const all = [];
+    for (const cell of cells) {
+        if (!cell.source.trim()) {
+            continue;
+        }
+        if (cell.kind === 'code') {
+            code.push(cell.source);
+            all.push(cell.source);
+            continue;
+        }
+        const markdown = asMarkdownComment(cell.source);
+        if (markdown.length <= MAX_CELL_CHARS) {
+            all.push(markdown);
+        }
+    }
+    if (all.length > MAX_NOTEBOOK_CELLS || totalBytes(all) > MAX_CELLS_BYTES) {
+        return code;
+    }
+    return all;
+}
+function asMarkdownComment(source) {
+    const lines = source
+        .replace(DATA_URI, 'data:…')
+        .split(/\r?\n/)
+        .map(line => (line ? `# ${line}` : '#'));
+    return ['# %% [markdown]', ...lines].join('\n');
+}
+/** UTF-8 size, which is what the server's byte limit counts. */
+function totalBytes(texts) {
+    var _a;
+    let bytes = 0;
+    for (const text of texts) {
+        for (const char of text) {
+            const point = (_a = char.codePointAt(0)) !== null && _a !== void 0 ? _a : 0;
+            bytes += point < 0x80 ? 1 : point < 0x800 ? 2 : point < 0x10000 ? 3 : 4;
+        }
+    }
+    return bytes;
+}
+//# sourceMappingURL=notebookContext.js.map

package/lib/notebookContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"notebookContext.js","sourceRoot":"","sources":["../src/notebookContext.ts"],"names":[],"mappings":"AAiBA,+EAA+E;AAC/E,6EAA6E;AAC7E,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAG,CAAC;AACtC,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC;AAErC,4EAA4E;AAC5E,8EAA8E;AAC9E,mEAAmE;AACnE,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,GAAG,IAAI,CAAC;AAE3C,+EAA+E;AAC/E,8EAA8E;AAC9E,2CAA2C;AAC3C,MAAM,QAAQ,GACZ,gEAAgE,CAAC;AAEnE,yEAAyE;AACzE,MAAM,UAAU,UAAU,CAAC,QAAkB;IAC3C,MAAM,KAAK,GAAmB,EAAE,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;QAC7B,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YAC3C,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IACD,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;IACnC,OAAO;QACL,KAAK;QACL,QAAQ,EAAE,MAAM;YACd,CAAC,CAAC;gBACE,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,EAAE;gBAC5C,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM;aACrC;YACH,CAAC,CAAC,IAAI;KACT,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,iBAAiB,CAAC,OAAgB;IAChD,MAAM,GAAG,GAAI,OAAsC,aAAtC,OAAO,uBAAP,OAAO,CAAiC,KAAK,CAAC;IAC3D,MAAM,KAAK,GAAmB,EAAE,CAAC;IACjC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACvB,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;YACvB,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,EAAE,CAG9C,CAAC;YACF,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC3C,SAAS;YACX,CAAC;YACD,kEAAkE;YAClE,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;gBAChC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjB,CAAC,CAAC,OAAO,MAAM,KAAK,QAAQ;oBAC1B,CAAC,CAAC,MAAM;oBACR,CAAC,CAAC,EAAE,CAAC;YACT,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AACnC,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAA8B;IAChE,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YACxB,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,SAAS;QACX,CAAC;QACD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,QAAQ,CAAC,MAAM,IAAI,cAAc,EAAE,CAAC;YACtC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,GAAG,kBAAkB,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,eAAe,EAAE,CAAC;QACzE,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc;IACvC,MAAM,KAAK,GAAG,MAAM;SACjB,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC;SAC3B,KAAK,CAAC,OAAO,CAAC;SACd,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3C,OAAO,CAAC,iBAAiB,EAAE,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAClD,CAAC;AAED,gEAAgE;AAChE,SAAS,UAAU,CAAC,KAAwB;;IAC1C,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,MAAA,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,mCAAI,CAAC,CAAC;YACvC,KAAK,IAAI,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/lib/notebookHost.d.ts

@@ -0,0 +1,6 @@
+import type { JupyterFrontEnd } from '@jupyterlab/application';
+import type { INotebookTracker } from '@jupyterlab/notebook';
+import type { INotebookHost } from './conversationPin';
+export declare const NOTEBOOK_WAIT_MS = 15000;
+/** JupyterLab as ConversationPin sees it. */
+export declare function notebookHost(app: JupyterFrontEnd, tracker: INotebookTracker, waitMs?: number): INotebookHost;

package/lib/notebookHost.js

@@ -0,0 +1,72 @@
+import { NotebookPanel } from '@jupyterlab/notebook';
+// Longest wait for a notebook to open, be created, or load. The pin takes one
+// step at a time, so a step that never finished would stall every chat.
+export const NOTEBOOK_WAIT_MS = 15000;
+/** `promise`, or null if it fails or has not settled within `ms`. */
+function within(promise, ms) {
+    return new Promise(resolve => {
+        const timer = setTimeout(() => resolve(null), ms);
+        promise.then(value => {
+            clearTimeout(timer);
+            resolve(value);
+        }, () => {
+            clearTimeout(timer);
+            resolve(null);
+        });
+    });
+}
+/**
+ * The notebook a command opened, once its contents have loaded; null if the
+ * command opened something else, or either step failed or took over `ms`.
+ *
+ * `docmanager:open` and `notebook:create-new` hand back the panel before the
+ * file has been read, and reading it REPLACES every cell: an answer placed
+ * before then is wiped when the file arrives.
+ */
+async function loaded(opened, ms) {
+    const widget = await within(opened, ms);
+    if (!(widget instanceof NotebookPanel)) {
+        return null;
+    }
+    return within(widget.context.ready.then(() => widget), ms);
+}
+/** JupyterLab as ConversationPin sees it. */
+export function notebookHost(app, tracker, waitMs = NOTEBOOK_WAIT_MS) {
+    return {
+        // `app.shell` tracks only main-area tabs, so its current widget is the
+        // active tab even while focus is in the sidebar.
+        activeTab: () => {
+            const widget = app.shell.currentWidget;
+            return widget && tracker.has(widget) ? widget : null;
+        },
+        // A new Python 3 notebook, made the active tab. `notebook:create-new`
+        // returns the panel it opens; anything else is a failure. Taking the next
+        // notebook to appear instead could take one the user opened meanwhile.
+        create: async () => {
+            const panel = await loaded(app.commands.execute('notebook:create-new', { kernelName: 'python3' }), waitMs);
+            if (panel) {
+                app.shell.activateById(panel.id);
+            }
+            return panel;
+        },
+        // Null when the file has been deleted, or has gone some other way: a new
+        // notebook then takes over.
+        reopen: async (path) => {
+            // Opened again by the user already: theirs, selection and all.
+            const open = tracker.find(panel => panel.context.path === path);
+            if (open) {
+                return loaded(Promise.resolve(open), waitMs);
+            }
+            const panel = await loaded(app.commands.execute('docmanager:open', { path }), waitMs);
+            if (panel) {
+                // It opens with its first cell selected, which nobody chose, and
+                // answers go below the selected cell: above all the user's work.
+                // Select the last cell, so they go at the end.
+                panel.content.activeCellIndex = panel.content.widgets.length - 1;
+            }
+            return panel;
+        },
+        activate: panel => app.shell.activateById(panel.id)
+    };
+}
+//# sourceMappingURL=notebookHost.js.map

package/lib/notebookHost.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"notebookHost.js","sourceRoot":"","sources":["../src/notebookHost.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAGrD,8EAA8E;AAC9E,wEAAwE;AACxE,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAEtC,qEAAqE;AACrE,SAAS,MAAM,CAAI,OAAmB,EAAE,EAAU;IAChD,OAAO,IAAI,OAAO,CAAW,OAAO,CAAC,EAAE;QACrC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QAClD,OAAO,CAAC,IAAI,CACV,KAAK,CAAC,EAAE;YACN,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC,EACD,GAAG,EAAE;YACH,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;GAOG;AACH,KAAK,UAAU,MAAM,CACnB,MAAwB,EACxB,EAAU;IAEV,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACxC,IAAI,CAAC,CAAC,MAAM,YAAY,aAAa,CAAC,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,MAAM,CACX,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EACvC,EAAE,CACH,CAAC;AACJ,CAAC;AAED,6CAA6C;AAC7C,MAAM,UAAU,YAAY,CAC1B,GAAoB,EACpB,OAAyB,EACzB,MAAM,GAAG,gBAAgB;IAEzB,OAAO;QACL,uEAAuE;QACvE,iDAAiD;QACjD,SAAS,EAAE,GAAG,EAAE;YACd,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC;YACvC,OAAO,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAE,MAAwB,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1E,CAAC;QACD,sEAAsE;QACtE,0EAA0E;QAC1E,uEAAuE;QACvE,MAAM,EAAE,KAAK,IAAI,EAAE;YACjB,MAAM,KAAK,GAAG,MAAM,MAAM,CACxB,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,qBAAqB,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,EACtE,MAAM,CACP,CAAC;YACF,IAAI,KAAK,EAAE,CAAC;gBACV,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,yEAAyE;QACzE,4BAA4B;QAC5B,MAAM,EAAE,KAAK,EAAC,IAAI,EAAC,EAAE;YACnB,+DAA+D;YAC/D,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAChE,IAAI,IAAI,EAAE,CAAC;gBACT,OAAO,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;YAC/C,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,MAAM,CACxB,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,CAAC,EACjD,MAAM,CACP,CAAC;YACF,IAAI,KAAK,EAAE,CAAC;gBACV,iEAAiE;gBACjE,iEAAiE;gBACjE,+CAA+C;gBAC/C,KAAK,CAAC,OAAO,CAAC,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;YACnE,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,QAAQ,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;KACpD,CAAC;AACJ,CAAC"}

package/lib/notebookRun.d.ts

@@ -0,0 +1,37 @@
+import type * as nbformat from '@jupyterlab/nbformat';
+/**
+ * What one code cell did when it ran, in the shape `/api/chat/observe` takes.
+ *
+ * The notebook counterpart of the sandbox's ExecResult. `message` is the
+ * agent's own explanation for the response that carried this code, so a
+ * capped final answer can say what the whole chain was doing.
+ */
+export interface ObservedRun {
+    code: string;
+    status: 'ok' | 'error';
+    stdout: string;
+    error: string | null;
+    message: string;
+}
+/** The part of a run the kernel decides; what `harvestOutputs` reads off it. */
+export type RunOutcome = Pick<ObservedRun, 'status' | 'stdout' | 'error'>;
+/**
+ * Most text kept per field. The server reads only the tail of stdout and a
+ * bounded traceback (OBSERVATION_STDOUT_CHARS), and every observe request
+ * re-sends every run so far, so anything past this is bytes shipped for
+ * nothing -- a cell printing in a loop can produce megabytes.
+ */
+export declare const MAX_OBSERVED_CHARS = 8000;
+/** Kernel tracebacks are coloured for a terminal; the model does not need that. */
+export declare function stripAnsi(text: string): string;
+/**
+ * Reduce a cell's outputs to what the observation prompt needs.
+ *
+ * Stream stdout is concatenated in order. An `error` output makes the run an
+ * error and its traceback the error text; the kernel's stderr stream is folded
+ * in after it, because warnings there are often the useful part of a failure.
+ * Rendered outputs (execute_result, display_data) are deliberately NOT sent:
+ * a DataFrame repr is kilobytes of dataset-derived text, and the agent gets
+ * the printed summary the code chose to make instead.
+ */
+export declare function harvestOutputs(outputs: readonly nbformat.IOutput[]): RunOutcome;

package/lib/notebookRun.js

@@ -0,0 +1,70 @@
+/**
+ * Most text kept per field. The server reads only the tail of stdout and a
+ * bounded traceback (OBSERVATION_STDOUT_CHARS), and every observe request
+ * re-sends every run so far, so anything past this is bytes shipped for
+ * nothing -- a cell printing in a loop can produce megabytes.
+ */
+export const MAX_OBSERVED_CHARS = 8000;
+// CSI (colour, cursor, including private-mode `?25l` from progress bars) and
+// OSC (titles, hyperlinks) sequences, per ECMA-48. An OSC body cannot contain
+// ESC, so the body class excludes it: otherwise an ST-terminated OSC followed
+// later by a BEL-terminated one would swallow the text between them.
+const ANSI = 
+// eslint-disable-next-line no-control-regex
+/\u001b\[[0-?]*[ -/]*[@-~]|\u001b\][^\u0007\u001b]*(?:\u0007|\u001b\\)/g;
+/** Kernel tracebacks are coloured for a terminal; the model does not need that. */
+export function stripAnsi(text) {
+    return text.replace(ANSI, '');
+}
+/** Keep the end: that is where the summary line, or the raised error, is. */
+function tail(text) {
+    if (text.length <= MAX_OBSERVED_CHARS) {
+        return text;
+    }
+    return `…[${text.length - MAX_OBSERVED_CHARS} chars truncated]\n${text.slice(-MAX_OBSERVED_CHARS)}`;
+}
+function asText(value) {
+    if (value === undefined) {
+        return '';
+    }
+    return Array.isArray(value) ? value.join('') : value;
+}
+/**
+ * Reduce a cell's outputs to what the observation prompt needs.
+ *
+ * Stream stdout is concatenated in order. An `error` output makes the run an
+ * error and its traceback the error text; the kernel's stderr stream is folded
+ * in after it, because warnings there are often the useful part of a failure.
+ * Rendered outputs (execute_result, display_data) are deliberately NOT sent:
+ * a DataFrame repr is kilobytes of dataset-derived text, and the agent gets
+ * the printed summary the code chose to make instead.
+ */
+export function harvestOutputs(outputs) {
+    const stdout = [];
+    const stderr = [];
+    let error = null;
+    for (const output of outputs) {
+        if (nbformatIsStream(output)) {
+            (output.name === 'stderr' ? stderr : stdout).push(asText(output.text));
+        }
+        else if (nbformatIsError(output)) {
+            const traceback = stripAnsi(output.traceback.join('\n'));
+            error = `${output.ename}: ${output.evalue}\n${traceback}`.trim();
+        }
+    }
+    if (error !== null && stderr.length > 0) {
+        error = `${error}\n${stripAnsi(stderr.join(''))}`.trim();
+    }
+    return {
+        status: error === null ? 'ok' : 'error',
+        stdout: tail(stripAnsi(stdout.join(''))),
+        error: error === null ? null : tail(error)
+    };
+}
+function nbformatIsStream(output) {
+    return output.output_type === 'stream';
+}
+function nbformatIsError(output) {
+    return output.output_type === 'error';
+}
+//# sourceMappingURL=notebookRun.js.map

package/lib/notebookRun.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"notebookRun.js","sourceRoot":"","sources":["../src/notebookRun.ts"],"names":[],"mappings":"AAoBA;;;;;GAKG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAEvC,6EAA6E;AAC7E,8EAA8E;AAC9E,8EAA8E;AAC9E,qEAAqE;AACrE,MAAM,IAAI;AACR,4CAA4C;AAC5C,wEAAwE,CAAC;AAE3E,mFAAmF;AACnF,MAAM,UAAU,SAAS,CAAC,IAAY;IACpC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAChC,CAAC;AAED,6EAA6E;AAC7E,SAAS,IAAI,CAAC,IAAY;IACxB,IAAI,IAAI,CAAC,MAAM,IAAI,kBAAkB,EAAE,CAAC;QACtC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,KAAK,IAAI,CAAC,MAAM,GAAG,kBAAkB,sBAAsB,IAAI,CAAC,KAAK,CAC1E,CAAC,kBAAkB,CACpB,EAAE,CAAC;AACN,CAAC;AAED,SAAS,MAAM,CAAC,KAA2C;IACzD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AACvD,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,cAAc,CAC5B,OAAoC;IAEpC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,GAAkB,IAAI,CAAC;IAEhC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACzE,CAAC;aAAM,IAAI,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;YACnC,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACzD,KAAK,GAAG,GAAG,MAAM,CAAC,KAAK,KAAK,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC;QACnE,CAAC;IACH,CAAC;IAED,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxC,KAAK,GAAG,GAAG,KAAK,KAAK,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;IAC3D,CAAC;IAED,OAAO;QACL,MAAM,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO;QACvC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACxC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,MAAwB;IAExB,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,CAAC;AACzC,CAAC;AAED,SAAS,eAAe,CAAC,MAAwB;IAC/C,OAAO,MAAM,CAAC,WAAW,KAAK,OAAO,CAAC;AACxC,CAAC"}