@datalayer/agent-runtimes 1.0.21 → 1.0.26

package/lib/types/chat.d.ts

@@ -3,6 +3,7 @@
  *
  * @module types/chat
  */
+import type { ToolbarItem } from '@datalayer/primer-addons';
 import type { ComponentType, ReactNode } from 'react';
 import type { ICollaborationProvider } from '@datalayer/jupyter-react';
 import type { ChatMessage, MessageHandler } from './messages';
@@ -369,6 +370,17 @@ export interface ChatCommonProps {
      * Defaults to the toolbar from `@datalayer/jupyter-react` when omitted.
      */
     ephemeralNotebookToolbar?: EphemeralNotebookToolbarComponent;
+    /**
+     * Items added to the toolbar of the ephemeral notebook, e.g. the status of
+     * the sandbox it runs on. Merged with the items of the toolbar and ordered
+     * by the `order` of each item.
+     */
+    ephemeralNotebookToolbarExtraItems?: ToolbarItem[];
+    /**
+     * Items added to the toolbar of the ephemeral document, the same way as
+     * {@link ChatProps.ephemeralNotebookToolbarExtraItems} for the notebook.
+     */
+    ephemeralDocumentToolbarExtraItems?: ToolbarItem[];
     /**
      * Optional real-time collaboration provider for the ephemeral notebook.
      * When supplied, the notebook joins a shared collaborative room so its state
@@ -801,6 +813,17 @@ export interface ChatBaseProps {
      * Defaults to the toolbar from `@datalayer/jupyter-react` when omitted.
      */
     ephemeralNotebookToolbar?: EphemeralNotebookToolbarComponent;
+    /**
+     * Items added to the toolbar of the ephemeral notebook, e.g. the status of
+     * the sandbox it runs on. Merged with the items of the toolbar and ordered
+     * by the `order` of each item.
+     */
+    ephemeralNotebookToolbarExtraItems?: ToolbarItem[];
+    /**
+     * Items added to the toolbar of the ephemeral document, the same way as
+     * {@link ChatProps.ephemeralNotebookToolbarExtraItems} for the notebook.
+     */
+    ephemeralDocumentToolbarExtraItems?: ToolbarItem[];
     /**
      * Optional real-time collaboration provider for the ephemeral notebook.
      * When supplied, the notebook joins a shared collaborative room so its state

package/lib/types/tools.d.ts

@@ -192,6 +192,17 @@ export interface ToolCallRequest {
     toolCallId: string;
     toolName: string;
     args: Record<string, unknown>;
+    /**
+     * Whether `args` is the complete argument set.
+     *
+     * Protocols that stream arguments (AG-UI) emit a first tool-call with
+     * `{}` and the real args later — those mark `false`, and execution waits.
+     * Protocols whose tool-call event is terminal (Vercel) mark `true`: an
+     * empty `args` there means the model chose to pass nothing, which is
+     * legal for a tool whose parameters are all optional — waiting on more
+     * would wait forever. Absent, the executor falls back to heuristics.
+     */
+    argsComplete?: boolean;
 }
 /**
  * Tool registry entry (internal)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datalayer/agent-runtimes",
-  "version": "1.0.21",
+  "version": "1.0.26",
   "type": "module",
   "workspaces": [
     ".",
@@ -73,8 +73,8 @@
     "examples:fresh": "npm run clean:cache && npm run examples",
     "examples:jupyter": "run-p jupyter:start server:start server:start:codemode examples:vite",
     "examples:nextjs": "npm run dev --workspace=nextjs-notebook-example",
-    "examples:vite": "VITE_DATALAYER_URL=${VITE_DATALAYER_URL:-https://r1.datalayer.run} VITE_DATALAYER_RUNTIMES_URL=${VITE_DATALAYER_RUNTIMES_URL:-https://r1.datalayer.run} VITE_DATALAYER_AGENT_RUNTIMES_URL=${VITE_DATALAYER_AGENT_RUNTIMES_URL:-https://r1.datalayer.run} VITE_APP_TARGET=examples vite",
-    "examples2:vite": "VITE_DATALAYER_URL=${VITE_DATALAYER_URL:-https://r1.datalayer.run} VITE_DATALAYER_RUNTIMES_URL=${VITE_DATALAYER_RUNTIMES_URL:-https://r1.datalayer.run} VITE_DATALAYER_AGENT_RUNTIMES_URL=${VITE_DATALAYER_AGENT_RUNTIMES_URL:-https://r1.datalayer.run} VITE_APP_TARGET=examples2 vite",
+    "examples:vite": "VITE_DATALAYER_RUNTIMES_URL=${VITE_DATALAYER_RUNTIMES_URL:-https://r1.datalayer.run} VITE_DATALAYER_AGENT_RUNTIMES_URL=${VITE_DATALAYER_AGENT_RUNTIMES_URL:-https://r1.datalayer.run} VITE_APP_TARGET=examples vite",
+    "examples2:vite": "VITE_DATALAYER_RUNTIMES_URL=${VITE_DATALAYER_RUNTIMES_URL:-https://r1.datalayer.run} VITE_DATALAYER_AGENT_RUNTIMES_URL=${VITE_DATALAYER_AGENT_RUNTIMES_URL:-https://r1.datalayer.run} VITE_APP_TARGET=examples2 vite",
     "format": "prettier --write \"src/**/*.{js,jsx,ts,tsx,css,json,md}\" \"examples/**/*.{js,jsx,ts,tsx,css,json,md,mjs}\"",
     "format:all": "npm run format && npm run format --workspaces --if-present",
     "format:check": "prettier --check \"src/**/*.{js,jsx,ts,tsx,css,json,md}\" \"examples/**/*.{js,jsx,ts,tsx,css,json,md,mjs}\"",
