@oceanum/oceanumlab 4.6.0

package/lib/DatasourceItem.js

@@ -0,0 +1,47 @@
+import React, { useState } from 'react';
+import { caretDownIcon, caretRightIcon } from '@jupyterlab/ui-components';
+import { ArrowBendDownRight } from '@phosphor-icons/react';
+import '../style/index.css';
+/**
+ * A React component for expandable containers.
+ */
+export const DatasourceItem = ({ datasource, insertDatasource, onMouseDown }) => {
+    const [expanded, setExpandedValue] = useState(false);
+    const [isDragging, setIsDragging] = useState(false);
+    const handleToggleExpand = () => {
+        setExpandedValue(!expanded);
+    };
+    const startDrag = (e) => {
+        setIsDragging(true);
+        onMouseDown(e, datasource);
+    };
+    return (React.createElement("div", { className: expanded ? 'expanded' : '' },
+        React.createElement("div", { key: datasource.id, className: "datasource-item-title" },
+            React.createElement("button", { title: expanded ? 'Hide Details' : 'Show Details', onClick: handleToggleExpand, className: "datasource-item-expand" }, expanded ? (React.createElement(caretDownIcon.react, { tag: "span", elementPosition: "center", width: "20px" })) : (React.createElement(caretRightIcon.react, { tag: "span", elementPosition: "center", width: "20px" }))),
+            React.createElement("span", { title: datasource.description, className: isDragging
+                    ? 'datasource-item-name'
+                    : 'datasource-item-name draggable', onClick: handleToggleExpand, onMouseDown: startDrag, onMouseUp: () => setIsDragging(false) }, datasource.description),
+            React.createElement("button", { title: "Insert code to load datasource", onClick: handleToggleExpand, className: "datasource-item-action" },
+                React.createElement(ArrowBendDownRight, { size: 18, onClick: e => insertDatasource(datasource) }))),
+        expanded && (React.createElement("div", { className: 'datasource-item-details' },
+            React.createElement("div", { className: "datasource-query-field" },
+                "Datasource id: ",
+                React.createElement("span", null, datasource.datasource)),
+            datasource.geofilter && (React.createElement("div", { className: "datasource-query-field" },
+                "Geofilter type:",
+                React.createElement("span", null, datasource.geofilter.type))),
+            datasource.variables && (React.createElement("div", { className: "datasource-query-field" },
+                "Variables:",
+                React.createElement("span", null, datasource.variables.join(',')))),
+            datasource.timefilter && (React.createElement("div", { className: "datasource-query-field" },
+                "Timefilter:",
+                React.createElement("span", null, datasource.timefilter.times[0]),
+                React.createElement("span", null, "to"),
+                React.createElement("span", null, datasource.timefilter.times[1]))),
+            datasource.spatialref && (React.createElement("div", { className: "datasource-query-field" },
+                "Spatial ref:",
+                React.createElement("span", null, datasource.spatialref))),
+            React.createElement("div", null,
+                React.createElement("a", { href: `https://oceanum.io/datasets/${datasource.datasource}`, target: "_blank" }, "Datasource details"))))));
+};
+//# sourceMappingURL=DatasourceItem.js.map

