fix: resolve approval lookup after RUN_FINISHED and update smoke test state names

Fall back to toolCallToMessage map when resolving approval requests after
RUN_FINISHED clears activeMessageIds. Update smoke test to use current
tool call state names (input-complete, approval-responded, input-streaming).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: jherr

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

@@ -1142,11 +1142,15 @@ export class StreamProcessor {
         approval: { id: string; needsApproval: boolean }
       }
 
-      // Update the tool call part with approval state
-      if (messageId) {
+      // Resolve the message containing this tool call. After RUN_FINISHED,
+      // activeMessageIds is cleared, so fall back to the toolCallToMessage map
+      // which is populated during TOOL_CALL_START and preserved across finalize.
+      const resolvedMessageId =
+        messageId ?? this.toolCallToMessage.get(toolCallId)
+      if (resolvedMessageId) {
         this.messages = updateToolCallApproval(
           this.messages,
-          messageId,
+          resolvedMessageId,
           toolCallId,
           approval.id,
         )

packages/typescript/smoke-tests/e2e/src/routes/tools-test.tsx

@@ -186,8 +186,8 @@ function ToolsTestPage() {
       )
       const pendingCalls = allToolCalls.filter(
         (tc) =>
-          tc.state !== 'complete' &&
-          tc.state !== 'output-available' &&
+          tc.state !== 'input-complete' &&
+          tc.state !== 'approval-responded' &&
           tc.output === undefined &&
           !resultIds.has(tc.id),
       )
@@ -509,9 +509,10 @@ function ToolsTestPage() {
                         padding: '2px 6px',
                         borderRadius: '3px',
                         backgroundColor:
-                          tc.state === 'complete'
+                          tc.state === 'input-complete' ||
+                          tc.state === 'approval-responded'
                             ? '#28a745'
-                            : tc.state === 'streaming-arguments'
+                            : tc.state === 'input-streaming'
                               ? '#ffc107'
                               : tc.state === 'approval-requested'
                                 ? '#17a2b8'
@@ -572,8 +573,8 @@ function ToolsTestPage() {
         data-complete-tool-count={
           toolCalls.filter(
             (tc) =>
-              tc.state === 'complete' ||
-              tc.state === 'output-available' ||
+              tc.state === 'input-complete' ||
+              tc.state === 'approval-responded' ||
               tc.output !== undefined ||
               toolResultIds.has(tc.id),
           ).length