@@ -138,7 +138,7 @@
     "@agentclientprotocol/sdk": "^0.8.0",
     "@ai-sdk/react": "4.0.23",
     "@anthropic-ai/sdk": "^0.52.0",
-    "@datalayer/core": "^1.1.48",
+    "@datalayer/core": "^1.1.57",
     "@datalayer/icons-react": "^1.0.7",
     "@datalayer/jupyter-lexical": "^1.0.16",
     "@datalayer/lexical-loro": "^1.0.0",

package/lib/components/runtimes/RuntimePickerBase.d.ts

@@ -1,69 +0,0 @@
-import { ReactElement, ReactNode } from 'react';
-import { ISessionContext } from '@jupyterlab/apputils';
-import { ITranslator } from '@jupyterlab/translation';
-import { IRuntimeDesc } from '../../models';
-import { IMultiServiceManager } from '../../runtimes';
-type IDisplayMode = 'menu' | 'radio';
-/**
- * {@link RuntimePickerBase} properties
- */
-export interface IRuntimePickerBaseProps {
-    /**
-     * The display mode.
-     */
-    display: IDisplayMode;
-    /**
-     * Additional actions items to be placed at the top of the picker
-     */
-    preActions?: ReactNode;
-    /**
-     * Additional actions items to be placed at the bottom of the picker
-     */
-    postActions?: ReactNode;
-    /**
-     * Whether the picker is disabled or not.
-     */
-    disabled?: boolean;
-    /**
-     * Runtime description passing this filter function will be displayed.
-     */
-    filterRuntime?: (desc: IRuntimeDesc) => boolean;
-    /**
-     * Session preference.
-     */
-    preference?: {
-        id?: string;
-        kernelDisplayName?: string;
-        language?: string;
-        location?: string;
-    };
-    /**
-     * Runtime description.
-     */
-    runtimeDesc?: IRuntimeDesc;
-    /**
-     * Set runtime description.
-     */
-    setRuntimeDesc: (desc?: IRuntimeDesc) => void;
-    /**
-     * Document session context.
-     */
-    sessionContext?: ISessionContext;
-    /**
-     * Application service manager
-     */
-    multiServiceManager: IMultiServiceManager;
-    /**
-     * Application translator.
-     */
-    translator?: ITranslator;
-    /**
-     * Change the look depending on the menu integration.
-     */
-    variant?: 'document' | 'cell';
-}
-/**
- * Base Kernel Picker component.
- */
-export declare function RuntimePickerBase(props: IRuntimePickerBaseProps): ReactElement;
-export default RuntimePickerBase;

package/lib/components/runtimes/RuntimePickerBase.js