package/lib/DatasourceItem.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DatasourceItem.js","sourceRoot":"","sources":["../src/DatasourceItem.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE3D,OAAO,oBAAoB,CAAC;AAY5B;;GAEG;AAEH,MAAM,CAAC,MAAM,cAAc,GAAmC,CAAC,EAC7D,UAAU,EACV,gBAAgB,EAChB,WAAW,EACZ,EAAE,EAAE;IACH,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACrD,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEpD,MAAM,kBAAkB,GAAG,GAAS,EAAE;QACpC,gBAAgB,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,CAAC,CAAgD,EAAE,EAAE;QACrE,aAAa,CAAC,IAAI,CAAC,CAAC;QACpB,WAAW,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC7B,CAAC,CAAC;IAEF,OAAO,CACL,6BAAK,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;QACxC,6BAAK,GAAG,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAC,uBAAuB;YACxD,gCACE,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,EACjD,OAAO,EAAE,kBAAkB,EAC3B,SAAS,EAAC,wBAAwB,IAEjC,QAAQ,CAAC,CAAC,CAAC,CACV,oBAAC,aAAa,CAAC,KAAK,IAClB,GAAG,EAAC,MAAM,EACV,eAAe,EAAC,QAAQ,EACxB,KAAK,EAAC,MAAM,GACZ,CACH,CAAC,CAAC,CAAC,CACF,oBAAC,cAAc,CAAC,KAAK,IACnB,GAAG,EAAC,MAAM,EACV,eAAe,EAAC,QAAQ,EACxB,KAAK,EAAC,MAAM,GACZ,CACH,CACM;YACT,8BACE,KAAK,EAAE,UAAU,CAAC,WAAW,EAC7B,SAAS,EACP,UAAU;oBACR,CAAC,CAAC,sBAAsB;oBACxB,CAAC,CAAC,gCAAgC,EAEtC,OAAO,EAAE,kBAAkB,EAC3B,WAAW,EAAE,SAAS,EACtB,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,IAEpC,UAAU,CAAC,WAAW,CAClB;YAEP,gCACE,KAAK,EAAC,gCAAgC,EACtC,OAAO,EAAE,kBAAkB,EAC3B,SAAS,EAAC,wBAAwB;gBAElC,oBAAC,kBAAkB,IACjB,IAAI,EAAE,EAAE,EACR,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,UAAU,CAAC,GAC1C,CACK,CACL;QACL,QAAQ,IAAI,CACX,6BAAK,SAAS,EAAE,yBAAyB;YACvC,6BAAK,SAAS,EAAC,wBAAwB;;gBACtB,kCAAO,UAAU,CAAC,UAAU,CAAQ,CAC/C;YACL,UAAU,CAAC,SAAS,IAAI,CACvB,6BAAK,SAAS,EAAC,wBAAwB;;gBACtB,kCAAO,UAAU,CAAC,SAAS,CAAC,IAAI,CAAQ,CACnD,CACP;YACA,UAAU,CAAC,SAAS,IAAI,CACvB,6BAAK,SAAS,EAAC,wBAAwB;;gBAC3B,kCAAO,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAQ,CACnD,CACP;YACA,UAAU,CAAC,UAAU,IAAI,CACxB,6BAAK,SAAS,EAAC,wBAAwB;;gBAC1B,kCAAO,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAQ;gBACxD,uCAAe;gBACf,kCAAO,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAQ,CACzC,CACP;YACA,UAAU,CAAC,UAAU,IAAI,CACxB,6BAAK,SAAS,EAAC,wBAAwB;;gBACzB,kCAAO,UAAU,CAAC,UAAU,CAAQ,CAC5C,CACP;YACD;gBACE,2BACE,IAAI,EAAE,+BAA+B,UAAU,CAAC,UAAU,EAAE,EAC5D,MAAM,EAAC,QAAQ,yBAGb,CACA,CACF,CACP,CACG,CACP,CAAC;AACJ,CAAC,CAAC"}

package/lib/aiLoop.d.ts

@@ -0,0 +1,60 @@
+import type { ChatMessage, OceanumResponse } from './chatRouter';
+import type { ObservedRun } from './notebookRun';
+/** What `inject` reports back: the chat text, and what ran if anything did. */
+export interface PlacedResponse {
+    message: string;
+    runs: ObservedRun[];
+    /**
+     * A code cell could not be executed at all (no kernel, pending input, the
+     * notebook closed, Stop). Its run says so; observing it would only bring
+     * back more code that cannot run either.
+     */
+    halted?: boolean;
+}
+/** The pieces the loop needs, injected so it can be tested without a notebook. */
+export interface LoopDeps {
+    route(prompt: string, history: ChatMessage[], signal: AbortSignal): Promise<{
+        response: OceanumResponse;
+        hasCodeCellSelected: boolean;
+    }>;
+    observe(prompt: string, history: ChatMessage[], runs: ObservedRun[], signal: AbortSignal): Promise<OceanumResponse>;
+    place(response: OceanumResponse, options: {
+        replaceCodeCell: boolean;
+        autoRun: boolean;
+        signal: AbortSignal;
+    }): Promise<PlacedResponse>;
+}
+export interface LoopOptions {
+    /** Run code cells as they are placed. Workflow 1; required for 2. */
+    autoRunCode: boolean;
+    /** Send each run's output back and place what comes next. Workflow 2. */
+    iterate: boolean;
+    /** Safety net on observe requests; see `runChatLoop`. */
+    maxRounds: number;
+    signal: AbortSignal;
+}
+export declare const STOPPED = "Stopped.";
+/**
+ * One prompt, start to finish, under the two switches.
+ *
+ * The loop lives here rather than on the server because the CLIENT owns the
+ * kernel: only it can run a cell and read what came back. The server sees one
+ * turn at a time through `/api/chat/observe`.
+ *
+ *   autoRunCode off              -> place blocks, stop.           (workflow 3)
+ *   autoRunCode on, iterate off  -> place, run, stop.             (workflow 1)
+ *   autoRunCode on, iterate on   -> place, run, observe, repeat.  (workflow 2)
+ *
+ * `iterate` without `autoRunCode` is inert: nothing ran, so there is nothing
+ * to observe. That falls out of the data (no runs) rather than a guard.
+ *
+ * Rounds end when nothing ran (the agent answered without code, so it is
+ * done), or on Stop. The server enforces its own EXECUTE_MAX_ROUNDS: at the
+ * cap it still answers, explaining the last run with any code stripped, and
+ * that code-free answer ends the loop here. `maxRounds` is therefore only a
+ * safety net against a server that keeps sending code, checked AFTER the
+ * observe so the server's explanation turn is always requested. Every round's
+ * message is kept: a chain that took three steps should read as three steps,
+ * not as the last one.
+ */
+export declare function runChatLoop(prompt: string, history: ChatMessage[], deps: LoopDeps, options: LoopOptions): Promise<string>;

