@datalayer/agent-runtimes 1.0.2 → 1.0.4

package/lib/runtime/useAgentRuntime.d.ts

@@ -1,93 +0,0 @@
-import type { ServiceManager } from '@jupyterlab/services';
-import type { IRuntimeOptions, AgentConfig, RuntimeConnection, AgentConnection, AgentRuntimeStatus } from './types';
-export interface UseAgentRuntimeOptions {
-    /** Agent configuration */
-    agentConfig?: AgentConfig;
-    /** Auto-create agent when runtime connects */
-    autoCreateAgent?: boolean;
-}
-export interface UseAgentRuntimeReturn {
-    /** Current runtime connection */
-    runtime: RuntimeConnection | null;
-    /** Current agent connection */
-    agent: AgentConnection | null;
-    /** Runtime status */
-    status: AgentRuntimeStatus;
-    /** Error message if any */
-    error: string | null;
-    /** Whether the runtime is launching */
-    isLaunching: boolean;
-    /** Whether everything is ready (runtime + agent) */
-    isReady: boolean;
-    /** Agent endpoint URL (shortcut) */
-    endpoint: string | null;
-    /** ServiceManager for the runtime */
-    serviceManager: ServiceManager.IManager | null;
-    /** Launch a new runtime */
-    launchRuntime: (options: IRuntimeOptions) => Promise<RuntimeConnection>;
-    /** Connect to an existing runtime */
-    connectToRuntime: (options: {
-        podName: string;
-        environmentName: string;
-        serviceManager: ServiceManager.IManager;
-        kernelId?: string;
-    }) => void;
-    /** Create an agent (if not auto-created) */
-    createAgent: (config?: AgentConfig) => Promise<AgentConnection>;
-    /** Disconnect from the runtime */
-    disconnect: () => void;
-}
-/**
- * Combined hook for using a runtime with an AI agent.
- *
- * This hook provides everything needed to:
- * 1. Connect to an existing runtime (or launch a new one)
- * 2. Create an AI agent on the runtime
- *
- * Use this in conjunction with useNotebookTools or useLexicalTools
- * for frontend tool execution.
- *
- * @param options - Configuration options
- * @returns Complete agent runtime state and controls
- *
- * @example
- * ```tsx
- * // For notebooks
- * import { useAgentRuntime } from '@datalayer/agent-runtimes/lib/runtime';
- * import { useNotebookTools } from '@datalayer/agent-runtimes/lib/tools/adapters/agent-runtimes';
- *
- * function NotebookEditor({ notebookId }) {
- *   const {
- *     isReady,
- *     endpoint,
- *     error,
- *     connectToRuntime,
- *   } = useAgentRuntime({
- *     autoCreateAgent: true,
- *     agentConfig: {
- *       model: 'anthropic:claude-sonnet-4-5',
- *       systemPrompt: 'You help users with Jupyter notebooks.',
- *     },
- *   });
- *
- *   // Get tools separately
- *   const tools = useNotebookTools(notebookId);
- *
- *   // Connect when user assigns a runtime
- *   const onRuntimeAssigned = (serviceManager, podName, poolName) => {
- *     connectToRuntime({ serviceManager, podName, jupyterpoolName: poolName });
- *   };
- *
- *   return (
- *     <>
- *       <Notebook />
- *       {isReady && (
- *         <ChatFloating endpoint={endpoint} frontendTools={frontendTools} />
- *       )}
- *       {error && <ErrorBanner>{error}</ErrorBanner>}
- *     </>
- *   );
- * }
- * ```
- */
-export declare function useAgentRuntime(options?: UseAgentRuntimeOptions): UseAgentRuntimeReturn;

package/lib/runtime/useAgentRuntime.js

