@datalayer/agent-runtimes 1.0.18 → 1.0.19

package/lib/chat/notebook/EphemeralNotebook.js

@@ -21,7 +21,8 @@ import { jsx as _jsx } from "react/jsx-runtime";
  */
 import { useEffect, useMemo, useRef, useState } from 'react';
 import { ServerConnection, ServiceManager } from '@jupyterlab/services';
-import { Box, DatalayerThemeProvider, getThemeConfig, useSystemColorMode, useThemeStore, } from '@datalayer/primer-addons';
+import { DatalayerThemeProvider, getThemeConfig, useSystemColorMode, useThemeStore, } from '@datalayer/primer-addons';
+import { Box } from '@primer/react';
 import { Notebook, NotebookToolbar, CellSidebarExtension, CellSidebarButton, JupyterReactTheme, notebookStore, disposeServiceManager, } from '@datalayer/jupyter-react';
 import { useAgentsRuntimes } from '../../hooks/useAgentRuntimes';
 import { registerSandboxServiceManager } from '../../services/sandboxServiceManagers';
@@ -54,7 +55,7 @@ const EPHEMERAL_NOTEBOOK_CONTENT = {
 /**
  * Renders an in-memory notebook backed by a sandbox kernel.
  */
-export function EphemeralNotebook({ notebookId, runtimePodName, cellSidebarMargin = 120, nbformat, onNbformatChange, toolbarComponent, }) {
+export function EphemeralNotebook({ notebookId, runtimePodName, runtimeOverride, cellSidebarMargin = 120, nbformat, onNbformatChange, toolbarComponent, collaborationProvider, }) {
     // The `nbformat` passed to the `Notebook` component MUST stay a stable
     // reference for the lifetime of a given `notebookId`: the underlying
     // `useNotebookModel` hook rebuilds the notebook model (and its context /
@@ -77,18 +78,36 @@ export function EphemeralNotebook({ notebookId, runtimePodName, cellSidebarMargi
     // notebook must bind to exactly the runtime assigned to this agent, or to
     // none at all (straight path).
     const { runtimes, refetchRuntimes } = useAgentsRuntimes();
-    const selectedRuntime = useMemo(() => {
+    const podResolvedRuntime = useMemo(() => {
         const preferredPod = String(runtimePodName || '').trim();
         if (!preferredPod) {
             return undefined;
         }
         return runtimes.find(rt => String(rt?.pod_name || '') === preferredPod);
     }, [runtimePodName, runtimes]);
+    // An explicit endpoint override wins over the pod lookup: it lets a SaaS
+    // browser bind the kernel through the runtimes tunnel proxy (the node's
+    // Jupyter server is not directly reachable, so no pod exists in the list).
+    const selectedRuntime = useMemo(() => {
+        const overrideBaseUrl = String(runtimeOverride?.baseUrl || '').trim();
+        if (overrideBaseUrl) {
+            return {
+                url: overrideBaseUrl,
+                wsUrl: String(runtimeOverride?.wsUrl || '').trim() || undefined,
+                token: String(runtimeOverride?.token || '').trim(),
+                pod_name: String(runtimeOverride?.podName || '').trim() || 'agent-node-proxy',
+            };
+        }
+        return podResolvedRuntime;
+    }, [runtimeOverride, podResolvedRuntime]);
     // While the assigned pod has not yet appeared in the runtimes list, poll the
     // list quickly instead of waiting for the default (10s) refresh interval.
     // This is what makes the "Starting notebook…" state clear promptly once the
-    // agent runtime is ready, rather than lingering for several seconds.
-    const needsRuntimeLookup = Boolean(String(runtimePodName || '').trim() && !selectedRuntime);
+    // agent runtime is ready, rather than lingering for several seconds. The
+    // override path binds immediately, so no polling is needed there.
+    const needsRuntimeLookup = Boolean(!runtimeOverride?.baseUrl &&
+        String(runtimePodName || '').trim() &&
+        !selectedRuntime);
     useEffect(() => {
         if (!needsRuntimeLookup) {
             return;
@@ -120,9 +139,10 @@ export function EphemeralNotebook({ notebookId, runtimePodName, cellSidebarMargi
             }
             try {
                 const token = String(selectedRuntime?.token || '').trim();
+                const wsUrl = String(selectedRuntime?.wsUrl || '').trim() || baseUrl.replace(/^http/, 'ws');
                 const serverSettings = ServerConnection.makeSettings({
                     baseUrl,
-                    wsUrl: baseUrl.replace(/^http/, 'ws'),
+                    wsUrl,
                     token,
                     appendToken: true,
                 });
@@ -155,7 +175,12 @@ export function EphemeralNotebook({ notebookId, runtimePodName, cellSidebarMargi
                 disposeServiceManager(manager);
             }
         };
-    }, [selectedRuntime?.pod_name, selectedRuntime?.url, selectedRuntime?.token]);
+    }, [
+        selectedRuntime?.pod_name,
+        selectedRuntime?.url,
+        selectedRuntime?.token,
+        selectedRuntime?.wsUrl,
+    ]);
     // Bind strictly to the agent runtime sandbox; there is NO local fallback
     // kernel. The notebook executes on the agent's runtime or shows a waiting
     // state until the runtime is ready.
@@ -163,9 +188,17 @@ export function EphemeralNotebook({ notebookId, runtimePodName, cellSidebarMargi
     const activeKernelId = runtimeKernelId;
     const activeStartDefaultKernel = runtimeStartDefaultKernel;
     const ToolbarComponent = toolbarComponent || NotebookToolbar;
-    const isRuntimeStarting = Boolean(String(runtimePodName || '').trim() && !activeServiceManager);
+    const isRuntimeStarting = Boolean((String(runtimePodName || '').trim() ||
+        String(runtimeOverride?.baseUrl || '').trim()) &&
+        !activeServiceManager);
     useProgressTask(`ephemeral-notebook-start-${notebookId}`, isRuntimeStarting);
     useEffect(() => {
+        // When a collaboration provider is active the shared ydoc is the single
+        // source of truth and is synced remotely; the local in-memory persistence
+        // poll would fight it, so it is disabled in that mode.
+        if (collaborationProvider) {
+            return;
+        }
         // Read the CURRENT live notebook model straight from the notebook store.
         // The `NotebookAdapter` exposes `notebook` (the widget, whose `.model` is
         // the `INotebookModel`) and `panel` (`panel.content.model`). There is NO
@@ -247,6 +280,6 @@ export function EphemeralNotebook({ notebookId, runtimePodName, cellSidebarMargi
                             flex: '1 1 auto',
                             minHeight: 0,
                         },
-                    }, children: _jsx(Notebook, { nbformat: initialNbformat, id: notebookId, serviceManager: activeServiceManager, startDefaultKernel: activeStartDefaultKernel, kernelId: activeKernelId, height: "100%", cellSidebarMargin: cellSidebarMargin, extensions: extensions, Toolbar: ToolbarComponent }) }) }) })) : null }));
+                    }, children: _jsx(Notebook, { nbformat: initialNbformat, id: notebookId, serviceManager: activeServiceManager, startDefaultKernel: activeStartDefaultKernel, kernelId: activeKernelId, height: "100%", cellSidebarMargin: cellSidebarMargin, extensions: extensions, Toolbar: ToolbarComponent, collaborationProvider: collaborationProvider }) }) }) })) : null }));
 }
 export default EphemeralNotebook;