package/lib/aiLoop.js

@@ -0,0 +1,85 @@
+export const STOPPED = 'Stopped.';
+/**
+ * One prompt, start to finish, under the two switches.
+ *
+ * The loop lives here rather than on the server because the CLIENT owns the
+ * kernel: only it can run a cell and read what came back. The server sees one
+ * turn at a time through `/api/chat/observe`.
+ *
+ *   autoRunCode off              -> place blocks, stop.           (workflow 3)
+ *   autoRunCode on, iterate off  -> place, run, stop.             (workflow 1)
+ *   autoRunCode on, iterate on   -> place, run, observe, repeat.  (workflow 2)
+ *
+ * `iterate` without `autoRunCode` is inert: nothing ran, so there is nothing
+ * to observe. That falls out of the data (no runs) rather than a guard.
+ *
+ * Rounds end when nothing ran (the agent answered without code, so it is
+ * done), or on Stop. The server enforces its own EXECUTE_MAX_ROUNDS: at the
+ * cap it still answers, explaining the last run with any code stripped, and
+ * that code-free answer ends the loop here. `maxRounds` is therefore only a
+ * safety net against a server that keeps sending code, checked AFTER the
+ * observe so the server's explanation turn is always requested. Every round's
+ * message is kept: a chain that took three steps should read as three steps,
+ * not as the last one.
+ */
+export async function runChatLoop(prompt, history, deps, options) {
+    const { signal } = options;
+    const messages = [];
+    const runs = [];
+    let response;
+    let replaceCodeCell;
+    try {
+        const first = await deps.route(prompt, history, signal);
+        response = first.response;
+        replaceCodeCell = first.hasCodeCellSelected;
+    }
+    catch (err) {
+        if (signal.aborted) {
+            return STOPPED;
+        }
+        throw err;
+    }
+    if (signal.aborted) {
+        return STOPPED;
+    }
+    for (let round = 1;; round++) {
+        const placed = await deps.place(response, {
+            replaceCodeCell,
+            autoRun: options.autoRunCode,
+            signal
+        });
+        messages.push(placed.message);
+        runs.push(...placed.runs);
+        replaceCodeCell = false;
+        if (signal.aborted) {
+            messages.push(STOPPED);
+            break;
+        }
+        if (!options.iterate || placed.runs.length === 0 || placed.halted) {
+            // Nothing ran this round, or nothing could, so there is nothing to
+            // observe.
+            break;
+        }
+        if (round > options.maxRounds) {
+            break;
+        }
+        try {
+            response = await deps.observe(prompt, history, runs, signal);
+        }
+        catch (err) {
+            // Stop pressed while the request was in flight. What was already placed
+            // and explained stays in the chat; only the next step is dropped.
+            if (signal.aborted) {
+                messages.push(STOPPED);
+                break;
+            }
+            throw err;
+        }
+        if (signal.aborted) {
+            messages.push(STOPPED);
+            break;
+        }
+    }
+    return messages.filter(m => m.trim().length > 0).join('\n\n');
+}
+//# sourceMappingURL=aiLoop.js.map