@@ -1,157 +0,0 @@
-import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-import { useEffect, useMemo, useState } from 'react';
-import { nullTranslator } from '@jupyterlab/translation';
-//import { kernelIcon } from '@jupyterlab/ui-components';
-import { ActionList, ActionMenu, IconButton, Text, RadioGroup, Radio, FormControl, LabelGroup, Label, } from '@primer/react';
-import CloudUploadIcon from '@datalayer/icons-react/data1/CloudUploadIcon';
-import { Box } from '@datalayer/primer-addons';
-import { CpuIcon } from '@primer/octicons-react';
-import { BrowserIcon, LaptopSimpleIcon } from '@datalayer/icons-react';
-import { CreditsIndicator } from '../../components/progress';
-import { isRuntimeRemote } from '../../runtimes';
-import { getGroupedRuntimeDescs } from './RuntimeUtils';
-/**
- * Maximal runtime display name length after which it is trimmed.
- */
-const RUNTIME_DISPLAY_NAME_MAX_LENGTH = 25;
-/**
- * Base Kernel Picker component.
- */
-export function RuntimePickerBase(props) {
-    const { disabled, display, filterRuntime, multiServiceManager, postActions, preActions, preference, runtimeDesc, sessionContext, setRuntimeDesc, translator, variant, } = props;
-    const [groupedRuntimeDescs, setGroupedRuntimeDescs] = useState(getGroupedRuntimeDescs(multiServiceManager, preference?.id, translator, filterRuntime, variant));
-    const trans = useMemo(() => (translator ?? nullTranslator).load('jupyterlab'), [translator]);
-    const [defaultSet, setDefaultSet] = useState(false);
-    const [remoteModelsReady, setRemoteModelsReady] = useState(!multiServiceManager.remote);
-    // Trick because overflow is an unknown prop of ActionMenu.Overlay.
-    const overlayProps = {
-        maxHeight: 'large',
-        width: (variant === 'cell' ? 'small' : 'auto'),
-    };
-    useEffect(() => {
-        let disposed = false;
-        const updateGroupedRuntimeDescs = () => {
-            if (disposed) {
-                return;
-            }
-            setGroupedRuntimeDescs(getGroupedRuntimeDescs(multiServiceManager, preference?.id, translator, filterRuntime, variant));
-        };
-        const refreshRuntimeModels = multiServiceManager.remote?.runtimesManager.refreshRuntimeModels?.();
-        if (refreshRuntimeModels) {
-            refreshRuntimeModels
-                .catch(reason => {
-                console.error('Failed to resolve remote runtimes for the runtime picker.', reason);
-            })
-                .finally(() => {
-                if (disposed) {
-                    return;
-                }
-                updateGroupedRuntimeDescs();
-                setRemoteModelsReady(true);
-            });
-        }
-        else {
-            setRemoteModelsReady(true);
-        }
-        // Recompute immediately in case a runtime changed between the initial
-        // render and connecting the manager signals.
-        updateGroupedRuntimeDescs();
-        multiServiceManager.browser?.kernels.runningChanged.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.browser?.kernelspecs.specsChanged.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.browser?.sessions.runningChanged.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.local.kernels.runningChanged.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.local.kernelspecs.specsChanged.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.local.sessions.runningChanged.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.remote?.runtimesManager.changed.connect(updateGroupedRuntimeDescs);
-        multiServiceManager.remote?.environments.changed.connect(updateGroupedRuntimeDescs);
-        return () => {
-            disposed = true;
-            multiServiceManager.browser?.kernels.runningChanged.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.browser?.kernelspecs.specsChanged.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.browser?.sessions.runningChanged.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.local.kernels.runningChanged.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.local.kernelspecs.specsChanged.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.local.sessions.runningChanged.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.remote?.runtimesManager.changed.disconnect(updateGroupedRuntimeDescs);
-            multiServiceManager.remote?.environments.changed.disconnect(updateGroupedRuntimeDescs);
-        };
-    }, [multiServiceManager, preference?.id, translator, filterRuntime, variant]);
-    useEffect(() => {
-        if (defaultSet) {
-            return;
-        }
-        if (sessionContext && groupedRuntimeDescs) {
-            const kernelId = sessionContext.session?.kernel?.id;
-            if (kernelId) {
-                let matched = false;
-                Object.entries(groupedRuntimeDescs).forEach(([group, runtimeDescs]) => {
-                    runtimeDescs.forEach(runtimeDesc => {
-                        if (runtimeDesc.kernelId === kernelId) {
-                            matched = true;
-                            setRuntimeDesc(runtimeDesc);
-                        }
-                    });
-                });
-                if (!matched && !remoteModelsReady) {
-                    return;
-                }
-            }
-        }
-        setDefaultSet(true);
-    }, [
-        defaultSet,
-        groupedRuntimeDescs,
-        remoteModelsReady,
-        sessionContext,
-        setRuntimeDesc,
-    ]);
-    // For cell using submenu instead of group would be nice unfortunately the feature
-    // is not yet implemented in the component there has been a not-great demo story.
-    // https://github.com/primer/react/pull/3585
-    return (_jsx(_Fragment, { children: display === 'menu' ? (
-        /*
-         * Section for Menu display.
-         */
-        _jsxs(ActionMenu, { children: [variant === 'cell' ? (_jsx(ActionMenu.Anchor, { children: _jsx(IconButton, { disabled: disabled || groupedRuntimeDescs === null, 
-                        //                icon={() => <kernelIcon.react className="dla-Cell-runtime-icon" tag={'span'} />}
-                        icon: () => (_jsx("span", { className: "dla-Cell-runtime-icon", children: _jsx(CpuIcon, {}) })), "aria-label": trans.__('Assign a Runtime to the Cell.'), title: trans.__('Assign a Runtime to the Cell.'), size: "small", variant: "invisible" }) })) : (_jsxs(ActionMenu.Button, { variant: "default", disabled: disabled || groupedRuntimeDescs === null, children: [_jsx(Text, { fontWeight: 'bold', children: trans.__('Runtime:') }), ' ' + (runtimeDesc?.displayName ?? trans.__('No Runtime'))] })), _jsx(ActionMenu.Overlay, { ...overlayProps, width: "medium", sx: { overflowY: 'auto' }, side: variant === 'cell' ? 'outside-left' : 'outside-right', children: _jsxs(ActionList, { selectionVariant: "single", children: [variant === 'cell' && (_jsxs(ActionList.Item, { selected: runtimeDesc === undefined, onSelect: () => {
-                                    setRuntimeDesc(undefined);
-                                }, children: [preference?.location && (_jsx(ActionList.LeadingVisual, { children: preference.location === 'local' ? (_jsx(LaptopSimpleIcon, {})) : preference.location === 'browser' ? (_jsx(BrowserIcon, {})) : (_jsx(CloudUploadIcon, {})) })), trans.__('Assign the Notebook Runtime')] }, 'null')), !!preActions && preActions, Object.entries(groupedRuntimeDescs ?? {}).map(([group, runtimeDescs]) => (_jsxs(ActionList.Group, { children: [_jsx(ActionList.GroupHeading, { children: group }), runtimeDescs.map(candidateRuntimeDesc => {
-                                        const annotation = candidateRuntimeDesc.podName
-                                            ? ` - ${candidateRuntimeDesc.podName.split('-', 2).reverse()[0]}`
-                                            : candidateRuntimeDesc.kernelId
-                                                ? ` - ${candidateRuntimeDesc.kernelId}`
-                                                : '';
-                                        const fullDisplayName = (candidateRuntimeDesc.displayName ?? '') + annotation;
-                                        const displayName = (candidateRuntimeDesc.displayName?.length ?? 0) >
-                                            RUNTIME_DISPLAY_NAME_MAX_LENGTH
-                                            ? candidateRuntimeDesc.displayName.slice(0, RUNTIME_DISPLAY_NAME_MAX_LENGTH) + '…'
-                                            : (candidateRuntimeDesc.displayName ?? '');
-                                        return (_jsxs(ActionList.Item, { title: fullDisplayName, selected: (candidateRuntimeDesc.location ===
-                                                runtimeDesc?.location ||
-                                                (isRuntimeRemote(candidateRuntimeDesc.location) &&
-                                                    isRuntimeRemote(runtimeDesc?.location ?? 'local'))) &&
-                                                (candidateRuntimeDesc.kernelId ??
-                                                    candidateRuntimeDesc.name) ===
-                                                    (runtimeDesc?.kernelId ?? runtimeDesc?.name), onSelect: () => {
-                                                setRuntimeDesc(candidateRuntimeDesc);
-                                            }, children: [_jsx(ActionList.LeadingVisual, { children: candidateRuntimeDesc.location === 'local' ? (_jsx(LaptopSimpleIcon, {})) : candidateRuntimeDesc.location === 'browser' ? (_jsx(BrowserIcon, {})) : (_jsx(CloudUploadIcon, {})) }), displayName + annotation.slice(0, 10)] }, candidateRuntimeDesc.name));
-                                    })] }, group))), !!postActions && (_jsxs(_Fragment, { children: [_jsx(ActionList.Divider, {}), postActions] }))] }) })] })) : (
-        /*
-         * Section for Radio display.
-         */
-        _jsxs(_Fragment, { children: [defaultSet && (_jsx(RadioGroup, { name: "kernel-options", "aria-labelledby": "kernel-options", children: Object.entries(groupedRuntimeDescs ?? {}).map(([group, runtimeDescs]) => (_jsxs(Box, { children: [_jsx(Box, { as: "h4", style: { marginTop: 0 }, children: group }), runtimeDescs.map(k => {
-                                return (_jsxs(Box, { title: k.name, children: [_jsxs(FormControl, { children: [_jsx(Radio, { value: k.kernelId, onChange: () => {
-                                                        setRuntimeDesc(k);
-                                                    }, checked: (k.location === runtimeDesc?.location ||
-                                                        (isRuntimeRemote(k.location) &&
-                                                            isRuntimeRemote(runtimeDesc?.location ?? 'local'))) &&
-                                                        (k.kernelId ?? k.name) ===
-                                                            (runtimeDesc?.kernelId ?? runtimeDesc?.name) }), _jsx(FormControl.Label, { children: _jsxs(Box, { display: "flex", children: [_jsx(Box, { children: k.displayName }), k.kernelId && k.location === 'remote' && (_jsx(Box, { ml: 3, mt: 1, children: _jsx(CreditsIndicator, { kernelId: k.kernelId, serviceManager: multiServiceManager.remote }, "credits-indicator") }))] }) }), _jsx(FormControl.Caption, { children: _jsxs(LabelGroup, { sx: { marginTop: 1 }, children: [_jsx(Label, { variant: "secondary", children: k.name }), _jsx(Label, { variant: "secondary", sx: { marginLeft: 1 }, children: k.location }), k.burningRate && (_jsxs(Label, { variant: "sponsors", sx: { marginLeft: 1 }, children: [k.burningRate, " credits/second"] })), k.gpu && (_jsx(Label, { variant: "success", sx: { marginLeft: 1 }, children: "GPU" }))] }) })] }), _jsx(ActionList.Divider, {})] }, k.kernelId));
-                            })] }, group))) })), !!postActions && _jsx(_Fragment, { children: postActions })] })) }));
-}
-export default RuntimePickerBase;

