@datalayer/agent-runtimes 1.0.1 → 1.0.3

package/lib/renderers/a2ui/hooks/useA2UI.d.ts

@@ -1,36 +0,0 @@
-import { Types } from '@a2ui/lit/0.8';
-import type { A2UIActionEvent } from '../types/index.js';
-export interface UseA2UIReturn {
-    /** Map of all surfaces */
-    surfaces: ReadonlyMap<string, Types.Surface>;
-    /** Get a specific surface by ID */
-    getSurface: (surfaceId: string) => Types.Surface | undefined;
-    /** Process incoming A2UI messages from the server */
-    processMessages: (messages: Types.ServerToClientMessage[]) => void;
-    /** Send an action event to the server */
-    sendAction: (action: A2UIActionEvent) => void;
-    /** Get the root component tree for a surface */
-    getComponentTree: (surfaceId: string) => Types.AnyComponentNode | null;
-    /** Clear all surfaces */
-    clearSurfaces: () => void;
-}
-/**
- * Main hook for interacting with A2UI.
- * Provides access to surfaces, message processing, and action sending.
- *
- * @example
- * ```tsx
- * function MyComponent() {
- *   const { surfaces, processMessages, sendAction } = useA2UI();
- *
- *   // Process messages from server
- *   useEffect(() => {
- *     fetch('/api/a2ui').then(res => res.json()).then(processMessages);
- *   }, []);
- *
- *   // Render surfaces
- *   return <A2UIRenderer surfaceId="main" />;
- * }
- * ```
- */
-export declare function useA2UI(): UseA2UIReturn;

package/lib/renderers/a2ui/hooks/useA2UI.js

@@ -1,62 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/*
- Copyright 2025 Google LLC
-
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
-
-      https://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
-import { useCallback } from 'react';
-import { useA2UIContext } from '../context/A2UIContext.js';
-/**
- * Main hook for interacting with A2UI.
- * Provides access to surfaces, message processing, and action sending.
- *
- * @example
- * ```tsx
- * function MyComponent() {
- *   const { surfaces, processMessages, sendAction } = useA2UI();
- *
- *   // Process messages from server
- *   useEffect(() => {
- *     fetch('/api/a2ui').then(res => res.json()).then(processMessages);
- *   }, []);
- *
- *   // Render surfaces
- *   return <A2UIRenderer surfaceId="main" />;
- * }
- * ```
- */
-export function useA2UI() {
-    const { processor, surfaces, processMessages, sendAction, refresh } = useA2UIContext();
-    const getSurface = useCallback((surfaceId) => {
-        return surfaces.get(surfaceId);
-    }, [surfaces]);
-    const getComponentTree = useCallback((surfaceId) => {
-        const surface = surfaces.get(surfaceId);
-        return surface?.componentTree ?? null;
-    }, [surfaces]);
-    const clearSurfaces = useCallback(() => {
-        processor.clearSurfaces();
-        refresh();
-    }, [processor, refresh]);
-    return {
-        surfaces,
-        getSurface,
-        processMessages,
-        sendAction,
-        getComponentTree,
-        clearSurfaces,
-    };
-}

package/lib/renderers/a2ui/hooks/useDataBinding.d.ts

@@ -1,8 +0,0 @@
-import { Types, Primitives } from '@a2ui/lit/0.8';
-export declare function useDataBinding(node: Types.AnyComponentNode, surfaceId: string): {
-    resolveString: (value: Primitives.StringValue | null | undefined) => string | null;
-    resolveNumber: (value: Primitives.NumberValue | null | undefined) => number | null;
-    resolveBoolean: (value: Primitives.BooleanValue | null | undefined) => boolean | null;
-    setValue: (path: string, value: Types.DataValue) => void;
-    getValue: (path: string) => Types.DataValue | null;
-};

package/lib/renderers/a2ui/hooks/useDataBinding.js

