fix(ai): consider output-error in tool call helper

This commit updates the `lastAssistantMessageIsCompleteWithToolCalls`
helper function to also consider tool parts with a state of
`output-error` a completed tool call.

Co-Authored-By: Lars Grammel <[email protected]>

Author: aron

packages/ai/src/ui/last-assistant-message-is-complete-with-tool-calls.test.ts

@@ -2,7 +2,7 @@ import { lastAssistantMessageIsCompleteWithToolCalls } from './last-assistant-me
 import { describe, it, expect } from 'vitest';
 
 describe('lastAssistantMessageIsCompleteWithToolCalls', () => {
-  it('should return false if the last step of a multi-step sequency only has text', () => {
+  it('should return false if the last step of a multi-step sequence only has text', () => {
     expect(
       lastAssistantMessageIsCompleteWithToolCalls({
         messages: [
@@ -61,6 +61,36 @@ describe('lastAssistantMessageIsCompleteWithToolCalls', () => {
     ).toBe(true);
   });
 
+  it('should return true when the tool has a output-error state', () => {
+    expect(
+      lastAssistantMessageIsCompleteWithToolCalls({
+        messages: [
+          {
+            id: '1',
+            role: 'assistant',
+            parts: [
+              { type: 'step-start' },
+              {
+                type: 'tool-getWeatherInformation',
+                toolCallId: 'call_6iy0GxZ9R4VDI5MKohXxV48y',
+                state: 'output-error',
+                input: {
+                  city: 'New York',
+                },
+                errorText: 'Unable to get weather information',
+              },
+              {
+                type: 'text',
+                text: 'The current weather in New York is windy.',
+                state: 'done',
+              },
+            ],
+          },
+        ],
+      }),
+    ).toBe(true);
+  });
+
   it('should return true when dynamic tool call is complete', () => {
     expect(
       lastAssistantMessageIsCompleteWithToolCalls({
@@ -137,7 +167,7 @@ describe('lastAssistantMessageIsCompleteWithToolCalls', () => {
     ).toBe(false);
   });
 
-  it('should return false when dynamic tool call has an error', () => {
+  it('should return true when dynamic tool call has an error', () => {
     expect(
       lastAssistantMessageIsCompleteWithToolCalls({
         messages: [
@@ -160,7 +190,7 @@ describe('lastAssistantMessageIsCompleteWithToolCalls', () => {
           },
         ],
       }),
-    ).toBe(false);
+    ).toBe(true);
   });
 
   it('should return true when mixing regular and dynamic tool calls and all are complete', () => {

packages/ai/src/ui/last-assistant-message-is-complete-with-tool-calls.ts

@@ -1,4 +1,4 @@
-import { isToolOrDynamicToolUIPart, UIMessage } from './ui-messages';
+import { isToolOrDynamicToolUIPart, type UIMessage } from './ui-messages';
 
 /**
 Check if the message is an assistant message with completed tool calls.
@@ -30,6 +30,9 @@ export function lastAssistantMessageIsCompleteWithToolCalls({
 
   return (
     lastStepToolInvocations.length > 0 &&
-    lastStepToolInvocations.every(part => part.state === 'output-available')
+    lastStepToolInvocations.every(
+      part =>
+        part.state === 'output-available' || part.state === 'output-error',
+    )
   );
 }