package/lib/aiLoop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aiLoop.js","sourceRoot":"","sources":["../src/aiLoop.ts"],"names":[],"mappings":"AA+CA,MAAM,CAAC,MAAM,OAAO,GAAG,UAAU,CAAC;AAElC;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,MAAc,EACd,OAAsB,EACtB,IAAc,EACd,OAAoB;IAEpB,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAC3B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,IAAI,GAAkB,EAAE,CAAC;IAE/B,IAAI,QAAyB,CAAC;IAC9B,IAAI,eAAwB,CAAC;IAC7B,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC1B,eAAe,GAAG,KAAK,CAAC,mBAAmB,CAAC;IAC9C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,IAAI,KAAK,GAAG,CAAC,GAAI,KAAK,EAAE,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YACxC,eAAe;YACf,OAAO,EAAE,OAAO,CAAC,WAAW;YAC5B,MAAM;SACP,CAAC,CAAC;QACH,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1B,eAAe,GAAG,KAAK,CAAC;QAExB,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,MAAM;QACR,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClE,mEAAmE;YACnE,WAAW;YACX,MAAM;QACR,CAAC;QACD,IAAI,KAAK,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;YAC9B,MAAM;QACR,CAAC;QAED,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,wEAAwE;YACxE,kEAAkE;YAClE,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACvB,MAAM;YACR,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChE,CAAC"}

package/lib/chatRouter.d.ts

@@ -0,0 +1,64 @@
+import { ISettingRegistry } from '@jupyterlab/settingregistry';
+import type { Progress } from './progress';
+import type { ObservedRun } from './notebookRun';
+import type { INotebookSnapshot } from './notebookContext';
+/** One thing the backend asks us to place in the notebook. */
+export interface Block {
+    type: 'code' | 'markdown';
+    content: string;
+}
+/**
+ * What `/api/chat` answers with: one message, plus anything to place.
+ *
+ * This replaced a discriminated union of text/code/markdown responses
+ * (OCE-173). The union was exclusive, so the backend could not send a markdown
+ * table AND the query that produced it — it had to drop one. `message` is what
+ * goes in the chat window; `blocks` is what goes in the notebook, in order.
+ *
+ * `message` is always present now. On the old markdown variant it was optional,
+ * which is why the handoff used to fall back to 'Added markdown to notebook.'
+ */
+export interface OceanumResponse {
+    message: string;
+    blocks: Block[];
+}
+export interface ChatMessage {
+    role: 'user' | 'assistant';
+    content: string;
+}
+export interface RouteResult {
+    response: OceanumResponse;
+    hasCodeCellSelected: boolean;
+}
+export declare class ChatRouterError extends Error {
+    readonly statusCode?: number;
+    constructor(message: string, statusCode?: number);
+}
+export declare class ChatRouter {
+    private _settings;
+    private _notebook;
+    /**
+     * @param _notebook What the conversation's notebook holds -- the one its
+     *   answers are placed in -- or null if there is nothing to send. Asked on
+     *   every request, so it follows the conversation rather than whichever tab
+     *   happens to be active.
+     */
+    constructor(_settings: ISettingRegistry.ISettings, _notebook: () => Promise<INotebookSnapshot | null>);
+    route(prompt: string, chatHistory?: ChatMessage[], signal?: AbortSignal, onProgress?: (progress: Progress) => void): Promise<RouteResult>;
+    /**
+     * The notebook half of the execute loop: the code ran in the user's own
+     * kernel, here is what happened, what next? Same context as `route`, plus
+     * every run so far, so the agent sees the whole chain it is continuing.
+     */
+    observe(prompt: string, chatHistory: ChatMessage[], runs: ObservedRun[], signal?: AbortSignal, onProgress?: (progress: Progress) => void): Promise<OceanumResponse>;
+    private _gather;
+    private _send;
+    /**
+     * Read the streamed form, reporting progress, and return the final answer.
+     *
+     * The `done` event is authoritative: the events before it are for the user
+     * to look at, not for the client to reassemble. Anything else and a client
+     * that missed a frame would silently answer with less than the agent said.
+     */
+    private _drain;
+}

package/lib/chatRouter.js