@@ -1,83 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/*
- Copyright 2025 Google LLC
-
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
-
-      https://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
-import { useCallback } from 'react';
-import { useA2UIContext } from '../context/A2UIContext.js';
-export function useDataBinding(node, surfaceId) {
-    const { processor, refresh } = useA2UIContext();
-    const resolveString = useCallback((value) => {
-        if (!value)
-            return null;
-        if (value.literalString !== undefined) {
-            if (value.path) {
-                processor.setData(node, value.path, value.literalString, surfaceId);
-            }
-            return value.literalString;
-        }
-        if (value.path) {
-            const data = processor.getData(node, value.path, surfaceId);
-            return data !== null ? String(data) : null;
-        }
-        return null;
-    }, [processor, node, surfaceId]);
-    const resolveNumber = useCallback((value) => {
-        if (!value)
-            return null;
-        if (value.literalNumber !== undefined) {
-            if (value.path) {
-                processor.setData(node, value.path, value.literalNumber, surfaceId);
-            }
-            return value.literalNumber;
-        }
-        if (value.path) {
-            const data = processor.getData(node, value.path, surfaceId);
-            return data !== null ? Number(data) : null;
-        }
-        return null;
-    }, [processor, node, surfaceId]);
-    const resolveBoolean = useCallback((value) => {
-        if (!value)
-            return null;
-        if (value.literalBoolean !== undefined) {
-            if (value.path) {
-                processor.setData(node, value.path, value.literalBoolean, surfaceId);
-            }
-            return value.literalBoolean;
-        }
-        if (value.path) {
-            const data = processor.getData(node, value.path, surfaceId);
-            return data !== null ? Boolean(data) : null;
-        }
-        return null;
-    }, [processor, node, surfaceId]);
-    const setValue = useCallback((path, value) => {
-        processor.setData(node, path, value, surfaceId);
-        refresh();
-    }, [processor, node, surfaceId, refresh]);
-    const getValue = useCallback((path) => {
-        return processor.getData(node, path, surfaceId);
-    }, [processor, node, surfaceId]);
-    return {
-        resolveString,
-        resolveNumber,
-        resolveBoolean,
-        setValue,
-        getValue,
-    };
-}

package/lib/renderers/a2ui/index.d.ts

@@ -1,9 +0,0 @@
-export { A2UIProvider, useA2UIContext } from './context/A2UIContext.js';
-export { ThemeProvider, useTheme, defaultTheme, } from './context/ThemeContext.js';
-export { useA2UI } from './hooks/useA2UI.js';
-export { useDataBinding } from './hooks/useDataBinding.js';
-export { A2UIRenderer } from './components/A2UIRenderer.js';
-export * from './components/index.js';
-export * from './types/index.js';
-export { cn, classMapToString } from './lib/utils.js';
-export { Data, Types, Primitives } from '@a2ui/lit/0.8';

package/lib/renderers/a2ui/index.js

@@ -1,28 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/*
- Copyright 2025 Google LLC
-
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
-
-      https://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
-export { A2UIProvider, useA2UIContext } from './context/A2UIContext.js';
-export { ThemeProvider, useTheme, defaultTheme, } from './context/ThemeContext.js';
-export { useA2UI } from './hooks/useA2UI.js';
-export { useDataBinding } from './hooks/useDataBinding.js';
-export { A2UIRenderer } from './components/A2UIRenderer.js';
-export * from './components/index.js';
-export * from './types/index.js';
-export { cn, classMapToString } from './lib/utils.js';
-export { Data, Types, Primitives } from '@a2ui/lit/0.8';

package/lib/renderers/a2ui/lib/utils.d.ts

@@ -1,11 +0,0 @@
-import { type ClassValue } from "clsx";
-/**
- * Utility function to merge Tailwind CSS classes.
- * Combines clsx for conditional classes with tailwind-merge to handle conflicts.
- */
-export declare function cn(...inputs: ClassValue[]): string;
-/**
- * Converts a Record<string, boolean> (class map) to a className string.
- * This is used to convert Lit-style class maps to React className strings.
- */
-export declare function classMapToString(classMap: Record<string, boolean>): string;

package/lib/renderers/a2ui/lib/utils.js

@@ -1,38 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/*
- Copyright 2025 Google LLC
-
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
-
-      https://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
-import clsx from "clsx";
-import { twMerge } from "tailwind-merge";
-/**
- * Utility function to merge Tailwind CSS classes.
- * Combines clsx for conditional classes with tailwind-merge to handle conflicts.
- */
-export function cn(...inputs) {
-    return twMerge(clsx(inputs));
-}
-/**
- * Converts a Record<string, boolean> (class map) to a className string.
- * This is used to convert Lit-style class maps to React className strings.
- */
-export function classMapToString(classMap) {
-    return Object.entries(classMap)
-        .filter(([, enabled]) => enabled)
-        .map(([className]) => className)
-        .join(" ");
-}

package/lib/renderers/a2ui/types/index.d.ts

