@datalayer/agent-runtimes 1.0.2 → 1.0.3

package/lib/{components/chat/types/tool.d.ts → types/tools.d.ts}

@@ -1,9 +1,64 @@
 /**
- * Tool types for chat component.
- * Supports hybrid execution (frontend + backend) with per-tool annotation.
- *
- * @module components/chat/types/tool
+ * Runtime binding metadata for a tool implementation.
  */
+export interface ToolRuntimeSpec {
+    /** Implementation language */
+    language: 'python' | 'typescript';
+    /** Module/package containing the implementation */
+    package: string;
+    /** Callable/function name in the package */
+    method: string;
+}
+/**
+ * Specification for a runtime tool.
+ */
+export interface ToolSpec {
+    /** Unique tool identifier */
+    id: string;
+    /** Version */
+    version?: string;
+    /** Display name for the tool */
+    name: string;
+    /** Tool description */
+    description: string;
+    /** Tags for categorization */
+    tags: string[];
+    /** Whether the tool is enabled */
+    enabled: boolean;
+    /** Approval policy for this tool */
+    approval: 'auto' | 'manual';
+    /** Whether tool requires human approval before execution */
+    requiresApproval?: boolean;
+    /** Runtime binding metadata */
+    runtime: ToolRuntimeSpec;
+    /** Icon identifier */
+    icon?: string;
+    /** Emoji identifier */
+    emoji?: string;
+}
+/**
+ * Specification for a frontend tool set.
+ */
+export interface FrontendToolSpec {
+    /** Unique frontend tool identifier */
+    id: string;
+    /** Version */
+    version?: string;
+    /** Display name for the frontend tool */
+    name: string;
+    /** Frontend tool description */
+    description: string;
+    /** Tags for categorization */
+    tags: string[];
+    /** Whether the frontend tool is enabled */
+    enabled: boolean;
+    /** Which tools from the toolset to include ('all' or specific list) */
+    toolset: string;
+    /** Icon identifier */
+    icon?: string;
+    /** Emoji identifier */
+    emoji?: string;
+}
 import type { ReactNode } from 'react';
 /**
  * Tool execution location

package/lib/types/triggers.d.ts

@@ -0,0 +1,48 @@
+/**
+ * Dynamic field definition for a trigger type.
+ */
+export interface TriggerField {
+    /** Field key */
+    name: string;
+    /** Human-readable label */
+    label: string;
+    /** Field type */
+    type: 'string' | 'boolean' | 'number';
+    /** Whether the field is required */
+    required: boolean;
+    /** Placeholder text */
+    placeholder?: string;
+    /** Help text */
+    help?: string;
+    /** Font family hint (e.g., 'mono') */
+    font?: string;
+}
+/**
+ * Trigger type specification.
+ */
+export interface TriggerSpec {
+    /** Unique trigger identifier */
+    id: string;
+    /** Version */
+    version?: string;
+    /** Display name */
+    name: string;
+    /** Description of the trigger */
+    description: string;
+    /** Trigger type discriminator */
+    type: 'once' | 'schedule' | 'event';
+    /** Dynamic fields for this trigger type */
+    fields?: TriggerField[];
+}
+/**
+ * Trigger configuration for an agent spec.
+ */
+export interface AgentTriggerConfig {
+    type?: string;
+    cron?: string;
+    event_source?: string;
+    event?: string;
+    description?: string;
+    prompt?: string;
+    [key: string]: string | number | boolean | undefined;
+}

package/lib/types/triggers.js

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

package/lib/types/usage.d.ts

