fix(ai): restore onFinish callback in ToolLoopAgent stream and generate methods

.changeset/fix-tool-loop-agent-on-finish.md

@@ -0,0 +1,5 @@
+---
+"ai": patch
+---
+
+fix(ai): restore onFinish callback in ToolLoopAgent stream and generate methods

packages/ai/src/agent/tool-loop-agent.test.ts

@@ -673,6 +673,21 @@ describe('ToolLoopAgent', () => {
         ]);
       });
     });
+
+    it('should invoke onFinish when generation completes', async () => {
+      let onFinishCalled = false;
+
+      const agent = new ToolLoopAgent({ model: mockModel });
+
+      await agent.generate({
+        prompt: 'Hello, world!',
+        onFinish: async () => {
+          onFinishCalled = true;
+        },
+      });
+
+      expect(onFinishCalled).toBe(true);
+    });
   });
 
   describe('stream', () => {
@@ -1196,6 +1211,23 @@ describe('ToolLoopAgent', () => {
         },
       ]);
     });
+
+    it('should invoke onFinish when stream completes', async () => {
+      let onFinishCalled = false;
+
+      const agent = new ToolLoopAgent({ model: mockModel });
+
+      const result = await agent.stream({
+        prompt: 'Hello, world!',
+        onFinish: async () => {
+          onFinishCalled = true;
+        },
+      });
+
+      await result.consumeStream();
+
+      expect(onFinishCalled).toBe(true);
+    });
   });
 
   describe('experimental_onStart', () => {

packages/ai/src/agent/tool-loop-agent.ts

@@ -218,6 +218,7 @@ export class ToolLoopAgent<
         onToolExecutionEnd,
       ),
       onStepFinish: mergeCallbacks(this.settings.onStepFinish, onStepFinish),
+      onFinish: mergeCallbacks(this.settings.onFinish, onFinish),
       onEnd: mergeCallbacks(this.settings.onEnd, onEnd),
     };
 
@@ -277,6 +278,7 @@ export class ToolLoopAgent<
         onToolExecutionEnd,
       ),
       onStepFinish: mergeCallbacks(this.settings.onStepFinish, onStepFinish),
+      onFinish: mergeCallbacks(this.settings.onFinish, onFinish),
       onEnd: mergeCallbacks(this.settings.onEnd, onEnd),
     };