@@ -1,17 +0,0 @@
-import { Types } from '@a2ui/lit/0.8';
-export type MessageProcessor = Types.MessageProcessor;
-export type Theme = Types.Theme;
-export type Surface = Types.Surface;
-export type AnyComponentNode = Types.AnyComponentNode;
-export type ServerToClientMessage = Types.ServerToClientMessage;
-export type DataValue = Types.DataValue;
-export interface A2UIComponentProps {
-    node: Types.AnyComponentNode;
-    surfaceId: string;
-}
-export interface A2UIActionEvent {
-    type: 'a2ui.action';
-    actionId: string;
-    context: Record<string, unknown>;
-}
-export type SendActionCallback = (action: A2UIActionEvent) => void;

package/lib/runtime/index.d.ts

@@ -1,38 +0,0 @@
-/**
- * Runtime management module for agent-runtimes.
- *
- * Provides a Zustand store and hooks for launching and managing cloud runtimes
- * with integrated AI agent support.
- *
- * @module runtime
- *
- * @example
- * ```typescript
- * import { useRuntimeStore, useRuntime, useAgent } from '@datalayer/agent-runtimes/lib/runtime';
- *
- * // Launch a new runtime
- * const { launchRuntime, createAgent } = useRuntimeStore();
- * const runtime = await launchRuntime({ environmentName: 'python-simple', creditsLimit: 100 });
- * const agent = await createAgent({ model: 'anthropic:claude-sonnet-4-5' });
- *
- * // Use in ChatFloating
- * <ChatFloating endpoint={agent.endpoint} />
- *
- * // Or connect to an existing runtime
- * const { connectToRuntime } = useRuntimeStore();
- * connectToRuntime({
- *   podName: 'my-pod',
- *   environmentName: 'python-simple',
- *   serviceManager: myServiceManager,
- * });
- * ```
- */
-export { useRuntimeStore, useRuntime, useAgent, useRuntimeStatus, useRuntimeError, useIsLaunching, getRuntimeState, subscribeToRuntime, } from './runtimeStore';
-export type { RuntimeStore, RuntimeStoreState, RuntimeStoreActions, } from './runtimeStore';
-export { useAgentConnection } from './useAgentConnection';
-export type { UseAgentConnectionReturn, UseAgentConnectionOptions, } from './useAgentConnection';
-export { useAgentRuntime } from './useAgentRuntime';
-export type { UseAgentRuntimeReturn, UseAgentRuntimeOptions, } from './useAgentRuntime';
-export type { IRuntimeLocation, IRuntimeType, IRuntimeCapabilities, IRuntimePod, IRuntimeOptions, IRuntimeDesc, } from './types';
-export type { RuntimeConnection, AgentRuntimeStatus as RuntimeStatus, AgentConfig, AgentConnection, AgentRuntimeState, } from './types';
-export { DEFAULT_AGENT_CONFIG } from './types';

package/lib/runtime/index.js

@@ -1,40 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/**
- * Runtime management module for agent-runtimes.
- *
- * Provides a Zustand store and hooks for launching and managing cloud runtimes
- * with integrated AI agent support.
- *
- * @module runtime
- *
- * @example
- * ```typescript
- * import { useRuntimeStore, useRuntime, useAgent } from '@datalayer/agent-runtimes/lib/runtime';
- *
- * // Launch a new runtime
- * const { launchRuntime, createAgent } = useRuntimeStore();
- * const runtime = await launchRuntime({ environmentName: 'python-simple', creditsLimit: 100 });
- * const agent = await createAgent({ model: 'anthropic:claude-sonnet-4-5' });
- *
- * // Use in ChatFloating
- * <ChatFloating endpoint={agent.endpoint} />
- *
- * // Or connect to an existing runtime
- * const { connectToRuntime } = useRuntimeStore();
- * connectToRuntime({
- *   podName: 'my-pod',
- *   environmentName: 'python-simple',
- *   serviceManager: myServiceManager,
- * });
- * ```
- */
-// Zustand store
-export { useRuntimeStore, useRuntime, useAgent, useRuntimeStatus, useRuntimeError, useIsLaunching, getRuntimeState, subscribeToRuntime, } from './runtimeStore';
-// Hooks
-export { useAgentConnection } from './useAgentConnection';
-export { useAgentRuntime } from './useAgentRuntime';
-// Constants
-export { DEFAULT_AGENT_CONFIG } from './types';

package/lib/runtime/runtimeStore.d.ts