@@ -1,125 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-/**
- * Combined hook for using a runtime with an AI agent.
- *
- * This is the main entry point for consumers who want a simple,
- * all-in-one solution for cloud runtime + agent management.
- *
- * @module runtime/useAgentRuntime
- */
-import { useEffect, useRef, useCallback } from 'react';
-import { useRuntimeStore, useRuntime, useAgent, useRuntimeStatus, useRuntimeError, useIsLaunching, } from './runtimeStore';
-/**
- * Combined hook for using a runtime with an AI agent.
- *
- * This hook provides everything needed to:
- * 1. Connect to an existing runtime (or launch a new one)
- * 2. Create an AI agent on the runtime
- *
- * Use this in conjunction with useNotebookTools or useLexicalTools
- * for frontend tool execution.
- *
- * @param options - Configuration options
- * @returns Complete agent runtime state and controls
- *
- * @example
- * ```tsx
- * // For notebooks
- * import { useAgentRuntime } from '@datalayer/agent-runtimes/lib/runtime';
- * import { useNotebookTools } from '@datalayer/agent-runtimes/lib/tools/adapters/agent-runtimes';
- *
- * function NotebookEditor({ notebookId }) {
- *   const {
- *     isReady,
- *     endpoint,
- *     error,
- *     connectToRuntime,
- *   } = useAgentRuntime({
- *     autoCreateAgent: true,
- *     agentConfig: {
- *       model: 'anthropic:claude-sonnet-4-5',
- *       systemPrompt: 'You help users with Jupyter notebooks.',
- *     },
- *   });
- *
- *   // Get tools separately
- *   const tools = useNotebookTools(notebookId);
- *
- *   // Connect when user assigns a runtime
- *   const onRuntimeAssigned = (serviceManager, podName, poolName) => {
- *     connectToRuntime({ serviceManager, podName, jupyterpoolName: poolName });
- *   };
- *
- *   return (
- *     <>
- *       <Notebook />
- *       {isReady && (
- *         <ChatFloating endpoint={endpoint} frontendTools={frontendTools} />
- *       )}
- *       {error && <ErrorBanner>{error}</ErrorBanner>}
- *     </>
- *   );
- * }
- * ```
- */
-export function useAgentRuntime(options = {}) {
-    const { agentConfig, autoCreateAgent = true } = options;
-    // Get store state
-    const runtime = useRuntime();
-    const agent = useAgent();
-    const status = useRuntimeStatus();
-    const error = useRuntimeError();
-    const isLaunching = useIsLaunching();
-    // Get store actions
-    const launchRuntime = useRuntimeStore(state => state.launchRuntime);
-    const connectToRuntime = useRuntimeStore(state => state.connectToRuntime);
-    const disconnect = useRuntimeStore(state => state.disconnect);
-    const createAgentAction = useRuntimeStore(state => state.createAgent);
-    // Track if we've created the agent to prevent duplicates
-    const hasCreatedAgentRef = useRef(false);
-    const agentConfigRef = useRef(agentConfig);
-    agentConfigRef.current = agentConfig;
-    // Auto-create agent when runtime is ready
-    useEffect(() => {
-        if (autoCreateAgent &&
-            runtime &&
-            status === 'ready' &&
-            !agent &&
-            !hasCreatedAgentRef.current) {
-            hasCreatedAgentRef.current = true;
-            createAgentAction(agentConfigRef.current).catch(err => {
-                console.error('[useAgentRuntime] Failed to auto-create agent:', err);
-                hasCreatedAgentRef.current = false;
-            });
-        }
-    }, [autoCreateAgent, runtime, status, agent, createAgentAction]);
-    // Reset agent creation tracking on disconnect
-    useEffect(() => {
-        if (status === 'disconnected' || status === 'idle') {
-            hasCreatedAgentRef.current = false;
-        }
-    }, [status]);
-    // Memoized create agent function
-    const createAgent = useCallback((config) => createAgentAction(config || agentConfig), [createAgentAction, agentConfig]);
-    // Derived state
-    const isReady = status === 'ready' && !!agent?.isReady;
-    const endpoint = agent?.endpoint || null;
-    const serviceManager = runtime?.serviceManager || null;
-    return {
-        runtime,
-        agent,
-        status,
-        error,
-        isLaunching,
-        isReady,
-        endpoint,
-        serviceManager,
-        launchRuntime,
-        connectToRuntime,
-        createAgent,
-        disconnect,
-    };
-}

package/lib/specs/agents/codeai/agents.d.ts

@@ -1,28 +0,0 @@
-/**
- * Agent Library.
- *
- * Predefined agent specifications that can be instantiated as Agent Runtimes.
- * THIS FILE IS AUTO-GENERATED. DO NOT EDIT MANUALLY.
- * Generated from YAML specifications in specs/agents/
- */
-import type { AgentSpec } from '../../../types/Types';
-export declare const DATA_ACQUISITION_AGENT_SPEC: AgentSpec;
-export declare const SIMPLE_AGENT_SPEC: AgentSpec;
-export declare const AGENT_SPECS: Record<string, AgentSpec>;
-/**
- * Get an agent specification by ID.
- */
-export declare function getAgentSpecs(agentId: string): AgentSpec | undefined;
-/**
- * List all available agent specifications.
- *
- * @param prefix - If provided, only return specs whose ID starts with this prefix.
- */
-export declare function listAgentSpecs(prefix?: string): AgentSpec[];
-/**
- * Collect all required environment variables for an agent spec.
- *
- * Iterates over the spec's MCP servers and skills and returns the
- * deduplicated union of their `requiredEnvVars` arrays.
- */
-export declare function getAgentSpecRequiredEnvVars(spec: AgentSpec): string[];