@@ -0,0 +1,36 @@
+export interface AgentUsageSummary {
+    /** Agent ID */
+    agentId: string;
+    /** Total tokens (input + output) */
+    totalTokens: number;
+    /** Input tokens */
+    inputTokens: number;
+    /** Output tokens */
+    outputTokens: number;
+    /** Estimated cost in USD */
+    totalCostUsd: number;
+    /** Number of model requests */
+    requestCount: number;
+    /** Number of tool calls */
+    toolCallCount: number;
+    /** Period start */
+    periodStart: string;
+    /** Period end */
+    periodEnd: string;
+}
+export interface ContextUsage {
+    /** Agent ID */
+    agentId: string;
+    /** Current token count in the context window */
+    currentTokens: number;
+    /** Maximum tokens in the context window */
+    maxTokens: number;
+    /** Usage percentage (0-100) */
+    usagePercent: number;
+    /** Number of messages in context */
+    messageCount: number;
+    /** Number of archived/evicted messages */
+    archivedCount: number;
+    /** Number of auto-summarizations performed */
+    summarizationCount: number;
+}

package/lib/types/usage.js

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

package/lib/utils/index.d.ts

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

package/lib/utils/index.js

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

package/lib/utils/utils.d.ts

@@ -0,0 +1,60 @@
+/**
+ * Utility functions and constants for ChatBase.
+ *
+ * @module chat/utils
+ */
+import { QueryClient } from '@tanstack/react-query';
+import type { ChatMessage } from '../types/messages';
+import type { ProtocolConfig } from '../types/protocol';
+import { type BaseProtocolAdapter } from '../protocols';
+import type { ToolCallMessage, DisplayItem } from '../types/chat';
+/** Singleton QueryClient for ChatBase instances without external QueryClientProvider */
+export declare const internalQueryClient: QueryClient;
+/** Primer's default portal root ID */
+export declare const PRIMER_PORTAL_ROOT_ID = "__primerPortalRoot__";
+/**
+ * Check if an item is a tool call message
+ */
+export declare function isToolCallMessage(item: DisplayItem): item is ToolCallMessage;
+/**
+ * Strip low-level protocol/tool-call markup from assistant text content.
+ *
+ * Some models occasionally leak internal call formats (for example,
+ * <calls>...</calls> or <tml:invoke ...>) into natural-language responses.
+ * Keep user/system text untouched and sanitize assistant-visible text only.
+ */
+export declare function sanitizeAssistantContent(content: string): string;
+/**
+ * Extract text content from a ChatMessage
+ */
+export declare function getMessageText(message: ChatMessage): string;
+/**
+ * Convert history messages to display items.
+ *
+ * History returns:
+ * - Assistant messages with `toolCalls` array (tool invocations)
+ * - Tool messages (role='tool') with content (tool results)
+ *
+ * For display, we need to:
+ * 1. Keep user/assistant text messages as ChatMessage
+ * 2. Convert each toolCall from assistant messages into a ToolCallMessage
+ * 3. Match tool result messages (role='tool') to their ToolCallMessage and update result
+ * 4. Filter out raw tool messages (role='tool') from display — they're merged into ToolCallMessage
+ */
+export declare function convertHistoryToDisplayItems(messages: ChatMessage[]): DisplayItem[];
+/**
+ * Format token count for compact display
+ */
+export declare function formatTokenCount(tokens: number): string;
+/**
+ * Derive the API base URL from a configEndpoint.
+ * configEndpoint may look like:
+ *   "http://host:port/api/v1/config"     (from agentRuntimeConfig)
+ *   "http://host:port/api/v1/configure"   (from Chat component)
+ * We strip the trailing path segment to get "http://host:port/api/v1".
+ */
+export declare function getApiBaseFromConfig(configEndpoint: string): string;
+/**
+ * Create protocol adapter based on configuration
+ */
+export declare function createProtocolAdapter(config: ProtocolConfig): BaseProtocolAdapter | null;

package/lib/utils/utils.js