@@ -1,76 +0,0 @@
-import type { ServiceManager } from '@jupyterlab/services';
-import type { IRuntimeOptions, RuntimeConnection, AgentRuntimeStatus, AgentConfig, AgentConnection } from './types';
-/**
- * Runtime store state interface.
- */
-export interface RuntimeStoreState {
-    /** Current runtime connection */
-    runtime: RuntimeConnection | null;
-    /** Current agent connection */
-    agent: AgentConnection | null;
-    /** Current status */
-    status: AgentRuntimeStatus;
-    /** Error message if any */
-    error: string | null;
-    /** Whether a launch is in progress */
-    isLaunching: boolean;
-}
-/**
- * Runtime store actions interface.
- */
-export interface RuntimeStoreActions {
-    /** Launch a new runtime */
-    launchRuntime: (options: IRuntimeOptions) => Promise<RuntimeConnection>;
-    /** Connect to an existing runtime */
-    connectToRuntime: (connection: {
-        podName: string;
-        environmentName: string;
-        serviceManager: ServiceManager.IManager;
-        kernelId?: string;
-    }) => void;
-    /** Create an agent on the current runtime */
-    createAgent: (config?: AgentConfig) => Promise<AgentConnection>;
-    /** Disconnect from the current runtime */
-    disconnect: () => void;
-    /** Clear any errors */
-    clearError: () => void;
-    /** Set error */
-    setError: (error: string) => void;
-    /** Reset store to initial state */
-    reset: () => void;
-}
-export type RuntimeStore = RuntimeStoreState & RuntimeStoreActions;
-/**
- * Zustand store for runtime management.
- *
- * @example
- * ```typescript
- * import { useRuntimeStore } from '@datalayer/agent-runtimes/lib/runtime';
- *
- * // In a component
- * const { runtime, launchRuntime, status } = useRuntimeStore();
- *
- * // Launch a runtime
- * await launchRuntime({ environmentName: 'python-simple', creditsLimit: 100 });
- *
- * // Access runtime info
- * console.log(runtime?.agentBaseUrl);
- * ```
- */
-export declare const useRuntimeStore: import("zustand").UseBoundStore<import("zustand").StoreApi<RuntimeStore>>;
-/**
- * Selector hooks for common use cases.
- */
-export declare const useRuntime: () => RuntimeConnection | null;
-export declare const useAgent: () => AgentConnection | null;
-export declare const useRuntimeStatus: () => AgentRuntimeStatus;
-export declare const useRuntimeError: () => string | null;
-export declare const useIsLaunching: () => boolean;
-/**
- * Get runtime store state without React (for use outside components).
- */
-export declare const getRuntimeState: () => RuntimeStore;
-/**
- * Subscribe to runtime store changes (for use outside React).
- */
-export declare const subscribeToRuntime: (listener: (state: RuntimeStore, prevState: RuntimeStore) => void) => () => void;

package/lib/runtime/runtimeStore.js