package/lib/identity/AgentIdentity.js

@@ -201,11 +201,10 @@ export function IdentityCard({ identity, providerConfig, showExpirationDetails =
                             justifyContent: 'center',
                             backgroundColor: display.color,
                             color: 'white',
-                            flexShrink: 0,
                         }, children: _jsx(display.icon, { size: 16 }) }), _jsxs(Box, { sx: { flex: 1, display: 'flex', alignItems: 'center', gap: 2 }, children: [_jsx(Text, { sx: { fontWeight: 'semibold', fontSize: 1 }, children: display.name }), _jsx(Label, { size: "small", variant: "secondary", children: identity.authType === 'token' ? 'API Key' : 'OAuth' })] }), tokenStatus.isExpired ? (_jsx(Label, { variant: "danger", size: "small", children: _jsxs(Box, { sx: { display: 'inline-flex', alignItems: 'center', gap: 1 }, children: [_jsx(AlertIcon, { size: 12 }), " Expired"] }) })) : tokenStatus.isExpiringSoon ? (_jsx(Label, { variant: "attention", size: "small", children: _jsxs(Box, { sx: { display: 'inline-flex', alignItems: 'center', gap: 1 }, children: [_jsx(ClockIcon, { size: 12 }), " Expiring Soon"] }) })) : (_jsx(Label, { variant: "success", size: "small", children: _jsxs(Box, { sx: { display: 'inline-flex', alignItems: 'center', gap: 1 }, children: [_jsx(CheckCircleFillIcon, { size: 12 }), " Connected"] }) }))] }), _jsxs(Box, { sx: { display: 'flex', alignItems: 'center', gap: 2 }, children: [identity.userInfo?.avatarUrl && (_jsx("a", { href: identity.userInfo.profileUrl ||
                             (provider === 'github'
                                 ? `https://github.com/${identity.userInfo.username}`
-                                : undefined), target: "_blank", rel: "noopener noreferrer", style: { display: 'block', lineHeight: 0, flexShrink: 0 }, children: _jsx(Avatar, { src: identity.userInfo.avatarUrl, size: 32, alt: identity.userInfo.name ||
+                                : undefined), target: "_blank", rel: "noopener noreferrer", style: { display: 'block', lineHeight: 0 }, children: _jsx(Avatar, { src: identity.userInfo.avatarUrl, size: 32, alt: identity.userInfo.name ||
                                 identity.userInfo.username ||
                                 display.name, sx: { cursor: 'pointer' } }) })), _jsxs(Box, { sx: { flex: 1, minWidth: 0 }, children: [identity.userInfo?.name && (_jsx(Text, { sx: { fontWeight: 'semibold', fontSize: 1, display: 'block' }, children: identity.userInfo.name })), (identity.userInfo?.username || identity.userInfo?.email) && (_jsx("a", { href: identity.userInfo.profileUrl ||
                                     (provider === 'github'

package/lib/shims/primerAddonsBox.d.ts

@@ -0,0 +1 @@
+export { Box as default, Box } from '@primer/react';

package/lib/shims/primerAddonsBox.js

@@ -0,0 +1,7 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+// Runtime shim: force the primer-addons Box export to use Primer's Box.
+// This prevents `sx` from being forwarded to raw DOM nodes as an attribute.
+export { Box as default, Box } from '@primer/react';

package/lib/types/chat.d.ts

@@ -4,6 +4,7 @@
  * @module types/chat
  */
 import type { ComponentType, ReactNode } from 'react';
+import type { ICollaborationProvider } from '@datalayer/jupyter-react';
 import type { ChatMessage, MessageHandler } from './messages';
 import type { Protocol, ProtocolConfig } from './protocol';
 import type { McpServerSelection } from './inference';
@@ -12,6 +13,8 @@ import type { AgentRuntimeConfig } from './config';
 import type { SandboxWsStatus } from './sandbox';
 import type { FrontendToolDefinition } from './tools';
 import type { PoweredByTagProps } from '../chat/display/PoweredByTag';
+import type { EphemeralRuntimeOverride } from '../chat/notebook/EphemeralNotebook';
+import type { EphemeralDocumentCollaboration } from '../chat/document/EphemeralDocument';
 /**
  * Context passed to tool-call pre-hooks.
  * Fires when a tool call starts executing (backend or frontend).
@@ -366,6 +369,34 @@ export interface ChatCommonProps {
      * Defaults to the toolbar from `@datalayer/jupyter-react` when omitted.
      */
     ephemeralNotebookToolbar?: EphemeralNotebookToolbarComponent;
+    /**
+     * Optional real-time collaboration provider for the ephemeral notebook.
+     * When supplied, the notebook joins a shared collaborative room so its state
+     * transits over RTC (e.g. between an Agent Node and the SaaS UI) instead of
+     * the tunnel. The caller owns the provider lifecycle.
+     */
+    ephemeralNotebookCollaborationProvider?: ICollaborationProvider;
+    /**
+     * Explicit collaboration room / document id for the ephemeral notebook. When
+     * set it becomes the notebook id directly so multiple peers (e.g. an Agent
+     * Node and the SaaS UI) share the same collaborative room and tool scope.
+     */
+    ephemeralNotebookCollaborationDocumentId?: string;
+    /**
+     * Optional real-time collaboration configuration for the ephemeral document
+     * (Lexical). When supplied the document joins a shared Loro room over
+     * WebSocket so its rich-text state transits over RTC (e.g. between an Agent
+     * Node and the SaaS UI). Its `roomId` also becomes the document id so all
+     * peers share the same collaborative room and lexical tool scope.
+     */
+    ephemeralDocumentCollaboration?: EphemeralDocumentCollaboration;
+    /**
+     * Explicit runtime endpoint for the ephemeral notebook kernel. When set, the
+     * notebook binds its kernel to this endpoint directly instead of resolving a
+     * pod from the user's runtimes list — used to reach an Agent Node's Jupyter
+     * server through the runtimes tunnel HTTP/WebSocket proxy.
+     */
+    ephemeralRuntimeOverride?: EphemeralRuntimeOverride;
     /** Pre-hook: fires when a tool call starts executing */
     onToolCallStart?: (context: ToolCallStartContext) => void;
     /** Post-hook: fires when a tool result is received */
@@ -419,6 +450,18 @@ export interface ChatCommonProps {
      * so the indicator reflects the surface's runtime.
      */
     kernel?: import('@jupyterlab/services/lib/kernel/kernel').IKernelConnection | null;
+    /**
+     * Optional environment name displayed in the chat header's kernel
+     * indicator details (e.g. the agent's local sandbox name). Defaults to
+     * the indicator's own "browser-runtime" placeholder when omitted.
+     */
+    kernelEnvironmentName?: string;
+    /** Optional CPU info displayed in the chat header's kernel indicator. */
+    kernelCpu?: string;
+    /** Optional memory info displayed in the chat header's kernel indicator. */
+    kernelMemory?: string;
+    /** Optional GPU info displayed in the chat header's kernel indicator. */
+    kernelGpu?: string;
     /**
      * Disable ChatBase's internal JupyterReactTheme wrapper.
      * Use this when the host page already provides a JupyterReactTheme boundary.
@@ -746,6 +789,34 @@ export interface ChatBaseProps {
      * Defaults to the toolbar from `@datalayer/jupyter-react` when omitted.
      */
     ephemeralNotebookToolbar?: EphemeralNotebookToolbarComponent;
+    /**
+     * Optional real-time collaboration provider for the ephemeral notebook.
+     * When supplied, the notebook joins a shared collaborative room so its state
+     * transits over RTC (e.g. between an Agent Node and the SaaS UI) instead of
+     * the tunnel. The caller owns the provider lifecycle.
+     */
+    ephemeralNotebookCollaborationProvider?: ICollaborationProvider;
+    /**
+     * Explicit collaboration room / document id for the ephemeral notebook. When
+     * set it becomes the notebook id directly so multiple peers (e.g. an Agent
+     * Node and the SaaS UI) share the same collaborative room and tool scope.
+     */
+    ephemeralNotebookCollaborationDocumentId?: string;
+    /**
+     * Optional real-time collaboration configuration for the ephemeral document
+     * (Lexical). When supplied the document joins a shared Loro room over
+     * WebSocket so its rich-text state transits over RTC (e.g. between an Agent
+     * Node and the SaaS UI). Its `roomId` also becomes the document id so all
+     * peers share the same collaborative room and lexical tool scope.
+     */
+    ephemeralDocumentCollaboration?: EphemeralDocumentCollaboration;
+    /**
+     * Explicit runtime endpoint for the ephemeral notebook kernel. When set, the
+     * notebook binds its kernel to this endpoint directly instead of resolving a
+     * pod from the user's runtimes list — used to reach an Agent Node's Jupyter
+     * server through the runtimes tunnel HTTP/WebSocket proxy.
+     */
+    ephemeralRuntimeOverride?: EphemeralRuntimeOverride;
     /**
      * Callback when the agent requests authorization for an external service.
      * This is called when a tool needs OAuth access to a service like GitHub.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datalayer/agent-runtimes",
-  "version": "1.0.18",
+  "version": "1.0.19",
   "type": "module",
   "workspaces": [
     ".",
@@ -102,8 +102,8 @@
     "start:acp:vite": "VITE_ACP_WS_URL=ws://localhost:8765/api/v1/acp/ws vite",
     "start:agent": "run-p server:start start:agent:vite",
     "start:agent:vite": "vite --open /html/agent.html",
-    "start:agent-node": "run-p server:start:agent-node start:agent-node:vite",
-    "start:agent-node:vite": "vite --open /html/agent-node.html",
+    "start:agent-node": "run-p server:start:agent-node jupyter:start start:agent-node:vite",
+    "start:agent-node:vite": "VITE_JUPYTER_SANDBOX_URL=${VITE_JUPYTER_SANDBOX_URL:-http://localhost:8888/api/jupyter-server?token=60c1661cc408f978c309d04157af55c9588ff9557c9380e4fb50785750703da6} vite --open /html/agent-node.html",
     "start:agent-notebook": "run-p server:start jupyter:start start:agent-notebook:vite",
     "start:agent-notebook:vite": "vite --open /html/agent-notebook.html",
     "start:agent-document": "run-p server:start jupyter:start start:agent-document:vite",
@@ -141,6 +141,7 @@
     "@datalayer/core": "^1.1.48",
     "@datalayer/icons-react": "^1.0.7",
     "@datalayer/jupyter-lexical": "^1.0.16",
+    "@datalayer/lexical-loro": "^1.0.0",
     "@datalayer/jupyter-react": "^2.0.7",
     "@datalayer/primer-addons": "^1.0.15",
     "@datalayer/primer-rjsf": "^1.0.3",