diff --git a/packages/ai/src/chatTransport.test.ts b/packages/ai/src/chatTransport.test.ts index 3782dbe8d..89f67451d 100644 --- a/packages/ai/src/chatTransport.test.ts +++ b/packages/ai/src/chatTransport.test.ts @@ -809,7 +809,7 @@ describe("TriggerChatTransport", function () { onTriggeredRun: async function onTriggeredRun(state) { await sleep(1); observedRunId = state.runId; - observedState = { ...state }; + observedState = state; callbackCompleted = true; }, }); diff --git a/packages/ai/src/chatTransport.ts b/packages/ai/src/chatTransport.ts index fec8c75f0..d80b2fdeb 100644 --- a/packages/ai/src/chatTransport.ts +++ b/packages/ai/src/chatTransport.ts @@ -177,7 +177,9 @@ export class TriggerChatTransport< if (this.onTriggeredRun) { try { - await this.onTriggeredRun(runState); + await this.onTriggeredRun({ + ...runState, + }); } catch { // Ignore callback errors so chat streaming can continue. }