@@ -1,184 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/**
- * Zustand store for runtime management.
- *
- * Provides a global store for managing cloud runtimes and agent connections.
- *
- * @module runtime/runtimeStore
- */
-import { create } from 'zustand';
-const initialState = {
-    runtime: null,
-    agent: null,
-    status: 'idle',
-    error: null,
-    isLaunching: false,
-};
-/**
- * Create an agent on a runtime.
- */
-async function createAgentOnRuntime(agentBaseUrl, agentId, config = {}) {
-    const response = await fetch(`${agentBaseUrl}/api/v1/agents`, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({
-            name: config.name || agentId,
-            description: config.description || 'AI assistant',
-            agent_library: config.agentLibrary || 'pydantic-ai',
-            transport: config.transport || 'ag-ui',
-            model: config.model || 'anthropic:claude-sonnet-4-5',
-            system_prompt: config.systemPrompt || 'You are a helpful AI assistant.',
-        }),
-    });
-    if (response.ok || response.status === 400) {
-        // 400 means agent already exists, which is fine
-        const endpoint = `${agentBaseUrl}/api/v1/ag-ui/${agentId}/`;
-        return {
-            agentId,
-            endpoint,
-            isReady: true,
-        };
-    }
-    const errorData = await response.json().catch(() => ({}));
-    throw new Error(errorData.detail || `Failed to create agent: ${response.status}`);
-}
-/**
- * Zustand store for runtime management.
- *
- * @example
- * ```typescript
- * import { useRuntimeStore } from '@datalayer/agent-runtimes/lib/runtime';
- *
- * // In a component
- * const { runtime, launchRuntime, status } = useRuntimeStore();
- *
- * // Launch a runtime
- * await launchRuntime({ environmentName: 'python-simple', creditsLimit: 100 });
- *
- * // Access runtime info
- * console.log(runtime?.agentBaseUrl);
- * ```
- */
-export const useRuntimeStore = create((set, get) => ({
-    ...initialState,
-    connectToRuntime: connection => {
-        const baseUrl = connection.serviceManager.serverSettings.baseUrl;
-        const agentBaseUrl = baseUrl.replace('/jupyter/server/', '/agent-runtimes/');
-        const fullConnection = {
-            podName: connection.podName,
-            environmentName: connection.environmentName,
-            jupyterBaseUrl: baseUrl,
-            agentBaseUrl,
-            serviceManager: connection.serviceManager,
-            status: 'ready',
-            kernelId: connection.kernelId,
-        };
-        set({
-            runtime: fullConnection,
-            status: 'ready',
-            error: null,
-        });
-    },
-    launchRuntime: async (config) => {
-        set({ status: 'launching', error: null, isLaunching: true });
-        try {
-            // Import @datalayer/core dynamically to avoid circular dependencies
-            const { createRuntime, makeDatalayerSettings } = await import('@datalayer/core/lib/api');
-            const { ServiceManager } = await import('@jupyterlab/services');
-            // Create the runtime using IRuntimeOptions from @datalayer/core
-            const runtimePod = await createRuntime({
-                environmentName: config.environmentName,
-                creditsLimit: config.creditsLimit,
-                type: config.type || 'notebook',
-                givenName: config.givenName,
-                capabilities: config.capabilities,
-                snapshot: config.snapshot,
-            });
-            set({ status: 'connecting' });
-            // Create service manager for the runtime
-            const serverSettings = makeDatalayerSettings(runtimePod.ingress, runtimePod.token);
-            const serviceManager = new ServiceManager({ serverSettings });
-            // Wait for the service manager to be ready
-            await serviceManager.ready;
-            // Construct URLs
-            const jupyterBaseUrl = serverSettings.baseUrl;
-            const agentBaseUrl = jupyterBaseUrl.replace('/jupyter/server/', '/agent-runtimes/');
-            const connection = {
-                podName: runtimePod.pod_name,
-                environmentName: runtimePod.environment_name,
-                jupyterBaseUrl,
-                agentBaseUrl,
-                serviceManager,
-                status: 'ready',
-            };
-            set({
-                runtime: connection,
-                status: 'ready',
-                isLaunching: false,
-            });
-            return connection;
-        }
-        catch (err) {
-            const errorMessage = err instanceof Error ? err.message : 'Failed to launch runtime';
-            set({
-                status: 'error',
-                error: errorMessage,
-                isLaunching: false,
-            });
-            throw err;
-        }
-    },
-    createAgent: async (config = {}) => {
-        const { runtime } = get();
-        if (!runtime) {
-            throw new Error('No runtime connected. Launch or connect to a runtime first.');
-        }
-        try {
-            const agentId = config.name || runtime.podName;
-            const agentConnection = await createAgentOnRuntime(runtime.agentBaseUrl, agentId, config);
-            set({ agent: agentConnection });
-            return agentConnection;
-        }
-        catch (err) {
-            const errorMessage = err instanceof Error ? err.message : 'Failed to create agent';
-            set({ error: errorMessage });
-            throw err;
-        }
-    },
-    disconnect: () => {
-        set({
-            runtime: null,
-            agent: null,
-            status: 'disconnected',
-            error: null,
-        });
-    },
-    clearError: () => {
-        set({ error: null });
-    },
-    setError: error => {
-        set({ error, status: 'error' });
-    },
-    reset: () => {
-        set(initialState);
-    },
-}));
-/**
- * Selector hooks for common use cases.
- */
-export const useRuntime = () => useRuntimeStore(state => state.runtime);
-export const useAgent = () => useRuntimeStore(state => state.agent);
-export const useRuntimeStatus = () => useRuntimeStore(state => state.status);
-export const useRuntimeError = () => useRuntimeStore(state => state.error);
-export const useIsLaunching = () => useRuntimeStore(state => state.isLaunching);
-/**
- * Get runtime store state without React (for use outside components).
- */
-export const getRuntimeState = () => useRuntimeStore.getState();
-/**
- * Subscribe to runtime store changes (for use outside React).
- */
-export const subscribeToRuntime = useRuntimeStore.subscribe;