@tanstack/ai-client 0.29.0 → 0.29.1

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

@@ -123,6 +123,12 @@ export declare class ChatClient<TTools extends ReadonlyArray<AnyClientTool> = an
     private continuationPending;
     private subscriptionAbortController;
     private processingResolve;
+    /**
+     * `connect()` adapters push the full HTTP body into the subscribe queue, then
+     * wait until that queue is idle. After `send()` returns, every chunk from this
+     * request has been processed. Subscribe/send sockets do not drain that way.
+     */
+    private connectionDrainsOnSend;
     private errorReportedGeneration;
     private streamGeneration;
     private continuationGeneration;

package/dist/esm/chat-client.js

@@ -22,6 +22,16 @@ function resolveTransport(transport) {
 	if (fetcher) return fetcherToConnectionAdapter(fetcher);
 	throw new Error("ChatClient: either `connection` or `fetcher` is required.");
 }
+function connectionDrainsOnSend(connection) {
+	return "connect" in connection;
+}
+function isIntermediateToolTurn(chunk) {
+	if (chunk.type !== "RUN_FINISHED") return false;
+	if (chunk.outcome?.type === "interrupt") return false;
+	const extra = chunk;
+	if (extra.finishReason !== void 0) return extra.finishReason === "tool_calls";
+	return tanstackMetadata(chunk)?.finishReason === "tool_calls";
+}
 function normalizeQueueOption(option) {
 	const base = {
 		whenBusy: "queue",
@@ -204,6 +214,12 @@ var ChatClient = class {
 	continuationPending = false;
 	subscriptionAbortController = null;
 	processingResolve = null;
+	/**
+	* `connect()` adapters push the full HTTP body into the subscribe queue, then
+	* wait until that queue is idle. After `send()` returns, every chunk from this
+	* request has been processed. Subscribe/send sockets do not drain that way.
+	*/
+	connectionDrainsOnSend = false;
 	errorReportedGeneration = null;
 	streamGeneration = 0;
 	continuationGeneration = 0;
@@ -239,7 +255,9 @@ var ChatClient = class {
 		this.byokProvider = options.byokProvider;
 		this.context = options.context;
 		this.queueConfig = normalizeQueueOption(options.queue);
-		this.connection = normalizeConnectionAdapter(resolveTransport(options));
+		const transport = resolveTransport(options);
+		this.connectionDrainsOnSend = connectionDrainsOnSend(transport);
+		this.connection = normalizeConnectionAdapter(transport);
 		this.clientToolsRef = { current: /* @__PURE__ */ new Map() };
 		if (options.tools) for (const tool of options.tools) this.clientToolsRef.current.set(tool.name, tool);
 		this.devtoolsBridge = (options.devtoolsBridgeFactory ?? createNoOpChatDevtoolsBridge)(this.buildDevtoolsBridgeOptions(options.devtools));
@@ -374,6 +392,7 @@ var ChatClient = class {
 				},
 				onCustomEvent: (eventType, data, context) => {
 					if (eventType === "memory:state") this.devtoolsBridge.recordMemoryState(data);
+					if (eventType === "skills:state") this.devtoolsBridge.recordSkillsState(data);
 					this.callbacksRef.current.onCustomEvent(eventType, data, context);
 				}
 			}
@@ -562,7 +581,8 @@ var ChatClient = class {
 			this.clearedStreamTracker.onSessionRunError();
 		}
 		this.setSessionGenerating(this.activeRunIds.size > 0);
-		if (options?.resolveProcessing !== false) this.resolveProcessing();
+		const skipProcessingResolve = chunk.type === "RUN_FINISHED" && isIntermediateToolTurn(chunk);
+		if (options?.resolveProcessing !== false && !skipProcessingResolve) this.resolveProcessing();
 	}
 	/**
 	* Track interrupt state off the stream's terminal events. A RUN_FINISHED with
@@ -1302,6 +1322,7 @@ var ChatClient = class {
 			this.devtoolsBridge.emitSnapshot();
 			await this.connection.send(messages, mergedBody, signal, runContext);
 			if (generation !== this.streamGeneration || signal.aborted) return false;
+			if (this.connectionDrainsOnSend) this.resolveProcessing();
 			await processingComplete;
 			if (generation !== this.streamGeneration) return false;
 			if (this.status === "error") {
@@ -1710,10 +1731,12 @@ var ChatClient = class {
 			this.resetSessionGenerating();
 			this.setIsSubscribed(false);
 			this.setConnectionStatus("disconnected");
-			this.connection = normalizeConnectionAdapter(resolveTransport({
+			const transport = resolveTransport({
 				connection: options.connection,
 				fetcher: options.fetcher
-			}));
+			});
+			this.connectionDrainsOnSend = connectionDrainsOnSend(transport);
+			this.connection = normalizeConnectionAdapter(transport);
 			if (wasSubscribed) this.subscribe();
 		}
 		if (options.body !== void 0) this.bodyOption = options.body;