@datalayer/agent-runtimes 1.0.18 → 1.0.20

package/lib/specs/models.d.ts

@@ -22,6 +22,7 @@ export declare const AIModels: {
     readonly BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_8: "bedrock:us.anthropic.claude-opus-4-8";
     readonly BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_20250514_V1_0: "bedrock:us.anthropic.claude-opus-4-20250514-v1:0";
     readonly BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_5_20250929_V1_0: "bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0";
+    readonly BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_6: "bedrock:us.anthropic.claude-sonnet-4-6";
     readonly BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_20250514_V1_0: "bedrock:us.anthropic.claude-sonnet-4-20250514-v1:0";
     readonly OPENAI_GPT_4_1_MINI: "openai:gpt-4.1-mini";
     readonly OPENAI_GPT_4_1_NANO: "openai:gpt-4.1-nano";
@@ -45,6 +46,7 @@ export declare const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_6_V1_0_0_1: AIModel;
 export declare const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_8_0_0_1: AIModel;
 export declare const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_20250514_V1_0_0_0_1: AIModel;
 export declare const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_5_20250929_V1_0_0_0_1: AIModel;
+export declare const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_6_0_0_1: AIModel;
 export declare const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_20250514_V1_0_0_0_1: AIModel;
 export declare const OPENAI_GPT_4_1_MINI_0_0_1: AIModel;
 export declare const OPENAI_GPT_4_1_NANO_0_0_1: AIModel;

package/lib/specs/models.js

@@ -20,6 +20,7 @@ export const AIModels = {
     BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_8: 'bedrock:us.anthropic.claude-opus-4-8',
     BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_20250514_V1_0: 'bedrock:us.anthropic.claude-opus-4-20250514-v1:0',
     BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_5_20250929_V1_0: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
+    BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_6: 'bedrock:us.anthropic.claude-sonnet-4-6',
     BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_20250514_V1_0: 'bedrock:us.anthropic.claude-sonnet-4-20250514-v1:0',
     OPENAI_GPT_4_1_MINI: 'openai:gpt-4.1-mini',
     OPENAI_GPT_4_1_NANO: 'openai:gpt-4.1-nano',
@@ -39,6 +40,7 @@ export const ANTHROPIC_CLAUDE_3_5_HAIKU_20241022_0_0_1 = {
     provider: 'anthropic',
     default: false,
     requiredEnvVars: ['ANTHROPIC_API_KEY'],
+    tokensLimit: 8192,
 };
 export const ANTHROPIC_CLAUDE_OPUS_4_20250514_0_0_1 = {
     id: 'anthropic:claude-opus-4-20250514',
@@ -48,6 +50,7 @@ export const ANTHROPIC_CLAUDE_OPUS_4_20250514_0_0_1 = {
     provider: 'anthropic',
     default: false,
     requiredEnvVars: ['ANTHROPIC_API_KEY'],
+    tokensLimit: 32000,
 };
 export const ANTHROPIC_CLAUDE_SONNET_4_5_20250514_0_0_1 = {
     id: 'anthropic:claude-sonnet-4-5-20250514',
@@ -57,6 +60,7 @@ export const ANTHROPIC_CLAUDE_SONNET_4_5_20250514_0_0_1 = {
     provider: 'anthropic',
     default: false,
     requiredEnvVars: ['ANTHROPIC_API_KEY'],
+    tokensLimit: 64000,
 };
 export const ANTHROPIC_CLAUDE_SONNET_4_20250514_0_0_1 = {
     id: 'anthropic:claude-sonnet-4-20250514',
@@ -66,6 +70,7 @@ export const ANTHROPIC_CLAUDE_SONNET_4_20250514_0_0_1 = {
     provider: 'anthropic',
     default: false,
     requiredEnvVars: ['ANTHROPIC_API_KEY'],
+    tokensLimit: 64000,
 };
 export const AZURE_OPENAI_GPT_4_1_MINI_0_0_1 = {
     id: 'azure-openai:gpt-4.1-mini',
@@ -75,6 +80,7 @@ export const AZURE_OPENAI_GPT_4_1_MINI_0_0_1 = {
     provider: 'azure-openai',
     default: false,
     requiredEnvVars: ['AZURE_OPENAI_API_KEY', 'AZURE_OPENAI_ENDPOINT'],
+    tokensLimit: 32768,
 };
 export const AZURE_OPENAI_GPT_4_1_NANO_0_0_1 = {
     id: 'azure-openai:gpt-4.1-nano',
@@ -84,6 +90,7 @@ export const AZURE_OPENAI_GPT_4_1_NANO_0_0_1 = {
     provider: 'azure-openai',
     default: false,
     requiredEnvVars: ['AZURE_OPENAI_API_KEY', 'AZURE_OPENAI_ENDPOINT'],
+    tokensLimit: 32768,
 };
 export const AZURE_OPENAI_GPT_4_1_0_0_1 = {
     id: 'azure-openai:gpt-4.1',
@@ -93,6 +100,7 @@ export const AZURE_OPENAI_GPT_4_1_0_0_1 = {
     provider: 'azure-openai',
     default: false,
     requiredEnvVars: ['AZURE_OPENAI_API_KEY', 'AZURE_OPENAI_ENDPOINT'],
+    tokensLimit: 32768,
 };
 export const AZURE_OPENAI_GPT_4O_MINI_0_0_1 = {
     id: 'azure-openai:gpt-4o-mini',
@@ -102,6 +110,7 @@ export const AZURE_OPENAI_GPT_4O_MINI_0_0_1 = {
     provider: 'azure-openai',
     default: false,
     requiredEnvVars: ['AZURE_OPENAI_API_KEY', 'AZURE_OPENAI_ENDPOINT'],
+    tokensLimit: 16384,
 };
 export const AZURE_OPENAI_GPT_4O_0_0_1 = {
     id: 'azure-openai:gpt-4o',
@@ -111,6 +120,7 @@ export const AZURE_OPENAI_GPT_4O_0_0_1 = {
     provider: 'azure-openai',
     default: false,
     requiredEnvVars: ['AZURE_OPENAI_API_KEY', 'AZURE_OPENAI_ENDPOINT'],
+    tokensLimit: 16384,
 };
 export const BEDROCK_US_ANTHROPIC_CLAUDE_FABLE_5_0_0_1 = {
     id: 'bedrock:us.anthropic.claude-fable-5',
@@ -124,6 +134,7 @@ export const BEDROCK_US_ANTHROPIC_CLAUDE_FABLE_5_0_0_1 = {
         'AWS_SECRET_ACCESS_KEY',
         'AWS_DEFAULT_REGION',
     ],
+    tokensLimit: 64000,
 };
 export const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_6_V1_0_0_1 = {
     id: 'bedrock:us.anthropic.claude-opus-4-6-v1',
@@ -137,6 +148,7 @@ export const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_6_V1_0_0_1 = {
         'AWS_SECRET_ACCESS_KEY',
         'AWS_DEFAULT_REGION',
     ],
+    tokensLimit: 32000,
 };
 export const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_8_0_0_1 = {
     id: 'bedrock:us.anthropic.claude-opus-4-8',
@@ -150,6 +162,7 @@ export const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_8_0_0_1 = {
         'AWS_SECRET_ACCESS_KEY',
         'AWS_DEFAULT_REGION',
     ],
+    tokensLimit: 32000,
 };
 export const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_20250514_V1_0_0_0_1 = {
     id: 'bedrock:us.anthropic.claude-opus-4-20250514-v1:0',
@@ -163,6 +176,7 @@ export const BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_20250514_V1_0_0_0_1 = {
         'AWS_SECRET_ACCESS_KEY',
         'AWS_DEFAULT_REGION',
     ],
+    tokensLimit: 32000,
 };
 export const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_5_20250929_V1_0_0_0_1 = {
     id: 'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0',
@@ -176,6 +190,21 @@ export const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_5_20250929_V1_0_0_0_1 = {
         'AWS_SECRET_ACCESS_KEY',
         'AWS_DEFAULT_REGION',
     ],
+    tokensLimit: 64000,
+};
+export const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_6_0_0_1 = {
+    id: 'bedrock:us.anthropic.claude-sonnet-4-6',
+    version: '0.0.1',
+    name: 'Bedrock Claude Sonnet 4.6',
+    description: 'Claude Sonnet 4.6 via AWS Bedrock - balanced performance',
+    provider: 'bedrock',
+    default: true,
+    requiredEnvVars: [
+        'AWS_ACCESS_KEY_ID',
+        'AWS_SECRET_ACCESS_KEY',
+        'AWS_DEFAULT_REGION',
+    ],
+    tokensLimit: 64000,
 };
 export const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_20250514_V1_0_0_0_1 = {
     id: 'bedrock:us.anthropic.claude-sonnet-4-20250514-v1:0',
@@ -189,6 +218,7 @@ export const BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_20250514_V1_0_0_0_1 = {
         'AWS_SECRET_ACCESS_KEY',
         'AWS_DEFAULT_REGION',
     ],
+    tokensLimit: 64000,
 };
 export const OPENAI_GPT_4_1_MINI_0_0_1 = {
     id: 'openai:gpt-4.1-mini',
@@ -198,6 +228,7 @@ export const OPENAI_GPT_4_1_MINI_0_0_1 = {
     provider: 'openai',
     default: false,
     requiredEnvVars: ['OPENAI_API_KEY'],
+    tokensLimit: 32768,
 };
 export const OPENAI_GPT_4_1_NANO_0_0_1 = {
     id: 'openai:gpt-4.1-nano',
@@ -207,6 +238,7 @@ export const OPENAI_GPT_4_1_NANO_0_0_1 = {
     provider: 'openai',
     default: false,
     requiredEnvVars: ['OPENAI_API_KEY'],
+    tokensLimit: 32768,
 };
 export const OPENAI_GPT_4_1_0_0_1 = {
     id: 'openai:gpt-4.1',
@@ -216,6 +248,7 @@ export const OPENAI_GPT_4_1_0_0_1 = {
     provider: 'openai',
     default: false,
     requiredEnvVars: ['OPENAI_API_KEY'],
+    tokensLimit: 32768,
 };
 export const OPENAI_GPT_4O_MINI_0_0_1 = {
     id: 'openai:gpt-4o-mini',
@@ -225,6 +258,7 @@ export const OPENAI_GPT_4O_MINI_0_0_1 = {
     provider: 'openai',
     default: false,
     requiredEnvVars: ['OPENAI_API_KEY'],
+    tokensLimit: 16384,
 };
 export const OPENAI_GPT_4O_0_0_1 = {
     id: 'openai:gpt-4o',
@@ -234,6 +268,7 @@ export const OPENAI_GPT_4O_0_0_1 = {
     provider: 'openai',
     default: false,
     requiredEnvVars: ['OPENAI_API_KEY'],
+    tokensLimit: 16384,
 };
 export const OPENAI_O3_MINI_0_0_1 = {
     id: 'openai:o3-mini',
@@ -243,6 +278,7 @@ export const OPENAI_O3_MINI_0_0_1 = {
     provider: 'openai',
     default: false,
     requiredEnvVars: ['OPENAI_API_KEY'],
+    tokensLimit: 100000,
 };
 // ============================================================================
 // AI Model Catalog
@@ -262,6 +298,7 @@ export const AI_MODEL_CATALOGUE = {
     'bedrock:us.anthropic.claude-opus-4-8': BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_8_0_0_1,
     'bedrock:us.anthropic.claude-opus-4-20250514-v1:0': BEDROCK_US_ANTHROPIC_CLAUDE_OPUS_4_20250514_V1_0_0_0_1,
     'bedrock:us.anthropic.claude-sonnet-4-5-20250929-v1:0': BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_5_20250929_V1_0_0_0_1,
+    'bedrock:us.anthropic.claude-sonnet-4-6': BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_6_0_0_1,
     'bedrock:us.anthropic.claude-sonnet-4-20250514-v1:0': BEDROCK_US_ANTHROPIC_CLAUDE_SONNET_4_20250514_V1_0_0_0_1,
     'openai:gpt-4.1-mini': OPENAI_GPT_4_1_MINI_0_0_1,
     'openai:gpt-4.1-nano': OPENAI_GPT_4_1_NANO_0_0_1,

package/lib/stores/agentRuntimeStore.d.ts

@@ -2,7 +2,7 @@ import type { ServiceManager } from '@jupyterlab/services';
 import type { INotebookContent } from '@jupyterlab/nbformat';
 import type { IRuntimeOptions } from '../runtimes/apis';
 import type { AgentStatus, AgentConnection, AgentConfig, Protocol } from '../types';
-import type { AgentStreamSnapshotPayload, AgentStreamToolApprovalPayload, CodemodeStatusData } from '../types/stream';
+import type { AgentStreamSnapshotPayload, AgentStreamSubagentPayload, AgentStreamCompactionPayload, AgentStreamToolApprovalPayload, CodemodeStatusData } from '../types/stream';
 import type { ContextSnapshotData } from '../types/context';
 import type { McpToolsetsStatusResponse } from '../types/mcp';
 import type { LoadedSkillInfo } from '../types/skills';
@@ -54,6 +54,17 @@ export interface AgentRuntimeStoreState {
     wsState: AgentRuntimeWsState;
     approvals: AgentStreamToolApprovalPayload[];
     pendingApprovalCount: number;
+    /**
+     * Live subagent (`delegate_task`) activity keyed by the parent delegation
+     * tool call id. Each entry is the ordered stream of interaction events for
+     * that subagent run.
+     */
+    subagentActivity: Record<string, AgentStreamSubagentPayload[]>;
+    /**
+     * Latest history-compaction activity for the connected agent, or `null` when
+     * no compaction has occurred on the current stream.
+     */
+    compaction: AgentStreamCompactionPayload | null;
     contextSnapshot: ContextSnapshotData | null;
     costUsage: ContextSnapshotData['costUsage'] | null;
     mcpStatus: McpToolsetsStatusResponse | null;
@@ -113,6 +124,9 @@ export interface AgentRuntimeStoreActions {
     setWsState: (state: AgentRuntimeWsState) => void;
     setWs: (ws: WebSocket | null, agentId?: string) => void;
     applySnapshot: (payload: AgentStreamSnapshotPayload) => void;
+    appendSubagentEvent: (event: AgentStreamSubagentPayload) => void;
+    clearSubagentActivity: () => void;
+    setCompaction: (payload: AgentStreamCompactionPayload | null) => void;
     upsertApproval: (approval: AgentStreamToolApprovalPayload) => void;
     removeApproval: (approvalId: string) => void;
     sendDecision: (approvalId: string, approved: boolean, note?: string, toolCallId?: string, agentId?: string) => boolean;
@@ -221,6 +235,23 @@ export declare const useAgentRuntimeCostUsage: () => {
 export declare const useAgentRuntimeCodemodeStatus: () => CodemodeStatusData | null;
 export declare const useAgentRuntimeWsState: () => AgentRuntimeWsState;
 export declare const useAgentRuntimeLoadedSkills: (agentId?: string) => LoadedSkillInfo[];
+/** Live subagent activity for a given parent `delegate_task` tool call id. */
+export declare const useAgentRuntimeSubagentActivity: (toolCallId?: string) => readonly AgentStreamSubagentPayload[];
+/**
+ * Live subagent activity resolved by tool-call id with a subagent-name
+ * fallback. The backend keys events by the parent run's `tool_call_id`, which
+ * can differ from the id surfaced in the chat transport stream; when the id
+ * misses we fall back to the most recent run for the named subagent.
+ */
+export declare const useAgentRuntimeSubagentActivityByToolCall: (toolCallId?: string, subagentName?: string) => readonly AgentStreamSubagentPayload[];
+/**
+ * Key of the currently active (running) subagent run, or `null` when none is
+ * active. A run is active while its event list has no `end`/`error` phase; when
+ * several are active the most recently started one wins.
+ */
+export declare const useAgentRuntimeActiveSubagentToolCallId: () => string | null;
+/** Latest history-compaction activity for the connected agent. */
+export declare const useAgentRuntimeCompaction: () => AgentStreamCompactionPayload | null;
 export declare const getAgentRuntimeState: () => AgentRuntimeStore;
 export declare const subscribeToAgentRuntime: {
     (listener: (selectedState: AgentRuntimeStore, previousSelectedState: AgentRuntimeStore) => void): () => void;

package/lib/stores/agentRuntimeStore.js

@@ -106,6 +106,8 @@ const initialWsState = {
     wsState: 'closed',
     approvals: [],
     pendingApprovalCount: 0,
+    subagentActivity: {},
+    compaction: null,
     contextSnapshot: null,
     costUsage: null,
     mcpStatus: null,
@@ -332,7 +334,7 @@ export const agentRuntimeStore = createStore()(subscribeWithSelector(persist((se
             const agentBaseUrl = jupyterBaseUrl.replace('/jupyter/server/', '/agent-runtimes/');
             const conn = {
                 podName: runtimePod.pod_name,
-                environmentName: runtimePod.environment_name,
+                environmentName: runtimePod.environment.name,
                 jupyterBaseUrl,
                 agentBaseUrl,
                 status: 'ready',
@@ -400,6 +402,18 @@ export const agentRuntimeStore = createStore()(subscribeWithSelector(persist((se
         codemodeStatus: payload.codemodeStatus ?? null,
         fullContext: payload.fullContext ?? null,
     })),
+    appendSubagentEvent: event => set(state => {
+        const key = event.toolCallId ?? event.subagentName;
+        const existing = state.subagentActivity[key] ?? [];
+        return {
+            subagentActivity: {
+                ...state.subagentActivity,
+                [key]: [...existing, event],
+            },
+        };
+    }),
+    clearSubagentActivity: () => set({ subagentActivity: {} }),
+    setCompaction: payload => set({ compaction: payload }),
     upsertApproval: approval => set(state => {
         const filtered = state.approvals.filter(a => a.id !== approval.id);
         const approvals = [approval, ...filtered];
@@ -738,6 +752,56 @@ export const useAgentRuntimeCostUsage = () => useAgentRuntimeStore(s => s.costUs
 export const useAgentRuntimeCodemodeStatus = () => useAgentRuntimeStore(s => s.codemodeStatus);
 export const useAgentRuntimeWsState = () => useAgentRuntimeStore(s => s.wsState);
 export const useAgentRuntimeLoadedSkills = (agentId) => useAgentRuntimeStore(s => agentId ? (s.loadedSkillsByAgentId[agentId] ?? []) : []);
+const EMPTY_SUBAGENT_EVENTS = [];
+/** Live subagent activity for a given parent `delegate_task` tool call id. */
+export const useAgentRuntimeSubagentActivity = (toolCallId) => useAgentRuntimeStore(s => toolCallId
+    ? (s.subagentActivity[toolCallId] ?? EMPTY_SUBAGENT_EVENTS)
+    : EMPTY_SUBAGENT_EVENTS);
+/**
+ * Live subagent activity resolved by tool-call id with a subagent-name
+ * fallback. The backend keys events by the parent run's `tool_call_id`, which
+ * can differ from the id surfaced in the chat transport stream; when the id
+ * misses we fall back to the most recent run for the named subagent.
+ */
+export const useAgentRuntimeSubagentActivityByToolCall = (toolCallId, subagentName) => useAgentRuntimeStore(s => {
+    if (toolCallId) {
+        const byId = s.subagentActivity[toolCallId];
+        if (byId && byId.length > 0)
+            return byId;
+    }
+    if (subagentName) {
+        const byNameKey = s.subagentActivity[subagentName];
+        if (byNameKey && byNameKey.length > 0)
+            return byNameKey;
+        let match;
+        for (const events of Object.values(s.subagentActivity)) {
+            if (events.length > 0 && events[0]?.subagentName === subagentName) {
+                match = events;
+            }
+        }
+        if (match)
+            return match;
+    }
+    return EMPTY_SUBAGENT_EVENTS;
+});
+/**
+ * Key of the currently active (running) subagent run, or `null` when none is
+ * active. A run is active while its event list has no `end`/`error` phase; when
+ * several are active the most recently started one wins.
+ */
+export const useAgentRuntimeActiveSubagentToolCallId = () => useAgentRuntimeStore(s => {
+    let activeKey = null;
+    for (const [key, events] of Object.entries(s.subagentActivity)) {
+        if (events.length === 0)
+            continue;
+        const done = events.some(e => e.phase === 'end' || e.phase === 'error');
+        if (!done)
+            activeKey = key;
+    }
+    return activeKey;
+});
+/** Latest history-compaction activity for the connected agent. */
+export const useAgentRuntimeCompaction = () => useAgentRuntimeStore(s => s.compaction);
 // ---------------------------------------------------------------------------
 // Non-React access
 // ---------------------------------------------------------------------------

package/lib/stores/index.d.ts

@@ -3,6 +3,6 @@
  *
  * @module store
  */
-export { agentRuntimeStore, useAgentRuntimeStore, useAgentRuntimeConnection, useAgentRuntimeStatus, useAgentRuntimeError, useAgentRuntimeIsLaunching, useAgentRuntimeApprovals, useAgentRuntimePendingCount, useAgentRuntimeMcpStatus, useAgentRuntimeFullContext, useAgentRuntimeContextSnapshot, useAgentRuntimeCostUsage, useAgentRuntimeCodemodeStatus, useAgentRuntimeWsState, useAgentRuntimeLoadedSkills, getAgentRuntimeState, subscribeToAgentRuntime, type AgentRegistryEntry, type AgentRuntimeWsState, type AgentRuntimeStoreState, type AgentRuntimeStoreActions, type AgentRuntimeStore, } from './agentRuntimeStore';
+export { agentRuntimeStore, useAgentRuntimeStore, useAgentRuntimeConnection, useAgentRuntimeStatus, useAgentRuntimeError, useAgentRuntimeIsLaunching, useAgentRuntimeApprovals, useAgentRuntimePendingCount, useAgentRuntimeMcpStatus, useAgentRuntimeFullContext, useAgentRuntimeContextSnapshot, useAgentRuntimeCostUsage, useAgentRuntimeCodemodeStatus, useAgentRuntimeWsState, useAgentRuntimeLoadedSkills, useAgentRuntimeActiveSubagentToolCallId, useAgentRuntimeSubagentActivity, useAgentRuntimeSubagentActivityByToolCall, useAgentRuntimeCompaction, getAgentRuntimeState, subscribeToAgentRuntime, type AgentRegistryEntry, type AgentRuntimeWsState, type AgentRuntimeStoreState, type AgentRuntimeStoreActions, type AgentRuntimeStore, } from './agentRuntimeStore';
 export { useChatStore, useChatMessages, useChatLoading, useChatStreaming, useChatError, useChatTools, useChatOpen, useChatConfig, useChatReady, useChatInferenceProvider, useChatExtensionRegistry, defaultChatConfig, type ChatStore, type ChatState, type ChatActions, type ChatConfig, type ToolCallState, } from './chatStore';
 export { useConversationStore, useConversationMessages, useNeedsFetch, useIsFetching, type ConversationStore, type ConversationData, } from './conversationStore';

package/lib/stores/index.js

@@ -7,6 +7,6 @@
  *
  * @module store
  */
-export { agentRuntimeStore, useAgentRuntimeStore, useAgentRuntimeConnection, useAgentRuntimeStatus, useAgentRuntimeError, useAgentRuntimeIsLaunching, useAgentRuntimeApprovals, useAgentRuntimePendingCount, useAgentRuntimeMcpStatus, useAgentRuntimeFullContext, useAgentRuntimeContextSnapshot, useAgentRuntimeCostUsage, useAgentRuntimeCodemodeStatus, useAgentRuntimeWsState, useAgentRuntimeLoadedSkills, getAgentRuntimeState, subscribeToAgentRuntime, } from './agentRuntimeStore';
+export { agentRuntimeStore, useAgentRuntimeStore, useAgentRuntimeConnection, useAgentRuntimeStatus, useAgentRuntimeError, useAgentRuntimeIsLaunching, useAgentRuntimeApprovals, useAgentRuntimePendingCount, useAgentRuntimeMcpStatus, useAgentRuntimeFullContext, useAgentRuntimeContextSnapshot, useAgentRuntimeCostUsage, useAgentRuntimeCodemodeStatus, useAgentRuntimeWsState, useAgentRuntimeLoadedSkills, useAgentRuntimeActiveSubagentToolCallId, useAgentRuntimeSubagentActivity, useAgentRuntimeSubagentActivityByToolCall, useAgentRuntimeCompaction, getAgentRuntimeState, subscribeToAgentRuntime, } from './agentRuntimeStore';
 export { useChatStore, useChatMessages, useChatLoading, useChatStreaming, useChatError, useChatTools, useChatOpen, useChatConfig, useChatReady, useChatInferenceProvider, useChatExtensionRegistry, defaultChatConfig, } from './chatStore';
 export { useConversationStore, useConversationMessages, useNeedsFetch, useIsFetching, } from './conversationStore';

package/lib/types/agents.d.ts

@@ -30,7 +30,7 @@ export interface AgentRuntimeState {
 /**
  * Agent Runtime data type (mapped from runtimes service).
  *
- * Backend RuntimePod fields: pod_name, environment_name, environment_title, uid,
+ * Backend RuntimePod fields: pod_name, environment{name,title,cpu,memory,gpu,resources}, uid,
  * type, given_name, token, ingress, reservation_id, started_at, expired_at, burning_rate.
  *
  * We map `ingress` to `url` for consistency with the UI.
@@ -39,8 +39,14 @@ export type AgentRuntimeData = {
     pod_name: string;
     id: string;
     name: string;
-    environment_name: string;
-    environment_title?: string;
+    environment: {
+        name: string;
+        title?: string;
+        cpu?: string;
+        memory?: string;
+        gpu?: string;
+        resources?: Record<string, string>;
+    };
     given_name: string;
     type: string;
     started_at?: string;

package/lib/types/agentspecs.d.ts

@@ -29,6 +29,8 @@ export interface Agentspec {
     systemPromptCodemodeAddons?: string;
     /** Tags for categorization */
     tags: string[];
+    /** Domain used to group agents in the gallery */
+    domain?: string;
     /** Whether the agent is enabled */
     enabled: boolean;
     /** AI model identifier to use for this agent */

package/lib/types/chat.d.ts

@@ -4,6 +4,7 @@
  * @module types/chat
  */
 import type { ComponentType, ReactNode } from 'react';
+import type { ICollaborationProvider } from '@datalayer/jupyter-react';
 import type { ChatMessage, MessageHandler } from './messages';
 import type { Protocol, ProtocolConfig } from './protocol';
 import type { McpServerSelection } from './inference';
@@ -12,6 +13,8 @@ import type { AgentRuntimeConfig } from './config';
 import type { SandboxWsStatus } from './sandbox';
 import type { FrontendToolDefinition } from './tools';
 import type { PoweredByTagProps } from '../chat/display/PoweredByTag';
+import type { EphemeralRuntimeOverride } from '../chat/notebook/EphemeralNotebook';
+import type { EphemeralDocumentCollaboration } from '../chat/document/EphemeralDocument';
 /**
  * Context passed to tool-call pre-hooks.
  * Fires when a tool call starts executing (backend or frontend).
@@ -366,6 +369,34 @@ export interface ChatCommonProps {
      * Defaults to the toolbar from `@datalayer/jupyter-react` when omitted.
      */
     ephemeralNotebookToolbar?: EphemeralNotebookToolbarComponent;
+    /**
+     * Optional real-time collaboration provider for the ephemeral notebook.
+     * When supplied, the notebook joins a shared collaborative room so its state
+     * transits over RTC (e.g. between an Agent Node and the SaaS UI) instead of
+     * the tunnel. The caller owns the provider lifecycle.
+     */
+    ephemeralNotebookCollaborationProvider?: ICollaborationProvider;
+    /**
+     * Explicit collaboration room / document id for the ephemeral notebook. When
+     * set it becomes the notebook id directly so multiple peers (e.g. an Agent
+     * Node and the SaaS UI) share the same collaborative room and tool scope.
+     */
+    ephemeralNotebookCollaborationDocumentId?: string;
+    /**
+     * Optional real-time collaboration configuration for the ephemeral document
+     * (Lexical). When supplied the document joins a shared Loro room over
+     * WebSocket so its rich-text state transits over RTC (e.g. between an Agent
+     * Node and the SaaS UI). Its `roomId` also becomes the document id so all
+     * peers share the same collaborative room and lexical tool scope.
+     */
+    ephemeralDocumentCollaboration?: EphemeralDocumentCollaboration;
+    /**
+     * Explicit runtime endpoint for the ephemeral notebook kernel. When set, the
+     * notebook binds its kernel to this endpoint directly instead of resolving a
+     * pod from the user's runtimes list — used to reach an Agent Node's Jupyter
+     * server through the runtimes tunnel HTTP/WebSocket proxy.
+     */
+    ephemeralRuntimeOverride?: EphemeralRuntimeOverride;
     /** Pre-hook: fires when a tool call starts executing */
     onToolCallStart?: (context: ToolCallStartContext) => void;
     /** Post-hook: fires when a tool result is received */
@@ -413,12 +444,30 @@ export interface ChatCommonProps {
     codemodeStatusData?: import('./stream').CodemodeStatusData | null;
     /** Optional sandbox status override for immediate UI updates. */
     sandboxStatusData?: SandboxWsStatus | null;
+    /** Horizontal placement for the header kernel indicator. @default 'left' */
+    kernelIndicatorPlacement?: 'left' | 'center' | 'right';
     /**
      * Live kernel connection of the companion notebook/document sandbox.
      * When provided, it is forwarded to the chat header's `<KernelIndicator>`
      * so the indicator reflects the surface's runtime.
      */
     kernel?: import('@jupyterlab/services/lib/kernel/kernel').IKernelConnection | null;
+    /**
+     * Optional environment name displayed in the chat header's kernel
+     * indicator details (e.g. the agent's local sandbox name). Defaults to
+     * the indicator's own "browser-runtime" placeholder when omitted.
+     */
+    kernelEnvironmentName?: string;
+    /** Optional CPU info displayed in the chat header's kernel indicator. */
+    kernelCpu?: string;
+    /** Optional memory info displayed in the chat header's kernel indicator. */
+    kernelMemory?: string;
+    /** Optional GPU info displayed in the chat header's kernel indicator. */
+    kernelGpu?: string;
+    /** Optional theme variant override for companion notebook/document surfaces. */
+    themeVariant?: string;
+    /** Optional color mode override for companion notebook/document surfaces. */
+    colorMode?: 'light' | 'dark' | 'auto';
     /**
      * Disable ChatBase's internal JupyterReactTheme wrapper.
      * Use this when the host page already provides a JupyterReactTheme boundary.
@@ -553,6 +602,8 @@ export interface ChatBaseProps {
     headerActions?: React.ReactNode;
     /** Notebook kernel indicator state override for the chat header. */
     kernelIndicatorState?: import('@datalayer/jupyter-react').ExecutionState;
+    /** Horizontal placement for the header kernel indicator. @default 'left' */
+    kernelIndicatorPlacement?: 'left' | 'center' | 'right';
     /**
      * Live notebook kernel connection. When provided, the chat header
      * renders the same `<KernelIndicator>` as the notebook toolbar so
@@ -573,6 +624,10 @@ export interface ChatBaseProps {
     kernelMemory?: string;
     /** Optional GPU info displayed in kernel indicator details. */
     kernelGpu?: string;
+    /** Optional theme variant override for companion notebook/document surfaces. */
+    themeVariant?: string;
+    /** Optional color mode override for companion notebook/document surfaces. */
+    colorMode?: 'light' | 'dark' | 'auto';
     /**
      * Current chat view mode.
      * When provided, a segmented view-mode toggle is rendered in the header
@@ -746,6 +801,34 @@ export interface ChatBaseProps {
      * Defaults to the toolbar from `@datalayer/jupyter-react` when omitted.
      */
     ephemeralNotebookToolbar?: EphemeralNotebookToolbarComponent;
+    /**
+     * Optional real-time collaboration provider for the ephemeral notebook.
+     * When supplied, the notebook joins a shared collaborative room so its state
+     * transits over RTC (e.g. between an Agent Node and the SaaS UI) instead of
+     * the tunnel. The caller owns the provider lifecycle.
+     */
+    ephemeralNotebookCollaborationProvider?: ICollaborationProvider;
+    /**
+     * Explicit collaboration room / document id for the ephemeral notebook. When
+     * set it becomes the notebook id directly so multiple peers (e.g. an Agent
+     * Node and the SaaS UI) share the same collaborative room and tool scope.
+     */
+    ephemeralNotebookCollaborationDocumentId?: string;
+    /**
+     * Optional real-time collaboration configuration for the ephemeral document
+     * (Lexical). When supplied the document joins a shared Loro room over
+     * WebSocket so its rich-text state transits over RTC (e.g. between an Agent
+     * Node and the SaaS UI). Its `roomId` also becomes the document id so all
+     * peers share the same collaborative room and lexical tool scope.
+     */
+    ephemeralDocumentCollaboration?: EphemeralDocumentCollaboration;
+    /**
+     * Explicit runtime endpoint for the ephemeral notebook kernel. When set, the
+     * notebook binds its kernel to this endpoint directly instead of resolving a
+     * pod from the user's runtimes list — used to reach an Agent Node's Jupyter
+     * server through the runtimes tunnel HTTP/WebSocket proxy.
+     */
+    ephemeralRuntimeOverride?: EphemeralRuntimeOverride;
     /**
      * Callback when the agent requests authorization for an external service.
      * This is called when a tool needs OAuth access to a service like GitHub.

package/lib/types/models.d.ts

@@ -17,6 +17,11 @@ export interface AIModel {
     default: boolean;
     /** Required environment variable names */
     requiredEnvVars: string[];
+    /**
+     * Maximum output tokens the model can generate in a single run
+     * (maps to pydantic-ai output_tokens_limit).
+     */
+    tokensLimit?: number;
 }
 /**
  * Configuration for an AI model runtime (as returned by the server).

package/lib/types/stream.d.ts

@@ -1,7 +1,64 @@
 import type { ContextSnapshotData } from './context';
 import type { McpToolsetsStatusResponse } from './mcp';
 import type { SkillStatus } from './skills';
-export type AgentStreamEventType = 'agent.snapshot' | 'tool_approval_created' | 'tool_approval_approved' | 'tool_approval_rejected';
+export type AgentStreamEventType = 'agent.snapshot' | 'agent.subagent' | 'agent.compaction' | 'tool_approval_created' | 'tool_approval_approved' | 'tool_approval_rejected';
+/** Phase of a streamed history-compaction interaction. */
+export type AgentCompactionPhase = 'start' | 'end';
+/**
+ * History-compaction activity pushed on the agent's monitoring stream when the
+ * conversation is summarized to stay under the configured token budget. A
+ * `start` event precedes summarization; the matching `end` event reports the
+ * before/after token and message counts plus the elapsed time.
+ */
+export interface AgentStreamCompactionPayload {
+    /** Interaction phase. */
+    phase: AgentCompactionPhase;
+    /** Token ceiling driving compaction. */
+    budget: number;
+    /** Estimated history tokens before compaction. */
+    beforeTokens: number;
+    /** Estimated history tokens after compaction (`end`). */
+    afterTokens?: number;
+    /** Message count before compaction. */
+    beforeMessages: number;
+    /** Message count after compaction (`end`). */
+    afterMessages?: number;
+    /** Elapsed summarization time in milliseconds (`end`). */
+    durationMs?: number;
+    /** Cumulative number of compactions performed (`end`). */
+    compactionCount?: number;
+    /** Whether history was actually reduced (`end`). */
+    reduced?: boolean;
+}
+/** Phase of a streamed subagent interaction. */
+export type AgentSubagentPhase = 'start' | 'text' | 'thinking' | 'tool_call' | 'tool_result' | 'end' | 'error';
+/**
+ * Incremental subagent activity pushed on the parent agent's monitoring
+ * stream while a `delegate_task` call runs. Consumers key events by
+ * `toolCallId` (the parent delegation tool call) to render a live timeline.
+ */
+export interface AgentStreamSubagentPayload {
+    /** Name of the subagent producing the activity. */
+    subagentName: string;
+    /** Parent `delegate_task` tool call id, when resolvable. */
+    toolCallId?: string | null;
+    /** Interaction phase. */
+    phase: AgentSubagentPhase;
+    /** Delegated task description (`start`). */
+    task?: string;
+    /** Text or thinking delta (`text` / `thinking`). */
+    text?: string;
+    /** Tool name (`tool_call` / `tool_result`). */
+    toolName?: string;
+    /** Tool call arguments (`tool_call`). */
+    toolArgs?: Record<string, unknown>;
+    /** Tool result preview (`tool_result`). */
+    result?: string;
+    /** Final subagent output (`end`). */
+    output?: string;
+    /** Failure message (`error`). */
+    error?: string;
+}
 export interface AgentStreamMessage<TPayload = Record<string, unknown>> {
     version: string;
     type: AgentStreamEventType | string;