@@ -0,0 +1,225 @@
+import { OCEANUM_AI_BACKEND_URL } from './constants';
+import { readSse } from './sse';
+import { formatNotebookCells } from './notebookContext';
+export class ChatRouterError extends Error {
+    constructor(message, statusCode) {
+        super(message);
+        this.statusCode = statusCode;
+        this.name = 'ChatRouterError';
+    }
+}
+/**
+ * Check the body is the shape we expect before anything reads it.
+ *
+ * A bare `as OceanumResponse` cast used to be enough because the shape had not
+ * changed since the extension was written. It has now (OCE-173), and an
+ * extension pinned to one contract will meet a backend serving the other during
+ * any deploy where the two are not released together: the old body has no
+ * `blocks`, so `response.blocks.length` throws deep inside the handoff and the
+ * user sees nothing happen. One check turns that into the error the chat window
+ * already knows how to show.
+ */
+function asOceanumResponse(data) {
+    const body = data;
+    if (!body ||
+        typeof body.message !== 'string' ||
+        !Array.isArray(body.blocks)) {
+        throw new ChatRouterError('Unexpected response from the AI backend. It may be running an ' +
+            'incompatible version of the chat API.');
+    }
+    return { message: body.message, blocks: body.blocks };
+}
+export class ChatRouter {
+    /**
+     * @param _notebook What the conversation's notebook holds -- the one its
+     *   answers are placed in -- or null if there is nothing to send. Asked on
+     *   every request, so it follows the conversation rather than whichever tab
+     *   happens to be active.
+     */
+    constructor(_settings, _notebook) {
+        this._settings = _settings;
+        this._notebook = _notebook;
+    }
+    async route(prompt, chatHistory = [], signal, onProgress) {
+        const { payload, isCodeCell, context } = await this._gather(prompt, chatHistory);
+        const data = await this._send('/api/chat', payload, signal, onProgress);
+        return {
+            response: asOceanumResponse(data),
+            hasCodeCellSelected: isCodeCell && context.trim().length > 0
+        };
+    }
+    /**
+     * The notebook half of the execute loop: the code ran in the user's own
+     * kernel, here is what happened, what next? Same context as `route`, plus
+     * every run so far, so the agent sees the whole chain it is continuing.
+     */
+    async observe(prompt, chatHistory, runs, signal, onProgress) {
+        const { payload } = await this._gather(prompt, chatHistory);
+        const data = await this._send('/api/chat/observe', { ...payload, runs }, signal, onProgress);
+        return asOceanumResponse(data);
+    }
+    async _gather(prompt, chatHistory) {
+        let context = '';
+        let isCodeCell = false;
+        let notebookCells = [];
+        try {
+            const snapshot = await this._notebook();
+            if (snapshot) {
+                notebookCells = formatNotebookCells(snapshot.cells);
+                // Answers are placed in this same notebook, so its selected cell is
+                // the one a code answer may replace.
+                if (snapshot.selected) {
+                    context = snapshot.selected.source;
+                    isCodeCell = snapshot.selected.isCode;
+                }
+            }
+        }
+        catch (_a) {
+            // context is optional — never throw
+        }
+        // Build payload with all context
+        const payload = { prompt };
+        if (context) {
+            if (isCodeCell) {
+                payload.codeContext = context;
+            }
+            else {
+                payload.context = context;
+            }
+        }
+        // Include chat history (excluding the current prompt which is already in 'prompt')
+        if (chatHistory.length > 0) {
+            payload.chatHistory = chatHistory;
+        }
+        // Include the conversation's notebook cells
+        if (notebookCells.length > 0) {
+            payload.notebookCells = notebookCells;
+        }
+        return { payload, isCodeCell, context };
+    }
+    async _send(path, payload, signal, onProgress) {
+        var _a;
+        const token = this._settings.get('datameshToken').composite;
+        if (!token) {
+            throw new ChatRouterError('Datamesh token not configured. Set your token in Settings → Oceanum.io.');
+        }
+        let response;
+        try {
+            response = await fetch(`${OCEANUM_AI_BACKEND_URL}${path}`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'X-Datamesh-Token': token,
+                    // Ask for the streamed form only when someone is listening. The
+                    // server requires the media type to be NAMED -- a wildcard never
+                    // selects it -- so omitting this is what keeps the plain JSON path
+                    // byte-identical to what it always was.
+                    ...(onProgress ? { Accept: 'text/event-stream' } : {})
+                },
+                body: JSON.stringify(payload),
+                signal
+            });
+        }
+        catch (err) {
+            // The user pressed Stop. Not a failure, and not "could not reach"; the
+            // caller checks `signal.aborted` and stays quiet.
+            if (signal === null || signal === void 0 ? void 0 : signal.aborted) {
+                throw err;
+            }
+            throw new ChatRouterError(`Could not reach Oceanum AI backend at ${OCEANUM_AI_BACKEND_URL}. Is it running?`);
+        }
+        if (response.status === 401) {
+            throw new ChatRouterError('Invalid or expired Datamesh token.', 401);
+        }
+        if (!response.ok) {
+            let detail = '';
+            try {
+                const body = await response.json();
+                detail = (_a = body.detail) !== null && _a !== void 0 ? _a : JSON.stringify(body);
+            }
+            catch (_b) {
+                detail = response.statusText;
+            }
+            throw new ChatRouterError(`Backend error: ${detail}`, response.status);
+        }
+        // Stop can also land while the body is still being read; that rejects
+        // here, not in the fetch above, and the caller treats it the same way.
+        if (onProgress && isEventStream(response)) {
+            return this._drain(response, onProgress, signal);
+        }
+        return response.json();
+    }
+    /**
+     * Read the streamed form, reporting progress, and return the final answer.
+     *
+     * The `done` event is authoritative: the events before it are for the user
+     * to look at, not for the client to reassemble. Anything else and a client
+     * that missed a frame would silently answer with less than the agent said.
+     */
+    async _drain(response, onProgress, signal) {
+        var _a;
+        if (!response.body) {
+            // A proxy buffered the whole response, or this runtime has no
+            // ReadableStream. Either way there is nothing to stream; the body is
+            // still the answer, so fall back rather than fail.
+            return response.json();
+        }
+        let answer;
+        try {
+            for await (const event of readSse(response.body, signal)) {
+                if (event.event === 'status') {
+                    const data = parseJson(event.data);
+                    if (data === null || data === void 0 ? void 0 : data.phase) {
+                        onProgress(data);
+                    }
+                }
+                else if (event.event === 'done') {
+                    answer = parseJson(event.data);
+                }
+                else if (event.event === 'error') {
+                    const data = parseJson(event.data);
+                    // The status was 200 before this could happen, so the failure travels
+                    // in the frame and there was nothing for the check above to throw on.
+                    throw new ChatRouterError(`Backend error: ${(_a = data === null || data === void 0 ? void 0 : data.detail) !== null && _a !== void 0 ? _a : 'the request failed'}`, data === null || data === void 0 ? void 0 : data.status_code);
+                }
+            }
+        }
+        catch (err) {
+            // Stop fails the read with an AbortError, which the caller reports as
+            // Stop, and an `error` event is already the chat's error. Anything else
+            // is the connection going away mid-answer -- a redeploy, the network --
+            // which fetch reports as a bare TypeError ("terminated"). Left as it
+            // was, it reached the command as unexpected and the chat showed nothing.
+            if ((signal === null || signal === void 0 ? void 0 : signal.aborted) || err instanceof ChatRouterError) {
+                throw err;
+            }
+            throw new ChatRouterError('The connection to Oceanum AI was lost before the response was complete.');
+        }
+        if (answer === undefined) {
+            // The stream ended without its terminal event -- a dropped connection,
+            // or a proxy closing a response that looked idle. Treated as a failure
+            // rather than a partial answer, which is indistinguishable from a
+            // complete one once it reaches the chat.
+            if (signal === null || signal === void 0 ? void 0 : signal.aborted) {
+                throw new DOMException('Aborted', 'AbortError');
+            }
+            throw new ChatRouterError('The response ended before it was complete.');
+        }
+        return answer;
+    }
+}
+function isEventStream(response) {
+    var _a, _b;
+    return ((_b = (_a = response.headers.get('content-type')) === null || _a === void 0 ? void 0 : _a.includes('text/event-stream')) !== null && _b !== void 0 ? _b : false);
+}
+/** Parse one event's data, returning null rather than throwing: a malformed
+ * frame should cost a progress update, not the answer. */
+function parseJson(data) {
+    try {
+        return JSON.parse(data);
+    }
+    catch (_a) {
+        return null;
+    }
+}
+//# sourceMappingURL=chatRouter.js.map