package/lib/specs/agents/codeai/agents.js

@@ -1,179 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-import { FILESYSTEM_MCP_SERVER, KAGGLE_MCP_SERVER, TAVILY_MCP_SERVER, } from '../../mcpServers';
-import { GITHUB_SKILL_SPEC } from '../../skills';
-// ============================================================================
-// MCP Server Lookup
-// ============================================================================
-const MCP_SERVER_MAP = {
-    filesystem: FILESYSTEM_MCP_SERVER,
-    kaggle: KAGGLE_MCP_SERVER,
-    tavily: TAVILY_MCP_SERVER,
-};
-/**
- * Map skill IDs to SkillSpec objects, converting to AgentSkillSpec shape.
- */
-const SKILL_MAP = {
-    github: GITHUB_SKILL_SPEC,
-};
-function toAgentSkillSpec(skill) {
-    return {
-        id: skill.id,
-        name: skill.name,
-        description: skill.description,
-        version: '1.0.0',
-        tags: skill.tags,
-        enabled: skill.enabled,
-        requiredEnvVars: skill.requiredEnvVars,
-    };
-}
-// ============================================================================
-// Agent Specs
-// ============================================================================
-// Codeai Agents
-// ============================================================================
-export const DATA_ACQUISITION_AGENT_SPEC = {
-    id: 'codeai/data-acquisition',
-    name: 'Data Acquisition Agent',
-    description: `Acquires and manages data from various sources including Kaggle datasets and local filesystem operations.`,
-    tags: ['data', 'acquisition', 'kaggle', 'filesystem'],
-    enabled: true,
-    model: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
-    mcpServers: [
-        MCP_SERVER_MAP['kaggle'],
-        MCP_SERVER_MAP['filesystem'],
-        MCP_SERVER_MAP['tavily'],
-    ],
-    skills: [toAgentSkillSpec(SKILL_MAP['github'])],
-    environmentName: 'ai-agents-env',
-    icon: 'database',
-    emoji: '📊',
-    color: '#3B82F6',
-    suggestions: [
-        'Find popular machine learning datasets on Kaggle',
-        'Download and explore a dataset for sentiment analysis',
-        'List available files in my workspace',
-        'Search Kaggle for time series forecasting competitions',
-    ],
-    sandboxVariant: 'jupyter',
-    systemPrompt: `You are a data acquisition specialist with access to Kaggle datasets and filesystem tools. You can search for datasets, download data, read and write files, and help users prepare data for analysis. Guide users through finding relevant datasets and organizing their workspace efficiently.
-`,
-    systemPromptCodemodeAddons: `## IMPORTANT: Be Honest About Your Capabilities NEVER claim to have tools or capabilities you haven't verified.
-## Core Codemode Tools Use these 4 tools to accomplish any task: 1. **list_servers** - List available MCP servers
-   Use this to see what MCP servers you can access.
-
-2. **search_tools** - Progressive tool discovery by natural language query
-   Use this to find relevant tools before executing tasks.
-
-3. **get_tool_details** - Get full tool schema and documentation
-   Use this to understand tool parameters before calling them.
-
-4. **execute_code** - Run Python code that composes multiple tools
-   Use this for complex multi-step operations. Code runs in a PERSISTENT sandbox.
-   Variables, functions, and state PERSIST between execute_code calls.
-   Import tools using: \`from generated.servers.<server_name> import <function_name>\`
-   NEVER use \`import *\` - always use explicit named imports.
-
-## Recommended Workflow 1. **Discover**: Use list_servers and search_tools to find relevant tools 2. **Understand**: Use get_tool_details to check parameters 3. **Execute**: Use execute_code to perform multi-step tasks, calling tools as needed
-## Token Efficiency When possible, chain multiple tool calls in a single execute_code block. This reduces output tokens by processing intermediate results in code rather than returning them. If you want to examine results, print subsets, preview (maximum 20 first characters) and/or counts instead of full data, this is really important.
-`,
-    goal: undefined,
-    protocol: undefined,
-    uiExtension: undefined,
-    trigger: undefined,
-    modelConfig: undefined,
-    mcpServerTools: undefined,
-    guardrails: undefined,
-    evals: undefined,
-    codemode: undefined,
-    output: undefined,
-    advanced: undefined,
-    authorizationPolicy: undefined,
-    notifications: undefined,
-    team: undefined,
-};
-export const SIMPLE_AGENT_SPEC = {
-    id: 'codeai/simple',
-    name: 'A Simple Agent',
-    description: `A simple conversational agent. No tools, no MCP servers, no skills — just a helpful AI assistant you can chat with.`,
-    tags: ['simple', 'chat', 'assistant'],
-    enabled: true,
-    model: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
-    mcpServers: [],
-    skills: [],
-    environmentName: 'ai-agents-env',
-    icon: 'share-2',
-    emoji: '🤖',
-    color: '#6366F1',
-    suggestions: [
-        'Tell me a joke',
-        'Explain quantum computing in simple terms',
-        'Help me brainstorm ideas for a weekend project',
-        'Summarize the key points of a topic I describe',
-    ],
-    sandboxVariant: 'jupyter',
-    systemPrompt: `You are a helpful, friendly AI assistant. You do not have access to any external tools, MCP servers, or skills. Answer questions using your training knowledge, be concise, and let the user know if a question is outside your knowledge.
-`,
-    systemPromptCodemodeAddons: undefined,
-    goal: undefined,
-    protocol: undefined,
-    uiExtension: undefined,
-    trigger: undefined,
-    modelConfig: undefined,
-    mcpServerTools: undefined,
-    guardrails: undefined,
-    evals: undefined,
-    codemode: undefined,
-    output: undefined,
-    advanced: undefined,
-    authorizationPolicy: undefined,
-    notifications: undefined,
-    team: undefined,
-};
-// ============================================================================
-// Agent Specs Registry
-// ============================================================================
-export const AGENT_SPECS = {
-    // Codeai
-    'codeai/data-acquisition': DATA_ACQUISITION_AGENT_SPEC,
-    'codeai/simple': SIMPLE_AGENT_SPEC,
-};
-/**
- * Get an agent specification by ID.
- */
-export function getAgentSpecs(agentId) {
-    return AGENT_SPECS[agentId];
-}
-/**
- * List all available agent specifications.
- *
- * @param prefix - If provided, only return specs whose ID starts with this prefix.
- */
-export function listAgentSpecs(prefix) {
-    const specs = Object.values(AGENT_SPECS);
-    return prefix !== undefined
-        ? specs.filter(s => s.id.startsWith(prefix))
-        : specs;
-}
-/**
- * Collect all required environment variables for an agent spec.
- *
- * Iterates over the spec's MCP servers and skills and returns the
- * deduplicated union of their `requiredEnvVars` arrays.
- */
-export function getAgentSpecRequiredEnvVars(spec) {
-    const vars = new Set();
-    for (const server of spec.mcpServers) {
-        for (const v of server.requiredEnvVars ?? []) {
-            vars.add(v);
-        }
-    }
-    for (const skill of spec.skills) {
-        for (const v of skill.requiredEnvVars ?? []) {
-            vars.add(v);
-        }
-    }
-    return Array.from(vars);
-}

