@datalayer/agent-runtimes 1.0.4 → 1.0.5

package/lib/specs/tools.js

@@ -10,7 +10,7 @@ export const RUNTIME_ECHO_TOOL_SPEC_0_0_1 = {
     version: '0.0.1',
     name: 'Runtime Echo',
     description: 'Echo text back to the caller for quick runtime verification.',
-    tags: ["runtime", "utility"],
+    tags: ['runtime', 'utility'],
     enabled: true,
     approval: 'auto',
     timeout: undefined,
@@ -28,7 +28,7 @@ export const RUNTIME_SEND_MAIL_TOOL_SPEC_0_0_1 = {
     version: '0.0.1',
     name: 'Runtime Send Mail (Fake)',
     description: 'Fake mail sender for tool approval demos; returns a simulated send receipt.',
-    tags: ["runtime", "approval", "mail"],
+    tags: ['runtime', 'approval', 'mail'],
     enabled: true,
     approval: 'manual',
     timeout: undefined,
@@ -46,10 +46,10 @@ export const RUNTIME_SENSITIVE_ECHO_TOOL_SPEC_0_0_1 = {
     version: '0.0.1',
     name: 'Runtime Sensitive Echo',
     description: 'Echo text with a manual approval checkpoint before execution.',
-    tags: ["runtime", "approval"],
+    tags: ['runtime', 'approval'],
     enabled: true,
     approval: 'manual',
-    timeout: '5h0m0s',
+    timeout: undefined,
     requiresApproval: true,
     runtime: {
         language: 'python',

package/lib/specs/triggers.js

@@ -12,10 +12,35 @@ export const EVENT_TRIGGER_SPEC_0_0_1 = {
     description: 'Trigger on specific events such as a webhook call, API request, database change, file upload, or email arrival.',
     type: 'event',
     fields: [
-        { name: 'event_source', label: 'Event Source URL', type: 'string', required: false, placeholder: 'https://helpdesk.example.com/webhooks', help: 'Allowed event source URL (leave empty to allow any source)' },
-        { name: 'event', label: 'Event Name', type: 'string', required: false, placeholder: 'email_received' },
-        { name: 'description', label: 'Description', type: 'string', required: false, placeholder: 'Description (e.g. \'Triggered on incoming email\')' },
-        { name: 'prompt', label: 'Trigger Prompt', type: 'string', required: false, placeholder: 'Handle the incoming event and execute the agent end-to-end.' },
+        {
+            name: 'event_source',
+            label: 'Event Source URL',
+            type: 'string',
+            required: false,
+            placeholder: 'https://helpdesk.example.com/webhooks',
+            help: 'Allowed event source URL (leave empty to allow any source)',
+        },
+        {
+            name: 'event',
+            label: 'Event Name',
+            type: 'string',
+            required: false,
+            placeholder: 'email_received',
+        },
+        {
+            name: 'description',
+            label: 'Description',
+            type: 'string',
+            required: false,
+            placeholder: "Description (e.g. 'Triggered on incoming email')",
+        },
+        {
+            name: 'prompt',
+            label: 'Trigger Prompt',
+            type: 'string',
+            required: false,
+            placeholder: 'Handle the incoming event and execute the agent end-to-end.',
+        },
     ],
 };
 export const ONCE_TRIGGER_SPEC_0_0_1 = {
@@ -25,7 +50,13 @@ export const ONCE_TRIGGER_SPEC_0_0_1 = {
     description: 'Execute agent immediately after deployment.',
     type: 'once',
     fields: [
-        { name: 'prompt', label: 'Trigger Prompt', type: 'string', required: false, placeholder: 'Start when requested by a user and complete the agent once.' },
+        {
+            name: 'prompt',
+            label: 'Trigger Prompt',
+            type: 'string',
+            required: false,
+            placeholder: 'Start when requested by a user and complete the agent once.',
+        },
     ],
 };
 export const SCHEDULE_TRIGGER_SPEC_0_0_1 = {
@@ -35,18 +66,37 @@ export const SCHEDULE_TRIGGER_SPEC_0_0_1 = {
     description: 'Run on a recurring schedule using a cron expression (e.g. daily at 9 AM, every Monday, monthly on the 1st).',
     type: 'schedule',
     fields: [
-        { name: 'cron', label: 'Cron Expression', type: 'string', required: true, placeholder: '0 9 * * * (every day at 9 AM)', font: 'mono' },
-        { name: 'description', label: 'Description', type: 'string', required: false, placeholder: 'Description (e.g. \'Monthly sales report\')' },
-        { name: 'prompt', label: 'Trigger Prompt', type: 'string', required: false, placeholder: 'Run the scheduled agent and produce the configured deliverable.' },
+        {
+            name: 'cron',
+            label: 'Cron Expression',
+            type: 'string',
+            required: true,
+            placeholder: '0 9 * * * (every day at 9 AM)',
+            font: 'mono',
+        },
+        {
+            name: 'description',
+            label: 'Description',
+            type: 'string',
+            required: false,
+            placeholder: "Description (e.g. 'Monthly sales report')",
+        },
+        {
+            name: 'prompt',
+            label: 'Trigger Prompt',
+            type: 'string',
+            required: false,
+            placeholder: 'Run the scheduled agent and produce the configured deliverable.',
+        },
     ],
 };
 // ============================================================================
 // Trigger Catalog
 // ============================================================================
 export const TRIGGER_CATALOG = {
-    'event': EVENT_TRIGGER_SPEC_0_0_1,
-    'once': ONCE_TRIGGER_SPEC_0_0_1,
-    'schedule': SCHEDULE_TRIGGER_SPEC_0_0_1,
+    event: EVENT_TRIGGER_SPEC_0_0_1,
+    once: ONCE_TRIGGER_SPEC_0_0_1,
+    schedule: SCHEDULE_TRIGGER_SPEC_0_0_1,
 };
 export function getTriggerSpecs() {
     return Object.values(TRIGGER_CATALOG);

package/lib/stores/agentRuntimeStore.d.ts

@@ -0,0 +1,204 @@
+import type { ServiceManager } from '@jupyterlab/services';
+import type { IRuntimeOptions } from '@datalayer/core/lib/stateful/runtimes/apis';
+import type { AgentStatus, AgentConnection, AgentConfig, Protocol } from '../types';
+import type { AgentStreamSnapshotPayload, AgentStreamToolApprovalPayload, CodemodeStatusData } from '../types/stream';
+import type { ContextSnapshotData } from '../types/context';
+import type { McpToolsetsStatusResponse } from '../types/mcp';
+import type { LoadedSkillInfo } from '../types/skills';
+/**
+ * A URL-addressable agent that has been previously connected.
+ * May originate from an agent runtime pod OR from the stable agents service.
+ */
+export interface AgentRegistryEntry {
+    id: string;
+    name: string;
+    description: string;
+    baseUrl: string;
+    protocol: Protocol;
+    status: AgentStatus;
+    error?: string | null;
+    lastUpdated: number;
+    documentId?: string;
+    runtimeId?: string;
+    author?: string;
+    avatarUrl?: string;
+}
+export type AgentRuntimeWsState = 'closed' | 'connecting' | 'connected';
+export interface LocalTokenTurn {
+    turnNumber: number;
+    timestampMs: number;
+    systemPromptTokens: number;
+    toolsDescriptionTokens: number;
+    userMessageTokens: number;
+    aiMessageTokens: number;
+    toolsUsageTokens: number;
+    totalTokens: number;
+}
+export interface LocalCostPoint {
+    timestampMs: number;
+    cumulativeUsd: number;
+}
+export interface MonitoringCacheEntry {
+    tokenTurns: LocalTokenTurn[];
+    costPoints: LocalCostPoint[];
+}
+export declare function getMonitoringCacheKey(serviceName?: string, agentId?: string): string;
+export interface AgentRuntimeStoreState {
+    agents: readonly AgentRegistryEntry[];
+    runtime: AgentConnection | null;
+    status: AgentStatus;
+    error: string | null;
+    isLaunching: boolean;
+    wsState: AgentRuntimeWsState;
+    approvals: AgentStreamToolApprovalPayload[];
+    pendingApprovalCount: number;
+    contextSnapshot: ContextSnapshotData | null;
+    costUsage: ContextSnapshotData['costUsage'] | null;
+    mcpStatus: McpToolsetsStatusResponse | null;
+    codemodeStatus: CodemodeStatusData | null;
+    fullContext: Record<string, unknown> | null;
+    monitoringCache: Record<string, MonitoringCacheEntry>;
+    loadedSkillsByAgentId: Record<string, LoadedSkillInfo[]>;
+}
+export interface AgentRuntimeStoreActions {
+    upsertAgent: (agent: Partial<AgentRegistryEntry> & {
+        id: string;
+        baseUrl: string;
+        protocol: Protocol;
+    }) => void;
+    getAgentById: (id: string) => AgentRegistryEntry | undefined;
+    getAgentByUrl: (baseUrl: string, protocol: Protocol) => AgentRegistryEntry | undefined;
+    updateAgentStatus: (id: string, status: AgentStatus, error?: string | null) => void;
+    toggleAgentStatus: (id: string) => void;
+    deleteAgent: (id: string) => void;
+    clearAgents: () => void;
+    setLoadedSkillsForAgent: (agentId: string, skills: LoadedSkillInfo[]) => void;
+    getLoadedSkillsForAgent: (agentId: string) => LoadedSkillInfo[];
+    clearLoadedSkillsForAgent: (agentId: string) => void;
+    launchAgent: (options: IRuntimeOptions) => Promise<AgentConnection>;
+    connectAgent: (connection: {
+        podName: string;
+        environmentName: string;
+        serviceManager?: ServiceManager.IManager;
+        jupyterBaseUrl?: string;
+        kernelId?: string;
+    }) => void;
+    createAgent: (config?: AgentConfig) => Promise<Pick<AgentConnection, 'agentId' | 'endpoint' | 'isReady'>>;
+    disconnect: () => void;
+    clearError: () => void;
+    setError: (error: string) => void;
+    setWsState: (state: AgentRuntimeWsState) => void;
+    setWs: (ws: WebSocket | null, agentId?: string) => void;
+    applySnapshot: (payload: AgentStreamSnapshotPayload) => void;
+    upsertApproval: (approval: AgentStreamToolApprovalPayload) => void;
+    removeApproval: (approvalId: string) => void;
+    sendDecision: (approvalId: string, approved: boolean, note?: string, toolCallId?: string, agentId?: string) => boolean;
+    requestRefresh: (agentId?: string) => boolean;
+    sendRawMessage: (payload: Record<string, unknown>, agentId?: string) => boolean;
+    appendLocalTokenTurn: (params: {
+        serviceName?: string;
+        agentId?: string;
+        timestampMs: number;
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }) => void;
+    mergeTokenTurns: (params: {
+        serviceName?: string;
+        agentId?: string;
+        turns: LocalTokenTurn[];
+    }) => void;
+    appendLocalTokenTurnFull: (params: {
+        serviceName?: string;
+        agentId?: string;
+        timestampMs: number;
+        systemPromptTokens: number;
+        toolsDescriptionTokens: number;
+        userMessageTokens: number;
+        aiMessageTokens: number;
+        toolsUsageTokens: number;
+        totalTokens: number;
+    }) => void;
+    upsertLocalCostPoint: (params: {
+        serviceName?: string;
+        agentId?: string;
+        timestampMs: number;
+        cumulativeUsd: number;
+    }) => void;
+    mergeCostPoints: (params: {
+        serviceName?: string;
+        agentId?: string;
+        points: LocalCostPoint[];
+    }) => void;
+    reset: () => void;
+    resetWs: () => void;
+}
+export type AgentRuntimeStore = AgentRuntimeStoreState & AgentRuntimeStoreActions;
+export declare const agentRuntimeStore: Omit<Omit<import("zustand").StoreApi<AgentRuntimeStore>, "subscribe"> & {
+    subscribe: {
+        (listener: (selectedState: AgentRuntimeStore, previousSelectedState: AgentRuntimeStore) => void): () => void;
+        <U>(selector: (state: AgentRuntimeStore) => U, listener: (selectedState: U, previousSelectedState: U) => void, options?: {
+            equalityFn?: ((a: U, b: U) => boolean) | undefined;
+            fireImmediately?: boolean;
+        } | undefined): () => void;
+    };
+}, "persist"> & {
+    persist: {
+        setOptions: (options: Partial<import("zustand/middleware").PersistOptions<AgentRuntimeStore, unknown>>) => void;
+        clearStorage: () => void;
+        rehydrate: () => Promise<void> | void;
+        hasHydrated: () => boolean;
+        onHydrate: (fn: (state: AgentRuntimeStore) => void) => () => void;
+        onFinishHydration: (fn: (state: AgentRuntimeStore) => void) => () => void;
+        getOptions: () => Partial<import("zustand/middleware").PersistOptions<AgentRuntimeStore, unknown>>;
+    };
+};
+export declare function useAgentRuntimeStore(): AgentRuntimeStore;
+export declare function useAgentRuntimeStore<T>(selector: (state: AgentRuntimeStore) => T): T;
+export declare const useAgentRuntimeConnection: () => AgentConnection | null;
+export declare const useAgentRuntimeStatus: () => AgentStatus;
+export declare const useAgentRuntimeError: () => string | null;
+export declare const useAgentRuntimeIsLaunching: () => boolean;
+export declare const useAgentRuntimeApprovals: () => AgentStreamToolApprovalPayload[];
+export declare const useAgentRuntimePendingCount: () => number;
+export declare const useAgentRuntimeMcpStatus: () => McpToolsetsStatusResponse | null;
+export declare const useAgentRuntimeFullContext: () => Record<string, unknown> | null;
+export declare const useAgentRuntimeContextSnapshot: () => ContextSnapshotData | null;
+export declare const useAgentRuntimeCostUsage: () => {
+    lastTurnCostUsd: number;
+    cumulativeCostUsd: number;
+    perRunBudgetUsd: number | null;
+    cumulativeBudgetUsd: number | null;
+    requestCount: number;
+    totalTokensUsed: number;
+    modelBreakdown: Array<{
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        costUsd: number;
+        requests: number;
+    }>;
+    runs: Array<{
+        timestamp: string;
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        runCostUsd: number;
+        cumulativeCostUsd: number;
+        pricePerInputToken: number | null;
+        pricePerOutputToken: number | null;
+        pricingResolved: boolean;
+    }>;
+    lastUpdated: string | null;
+} | null | undefined;
+export declare const useAgentRuntimeCodemodeStatus: () => CodemodeStatusData | null;
+export declare const useAgentRuntimeWsState: () => AgentRuntimeWsState;
+export declare const useAgentRuntimeLoadedSkills: (agentId?: string) => LoadedSkillInfo[];
+export declare const getAgentRuntimeState: () => AgentRuntimeStore;
+export declare const subscribeToAgentRuntime: {
+    (listener: (selectedState: AgentRuntimeStore, previousSelectedState: AgentRuntimeStore) => void): () => void;
+    <U>(selector: (state: AgentRuntimeStore) => U, listener: (selectedState: U, previousSelectedState: U) => void, options?: {
+        equalityFn?: ((a: U, b: U) => boolean) | undefined;
+        fireImmediately?: boolean;
+    } | undefined): () => void;
+};