package/lib/components/runtimes/RuntimePickerNotebook.d.ts

@@ -1,42 +0,0 @@
-import { ITranslator } from '@jupyterlab/translation';
-import { CommandRegistry } from '@lumino/commands';
-import type { IMultiServiceManager, IDatalayerSessionContext } from '../../runtimes';
-import { RuntimeTransfer } from './RuntimeTransfer';
-/**
- * {@link RuntimePickerNotebook} properties
- */
-export interface IRuntimePickerNotebookProps {
-    /**
-     * Callback to allow the user to login
-     */
-    logIn?: () => void;
-    /**
-     * Document session context
-     */
-    sessionContext: IDatalayerSessionContext;
-    /**
-     * Multi service manager
-     */
-    multiServiceManager: IMultiServiceManager;
-    /**
-     * Set the selected kernel
-     */
-    setValue: (v: RuntimeTransfer | Error) => void;
-    /**
-     * Close the dialog
-     */
-    close: () => void;
-    /**
-     * Command registry
-     */
-    commands: CommandRegistry;
-    /**
-     * Application translator
-     */
-    translator?: ITranslator;
-}
-/**
- * Runtime Picker components for a Notebook.
- */
-export declare function RuntimePickerNotebook(props: IRuntimePickerNotebookProps): JSX.Element;
-export default RuntimePickerNotebook;