package/lib/chatRouter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatRouter.js","sourceRoot":"","sources":["../src/chatRouter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAGhC,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAkCxD,MAAM,OAAO,eAAgB,SAAQ,KAAK;IACxC,YACE,OAAe,EACC,UAAmB;QAEnC,KAAK,CAAC,OAAO,CAAC,CAAC;QAFC,eAAU,GAAV,UAAU,CAAS;QAGnC,IAAI,CAAC,IAAI,GAAG,iBAAiB,CAAC;IAChC,CAAC;CACF;AAED;;;;;;;;;;GAUG;AACH,SAAS,iBAAiB,CAAC,IAAa;IACtC,MAAM,IAAI,GAAG,IAAuC,CAAC;IACrD,IACE,CAAC,IAAI;QACL,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ;QAChC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAC3B,CAAC;QACD,MAAM,IAAI,eAAe,CACvB,gEAAgE;YAC9D,uCAAuC,CAC1C,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;AACxD,CAAC;AAED,MAAM,OAAO,UAAU;IACrB;;;;;OAKG;IACH,YACU,SAAqC,EACrC,SAAkD;QADlD,cAAS,GAAT,SAAS,CAA4B;QACrC,cAAS,GAAT,SAAS,CAAyC;IACzD,CAAC;IAEJ,KAAK,CAAC,KAAK,CACT,MAAc,EACd,cAA6B,EAAE,EAC/B,MAAoB,EACpB,UAAyC;QAEzC,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CACzD,MAAM,EACN,WAAW,CACZ,CAAC;QACF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACxE,OAAO;YACL,QAAQ,EAAE,iBAAiB,CAAC,IAAI,CAAC;YACjC,mBAAmB,EAAE,UAAU,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;SAC7D,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO,CACX,MAAc,EACd,WAA0B,EAC1B,IAAmB,EACnB,MAAoB,EACpB,UAAyC;QAEzC,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAC3B,mBAAmB,EACnB,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,EACpB,MAAM,EACN,UAAU,CACX,CAAC;QACF,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,MAAc,EACd,WAA0B;QAE1B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,IAAI,aAAa,GAAa,EAAE,CAAC;QAEjC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACxC,IAAI,QAAQ,EAAE,CAAC;gBACb,aAAa,GAAG,mBAAmB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACpD,oEAAoE;gBACpE,qCAAqC;gBACrC,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;oBACtB,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;oBACnC,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACxC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,WAAM,CAAC;YACP,oCAAoC;QACtC,CAAC;QAED,iCAAiC;QACjC,MAAM,OAAO,GAAgB,EAAE,MAAM,EAAE,CAAC;QAExC,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,mFAAmF;QACnF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;QACpC,CAAC;QAED,4CAA4C;QAC5C,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC;QACxC,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,IAAY,EACZ,OAAgB,EAChB,MAAoB,EACpB,UAAyC;;QAEzC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,SAAmB,CAAC;QACtE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,eAAe,CACvB,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QAED,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,sBAAsB,GAAG,IAAI,EAAE,EAAE;gBACzD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,kBAAkB,EAAE,KAAK;oBACzB,gEAAgE;oBAChE,iEAAiE;oBACjE,mEAAmE;oBACnE,wCAAwC;oBACxC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvD;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7B,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,uEAAuE;YACvE,kDAAkD;YAClD,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,MAAM,IAAI,eAAe,CACvB,yCAAyC,sBAAsB,kBAAkB,CAClF,CAAC;QACJ,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,eAAe,CAAC,oCAAoC,EAAE,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnC,MAAM,GAAG,MAAA,IAAI,CAAC,MAAM,mCAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC/C,CAAC;YAAC,WAAM,CAAC;gBACP,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;YAC/B,CAAC;YACD,MAAM,IAAI,eAAe,CAAC,kBAAkB,MAAM,EAAE,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACzE,CAAC;QAED,sEAAsE;QACtE,uEAAuE;QACvE,IAAI,UAAU,IAAI,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1C,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,MAAM,CAClB,QAAkB,EAClB,UAAwC,EACxC,MAAoB;;QAEpB,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,8DAA8D;YAC9D,qEAAqE;YACrE,mDAAmD;YACnD,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC;QAED,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC;gBACzD,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAC7B,MAAM,IAAI,GAAG,SAAS,CAAW,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC7C,IAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,EAAE,CAAC;wBAChB,UAAU,CAAC,IAAI,CAAC,CAAC;oBACnB,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;oBAClC,MAAM,GAAG,SAAS,CAAU,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC1C,CAAC;qBAAM,IAAI,KAAK,CAAC,KAAK,KAAK,OAAO,EAAE,CAAC;oBACnC,MAAM,IAAI,GAAG,SAAS,CACpB,KAAK,CAAC,IAAI,CACX,CAAC;oBACF,sEAAsE;oBACtE,sEAAsE;oBACtE,MAAM,IAAI,eAAe,CACvB,kBAAkB,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,MAAM,mCAAI,oBAAoB,EAAE,EACxD,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,WAAW,CAClB,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,sEAAsE;YACtE,wEAAwE;YACxE,wEAAwE;YACxE,qEAAqE;YACrE,yEAAyE;YACzE,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,KAAI,GAAG,YAAY,eAAe,EAAE,CAAC;gBACtD,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,MAAM,IAAI,eAAe,CACvB,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,uEAAuE;YACvE,uEAAuE;YACvE,kEAAkE;YAClE,yCAAyC;YACzC,IAAI,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAClD,CAAC;YACD,MAAM,IAAI,eAAe,CAAC,4CAA4C,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,SAAS,aAAa,CAAC,QAAkB;;IACvC,OAAO,CACL,MAAA,MAAA,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,0CAAE,QAAQ,CAAC,mBAAmB,CAAC,mCAAI,KAAK,CAC7E,CAAC;AACJ,CAAC;AAED;0DAC0D;AAC1D,SAAS,SAAS,CAAI,IAAY;IAChC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAM,CAAC;IAC/B,CAAC;IAAC,WAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