package/lib/specs/agents/codeai/index.d.ts

@@ -1 +0,0 @@
-export * from './agents';

package/lib/specs/agents/codemode-paper/agents.d.ts

@@ -1,31 +0,0 @@
-/**
- * Agent Library.
- *
- * Predefined agent specifications that can be instantiated as Agent Runtimes.
- * THIS FILE IS AUTO-GENERATED. DO NOT EDIT MANUALLY.
- * Generated from YAML specifications in specs/agents/
- */
-import type { AgentSpec } from '../../../types/Types';
-export declare const CRAWLER_AGENT_SPEC: AgentSpec;
-export declare const DATA_ACQUISITION_AGENT_SPEC: AgentSpec;
-export declare const FINANCIAL_VIZ_AGENT_SPEC: AgentSpec;
-export declare const GITHUB_AGENT_SPEC: AgentSpec;
-export declare const INFORMATION_ROUTING_AGENT_SPEC: AgentSpec;
-export declare const AGENT_SPECS: Record<string, AgentSpec>;
-/**
- * Get an agent specification by ID.
- */
-export declare function getAgentSpecs(agentId: string): AgentSpec | undefined;
-/**
- * List all available agent specifications.
- *
- * @param prefix - If provided, only return specs whose ID starts with this prefix.
- */
-export declare function listAgentSpecs(prefix?: string): AgentSpec[];
-/**
- * Collect all required environment variables for an agent spec.
- *
- * Iterates over the spec's MCP servers and skills and returns the
- * deduplicated union of their `requiredEnvVars` arrays.
- */
-export declare function getAgentSpecRequiredEnvVars(spec: AgentSpec): string[];