package/lib/components/runtimes/RuntimePickerNotebook.js

@@ -1,232 +0,0 @@
-import { Fragment as _Fragment, jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-import { useCallback, useEffect, useState } from 'react';
-import { ActionList, FormControl, ToggleSwitch, Tooltip, IconButton, } from '@primer/react';
-import { Box } from '@datalayer/primer-addons';
-import { AlertIcon } from '@primer/octicons-react';
-import { JSONExt } from '@lumino/coreutils';
-import { KernelExecutor } from '@datalayer/jupyter-react';
-import { RuntimeSnippetsFacade } from '../../jupyter';
-import { ExternalTokenSilentLogin } from '@datalayer/core/lib/components/iam';
-import { useCoreStore, useIAMStore } from '../../state/substates';
-import { RuntimeReservationControl, MAXIMAL_RUNTIME_TIME_RESERVATION_MINUTES, } from './RuntimeReservationControl';
-import { RuntimeVariables } from './RuntimeVariables';
-import { RuntimePickerBase } from './RuntimePickerBase';
-/**
- * Runtime Picker components for a Notebook.
- */
-export function RuntimePickerNotebook(props) {
-    const { multiServiceManager, sessionContext, setValue, translator } = props;
-    const { configuration } = useCoreStore();
-    const { credits, refreshCredits, token, user } = useIAMStore();
-    const [selectedRuntimeDesc, setSelectedRuntimeDesc] = useState();
-    const [timeLimit, setTimeLimit] = useState(10);
-    const [userStorage, setUserStorage] = useState(false);
-    const [canTransferFrom, setTransferFrom] = useState(false);
-    const [canTransferTo, setTransferTo] = useState(false);
-    const [transferVariables, setTransferVariables] = useState(false);
-    const [hasLoadedVariables, setHasLoadedVariables] = useState(false);
-    const [kernelVariables, setRuntimeVariables] = useState();
-    const [toTransfer, setToTransfer] = useState([]);
-    useEffect(() => {
-        refreshCredits();
-    }, []);
-    useEffect(() => {
-        const specs = sessionContext.specsManager.specs?.kernelspecs;
-        if (sessionContext.session?.kernel?.name && specs) {
-            const spec = Object.values(specs).find(spec => spec?.name === sessionContext.session.kernel.name);
-            if (spec) {
-                setSelectedRuntimeDesc({
-                    name: spec.name,
-                    kernelId: sessionContext.session.kernel.id,
-                    location: sessionContext.location,
-                    language: spec.language,
-                    displayName: sessionContext.kernelDisplayName,
-                });
-                setTransferFrom(RuntimeSnippetsFacade.supports(spec.language));
-            }
-        }
-    }, [sessionContext]);
-    const listRuntimeVariables = useCallback(async () => {
-        if (hasLoadedVariables) {
-            return Promise.resolve();
-        }
-        setHasLoadedVariables(true);
-        const connection = sessionContext.session.kernel;
-        const spec = sessionContext.specsManager.specs.kernelspecs[connection.model.name];
-        const snippets = new RuntimeSnippetsFacade(spec.language);
-        const outputs = await new KernelExecutor({ connection }).execute(snippets.listVariables());
-        const content = outputs.get(0).data['text/plain'];
-        if (content) {
-            setRuntimeVariables(JSON.parse(
-            // We need to remove the quotes prior to parsing.
-            content.slice(1, content.length - 1)));
-            if (toTransfer.length) {
-                const candidates = Object.keys(kernelVariables ?? {});
-                setToTransfer(toTransfer.filter(n => candidates.includes(n)));
-            }
-            else {
-                // By default select all variables.
-                setToTransfer(Object.keys(kernelVariables ?? {}));
-            }
-        }
-        else {
-            setRuntimeVariables({});
-        }
-    }, [hasLoadedVariables, kernelVariables, toTransfer]);
-    const setSelectedVariables = useCallback((l) => {
-        if (!JSONExt.deepEqual(toTransfer, l)) {
-            setToTransfer(l);
-        }
-    }, [toTransfer]);
-    const setTransferVariable = useCallback((value) => {
-        if (transferVariables !== value) {
-            if (value) {
-                listRuntimeVariables().catch(error => {
-                    console.error('Failed to list the runtime variable', error);
-                    setHasLoadedVariables(false);
-                });
-            }
-            setTransferVariables(value);
-        }
-    }, [transferVariables]);
-    const setRuntimeDesc = useCallback((runtimeDesc) => {
-        if (!runtimeDesc) {
-            if (selectedRuntimeDesc) {
-                setSelectedRuntimeDesc(undefined);
-                setTransferTo(false);
-            }
-            return;
-        }
-        if (selectedRuntimeDesc?.displayName !== runtimeDesc.displayName ||
-            selectedRuntimeDesc?.kernelId !== runtimeDesc.kernelId) {
-            setSelectedRuntimeDesc({ ...runtimeDesc });
-            setTransferTo(RuntimeSnippetsFacade.supports(runtimeDesc.language));
-        }
-    }, [selectedRuntimeDesc]);
-    const handleUserStorageChange = useCallback((e) => {
-        e.preventDefault();
-        setUserStorage(!userStorage);
-    }, [userStorage]);
-    useEffect(() => {
-        const resolvedBurningRate = selectedRuntimeDesc?.burningRate ??
-            multiServiceManager.remote?.environments
-                .get()
-                .find(env => env.name === selectedRuntimeDesc?.name)?.burning_rate;
-        const includedRuns = user?.subscription?.usage?.included_runs ??
-            user?.subscription?.included_runs;
-        const currentRuns = user?.subscription?.usage?.current_runs ??
-            user?.subscription?.current_runs ??
-            user?.subscription?.used_runs;
-        const hasKnownRunAllowance = typeof includedRuns === 'number';
-        const hasRemainingRuns = hasKnownRunAllowance &&
-            typeof currentRuns === 'number' &&
-            includedRuns > 0 &&
-            currentRuns < includedRuns;
-        const hasKnownCredits = typeof credits?.available === 'number';
-        const maxMinutes = selectedRuntimeDesc?.location === 'remote' && resolvedBurningRate
-            ? Math.floor((credits?.available ?? 0) / resolvedBurningRate / 60.0)
-            : undefined;
-        const effectiveTimeLimit = selectedRuntimeDesc?.location === 'remote'
-            ? Math.max(1, Math.min(timeLimit, maxMinutes && maxMinutes > 0 ? maxMinutes : 10))
-            : timeLimit;
-        const creditsLimit = selectedRuntimeDesc?.location === 'remote' && resolvedBurningRate
-            ? Math.min(effectiveTimeLimit, MAXIMAL_RUNTIME_TIME_RESERVATION_MINUTES) *
-                resolvedBurningRate *
-                60
-            : undefined;
-        const requiresRuntimeStart = !!selectedRuntimeDesc && !selectedRuntimeDesc.kernelId;
-        if (requiresRuntimeStart && selectedRuntimeDesc.location === 'remote') {
-            if (!resolvedBurningRate || !Number.isFinite(resolvedBurningRate)) {
-                setValue({ runtime: null, selectedVariables: toTransfer });
-                return;
-            }
-            if (hasKnownCredits &&
-                hasKnownRunAllowance &&
-                !hasRemainingRuns &&
-                (!creditsLimit || creditsLimit <= 0)) {
-                setValue({ runtime: null, selectedVariables: toTransfer });
-                return;
-            }
-        }
-        setValue({
-            runtime: selectedRuntimeDesc
-                ? {
-                    environmentName: ['browser', 'remote'].includes(selectedRuntimeDesc.location)
-                        ? `${selectedRuntimeDesc.location}-${selectedRuntimeDesc.name}`
-                        : selectedRuntimeDesc.name,
-                    id: selectedRuntimeDesc.kernelId,
-                    creditsLimit,
-                    capabilities: userStorage ? ['user_storage'] : undefined,
-                }
-                : null,
-            selectedVariables: toTransfer,
-        });
-    }, [
-        selectedRuntimeDesc,
-        userStorage,
-        toTransfer,
-        timeLimit,
-        multiServiceManager.remote,
-        credits?.available,
-        user,
-    ]);
-    const { kernelPreference: { canStart }, } = sessionContext;
-    const resolvedBurningRate = selectedRuntimeDesc?.burningRate ??
-        multiServiceManager.remote?.environments
-            .get()
-            .find(env => env.name === selectedRuntimeDesc?.name)?.burning_rate;
-    const maxFromCredits = resolvedBurningRate
-        ? Math.floor((credits?.available ?? 0) / resolvedBurningRate / 60.0)
-        : -1;
-    const includedRuns = user?.subscription?.usage?.included_runs ??
-        user?.subscription?.included_runs;
-    const currentRuns = user?.subscription?.usage?.current_runs ??
-        user?.subscription?.current_runs ??
-        user?.subscription?.used_runs;
-    const hasKnownRunAllowance = typeof includedRuns === 'number';
-    const hasRemainingRuns = hasKnownRunAllowance &&
-        typeof currentRuns === 'number' &&
-        includedRuns > 0 &&
-        currentRuns < includedRuns;
-    const hasKnownCredits = typeof credits?.available === 'number';
-    const effectiveMaxMinutes = selectedRuntimeDesc?.location === 'remote'
-        ? hasKnownCredits && hasKnownRunAllowance && !hasRemainingRuns
-            ? Math.max(1, maxFromCredits)
-            : Math.max(10, maxFromCredits > 0 ? maxFromCredits : 0)
-        : Math.max(1, maxFromCredits);
-    const outOfCredits = hasKnownCredits &&
-        hasKnownRunAllowance &&
-        !hasRemainingRuns &&
-        maxFromCredits < Number.EPSILON;
-    return (_jsxs(Box, { as: "form", className: "dla-Runtimes-picker", children: [_jsx(Box, { sx: { padding: 'var(--stack-padding-condensed) 0' }, children: _jsx(RuntimePickerBase, { display: "radio", disabled: canStart === false, preference: {
-                        id: sessionContext.session?.id,
-                        kernelDisplayName: sessionContext.kernelPreference.shouldStart
-                            ? sessionContext.kernelDisplayName
-                            : undefined,
-                    }, sessionContext: sessionContext, multiServiceManager: multiServiceManager, translator: translator, runtimeDesc: selectedRuntimeDesc, setRuntimeDesc: setRuntimeDesc, postActions: token || !props.logIn ? (
-                    /*
-                  <Button
-                    variant="default"
-                    onClick={e => {
-                      e.preventDefault();
-                      commands.execute(CommandIDs.launchRemoteRuntime);
-                      close();
-                    }}
-                  >
-                    Launch a New Runtime
-                  </Button>
-                  */
-                    _jsx(_Fragment, {})) : (_jsx(ActionList.Item, { onSelect: props.logIn, title: 'Runtime Provider.', children: _jsx(ExternalTokenSilentLogin, { message: "Click to sign in." }) })) }) }), !selectedRuntimeDesc?.kernelId &&
-                selectedRuntimeDesc?.location === 'remote' && (_jsxs(_Fragment, { children: [_jsx(RuntimeReservationControl, { disabled: outOfCredits || selectedRuntimeDesc?.location !== 'remote', label: 'Time reservation', max: effectiveMaxMinutes, time: timeLimit, onTimeChange: setTimeLimit, error: outOfCredits && maxFromCredits >= 0
-                            ? 'You must add credits to your account.'
-                            : timeLimit === 0
-                                ? 'You must set a time limit.'
-                                : undefined, burningRate: selectedRuntimeDesc.burningRate }), !configuration.whiteLabel && (_jsxs(FormControl, { disabled: !!selectedRuntimeDesc?.kernelId ||
-                            selectedRuntimeDesc?.location !== 'remote', layout: "horizontal", children: [_jsxs(FormControl.Label, { id: "user-storage-picker-label", children: ["User storage", _jsx(Tooltip, { text: "The runtime will be slower to start.", direction: "e", style: { marginLeft: 3 }, children: _jsx(IconButton, { icon: AlertIcon, "aria-label": "", variant: "invisible" }) })] }), _jsx(ToggleSwitch, { disabled: !!selectedRuntimeDesc?.kernelId ||
-                                    selectedRuntimeDesc?.location !== 'remote', checked: userStorage, size: "small", onClick: handleUserStorageChange, "aria-labelledby": "user-storage-picker-label" })] }))] })), canTransferFrom && canTransferTo && (_jsx(RuntimeVariables, { selectedVariables: toTransfer, setSelectVariable: setSelectedVariables, transferVariables: transferVariables, setTransferVariable: setTransferVariable, kernelVariables: kernelVariables, translator: translator }))] }));
-}
-export default RuntimePickerNotebook;

package/lib/components/runtimes/RuntimeSimplePicker.d.ts

@@ -1,38 +0,0 @@
-import { Session } from '@jupyterlab/services';
-import { IRuntimeModel, type IRuntimeDesc } from '../../models';
-export interface IRuntimeAssignOptions {
-    /**
-     * Runtime description.
-     */
-    runtimeDesc?: IRuntimeDesc;
-    /**
-     * Runtime model to connect to.
-     */
-    runtimeModel?: IRuntimeModel;
-    /**
-     * Whether to transfer the state to the runtime model.
-     */
-    transferState?: boolean;
-}
-/**
- * Runtime picker component properties.
- */
-interface IRuntimeSimplePickerProps {
-    /**
-     * Runtime assignment callback.
-     */
-    assignRuntime: (options: IRuntimeAssignOptions) => Promise<void>;
-    /**
-     * Connection to the active runtime.
-     */
-    sessionConnection?: Session.ISessionConnection;
-    /**
-     * Whether browser runtime is available.
-     */
-    browserRuntimeAvailable?: boolean;
-}
-/**
- * Runtime simple picker component.
- */
-export declare function RuntimeSimplePicker(props: IRuntimeSimplePickerProps): JSX.Element;
-export {};