fix: unresponsive chat after client tool execution

.changeset/fix-silent-chat-continuation.md

@@ -0,0 +1,6 @@
+---
+"@tanstack/ai": patch
+"@tanstack/ai-client": patch
+---
+
+fix: Continue conversation after client tool execution

packages/typescript/ai-client/src/chat-client.ts

@@ -290,6 +290,7 @@ export class ChatClient {
     this.setIsLoading(true)
     this.setError(undefined)
     this.abortController = new AbortController()
+    let streamCompletedSuccessfully = false
 
     try {
       // Get model messages for the LLM
@@ -312,6 +313,7 @@ export class ChatClient {
       )
 
       await this.processStream(stream)
+      streamCompletedSuccessfully = true
     } catch (err) {
       if (err instanceof Error) {
         if (err.name === 'AbortError') {
@@ -323,6 +325,21 @@ export class ChatClient {
     } finally {
       this.abortController = null
       this.setIsLoading(false)
+
+      // Continue conversation if the stream ended with a tool result
+      if (streamCompletedSuccessfully) {
+        const messages = this.processor.getMessages()
+        const lastMessage = messages[messages.length - 1]
+        const lastPart = lastMessage?.parts[lastMessage.parts.length - 1]
+
+        if (lastPart?.type === 'tool-result' && this.shouldAutoSend()) {
+          try {
+            await this.continueFlow()
+          } catch (error) {
+            console.error('Failed to continue flow after tool result:', error)
+          }
+        }
+      }
     }
   }
 

packages/typescript/ai/src/stream/processor.ts

@@ -355,11 +355,21 @@ export class StreamProcessor {
 
     if (toolParts.length === 0) return true
 
+    // Check for server tool completions via tool-result parts
+    const toolResultParts = lastAssistant.parts.filter(
+      (p): p is Extract<typeof p, { type: 'tool-result' }> =>
+        p.type === 'tool-result',
+    )
+    const completedToolCallIds = new Set(
+      toolResultParts.map((p) => p.toolCallId),
+    )
+
     // All tool calls must be in a terminal state
     return toolParts.every(
       (part) =>
         part.state === 'approval-responded' ||
-        (part.output !== undefined && !part.approval),
+        (part.output !== undefined && !part.approval) ||
+        completedToolCallIds.has(part.id),
     )
   }