@tanstack/ai-client 0.33.2 → 0.34.0

package/dist/esm/chat-client.d.ts

@@ -1,7 +1,7 @@
 import { AnyClientTool, InterruptDefinition, ModelMessage, RunAgentResumeItem, StreamChunk } from '@tanstack/ai/client';
 import { ByokClient } from './byok.js';
 import { ConnectionAdapter } from './connection-adapters.js';
-import { BoundInterrupts, ChatClientOptions, ChatClientState, ChatFetcher, ChatInterruptState, ResolvableChatInterrupt, ChatResumeState, ConnectionStatus, MultimodalContent, QueueOption, QueueStrategy, QueuedMessage, SendMessageOptions, UIMessage, WhenBusy } from './types.js';
+import { BoundInterrupts, ChatClientOptions, ChatClientState, ChatFetcher, ChatInterruptState, ResolvableChatInterrupt, ChatResumeState, ConnectionStatus, MultimodalContent, QueueOption, QueueStrategy, QueuedMessage, SendMessageOptions, SubagentHandle, UIMessage, WhenBusy } from './types.js';
 type ChatClientUpdateOptionsWithoutContext<TTools extends ReadonlyArray<AnyClientTool>, TInterrupts extends ReadonlyArray<InterruptDefinition<any, any, any, any>> = readonly []> = {
     connection?: ConnectionAdapter;
     fetcher?: ChatFetcher;
@@ -51,6 +51,8 @@ export interface NormalizedQueueConfig {
 export declare function normalizeQueueOption(option: QueueOption | undefined): NormalizedQueueConfig;
 export declare class ChatClient<TTools extends ReadonlyArray<AnyClientTool> = any, TContext = unknown, TInterrupts extends ReadonlyArray<InterruptDefinition<any, any, any, any>> = any> {
     private readonly processor;
+    private readonly subagentHandles;
+    private readonly stoppedSubagentIds;
     private connection;
     private uniqueId;
     private threadId;
@@ -484,6 +486,12 @@ export declare class ChatClient<TTools extends ReadonlyArray<AnyClientTool> = an
      * Get current messages
      */
     getMessages(): Array<UIMessage<TTools>>;
+    getSubagents(): SubagentHandle[];
+    private devtoolsSnapshotQueued;
+    /** One devtools snapshot per microtask, however many chunks arrive. */
+    private queueDevtoolsSnapshot;
+    private syncSubagentHandles;
+    private stopSubagent;
     /**
      * True when the last hydrate or older-page response said more messages exist.
      */

package/dist/esm/chat-client.js

@@ -6,7 +6,7 @@ import { fetcherToConnectionAdapter, getChunkRunId as getChunkRunId$1, normalize
 import { ChatPersistor } from "./client-persistor.js";
 import { ClearedStreamTracker } from "./cleared-stream-tracker.js";
 import { InterruptManager } from "./interrupt-manager.js";
-import { StreamProcessor, convertSchemaToJsonSchema, generateMessageId, isStandardSchema, mergeMetadata, normalizeToUIMessage, parseWithStandardSchema, restoreInboundChunk, tanstackMetadata } from "@tanstack/ai/client";
+import { EventType, StreamProcessor, convertSchemaToJsonSchema, generateMessageId, isStandardSchema, mergeMetadata, normalizeToUIMessage, parseWithStandardSchema, restoreInboundChunk, tanstackMetadata } from "@tanstack/ai/client";
 import { ByokBlockedError, ByokMissingError, ByokUnresolvedProviderError } from "@tanstack/ai/byok";
 //#region src/chat-client.ts
 function assertUniqueInterruptDefinitions(interrupts) {
@@ -139,11 +139,35 @@ var REJOIN_REBUILD_TRIGGERS = /* @__PURE__ */ new Set([
 	"TEXT_MESSAGE_START",
 	"TEXT_MESSAGE_CONTENT",
 	"TOOL_CALL_START",
-	"MESSAGES_SNAPSHOT"
+	"MESSAGES_SNAPSHOT",
+	"SUBAGENT_STARTED"
 ]);
+/** Every subagent card in the messages, nested cards included. */
+function collectSubagentParts(messages) {
+	const cards = [];
+	for (const message of messages) for (const part of message.parts) {
+		if (part.type !== "subagent") continue;
+		cards.push(part);
+		cards.push(...collectSubagentParts(part.subagent.messages));
+	}
+	return cards;
+}
+function readSubagentRunId(chunk) {
+	if ("subagentRunId" in chunk && typeof chunk.subagentRunId === "string") return chunk.subagentRunId;
+}
 var MAX_RUN_LINEAGE_DEPTH = 64;
+function copyForDevtools(value) {
+	if (Array.isArray(value)) return value.map(copyForDevtools);
+	if (value instanceof Date) return new Date(value.getTime());
+	if (value === null || typeof value !== "object") return value;
+	const out = {};
+	for (const [key, item] of Object.entries(value)) if (typeof item !== "function") out[key] = copyForDevtools(item);
+	return out;
+}
 var ChatClient = class {
 	processor;
+	subagentHandles = /* @__PURE__ */ new Map();
+	stoppedSubagentIds = /* @__PURE__ */ new Set();
 	connection;
 	uniqueId;
 	threadId;
@@ -321,8 +345,10 @@ var ChatClient = class {
 			...initialMessages ? { initialMessages } : {},
 			events: {
 				onMessagesChange: (messages) => {
+					this.syncSubagentHandles();
 					this.persistor?.notifyMessagesChanged(messages);
 					this.callbacksRef.current.onMessagesChange(messages);
+					this.queueDevtoolsSnapshot();
 				},
 				onStreamStart: () => {
 					this.setStatus("streaming");
@@ -414,6 +440,7 @@ var ChatClient = class {
 				}
 			}
 		});
+		this.syncSubagentHandles();
 		this.persistor?.hydrateAsync(persistedState);
 		this.rejoinRunId = rejoinRunId;
 		this.cachesMessages = cachesMessages;
@@ -882,7 +909,7 @@ var ChatClient = class {
 	}
 	getDevtoolsSnapshot() {
 		return {
-			messages: this.processor.getMessages(),
+			messages: copyForDevtools(this.processor.getMessages()),
 			status: this.status,
 			isLoading: this.isLoading,
 			isSubscribed: this.isSubscribed,
@@ -1061,7 +1088,15 @@ var ChatClient = class {
 		}
 		this.callbacksRef.current.onChunk(chunk);
 		this.devtoolsBridge.observeChunk(chunk);
+		const attributedId = readSubagentRunId(chunk);
+		if (attributedId && chunk.type === EventType.SUBAGENT_STARTED) this.stoppedSubagentIds.delete(attributedId);
+		if (attributedId && this.stoppedSubagentIds.has(attributedId)) {
+			this.updateRunLifecycle(chunk);
+			this.resolveJoinedRun(chunk);
+			return;
+		}
 		this.processor.processChunk(chunk);
+		this.syncSubagentHandles();
 		this.updateRunLifecycle(chunk);
 		this.observeInterruptState(chunk);
 		if (options?.defer !== false && (typeof document === "undefined" || !document.hidden)) await new Promise((resolve) => setTimeout(resolve, 0));
@@ -1502,6 +1537,7 @@ var ChatClient = class {
 		this.resetHistoryPaging();
 		this.discardPendingSends();
 		this.persistor?.remove();
+		this.stoppedSubagentIds.clear();
 		this.lastResume = null;
 		this.interruptManager.reset();
 		this.pendingResumeThreadId = null;
@@ -1639,6 +1675,59 @@ var ChatClient = class {
 	getMessages() {
 		return this.processor.getMessages();
 	}
+	getSubagents() {
+		return [...this.subagentHandles.values()];
+	}
+	devtoolsSnapshotQueued = false;
+	/** One devtools snapshot per microtask, however many chunks arrive. */
+	queueDevtoolsSnapshot() {
+		if (this.devtoolsSnapshotQueued) return;
+		this.devtoolsSnapshotQueued = true;
+		queueMicrotask(() => {
+			this.devtoolsSnapshotQueued = false;
+			this.devtoolsBridge.emitSnapshot();
+		});
+	}
+	syncSubagentHandles() {
+		const cards = collectSubagentParts(this.processor.getMessages());
+		const present = new Set(cards.map((part) => part.subagent.id));
+		for (const id of this.subagentHandles.keys()) if (!present.has(id)) this.subagentHandles.delete(id);
+		for (const part of cards) {
+			const id = part.subagent.id;
+			const existing = this.subagentHandles.get(id);
+			if (existing) {
+				if (part.subagent === existing) continue;
+				const { stop } = existing;
+				for (const key of Object.keys(existing)) Reflect.deleteProperty(existing, key);
+				Object.assign(existing, part.subagent, { stop });
+				part.subagent = existing;
+				continue;
+			}
+			const handle = {
+				...part.subagent,
+				stop: () => {
+					this.stopSubagent(id);
+				}
+			};
+			this.subagentHandles.set(id, handle);
+			part.subagent = handle;
+		}
+	}
+	stopSubagent(id) {
+		const handle = this.subagentHandles.get(id);
+		if (!handle || handle.status !== "running") return;
+		this.stoppedSubagentIds.add(id);
+		this.processor.processChunk({
+			type: EventType.SUBAGENT_ERROR,
+			subagentRunId: id,
+			message: "Stopped",
+			timestamp: Date.now()
+		});
+		this.syncSubagentHandles();
+		handle.status = "error";
+		handle.error = { message: "Stopped" };
+		this.abortController?.abort();
+	}
 	/**
 	* True when the last hydrate or older-page response said more messages exist.
 	*/