@@ -0,0 +1,205 @@
+/*
+ * Copyright (c) 2025-2026 Datalayer, Inc.
+ * Distributed under the terms of the Modified BSD License.
+ */
+/**
+ * Utility functions and constants for ChatBase.
+ *
+ * @module chat/utils
+ */
+import { QueryClient } from '@tanstack/react-query';
+import { AGUIAdapter, A2AAdapter, VercelAIAdapter, ACPAdapter, } from '../protocols';
+// ---------------------------------------------------------------------------
+// Constants
+// ---------------------------------------------------------------------------
+/** Singleton QueryClient for ChatBase instances without external QueryClientProvider */
+export const internalQueryClient = new QueryClient({
+    defaultOptions: {
+        queries: {
+            staleTime: 5 * 60 * 1000, // 5 minutes
+            refetchOnWindowFocus: false,
+        },
+    },
+});
+/** Primer's default portal root ID */
+export const PRIMER_PORTAL_ROOT_ID = '__primerPortalRoot__';
+// ---------------------------------------------------------------------------
+// Type guards & helpers
+// ---------------------------------------------------------------------------
+/**
+ * Check if an item is a tool call message
+ */
+export function isToolCallMessage(item) {
+    return 'type' in item && item.type === 'tool-call';
+}
+/**
+ * Strip low-level protocol/tool-call markup from assistant text content.
+ *
+ * Some models occasionally leak internal call formats (for example,
+ * <calls>...</calls> or <tml:invoke ...>) into natural-language responses.
+ * Keep user/system text untouched and sanitize assistant-visible text only.
+ */
+export function sanitizeAssistantContent(content) {
+    if (!content || !content.includes('<')) {
+        return content;
+    }
+    let sanitized = content;
+    // Remove complete blocks first.
+    sanitized = sanitized
+        .replace(/<\s*calls\b[\s\S]*?<\s*\/\s*calls\s*>/gi, '')
+        .replace(/<\s*(?:an:)?tml:invoke\b[\s\S]*?<\s*\/\s*(?:an:)?tml:invoke\s*>/gi, '')
+        .replace(/<\s*(?:an:)?tml:parameter\b[\s\S]*?<\s*\/\s*(?:an:)?tml:parameter\s*>/gi, '');
+    // Remove standalone/open/close tags and malformed tag fragments.
+    sanitized = sanitized
+        .replace(/<\s*\/?\s*(?:an:[a-z_]+|tml:[a-z_]+|calls)\b[^>]*>?/gi, '')
+        .replace(/<\s*\/\s*>/g, '')
+        .replace(/^[ \t]+$/gm, '')
+        .replace(/\n{3,}/g, '\n\n')
+        .trim();
+    return sanitized;
+}
+/**
+ * Extract text content from a ChatMessage
+ */
+export function getMessageText(message) {
+    const text = typeof message.content === 'string'
+        ? message.content
+        : // Array of ContentPart - extract text parts
+            message.content
+                .filter((part) => part.type === 'text')
+                .map(part => part.text)
+                .join('');
+    if (message.role === 'assistant') {
+        return sanitizeAssistantContent(text);
+    }
+    return text;
+}
+// ---------------------------------------------------------------------------
+// History conversion
+// ---------------------------------------------------------------------------
+/**
+ * Convert history messages to display items.
+ *
+ * History returns:
+ * - Assistant messages with `toolCalls` array (tool invocations)
+ * - Tool messages (role='tool') with content (tool results)
+ *
+ * For display, we need to:
+ * 1. Keep user/assistant text messages as ChatMessage
+ * 2. Convert each toolCall from assistant messages into a ToolCallMessage
+ * 3. Match tool result messages (role='tool') to their ToolCallMessage and update result
+ * 4. Filter out raw tool messages (role='tool') from display — they're merged into ToolCallMessage
+ */
+export function convertHistoryToDisplayItems(messages) {
+    const displayItems = [];
+    const toolCallMap = new Map();
+    // First pass: collect all tool calls and build the initial display list
+    for (const msg of messages) {
+        if (msg.role === 'tool') {
+            // Tool result messages — will be merged later
+            const toolCallId = msg.metadata?.toolCallId;
+            if (toolCallId && toolCallMap.has(toolCallId)) {
+                // Update the existing tool call with the result
+                const toolCall = toolCallMap.get(toolCallId);
+                toolCall.result = msg.content;
+                toolCall.status = 'complete';
+            }
+            // Don't add tool messages to display — they're represented by ToolCallMessage
+            continue;
+        }
+        if (msg.role === 'assistant' && msg.toolCalls && msg.toolCalls.length > 0) {
+            // Assistant message with tool calls
+            // First add any text content as a regular message
+            const rawTextContent = typeof msg.content === 'string' ? msg.content.trim() : '';
+            const textContent = sanitizeAssistantContent(rawTextContent);
+            if (textContent) {
+                displayItems.push({
+                    ...msg,
+                    content: textContent,
+                    toolCalls: undefined, // Remove toolCalls from the text message
+                });
+            }
+            // Then add each tool call as a ToolCallMessage
+            // Map from message.ts ToolCallStatus to ChatBase ToolCallStatus
+            for (const tc of msg.toolCalls) {
+                let status = 'complete';
+                if (tc.status === 'pending' || tc.status === 'awaiting-approval') {
+                    status = 'inProgress';
+                }
+                else if (tc.status === 'executing') {
+                    status = 'executing';
+                }
+                else if (tc.status === 'failed') {
+                    status = 'error';
+                }
+                const toolCallMsg = {
+                    id: `tc-${tc.toolCallId}`,
+                    type: 'tool-call',
+                    toolCallId: tc.toolCallId,
+                    toolName: tc.toolName,
+                    args: tc.args || {},
+                    status,
+                    result: tc.result,
+                };
+                toolCallMap.set(tc.toolCallId, toolCallMsg);
+                displayItems.push(toolCallMsg);
+            }
+        }
+        else {
+            // Regular user/assistant/system message
+            displayItems.push(msg);
+        }
+    }
+    return displayItems;
+}
+// ---------------------------------------------------------------------------
+// Formatting
+// ---------------------------------------------------------------------------
+/**
+ * Format token count for compact display
+ */
+export function formatTokenCount(tokens) {
+    if (tokens >= 1_000_000)
+        return `${(tokens / 1_000_000).toFixed(1)}M`;
+    if (tokens >= 1_000)
+        return `${(tokens / 1_000).toFixed(1)}K`;
+    return tokens.toString();
+}
+/**
+ * Derive the API base URL from a configEndpoint.
+ * configEndpoint may look like:
+ *   "http://host:port/api/v1/config"     (from agentRuntimeConfig)
+ *   "http://host:port/api/v1/configure"   (from Chat component)
+ * We strip the trailing path segment to get "http://host:port/api/v1".
+ */
+export function getApiBaseFromConfig(configEndpoint) {
+    return configEndpoint.replace(/\/(config|configure)\/?$/, '');
+}
+// ---------------------------------------------------------------------------
+// Protocol adapter factory
+// ---------------------------------------------------------------------------
+/**
+ * Create protocol adapter based on configuration
+ */
+export function createProtocolAdapter(config) {
+    const adapterConfig = {
+        protocol: config.type,
+        baseUrl: config.endpoint,
+        authToken: config.authToken,
+        agentId: config.agentId,
+        ...config.options,
+    };
+    switch (config.type) {
+        case 'ag-ui':
+            return new AGUIAdapter(adapterConfig);
+        case 'a2a':
+            return new A2AAdapter(adapterConfig);
+        case 'vercel-ai':
+            return new VercelAIAdapter(adapterConfig);
+        case 'acp':
+            return new ACPAdapter(adapterConfig);
+        default:
+            console.warn(`[ChatBase] Unknown protocol type: ${config.type}`);
+            return null;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datalayer/agent-runtimes",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "type": "module",
   "workspaces": [
     ".",
@@ -89,7 +89,7 @@
     "preview": "vite preview",
     "rebuild:fresh": "npm run create:patches && npm install && npm run build && npm run clean:cache",
     "server": "python -m agent_runtimes serve",
-    "server:start": "PYTHONIOENCODING=utf-8 python -m agent_runtimes serve --port 8765 --debug",
+    "server:start": "PYTHONIOENCODING=utf-8 AGENT_RUNTIMES_DEBUG=false AGENT_RUNTIMES_LOG_LEVEL=info python -m agent_runtimes serve --port 8765 --log-level info",
     "start": "vite",
     "start:acp": "run-p server:start start:acp:vite",
     "start:acp:prod": "run-p server start:acp:vite",
@@ -120,18 +120,18 @@
   },
   "dependencies": {
     "@a2a-js/sdk": "^0.3.6",
-    "@a2ui/lit": "^0.8.1",
+    "@a2ui/react": "^0.8.0",
     "@ag-ui/client": "^0.0.42",
     "@ag-ui/core": "^0.0.42",
     "@ag-ui/encoder": "^0.0.42",
     "@ag-ui/proto": "^0.0.42",
     "@agentclientprotocol/sdk": "^0.8.0",
-    "@ai-sdk/react": "3.0.30",
+    "@ai-sdk/react": "^4.0.0-beta.33",
     "@anthropic-ai/sdk": "^0.52.0",
-    "@datalayer/core": "^1.0.1",
+    "@datalayer/core": "^1.0.13",
     "@datalayer/icons-react": "^1.0.6",
-    "@datalayer/jupyter-lexical": "^1.0.13",
-    "@datalayer/jupyter-react": "^2.0.3",
+    "@datalayer/jupyter-lexical": "^1.0.16",
+    "@datalayer/jupyter-react": "^2.0.7",
     "@datalayer/primer-addons": "^1.0.10",
     "@datalayer/primer-rjsf": "^1.0.1",
     "@excalidraw/excalidraw": "^0.18.0",
@@ -193,7 +193,7 @@
     "@tanstack/react-query": "^5.90.6",
     "@toon-format/toon": "^1.3.0",
     "@xyflow/react": "^12.10.0",
-    "ai": "6.0.28",
+    "ai": "^6.0.134",
     "ansi-to-html": "^0.7.2",
     "axios": "^1.7.7",
     "boring-avatars": "^2.0.1",
@@ -257,12 +257,11 @@
     "@storybook/builder-vite": "^9.1.1",
     "@storybook/react-vite": "^9.1.1",
     "@tailwindcss/postcss": "^4.1.16",
-    "@testing-library/react": "^16.3.0",
     "@testing-library/jest-dom": "^6.4.8",
+    "@testing-library/react": "^16.3.0",
     "@types/node": "^20.11.0",
     "@types/react": "18.3.20",
     "@types/react-dom": "18.3.6",
-    "@types/react-syntax-highlighter": "^15.5.13",
     "@vitejs/plugin-react": "^4.5.2",
     "@vitest/browser": "^3.2.4",
     "@vitest/coverage-v8": "^3.2.4",
@@ -302,7 +301,7 @@
     "vitest": "^3.2.4"
   },
   "resolutions": {
-    "@ai-sdk/react": "3.0.30",
+    "@ai-sdk/react": "^4.0.0-beta.33",
     "@microsoft/fast-colors": "5.3.1",
     "@microsoft/fast-components": "2.30.6",
     "@microsoft/fast-element": "1.13.0",
@@ -312,14 +311,14 @@
     "@microsoft/load-themed-styles": "1.10.295",
     "@types/react": "18.3.20",
     "@types/react-dom": "18.3.6",
-    "ai": "6.0.28",
+    "ai": "^6.0.134",
     "pyodide": "0.28.0",
     "react": "18.3.1",
     "react-dom": "18.3.1",
     "typescript": "^5.8.3"
   },
   "overrides": {
-    "@ai-sdk/react": "3.0.30",
+    "@ai-sdk/react": "^4.0.0-beta.33",
     "@microsoft/fast-colors": "5.3.1",
     "@microsoft/fast-components": "2.30.6",
     "@microsoft/fast-element": "1.13.0",
@@ -329,7 +328,7 @@
     "@microsoft/load-themed-styles": "1.10.295",
     "@types/react": "18.3.20",
     "@types/react-dom": "18.3.6",
-    "ai": "6.0.28",
+    "ai": "^6.0.134",
     "pyodide": "0.28.0",
     "react": "18.3.1",
     "react-dom": "18.3.1",