package/lib/constants.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Build-time constants for Oceanum AI integration.
+ * These values are compiled into the extension.
+ */
+/** URL of the Oceanum AI backend service */
+export declare const OCEANUM_AI_BACKEND_URL = "https://ai.oceanum.io";
+/**
+ * Safety net on observe requests per prompt. The server's EXECUTE_MAX_ROUNDS
+ * is what actually caps the chain (at its cap it answers without code, which
+ * ends the loop); this only stops a client talking to a server that keeps
+ * sending code, and is checked after each observe so the server's explanation
+ * turn is always requested.
+ */
+export declare const MAX_OBSERVE_ROUNDS = 8;

package/lib/constants.js

@@ -0,0 +1,15 @@
+/**
+ * Build-time constants for Oceanum AI integration.
+ * These values are compiled into the extension.
+ */
+/** URL of the Oceanum AI backend service */
+export const OCEANUM_AI_BACKEND_URL = 'https://ai.oceanum.io';
+/**
+ * Safety net on observe requests per prompt. The server's EXECUTE_MAX_ROUNDS
+ * is what actually caps the chain (at its cap it answers without code, which
+ * ends the loop); this only stops a client talking to a server that keeps
+ * sending code, and is checked after each observe so the server's explanation
+ * turn is always requested.
+ */
+export const MAX_OBSERVE_ROUNDS = 8;
+//# sourceMappingURL=constants.js.map

