@datalayer/agent-runtimes 1.0.9 → 1.0.17

package/lib/specs/tools.js

@@ -5,6 +5,258 @@
 // ============================================================================
 // Tool Definitions
 // ============================================================================
+export const CREATE_PLAN_TOOL_SPEC_0_0_1 = {
+    id: 'create-plan',
+    version: '0.0.1',
+    name: 'Create Plan',
+    description: 'Create a plan with multiple steps and emit an AG-UI state snapshot.',
+    tags: ['example', 'ag-ui', 'state'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'create_plan',
+    },
+    icon: 'list-unordered',
+    emoji: '📋',
+};
+export const CURRENT_TIME_TOOL_SPEC_0_0_1 = {
+    id: 'current-time',
+    version: '0.0.1',
+    name: 'Current Time',
+    description: 'Return the current time in ISO format for a given timezone.',
+    tags: ['example', 'ag-ui', 'time'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'current_time',
+    },
+    icon: 'clock',
+    emoji: '🕒',
+};
+export const DISPLAY_RECIPE_TOOL_SPEC_0_0_1 = {
+    id: 'display-recipe',
+    version: '0.0.1',
+    name: 'Display Recipe',
+    description: 'Update the shared recipe state and emit an AG-UI state snapshot.',
+    tags: ['example', 'ag-ui', 'shared-state'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'display_recipe',
+    },
+    icon: 'book',
+    emoji: '🍳',
+};
+export const EXAMPLE_CREATE_PLAN_TOOL_SPEC_0_0_1 = {
+    id: 'example-create-plan',
+    version: '0.0.1',
+    name: 'Create Plan',
+    description: 'Create a plan with multiple steps and emit an AG-UI state snapshot.',
+    tags: ['example', 'ag-ui', 'state'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'create_plan',
+    },
+    icon: 'list-unordered',
+    emoji: '📋',
+};
+export const EXAMPLE_CURRENT_TIME_TOOL_SPEC_0_0_1 = {
+    id: 'example-current-time',
+    version: '0.0.1',
+    name: 'Current Time',
+    description: 'Return the current time in ISO format for a given timezone.',
+    tags: ['example', 'ag-ui', 'time'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'current_time',
+    },
+    icon: 'clock',
+    emoji: '🕒',
+};
+export const EXAMPLE_DISPLAY_RECIPE_TOOL_SPEC_0_0_1 = {
+    id: 'example-display-recipe',
+    version: '0.0.1',
+    name: 'Display Recipe',
+    description: 'Update the shared recipe state and emit an AG-UI state snapshot.',
+    tags: ['example', 'ag-ui', 'shared-state'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'display_recipe',
+    },
+    icon: 'book',
+    emoji: '🍳',
+};
+export const EXAMPLE_GENERATE_HAIKU_TOOL_SPEC_0_0_1 = {
+    id: 'example-generate-haiku',
+    version: '0.0.1',
+    name: 'Generate Haiku',
+    description: 'Generate a haiku (Japanese + English + gradient) rendered as a card by the frontend.',
+    tags: ['example', 'ag-ui', 'generative-ui'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'generate_haiku',
+    },
+    icon: 'pencil',
+    emoji: '🖋️',
+};
+export const EXAMPLE_GENERATE_TASK_STEPS_TOOL_SPEC_0_0_1 = {
+    id: 'example-generate-task-steps',
+    version: '0.0.1',
+    name: 'Generate Task Steps',
+    description: 'Generate task steps for human review and emit an AG-UI state snapshot.',
+    tags: ['example', 'ag-ui', 'human-in-the-loop'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'generate_task_steps',
+    },
+    icon: 'tasklist',
+    emoji: '🧑‍⚖️',
+};
+export const EXAMPLE_GET_WEATHER_TOOL_SPEC_0_0_1 = {
+    id: 'example-get-weather',
+    version: '0.0.1',
+    name: 'Get Weather',
+    description: 'Fetch current weather for a location from the Open-Meteo API for frontend rendering.',
+    tags: ['example', 'ag-ui', 'weather'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'get_weather',
+    },
+    icon: 'sun',
+    emoji: '🌤️',
+};
+export const EXAMPLE_RENDER_A2UI_SURFACE_TOOL_SPEC_0_0_1 = {
+    id: 'example-render-a2ui-surface',
+    version: '0.0.1',
+    name: 'Render A2UI Surface',
+    description: 'Turn a declarative field spec into a validated A2UI v0.9 surface rendered live by the frontend as an interactive form/card.',
+    tags: ['example', 'ag-ui', 'a2ui', 'generative-ui'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.a2ui',
+        method: 'render_a2ui_surface',
+    },
+    icon: 'browser',
+    emoji: '🎛️',
+};
+export const EXAMPLE_UPDATE_PLAN_STEP_TOOL_SPEC_0_0_1 = {
+    id: 'example-update-plan-step',
+    version: '0.0.1',
+    name: 'Update Plan Step',
+    description: 'Update a plan step and emit an AG-UI state delta (JSON Patch RFC 6902).',
+    tags: ['example', 'ag-ui', 'state'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'update_plan_step',
+    },
+    icon: 'checklist',
+    emoji: '✅',
+};
+export const GENERATE_HAIKU_TOOL_SPEC_0_0_1 = {
+    id: 'generate-haiku',
+    version: '0.0.1',
+    name: 'Generate Haiku',
+    description: 'Generate a haiku (Japanese + English + gradient) rendered as a card by the frontend.',
+    tags: ['example', 'ag-ui', 'generative-ui'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'generate_haiku',
+    },
+    icon: 'pencil',
+    emoji: '🖋️',
+};
+export const GENERATE_TASK_STEPS_TOOL_SPEC_0_0_1 = {
+    id: 'generate-task-steps',
+    version: '0.0.1',
+    name: 'Generate Task Steps',
+    description: 'Generate task steps for human review and emit an AG-UI state snapshot.',
+    tags: ['example', 'ag-ui', 'human-in-the-loop'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'generate_task_steps',
+    },
+    icon: 'tasklist',
+    emoji: '🧑‍⚖️',
+};
+export const GET_WEATHER_TOOL_SPEC_0_0_1 = {
+    id: 'get-weather',
+    version: '0.0.1',
+    name: 'Get Weather',
+    description: 'Fetch current weather for a location from the Open-Meteo API for frontend rendering.',
+    tags: ['example', 'ag-ui', 'weather'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'get_weather',
+    },
+    icon: 'sun',
+    emoji: '🌤️',
+};
 export const RUNTIME_ECHO_TOOL_SPEC_0_0_1 = {
     id: 'runtime-echo',
     version: '0.0.1',
@@ -59,13 +311,46 @@ export const RUNTIME_SENSITIVE_ECHO_TOOL_SPEC_0_0_1 = {
     icon: 'shield',
     emoji: '🛡️',
 };
+export const UPDATE_PLAN_STEP_TOOL_SPEC_0_0_1 = {
+    id: 'update-plan-step',
+    version: '0.0.1',
+    name: 'Update Plan Step',
+    description: 'Update a plan step and emit an AG-UI state delta (JSON Patch RFC 6902).',
+    tags: ['example', 'ag-ui', 'state'],
+    enabled: true,
+    approval: 'auto',
+    timeout: undefined,
+    requiresApproval: false,
+    runtime: {
+        language: 'python',
+        package: 'agent_runtimes.examples.tools.ag_ui',
+        method: 'update_plan_step',
+    },
+    icon: 'checklist',
+    emoji: '✅',
+};
 // ============================================================================
 // Tool Catalog
 // ============================================================================
 export const TOOL_CATALOG = {
+    'create-plan': CREATE_PLAN_TOOL_SPEC_0_0_1,
+    'current-time': CURRENT_TIME_TOOL_SPEC_0_0_1,
+    'display-recipe': DISPLAY_RECIPE_TOOL_SPEC_0_0_1,
+    'example-create-plan': EXAMPLE_CREATE_PLAN_TOOL_SPEC_0_0_1,
+    'example-current-time': EXAMPLE_CURRENT_TIME_TOOL_SPEC_0_0_1,
+    'example-display-recipe': EXAMPLE_DISPLAY_RECIPE_TOOL_SPEC_0_0_1,
+    'example-generate-haiku': EXAMPLE_GENERATE_HAIKU_TOOL_SPEC_0_0_1,
+    'example-generate-task-steps': EXAMPLE_GENERATE_TASK_STEPS_TOOL_SPEC_0_0_1,
+    'example-get-weather': EXAMPLE_GET_WEATHER_TOOL_SPEC_0_0_1,
+    'example-render-a2ui-surface': EXAMPLE_RENDER_A2UI_SURFACE_TOOL_SPEC_0_0_1,
+    'example-update-plan-step': EXAMPLE_UPDATE_PLAN_STEP_TOOL_SPEC_0_0_1,
+    'generate-haiku': GENERATE_HAIKU_TOOL_SPEC_0_0_1,
+    'generate-task-steps': GENERATE_TASK_STEPS_TOOL_SPEC_0_0_1,
+    'get-weather': GET_WEATHER_TOOL_SPEC_0_0_1,
     'runtime-echo': RUNTIME_ECHO_TOOL_SPEC_0_0_1,
     'runtime-send-mail': RUNTIME_SEND_MAIL_TOOL_SPEC_0_0_1,
     'runtime-sensitive-echo': RUNTIME_SENSITIVE_ECHO_TOOL_SPEC_0_0_1,
+    'update-plan-step': UPDATE_PLAN_STEP_TOOL_SPEC_0_0_1,
 };
 export function getToolSpecs() {
     return Object.values(TOOL_CATALOG);

package/lib/state/index.d.ts

@@ -0,0 +1,4 @@
+export * from './substates';
+export * from '@datalayer/core/lib/state/storage';
+export * from '@datalayer/core/lib/state/State';
+export { coreStore, useCoreStore, iamStore, useIAMStore, RESERVATION_WARNING_TIME_MS, layoutStore, useLayoutStore, organizationStore, useOrganizationStore, surveysStore, useSurveysStore, teamStore, useTeamStore, datasourceStore, useDatasourceStore, runStore, useRunStore, DATALAYER_IAM_TOKEN_KEY, DATALAYER_IAM_USER_KEY, } from '@datalayer/core/lib/state';

package/lib/state/index.js

@@ -0,0 +1,10 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+// Export runtime/content substates owned by agent-runtimes.
+export * from './substates';
+// Re-export core state modules and stores that remain owned by core.
+export * from '@datalayer/core/lib/state/storage';
+export * from '@datalayer/core/lib/state/State';
+export { coreStore, useCoreStore, iamStore, useIAMStore, RESERVATION_WARNING_TIME_MS, layoutStore, useLayoutStore, organizationStore, useOrganizationStore, surveysStore, useSurveysStore, teamStore, useTeamStore, datasourceStore, useDatasourceStore, runStore, useRunStore, DATALAYER_IAM_TOKEN_KEY, DATALAYER_IAM_USER_KEY, } from '@datalayer/core/lib/state';

package/lib/state/substates/CellState.d.ts

@@ -0,0 +1,12 @@
+import { ICell } from '../../models';
+export type ICellState = {
+    cell?: ICell;
+};
+export type CellState = ICellState & {
+    create: (cell: ICell) => void;
+    update: (cell: Partial<ICell>) => void;
+};
+export declare const cellStore: import("zustand").StoreApi<CellState>;
+export declare function useCellStore(): CellState;
+export declare function useCellStore<T>(selector: (state: CellState) => T): T;
+export default useCellStore;

package/lib/state/substates/CellState.js

@@ -0,0 +1,22 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+import { createStore } from 'zustand/vanilla';
+import { useStore } from 'zustand';
+export const cellStore = createStore((set, get) => ({
+    cell: undefined,
+    create: (cell) => set((state) => ({ cell })),
+    update: (cell) => set((state) => ({
+        cell: cell
+            ? {
+                ...state.cell,
+                ...cell,
+            }
+            : undefined,
+    })),
+}));
+export function useCellStore(selector) {
+    return useStore(cellStore, selector);
+}
+export default useCellStore;

package/lib/state/substates/DocumentState.d.ts

@@ -0,0 +1,10 @@
+export interface IDocumentState {
+    documentSaveRequest?: Date;
+}
+export type DocumentState = IDocumentState & {
+    save: (documentSaveRequest: Date) => void;
+};
+export declare const documentStore: import("zustand").StoreApi<DocumentState>;
+export declare function useDocumentStore(): DocumentState;
+export declare function useDocumentStore<T>(selector: (state: DocumentState) => T): T;
+export default useDocumentStore;

package/lib/state/substates/DocumentState.js

@@ -0,0 +1,14 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+import { createStore } from 'zustand/vanilla';
+import { useStore } from 'zustand';
+export const documentStore = createStore((set, get) => ({
+    documentSaveRequest: undefined,
+    save: (documentSaveRequest) => set((state) => ({ documentSaveRequest })),
+}));
+export function useDocumentStore(selector) {
+    return useStore(documentStore, selector);
+}
+export default useDocumentStore;

package/lib/state/substates/GradeState.d.ts

@@ -0,0 +1,10 @@
+export type IGradeState = {
+    gradeRequest?: Date;
+};
+export type GradeState = IGradeState & {
+    grade: (gradeRequest: Date) => void;
+};
+export declare const gradeStore: import("zustand").StoreApi<GradeState>;
+export declare function useGradeStore(): GradeState;
+export declare function useGradeStore<T>(selector: (state: GradeState) => T): T;
+export default useGradeStore;

package/lib/state/substates/GradeState.js

@@ -0,0 +1,14 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+import { createStore } from 'zustand/vanilla';
+import { useStore } from 'zustand';
+export const gradeStore = createStore((set, get) => ({
+    gradeRequest: undefined,
+    grade: (gradeRequest) => set((state) => ({ gradeRequest })),
+}));
+export function useGradeStore(selector) {
+    return useStore(gradeStore, selector);
+}
+export default useGradeStore;

package/lib/state/substates/JupyterLabState.d.ts

@@ -0,0 +1,16 @@
+import { JupyterLabAppAdapter } from '@datalayer/jupyter-react';
+import { JupyterFrontEndPlugin } from '@jupyterlab/application';
+export type Plugin = JupyterFrontEndPlugin<any, any, any> & {
+    service: any;
+};
+export type IJupyterLabState = {
+    jupyterLabAdapter?: JupyterLabAppAdapter;
+};
+export type JupyterLabState = IJupyterLabState & {
+    setJupyterLabAdapter: (jupyterLabAdapter?: JupyterLabAppAdapter) => void;
+    plugin: (id: string) => Plugin | undefined;
+};
+export declare const jupyterLabStore: import("zustand").StoreApi<JupyterLabState>;
+export declare function useJupyterLabStore(): JupyterLabState;
+export declare function useJupyterLabStore<T>(selector: (state: JupyterLabState) => T): T;
+export default useJupyterLabStore;

package/lib/state/substates/JupyterLabState.js

@@ -0,0 +1,17 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+import { createStore } from 'zustand/vanilla';
+import { useStore } from 'zustand';
+export const jupyterLabStore = createStore((set, get) => ({
+    jupyterLabAdapter: undefined,
+    setJupyterLabAdapter: (jupyterLabAdapter) => set((state) => ({ jupyterLabAdapter })),
+    plugin: (id) => {
+        return get().jupyterLabAdapter?.plugin(id);
+    },
+}));
+export function useJupyterLabStore(selector) {
+    return useStore(jupyterLabStore, selector);
+}
+export default useJupyterLabStore;

package/lib/state/substates/NbformatState.d.ts

@@ -0,0 +1,13 @@
+export type SaveRequest = {
+    counter: number;
+};
+export type INbformatState = {
+    saveRequest: SaveRequest;
+};
+export type NbformatState = INbformatState & {
+    save: (saveRequest: SaveRequest) => void;
+};
+export declare const nbformatStore: import("zustand").StoreApi<NbformatState>;
+export declare function useNbformatStore(): NbformatState;
+export declare function useNbformatStore<T>(selector: (state: NbformatState) => T): T;
+export default useNbformatStore;

package/lib/state/substates/NbformatState.js

@@ -0,0 +1,16 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+import { createStore } from 'zustand/vanilla';
+import { useStore } from 'zustand';
+export const nbformatStore = createStore((set, get) => ({
+    saveRequest: {
+        counter: -1,
+    },
+    save: (saveRequest) => set((state) => ({ saveRequest })),
+}));
+export function useNbformatStore(selector) {
+    return useStore(nbformatStore, selector);
+}
+export default useNbformatStore;

package/lib/state/substates/RuntimesState.d.ts

@@ -0,0 +1,87 @@
+import type { IMultiServiceManager } from '../../runtimes';
+import type { IRuntimesConfiguration } from '@datalayer/core/lib/config';
+import type { IRuntimePod, ICodeSandboxSnapshot, IRuntimeModel } from '../../models';
+/**
+ * Datalayer Runtimes state.
+ */
+export type RuntimesState = {
+    /**
+     * Runtimes configuration
+     */
+    configuration: IRuntimesConfiguration;
+    setConfiguration: (config: IRuntimesConfiguration) => void;
+    /**
+     * Runtimes RUN URL.
+     */
+    runtimesUrl: string;
+    tab: number;
+    getIntTab: () => number;
+    setTab: (tab: number) => void;
+    /**
+     * Runtime pods.
+     */
+    runtimePods: IRuntimePod[];
+    /**
+     * Refresh the runtime pods.
+     */
+    refreshRuntimePods: () => Promise<void>;
+    /**
+     * Remove a runtime pod from the local state by its pod name.
+     *
+     * Used to prune a pod immediately after it is deleted, without waiting for
+     * the next `refreshRuntimePods()` poll tick. This lets the remote service
+     * managers dispose right away and stop polling the (now dead) pod ingress.
+     */
+    removeRuntimePod: (podName: string) => void;
+    /**
+     * Cached runtime models.
+     */
+    runtimeModels: readonly IRuntimeModel[];
+    /**
+     * Add a runtime model.
+     */
+    addRuntimeModel: (model: IRuntimeModel) => void;
+    /**
+     * Remove a runtime model by ID.
+     */
+    removeRuntimeModel: (id: string) => void;
+    /**
+     * Set the runtimes models list.
+     */
+    setRuntimeModels: (models: readonly IRuntimeModel[]) => void;
+    /**
+     * Jupyter service manager.
+     */
+    multiServiceManager?: IMultiServiceManager;
+    setMultiServiceManager: (multiServiceManager: IMultiServiceManager) => void;
+    showDisclaimer: boolean;
+    setShowDisclaimer: (showDisclaimer: boolean) => void;
+    /**
+     * Code Sandbox snapshots.
+     */
+    runtimeSnapshots: readonly ICodeSandboxSnapshot[];
+    /**
+     * Add a code sandbox snapshot.
+     */
+    addSandboxSnapshot: (snapshot: ICodeSandboxSnapshot) => void;
+    /**
+     * Remove a Code Sandbox Snapshot.
+     */
+    removeSandboxSnapshot: (id: string) => void;
+    /**
+     * Set Code Sandbox Snapshots.
+     */
+    setSandboxSnapshots: (snapshots: ICodeSandboxSnapshot[]) => void;
+    /**
+     * Package version.
+     */
+    version: string;
+    setVersion: (version: string) => void;
+};
+/**
+ * Kernel store
+ */
+export declare const runtimesStore: import("zustand").StoreApi<RuntimesState>;
+export declare function useRuntimesStore(): RuntimesState;
+export declare function useRuntimesStore<T>(selector: (state: RuntimesState) => T): T;
+export default useRuntimesStore;