@datalayer/agent-runtimes 1.0.2 → 1.0.3

package/lib/specs/agents/datalayer-ai/agents.js

@@ -1,352 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-import { ALPHAVANTAGE_MCP_SERVER, FILESYSTEM_MCP_SERVER, GOOGLE_WORKSPACE_MCP_SERVER, KAGGLE_MCP_SERVER, TAVILY_MCP_SERVER, } from '../../mcpServers';
-import { GITHUB_SKILL_SPEC } from '../../skills';
-// ============================================================================
-// MCP Server Lookup
-// ============================================================================
-const MCP_SERVER_MAP = {
-    alphavantage: ALPHAVANTAGE_MCP_SERVER,
-    filesystem: FILESYSTEM_MCP_SERVER,
-    'google-workspace': GOOGLE_WORKSPACE_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
-// ============================================================================
-// Datalayer Ai Agents
-// ============================================================================
-export const CRAWLER_AGENT_SPEC = {
-    id: 'datalayer-ai/crawler',
-    name: 'Crawler Agent',
-    description: `Web crawling and research agent that searches the web and GitHub repositories for information.`,
-    tags: ['web', 'search', 'research', 'crawler', 'github'],
-    enabled: false,
-    model: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
-    mcpServers: [MCP_SERVER_MAP['tavily']],
-    skills: [toAgentSkillSpec(SKILL_MAP['github'])],
-    environmentName: 'ai-agents-env',
-    icon: 'globe',
-    emoji: '🌐',
-    color: '#10B981',
-    suggestions: [
-        'Search the web for recent news about AI agents',
-        'Find trending open-source Python projects on GitHub',
-        'Research best practices for building RAG applications',
-        'Compare popular JavaScript frameworks in 2024',
-    ],
-    sandboxVariant: 'jupyter',
-    systemPrompt: `You are a web crawling and research assistant with access to Tavily search and GitHub tools. Use Tavily to search the web for current information and search GitHub repositories for relevant projects. Synthesize information from multiple sources and provide clear summaries with sources cited.
-`,
-    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 DATA_ACQUISITION_AGENT_SPEC = {
-    id: 'datalayer-ai/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 FINANCIAL_AGENT_SPEC = {
-    id: 'datalayer-ai/financial',
-    name: 'Financial Visualization Agent',
-    description: `Analyzes financial market data and creates visualizations and charts.`,
-    tags: ['finance', 'stocks', 'visualization', 'charts'],
-    enabled: false,
-    model: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
-    mcpServers: [MCP_SERVER_MAP['alphavantage']],
-    skills: [],
-    environmentName: 'ai-agents-env',
-    icon: 'trending-up',
-    emoji: '📈',
-    color: '#F59E0B',
-    suggestions: [
-        'Show me the stock price history for AAPL',
-        'Create a chart comparing MSFT and GOOGL over the last year',
-        'Analyze the trading volume trends for Tesla',
-        'Get the latest market news for tech stocks',
-    ],
-    sandboxVariant: 'jupyter',
-    systemPrompt: `You are a financial market analyst with access to Alpha Vantage market data tools. You can fetch stock prices, analyze trading volumes, create visualizations, and track market trends. Provide clear insights with relevant data points and suggest visualization approaches when appropriate.
-`,
-    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 GITHUB_AGENT_SPEC = {
-    id: 'datalayer-ai/github-agent',
-    name: 'GitHub Agent',
-    description: `Manages GitHub repositories, issues, and pull requests with email notification capabilities.`,
-    tags: ['github', 'git', 'code', 'email'],
-    enabled: false,
-    model: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
-    mcpServers: [MCP_SERVER_MAP['google-workspace']],
-    skills: [toAgentSkillSpec(SKILL_MAP['github'])],
-    environmentName: 'ai-agents-env',
-    icon: 'git-branch',
-    emoji: '🐙',
-    color: '#6366F1',
-    suggestions: [
-        'List my open pull requests across all repositories',
-        'Create an issue for a bug I found in datalayer/ui',
-        'Show recent commits on the main branch',
-        'Search for repositories related to Jupyter notebooks',
-    ],
-    sandboxVariant: 'jupyter',
-    systemPrompt: `You are a GitHub assistant with access to GitHub skills and Google Workspace for email notifications. You can list and search repositories, issues, and pull requests, create new issues, review PRs, search code, and send email notifications. Always confirm repository names before creating issues/PRs and provide clear summaries when listing multiple items.
-`,
-    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: 'datalayer-ai/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 = {
-    // Datalayer Ai
-    'datalayer-ai/crawler': CRAWLER_AGENT_SPEC,
-    'datalayer-ai/data-acquisition': DATA_ACQUISITION_AGENT_SPEC,
-    'datalayer-ai/financial': FINANCIAL_AGENT_SPEC,
-    'datalayer-ai/github-agent': GITHUB_AGENT_SPEC,
-    'datalayer-ai/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/datalayer-ai/index.d.ts

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

package/lib/specs/agents/datalayer-ai/index.js

@@ -1,5 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-export * from './agents';

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

@@ -1,43 +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 ANALYZE_CAMPAIGN_PERFORMANCE_AGENT_SPEC: AgentSpec;
-export declare const ANALYZE_SUPPORT_TICKETS_AGENT_SPEC: AgentSpec;
-export declare const AUDIT_INVENTORY_LEVELS_AGENT_SPEC: AgentSpec;
-export declare const AUTOMATE_REGULATORY_REPORTING_AGENT_SPEC: AgentSpec;
-export declare const CLASSIFY_ROUTE_EMAILS_AGENT_SPEC: AgentSpec;
-export declare const COMPREHENSIVE_SALES_ANALYTICS_AGENT_SPEC: AgentSpec;
-export declare const END_OF_MONTH_SALES_PERFORMANCE_AGENT_SPEC: AgentSpec;
-export declare const EXTRACT_DATA_FROM_FILES_AGENT_SPEC: AgentSpec;
-export declare const GENERATE_WEEKLY_REPORTS_AGENT_SPEC: AgentSpec;
-export declare const MONITOR_SALES_KPIS_AGENT_SPEC: AgentSpec;
-export declare const OPTIMIZE_DYNAMIC_PRICING_AGENT_SPEC: AgentSpec;
-export declare const OPTIMIZE_GRID_OPERATIONS_AGENT_SPEC: AgentSpec;
-export declare const PROCESS_CITIZEN_REQUESTS_AGENT_SPEC: AgentSpec;
-export declare const PROCESS_CLINICAL_TRIAL_DATA_AGENT_SPEC: AgentSpec;
-export declare const PROCESS_FINANCIAL_TRANSACTIONS_AGENT_SPEC: AgentSpec;
-export declare const SUMMARIZE_DOCUMENTS_AGENT_SPEC: AgentSpec;
-export declare const SYNC_CRM_CONTACTS_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/mocks/index.d.ts

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

package/lib/specs/agents/mocks/index.js

@@ -1,5 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-export * from './agents';

package/lib/state/index.d.ts

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

package/lib/state/index.js

@@ -1,5 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-export * from './substates';

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

@@ -1,80 +0,0 @@
-import type { Transport } from '../../components/chat/components/Chat';
-export type AgentStatus = 'running' | 'paused' | 'initializing' | 'error';
-/**
- * Unified Agent model combining runtime tracking and UI state
- */
-export interface Agent {
-    /** Unique agent identifier */
-    id: string;
-    /** Display name */
-    name: string;
-    /** Agent description */
-    description: string;
-    /** Base URL for the agent (for Jupyter: baseUrl, for FastAPI: baseUrl) */
-    baseUrl: string;
-    /** Transport protocol used */
-    transport: Transport;
-    /** Current status */
-    status: AgentStatus;
-    /** Last error message if status is 'error' */
-    error?: string | null;
-    /** Timestamp of last update */
-    lastUpdated: number;
-    /** Document ID (for document-based agents) */
-    documentId?: string;
-    /** Runtime ID (for Jupyter kernel-based agents) */
-    runtimeId?: string;
-    /** Author name (optional, for display) */
-    author?: string;
-    /** Avatar URL (optional, for display) */
-    avatarUrl?: string;
-}
-export type AgentState = {
-    /** All registered agents */
-    agents: readonly Agent[];
-    /** Add or update an agent */
-    upsertAgent: (agent: Partial<Agent> & {
-        id: string;
-        baseUrl: string;
-        transport: Transport;
-    }) => void;
-    /** Get agent by ID */
-    getAgentById: (id: string) => Agent | undefined;
-    /** Get agent by baseUrl and transport */
-    getAgentByUrl: (baseUrl: string, transport: Transport) => Agent | undefined;
-    /** Update agent status */
-    updateAgentStatus: (id: string, status: AgentStatus, error?: string | null) => void;
-    /** Toggle agent status between running/paused */
-    toggleAgentStatus: (id: string) => void;
-    /** Delete an agent */
-    deleteAgent: (id: string) => void;
-    /** Clear all agents */
-    clearAgents: () => void;
-};
-export declare const agentStore: Omit<import("zustand").StoreApi<AgentState>, "persist"> & {
-    persist: {
-        setOptions: (options: Partial<import("zustand/middleware").PersistOptions<AgentState, unknown>>) => void;
-        clearStorage: () => void;
-        rehydrate: () => Promise<void> | void;
-        hasHydrated: () => boolean;
-        onHydrate: (fn: (state: AgentState) => void) => () => void;
-        onFinishHydration: (fn: (state: AgentState) => void) => () => void;
-        getOptions: () => Partial<import("zustand/middleware").PersistOptions<AgentState, unknown>>;
-    };
-};
-export declare function useAgentStore(): AgentState;
-export declare function useAgentStore<T>(selector: (state: AgentState) => T): T;
-export type AIAgentState = AgentState;
-export declare const aiAgentStore: Omit<import("zustand").StoreApi<AgentState>, "persist"> & {
-    persist: {
-        setOptions: (options: Partial<import("zustand/middleware").PersistOptions<AgentState, unknown>>) => void;
-        clearStorage: () => void;
-        rehydrate: () => Promise<void> | void;
-        hasHydrated: () => boolean;
-        onHydrate: (fn: (state: AgentState) => void) => () => void;
-        onFinishHydration: (fn: (state: AgentState) => void) => () => void;
-        getOptions: () => Partial<import("zustand/middleware").PersistOptions<AgentState, unknown>>;
-    };
-};
-export declare const useAIAgentStore: typeof useAgentStore;
-export default useAgentStore;

package/lib/state/substates/AIAgentState.js

@@ -1,108 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-import { createStore } from 'zustand/vanilla';
-import { useStore } from 'zustand';
-import { persist, createJSONStorage } from 'zustand/middleware';
-export const agentStore = createStore()(persist((set, get) => ({
-    agents: [],
-    upsertAgent: agentData => {
-        set(state => {
-            const existingIndex = state.agents.findIndex(a => a.id === agentData.id);
-            const now = Date.now();
-            if (existingIndex >= 0) {
-                // Update existing agent
-                const updatedAgents = [...state.agents];
-                updatedAgents[existingIndex] = {
-                    ...updatedAgents[existingIndex],
-                    ...agentData,
-                    lastUpdated: now,
-                };
-                return { agents: updatedAgents };
-            }
-            else {
-                // Add new agent
-                const newAgent = {
-                    name: agentData.name || agentData.id,
-                    description: agentData.description || '',
-                    status: agentData.status || 'initializing',
-                    lastUpdated: now,
-                    ...agentData,
-                };
-                return { agents: [...state.agents, newAgent] };
-            }
-        });
-    },
-    getAgentById: (id) => {
-        const { agents } = get();
-        return agents.find(agent => agent.id === id);
-    },
-    getAgentByUrl: (baseUrl, transport) => {
-        const { agents } = get();
-        return agents.find(agent => agent.baseUrl === baseUrl && agent.transport === transport);
-    },
-    updateAgentStatus: (id, status, error = null) => {
-        set(state => {
-            const index = state.agents.findIndex(a => a.id === id);
-            if (index >= 0) {
-                const updatedAgents = [...state.agents];
-                updatedAgents[index] = {
-                    ...updatedAgents[index],
-                    status,
-                    error,
-                    lastUpdated: Date.now(),
-                };
-                return { agents: updatedAgents };
-            }
-            return {};
-        });
-    },
-    toggleAgentStatus: (id) => {
-        set(state => {
-            const index = state.agents.findIndex(a => a.id === id);
-            if (index >= 0) {
-                const updatedAgents = [...state.agents];
-                const currentStatus = updatedAgents[index].status;
-                updatedAgents[index] = {
-                    ...updatedAgents[index],
-                    status: currentStatus === 'running' ? 'paused' : 'running',
-                    lastUpdated: Date.now(),
-                };
-                return { agents: updatedAgents };
-            }
-            return {};
-        });
-    },
-    deleteAgent: (id) => {
-        set(state => ({
-            agents: state.agents.filter(a => a.id !== id),
-        }));
-    },
-    clearAgents: () => {
-        set({ agents: [] });
-    },
-}), {
-    name: 'agent-runtimes-storage',
-    storage: createJSONStorage(() => localStorage),
-    // Only persist essential fields
-    partialize: state => ({
-        agents: state.agents.map(agent => ({
-            id: agent.id,
-            name: agent.name,
-            description: agent.description,
-            baseUrl: agent.baseUrl,
-            transport: agent.transport,
-            status: agent.status,
-            lastUpdated: agent.lastUpdated,
-            documentId: agent.documentId,
-            runtimeId: agent.runtimeId,
-        })),
-    }),
-}));
-export function useAgentStore(selector) {
-    return useStore(agentStore, selector);
-}
-export const aiAgentStore = agentStore;
-export const useAIAgentStore = useAgentStore;
-export default useAgentStore;

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

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

package/lib/state/substates/index.js

@@ -1,5 +0,0 @@
-/*
- * Copyright (c) 2025-2026 Datalayer, Inc.
- * Distributed under the terms of the Modified BSD License.
- */
-export * from './AIAgentState';

package/lib/types/AIAgent.d.ts

@@ -1,17 +0,0 @@
-/**
- * AI Agent model
- */
-export type IAIAgent = {
-    /**
-     * ID of the document monitored by the agent.
-     */
-    documentId: string;
-    /**
-     * ID of the runtime connected to the agent.
-     *
-     * This is not the name of the remote pod but
-     * the Simple Kernel ID of the process within it.
-     */
-    runtimeId?: string;
-};
-export default IAIAgent;