package/lib/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,4CAA4C;AAC5C,MAAM,CAAC,MAAM,sBAAsB,GAAG,uBAAuB,CAAC;AAE9D;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC"}

package/lib/conversationPin.d.ts

@@ -0,0 +1,68 @@
+import type { NotebookPanel } from '@jupyterlab/notebook';
+/**
+ * What ConversationPin needs from JupyterLab -- passed in, so the pin can be
+ * tested without it. Both `create` and `reopen` must resolve only once the
+ * notebook's contents have loaded: loading replaces every cell, so anything
+ * placed before then is wiped.
+ */
+export interface INotebookHost {
+    /** The notebook in the active main-area tab, or null if that is not one. */
+    activeTab(): NotebookPanel | null;
+    /** A new notebook, made the active tab. Null if none can be created. */
+    create(): Promise<NotebookPanel | null>;
+    /** The notebook at `path`, opened again as a tab. Null if it cannot be. */
+    reopen(path: string): Promise<NotebookPanel | null>;
+    /** Bring `panel`'s tab to the front. */
+    activate(panel: NotebookPanel): void;
+}
+/**
+ * The notebook the current conversation lives in: what the agent is shown,
+ * and where its answers go.
+ *
+ * Pinned when the conversation starts -- New chat, or the first message after
+ * the panel opens -- to the notebook in the ACTIVE TAB or, when that tab is
+ * not a notebook, to a new notebook made the active tab. Kept for the whole
+ * conversation: switching tabs does not move it, a rename is followed, and a
+ * pinned notebook that has been closed is opened again when an answer has
+ * cells to place in it. The active tab rather than the tracker's current
+ * notebook, which is the LAST-FOCUSED notebook even with a terminal or a text
+ * file in front.
+ */
+export declare class ConversationPin {
+    private readonly _host;
+    private _panel;
+    private _path;
+    private _started;
+    private _queue;
+    constructor(_host: INotebookHost);
+    /** Start a new conversation. Resolves to its notebook's name. */
+    start(): Promise<string | null>;
+    /**
+     * Start the conversation if nothing has, and report its notebook's name
+     * either way.
+     */
+    ensure(): Promise<string | null>;
+    /**
+     * The conversation's notebook if it is open, starting the conversation if
+     * nothing has -- but NOT opening a closed one again: that is for placing an
+     * answer, not for asking a question. See path() for reading a closed one.
+     */
+    current(): Promise<NotebookPanel | null>;
+    /**
+     * The conversation's notebook, opened again if it has been closed. When it
+     * cannot be -- deleted, or gone some other way -- a new notebook takes its
+     * place, so the conversation always has one. Null only if none can be made.
+     */
+    notebook(): Promise<NotebookPanel | null>;
+    /** The conversation's notebook, brought to the front: see notebook(). */
+    show(): Promise<NotebookPanel | null>;
+    /** Where the notebook's file is -- its last known place while it is closed. */
+    path(): string | null;
+    /** The notebook's name -- its last known one while it is closed. */
+    name(): string | null;
+    private _live;
+    private _startNow;
+    private _pin;
+    private _onPathChanged;
+    private _serial;
+}