@copilotkit/runtime 1.66.2 → 1.66.3

package/dist/agent/converters/aisdk.cjs

@@ -1,5 +1,6 @@
 require("reflect-metadata");
 const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+const require_state_delta = require('../state-delta.cjs');
 let _copilotkit_shared = require("@copilotkit/shared");
 let _ag_ui_client = require("@ag-ui/client");
 
@@ -18,10 +19,11 @@ let _ag_ui_client = require("@ag-ui/client");
 * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).
 * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
 */
-async function* convertAISDKStream(fullStream, abortSignal, pendingInterrupts) {
+async function* convertAISDKStream(fullStream, abortSignal, pendingInterrupts, initialState) {
 	let messageId = (0, _copilotkit_shared.randomUUID)();
 	let reasoningMessageId = (0, _copilotkit_shared.randomUUID)();
 	let isInReasoning = false;
+	const normalizeStateEvent = require_state_delta.createStateEventNormalizer(initialState);
 	const toolCallStates = /* @__PURE__ */ new Map();
 	const ensureToolCallState = (toolCallId) => {
 		let state = toolCallStates.get(toolCallId);
@@ -220,16 +222,22 @@ async function* convertAISDKStream(fullStream, abortSignal, pendingInterrupts) {
 					toolCallStates.delete(p.toolCallId);
 					if (toolName === "AGUISendStateSnapshot" && toolResult && typeof toolResult === "object") {
 						const snapshot = toolResult.snapshot;
-						if (snapshot !== void 0) yield {
-							type: _ag_ui_client.EventType.STATE_SNAPSHOT,
-							snapshot
-						};
+						if (snapshot !== void 0) {
+							const stateSnapshotEvent = {
+								type: _ag_ui_client.EventType.STATE_SNAPSHOT,
+								snapshot
+							};
+							for (const event of normalizeStateEvent(stateSnapshotEvent)) yield event;
+						}
 					} else if (toolName === "AGUISendStateDelta" && toolResult && typeof toolResult === "object") {
 						const delta = toolResult.delta;
-						if (delta !== void 0) yield {
-							type: _ag_ui_client.EventType.STATE_DELTA,
-							delta
-						};
+						if (delta !== void 0) {
+							const stateDeltaEvent = {
+								type: _ag_ui_client.EventType.STATE_DELTA,
+								delta
+							};
+							for (const event of normalizeStateEvent(stateDeltaEvent)) yield event;
+						}
 					}
 					let serializedResult;
 					try {

package/dist/agent/converters/aisdk.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"aisdk.cjs","names":["EventType"],"sources":["../../../src/agent/converters/aisdk.ts"],"sourcesContent":["import type {\n  BaseEvent,\n  Interrupt,\n  ReasoningEndEvent,\n  ReasoningMessageContentEvent,\n  ReasoningMessageEndEvent,\n  ReasoningMessageStartEvent,\n  ReasoningStartEvent,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n  StateSnapshotEvent,\n  StateDeltaEvent,\n} from \"@ag-ui/client\";\nimport { EventType } from \"@ag-ui/client\";\nimport { randomUUID } from \"@copilotkit/shared\";\n\n/**\n * Converts an AI SDK `fullStream` into AG-UI `BaseEvent` objects.\n *\n * This is a pure converter — it does NOT emit lifecycle events\n * (RUN_STARTED / RUN_FINISHED / RUN_ERROR). The caller (Agent class)\n * is responsible for those.\n *\n * Terminal stream events (finish, error, abort) cause the generator to\n * return so the caller can handle lifecycle appropriately.\n *\n * `pendingInterrupts`, when provided, is filled with one AG-UI Interrupt per\n * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).\n * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.\n */\nexport async function* convertAISDKStream(\n  fullStream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n  pendingInterrupts?: Interrupt[],\n): AsyncGenerator<BaseEvent> {\n  let messageId = randomUUID();\n  let reasoningMessageId = randomUUID();\n  let isInReasoning = false;\n\n  const toolCallStates = new Map<\n    string,\n    {\n      started: boolean;\n      hasArgsDelta: boolean;\n      ended: boolean;\n      toolName?: string;\n    }\n  >();\n\n  const ensureToolCallState = (toolCallId: string) => {\n    let state = toolCallStates.get(toolCallId);\n    if (!state) {\n      state = { started: false, hasArgsDelta: false, ended: false };\n      toolCallStates.set(toolCallId, state);\n    }\n    return state;\n  };\n\n  /**\n   * Auto-close an open reasoning lifecycle.\n   * Some AI SDK providers (notably @ai-sdk/anthropic) never emit \"reasoning-end\",\n   * which leaves downstream state machines stuck. This helper emits the\n   * missing REASONING_MESSAGE_END + REASONING_END events so the stream\n   * can transition to text, tool-call, or finish phases.\n   */\n  function* closeReasoningIfOpen(): Generator<BaseEvent> {\n    if (!isInReasoning) return;\n    isInReasoning = false;\n    const reasoningMsgEnd: ReasoningMessageEndEvent = {\n      type: EventType.REASONING_MESSAGE_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningMsgEnd;\n    const reasoningEnd: ReasoningEndEvent = {\n      type: EventType.REASONING_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningEnd;\n  }\n\n  try {\n    for await (const part of fullStream) {\n      const p = part as Record<string, unknown>;\n\n      // Close any open reasoning lifecycle on every event except\n      // reasoning-delta, which arrives mid-block and must not interrupt it.\n      if (p.type !== \"reasoning-delta\") {\n        yield* closeReasoningIfOpen();\n      }\n\n      switch (p.type) {\n        case \"abort\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"reasoning-start\": {\n          // Use SDK-provided id, or generate a fresh UUID if id is falsy/\"0\"\n          // to prevent consecutive reasoning blocks from sharing a messageId\n          const providedId = \"id\" in p ? p.id : undefined;\n          reasoningMessageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          const reasoningStartEvent: ReasoningStartEvent = {\n            type: EventType.REASONING_START,\n            messageId: reasoningMessageId,\n          };\n          yield reasoningStartEvent;\n          const reasoningMessageStart: ReasoningMessageStartEvent = {\n            type: EventType.REASONING_MESSAGE_START,\n            messageId: reasoningMessageId,\n            role: \"reasoning\",\n          };\n          yield reasoningMessageStart;\n          isInReasoning = true;\n          break;\n        }\n\n        case \"reasoning-delta\": {\n          const delta = (p.text as string) ?? \"\";\n          if (!delta) break; // skip — @ag-ui/core schema requires delta to be non-empty\n          const reasoningDeltaEvent: ReasoningMessageContentEvent = {\n            type: EventType.REASONING_MESSAGE_CONTENT,\n            messageId: reasoningMessageId,\n            delta,\n          };\n          yield reasoningDeltaEvent;\n          break;\n        }\n\n        case \"reasoning-end\": {\n          // closeReasoningIfOpen() already called before the switch — no-op here\n          // if the SDK never emits this event (e.g. @ai-sdk/anthropic).\n          break;\n        }\n\n        case \"tool-input-start\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = p.toolName as string;\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n          break;\n        }\n\n        case \"tool-input-delta\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.hasArgsDelta = true;\n          const argsEvent: ToolCallArgsEvent = {\n            type: EventType.TOOL_CALL_ARGS,\n            toolCallId,\n            delta: p.delta as string,\n          };\n          yield argsEvent;\n          break;\n        }\n\n        case \"tool-input-end\": {\n          // No direct event – the subsequent \"tool-call\" part marks completion.\n          break;\n        }\n\n        case \"text-start\": {\n          // New text message starting - use the SDK-provided id\n          // Use randomUUID() if part.id is falsy or \"0\" to prevent message merging issues\n          const providedId = \"id\" in p ? p.id : undefined;\n          messageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          break;\n        }\n\n        case \"text-delta\": {\n          // AI SDK text-delta events use 'text' (not 'delta')\n          const textDelta = \"text\" in p ? (p.text as string) : \"\";\n          const textEvent: TextMessageChunkEvent = {\n            type: EventType.TEXT_MESSAGE_CHUNK,\n            role: \"assistant\",\n            messageId,\n            delta: textDelta,\n          };\n          yield textEvent;\n          break;\n        }\n\n        case \"tool-call\": {\n          const toolCallId = p.toolCallId as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = (p.toolName as string) ?? state.toolName;\n\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n\n          if (!state.hasArgsDelta && \"input\" in p && p.input !== undefined) {\n            let serializedInput = \"\";\n            if (typeof p.input === \"string\") {\n              serializedInput = p.input;\n            } else {\n              try {\n                serializedInput = JSON.stringify(p.input);\n              } catch {\n                serializedInput = String(p.input);\n              }\n            }\n\n            if (serializedInput.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serializedInput,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n          break;\n        }\n\n        case \"tool-approval-request\": {\n          // AI SDK native human-in-the-loop: a tool declared `needsApproval`\n          // was called. The preceding \"tool-call\" part already streamed the\n          // tool name + args; this part carries the toolCallId to pause on.\n          // (Some shapes nest the call under `toolCall` — read both.)\n          const nested = (p.toolCall ?? {}) as {\n            toolCallId?: string;\n            toolName?: string;\n            input?: unknown;\n          };\n          const toolCallId =\n            (p.toolCallId as string | undefined) ?? nested.toolCallId;\n          if (!toolCallId) {\n            throw new Error(\n              \"AI SDK tool-approval-request is missing toolCallId\",\n            );\n          }\n          const state = ensureToolCallState(toolCallId);\n          const toolName = state.toolName ?? nested.toolName;\n\n          // Defensive: emit the tool-call lifecycle if the \"tool-call\" part\n          // didn't precede this one, so the assistant tool-call is in history\n          // for the resume run (which keys the injected result by toolCallId).\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: toolName ?? \"\",\n            };\n            yield startEvent;\n          }\n          if (!state.hasArgsDelta && nested.input !== undefined) {\n            let serialized = \"\";\n            try {\n              serialized =\n                typeof nested.input === \"string\"\n                  ? nested.input\n                  : JSON.stringify(nested.input);\n            } catch {\n              serialized = String(nested.input);\n            }\n            if (serialized.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serialized,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n\n          pendingInterrupts?.push({\n            id: toolCallId,\n            toolCallId,\n            reason: \"tool_approval\",\n            message: toolName ? `Approve \"${toolName}\"?` : undefined,\n            ...(toolName ? { metadata: { toolName } } : {}),\n          });\n          break;\n        }\n\n        case \"tool-result\": {\n          // AI SDK tool-result uses \"output\"; older versions used \"result\" — check both\n          const toolResult =\n            \"output\" in p ? p.output : \"result\" in p ? p.result : null;\n          const toolName = \"toolName\" in p ? (p.toolName as string) : \"\";\n          toolCallStates.delete(p.toolCallId as string);\n\n          // Check if this is a state update tool\n          if (\n            toolName === \"AGUISendStateSnapshot\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const snapshot = (toolResult as Record<string, unknown>).snapshot;\n            if (snapshot !== undefined) {\n              const stateSnapshotEvent: StateSnapshotEvent = {\n                type: EventType.STATE_SNAPSHOT,\n                snapshot,\n              };\n              yield stateSnapshotEvent;\n            }\n          } else if (\n            toolName === \"AGUISendStateDelta\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const delta = (toolResult as { delta?: StateDeltaEvent[\"delta\"] })\n              .delta;\n            if (delta !== undefined) {\n              const stateDeltaEvent: StateDeltaEvent = {\n                type: EventType.STATE_DELTA,\n                delta,\n              };\n              yield stateDeltaEvent;\n            }\n          }\n\n          // Always emit the tool result event for the LLM\n          let serializedResult: string;\n          try {\n            serializedResult = JSON.stringify(toolResult);\n          } catch {\n            serializedResult = `[Unserializable tool result from ${toolName || \"unknown tool\"}]`;\n          }\n          const resultEvent: ToolCallResultEvent = {\n            type: EventType.TOOL_CALL_RESULT,\n            role: \"tool\",\n            messageId: randomUUID(),\n            toolCallId: p.toolCallId as string,\n            content: serializedResult,\n          };\n          yield resultEvent;\n          break;\n        }\n\n        case \"finish\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"error\": {\n          if (abortSignal.aborted) {\n            return;\n          }\n          // Re-throw so the caller can emit RUN_ERROR\n          const err = p.error ?? p.message ?? p.cause;\n          if (err instanceof Error) throw err;\n          throw new Error(\n            typeof err === \"string\"\n              ? err\n              : `AI SDK stream error: ${JSON.stringify(p)}`,\n          );\n        }\n\n        default:\n          // Unknown event types are silently ignored\n          break;\n      }\n    }\n  } finally {\n    // Always close reasoning on exit (normal or exceptional)\n    yield* closeReasoningIfOpen();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAiCA,gBAAuB,mBACrB,YACA,aACA,mBAC2B;CAC3B,IAAI,gDAAwB;CAC5B,IAAI,yDAAiC;CACrC,IAAI,gBAAgB;CAEpB,MAAM,iCAAiB,IAAI,KAQxB;CAEH,MAAM,uBAAuB,eAAuB;EAClD,IAAI,QAAQ,eAAe,IAAI,WAAW;AAC1C,MAAI,CAAC,OAAO;AACV,WAAQ;IAAE,SAAS;IAAO,cAAc;IAAO,OAAO;IAAO;AAC7D,kBAAe,IAAI,YAAY,MAAM;;AAEvC,SAAO;;;;;;;;;CAUT,UAAU,uBAA6C;AACrD,MAAI,CAAC,cAAe;AACpB,kBAAgB;AAKhB,QAJkD;GAChD,MAAMA,wBAAU;GAChB,WAAW;GACZ;AAMD,QAJwC;GACtC,MAAMA,wBAAU;GAChB,WAAW;GACZ;;AAIH,KAAI;AACF,aAAW,MAAM,QAAQ,YAAY;GACnC,MAAM,IAAI;AAIV,OAAI,EAAE,SAAS,kBACb,QAAO,sBAAsB;AAG/B,WAAQ,EAAE,MAAV;IACE,KAAK,QAEH;IAGF,KAAK,mBAAmB;KAGtB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,0BACE,cAAc,eAAe,MACxB,iDACW;AAKlB,WAJiD;MAC/C,MAAMA,wBAAU;MAChB,WAAW;MACZ;AAOD,WAL0D;MACxD,MAAMA,wBAAU;MAChB,WAAW;MACX,MAAM;MACP;AAED,qBAAgB;AAChB;;IAGF,KAAK,mBAAmB;KACtB,MAAM,QAAS,EAAE,QAAmB;AACpC,SAAI,CAAC,MAAO;AAMZ,WAL0D;MACxD,MAAMA,wBAAU;MAChB,WAAW;MACX;MACD;AAED;;IAGF,KAAK,gBAGH;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAW,EAAE;AACnB,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAMA,wBAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAGH;;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,eAAe;AAMrB,WALqC;MACnC,MAAMA,wBAAU;MAChB;MACA,OAAO,EAAE;MACV;AAED;;IAGF,KAAK,iBAEH;IAGF,KAAK,cAAc;KAGjB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,iBACE,cAAc,eAAe,MACxB,iDACW;AAClB;;IAGF,KAAK,cAAc;KAEjB,MAAM,YAAY,UAAU,IAAK,EAAE,OAAkB;AAOrD,WANyC;MACvC,MAAMA,wBAAU;MAChB,MAAM;MACN;MACA,OAAO;MACR;AAED;;IAGF,KAAK,aAAa;KAChB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAY,EAAE,YAAuB,MAAM;AAEjD,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAMA,wBAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAIH,SAAI,CAAC,MAAM,gBAAgB,WAAW,KAAK,EAAE,UAAU,QAAW;MAChE,IAAI,kBAAkB;AACtB,UAAI,OAAO,EAAE,UAAU,SACrB,mBAAkB,EAAE;UAEpB,KAAI;AACF,yBAAkB,KAAK,UAAU,EAAE,MAAM;cACnC;AACN,yBAAkB,OAAO,EAAE,MAAM;;AAIrC,UAAI,gBAAgB,SAAS,GAAG;AAM9B,aALqC;QACnC,MAAMA,wBAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAIzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAMA,wBAAU;OAChB;OACD;;AAGH;;IAGF,KAAK,yBAAyB;KAK5B,MAAM,SAAU,EAAE,YAAY,EAAE;KAKhC,MAAM,aACH,EAAE,cAAqC,OAAO;AACjD,SAAI,CAAC,WACH,OAAM,IAAI,MACR,qDACD;KAEH,MAAM,QAAQ,oBAAoB,WAAW;KAC7C,MAAM,WAAW,MAAM,YAAY,OAAO;AAK1C,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAMA,wBAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,YAAY;OAC3B;;AAGH,SAAI,CAAC,MAAM,gBAAgB,OAAO,UAAU,QAAW;MACrD,IAAI,aAAa;AACjB,UAAI;AACF,oBACE,OAAO,OAAO,UAAU,WACpB,OAAO,QACP,KAAK,UAAU,OAAO,MAAM;cAC5B;AACN,oBAAa,OAAO,OAAO,MAAM;;AAEnC,UAAI,WAAW,SAAS,GAAG;AAMzB,aALqC;QACnC,MAAMA,wBAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAGzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAMA,wBAAU;OAChB;OACD;;AAIH,wBAAmB,KAAK;MACtB,IAAI;MACJ;MACA,QAAQ;MACR,SAAS,WAAW,YAAY,SAAS,MAAM;MAC/C,GAAI,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE;MAC/C,CAAC;AACF;;IAGF,KAAK,eAAe;KAElB,MAAM,aACJ,YAAY,IAAI,EAAE,SAAS,YAAY,IAAI,EAAE,SAAS;KACxD,MAAM,WAAW,cAAc,IAAK,EAAE,WAAsB;AAC5D,oBAAe,OAAO,EAAE,WAAqB;AAG7C,SACE,aAAa,2BACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,WAAY,WAAuC;AACzD,UAAI,aAAa,OAKf,OAJ+C;OAC7C,MAAMA,wBAAU;OAChB;OACD;gBAIH,aAAa,wBACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,QAAS,WACZ;AACH,UAAI,UAAU,OAKZ,OAJyC;OACvC,MAAMA,wBAAU;OAChB;OACD;;KAML,IAAI;AACJ,SAAI;AACF,yBAAmB,KAAK,UAAU,WAAW;aACvC;AACN,yBAAmB,oCAAoC,YAAY,eAAe;;AASpF,WAPyC;MACvC,MAAMA,wBAAU;MAChB,MAAM;MACN,+CAAuB;MACvB,YAAY,EAAE;MACd,SAAS;MACV;AAED;;IAGF,KAAK,SAEH;IAGF,KAAK,SAAS;AACZ,SAAI,YAAY,QACd;KAGF,MAAM,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE;AACtC,SAAI,eAAe,MAAO,OAAM;AAChC,WAAM,IAAI,MACR,OAAO,QAAQ,WACX,MACA,wBAAwB,KAAK,UAAU,EAAE,GAC9C;;IAGH,QAEE;;;WAGE;AAER,SAAO,sBAAsB"}
+{"version":3,"file":"aisdk.cjs","names":["createStateEventNormalizer","EventType"],"sources":["../../../src/agent/converters/aisdk.ts"],"sourcesContent":["import type {\n  BaseEvent,\n  Interrupt,\n  ReasoningEndEvent,\n  ReasoningMessageContentEvent,\n  ReasoningMessageEndEvent,\n  ReasoningMessageStartEvent,\n  ReasoningStartEvent,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n  StateSnapshotEvent,\n  StateDeltaEvent,\n} from \"@ag-ui/client\";\nimport { EventType } from \"@ag-ui/client\";\nimport { randomUUID } from \"@copilotkit/shared\";\nimport { createStateEventNormalizer } from \"../state-delta\";\n\n/**\n * Converts an AI SDK `fullStream` into AG-UI `BaseEvent` objects.\n *\n * This is a pure converter — it does NOT emit lifecycle events\n * (RUN_STARTED / RUN_FINISHED / RUN_ERROR). The caller (Agent class)\n * is responsible for those.\n *\n * Terminal stream events (finish, error, abort) cause the generator to\n * return so the caller can handle lifecycle appropriately.\n *\n * `pendingInterrupts`, when provided, is filled with one AG-UI Interrupt per\n * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).\n * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.\n */\nexport async function* convertAISDKStream(\n  fullStream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n  pendingInterrupts?: Interrupt[],\n  initialState?: unknown,\n): AsyncGenerator<BaseEvent> {\n  let messageId = randomUUID();\n  let reasoningMessageId = randomUUID();\n  let isInReasoning = false;\n  const normalizeStateEvent = createStateEventNormalizer(initialState);\n\n  const toolCallStates = new Map<\n    string,\n    {\n      started: boolean;\n      hasArgsDelta: boolean;\n      ended: boolean;\n      toolName?: string;\n    }\n  >();\n\n  const ensureToolCallState = (toolCallId: string) => {\n    let state = toolCallStates.get(toolCallId);\n    if (!state) {\n      state = { started: false, hasArgsDelta: false, ended: false };\n      toolCallStates.set(toolCallId, state);\n    }\n    return state;\n  };\n\n  /**\n   * Auto-close an open reasoning lifecycle.\n   * Some AI SDK providers (notably @ai-sdk/anthropic) never emit \"reasoning-end\",\n   * which leaves downstream state machines stuck. This helper emits the\n   * missing REASONING_MESSAGE_END + REASONING_END events so the stream\n   * can transition to text, tool-call, or finish phases.\n   */\n  function* closeReasoningIfOpen(): Generator<BaseEvent> {\n    if (!isInReasoning) return;\n    isInReasoning = false;\n    const reasoningMsgEnd: ReasoningMessageEndEvent = {\n      type: EventType.REASONING_MESSAGE_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningMsgEnd;\n    const reasoningEnd: ReasoningEndEvent = {\n      type: EventType.REASONING_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningEnd;\n  }\n\n  try {\n    for await (const part of fullStream) {\n      const p = part as Record<string, unknown>;\n\n      // Close any open reasoning lifecycle on every event except\n      // reasoning-delta, which arrives mid-block and must not interrupt it.\n      if (p.type !== \"reasoning-delta\") {\n        yield* closeReasoningIfOpen();\n      }\n\n      switch (p.type) {\n        case \"abort\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"reasoning-start\": {\n          // Use SDK-provided id, or generate a fresh UUID if id is falsy/\"0\"\n          // to prevent consecutive reasoning blocks from sharing a messageId\n          const providedId = \"id\" in p ? p.id : undefined;\n          reasoningMessageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          const reasoningStartEvent: ReasoningStartEvent = {\n            type: EventType.REASONING_START,\n            messageId: reasoningMessageId,\n          };\n          yield reasoningStartEvent;\n          const reasoningMessageStart: ReasoningMessageStartEvent = {\n            type: EventType.REASONING_MESSAGE_START,\n            messageId: reasoningMessageId,\n            role: \"reasoning\",\n          };\n          yield reasoningMessageStart;\n          isInReasoning = true;\n          break;\n        }\n\n        case \"reasoning-delta\": {\n          const delta = (p.text as string) ?? \"\";\n          if (!delta) break; // skip — @ag-ui/core schema requires delta to be non-empty\n          const reasoningDeltaEvent: ReasoningMessageContentEvent = {\n            type: EventType.REASONING_MESSAGE_CONTENT,\n            messageId: reasoningMessageId,\n            delta,\n          };\n          yield reasoningDeltaEvent;\n          break;\n        }\n\n        case \"reasoning-end\": {\n          // closeReasoningIfOpen() already called before the switch — no-op here\n          // if the SDK never emits this event (e.g. @ai-sdk/anthropic).\n          break;\n        }\n\n        case \"tool-input-start\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = p.toolName as string;\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n          break;\n        }\n\n        case \"tool-input-delta\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.hasArgsDelta = true;\n          const argsEvent: ToolCallArgsEvent = {\n            type: EventType.TOOL_CALL_ARGS,\n            toolCallId,\n            delta: p.delta as string,\n          };\n          yield argsEvent;\n          break;\n        }\n\n        case \"tool-input-end\": {\n          // No direct event – the subsequent \"tool-call\" part marks completion.\n          break;\n        }\n\n        case \"text-start\": {\n          // New text message starting - use the SDK-provided id\n          // Use randomUUID() if part.id is falsy or \"0\" to prevent message merging issues\n          const providedId = \"id\" in p ? p.id : undefined;\n          messageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          break;\n        }\n\n        case \"text-delta\": {\n          // AI SDK text-delta events use 'text' (not 'delta')\n          const textDelta = \"text\" in p ? (p.text as string) : \"\";\n          const textEvent: TextMessageChunkEvent = {\n            type: EventType.TEXT_MESSAGE_CHUNK,\n            role: \"assistant\",\n            messageId,\n            delta: textDelta,\n          };\n          yield textEvent;\n          break;\n        }\n\n        case \"tool-call\": {\n          const toolCallId = p.toolCallId as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = (p.toolName as string) ?? state.toolName;\n\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n\n          if (!state.hasArgsDelta && \"input\" in p && p.input !== undefined) {\n            let serializedInput = \"\";\n            if (typeof p.input === \"string\") {\n              serializedInput = p.input;\n            } else {\n              try {\n                serializedInput = JSON.stringify(p.input);\n              } catch {\n                serializedInput = String(p.input);\n              }\n            }\n\n            if (serializedInput.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serializedInput,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n          break;\n        }\n\n        case \"tool-approval-request\": {\n          // AI SDK native human-in-the-loop: a tool declared `needsApproval`\n          // was called. The preceding \"tool-call\" part already streamed the\n          // tool name + args; this part carries the toolCallId to pause on.\n          // (Some shapes nest the call under `toolCall` — read both.)\n          const nested = (p.toolCall ?? {}) as {\n            toolCallId?: string;\n            toolName?: string;\n            input?: unknown;\n          };\n          const toolCallId =\n            (p.toolCallId as string | undefined) ?? nested.toolCallId;\n          if (!toolCallId) {\n            throw new Error(\n              \"AI SDK tool-approval-request is missing toolCallId\",\n            );\n          }\n          const state = ensureToolCallState(toolCallId);\n          const toolName = state.toolName ?? nested.toolName;\n\n          // Defensive: emit the tool-call lifecycle if the \"tool-call\" part\n          // didn't precede this one, so the assistant tool-call is in history\n          // for the resume run (which keys the injected result by toolCallId).\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: toolName ?? \"\",\n            };\n            yield startEvent;\n          }\n          if (!state.hasArgsDelta && nested.input !== undefined) {\n            let serialized = \"\";\n            try {\n              serialized =\n                typeof nested.input === \"string\"\n                  ? nested.input\n                  : JSON.stringify(nested.input);\n            } catch {\n              serialized = String(nested.input);\n            }\n            if (serialized.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serialized,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n\n          pendingInterrupts?.push({\n            id: toolCallId,\n            toolCallId,\n            reason: \"tool_approval\",\n            message: toolName ? `Approve \"${toolName}\"?` : undefined,\n            ...(toolName ? { metadata: { toolName } } : {}),\n          });\n          break;\n        }\n\n        case \"tool-result\": {\n          // AI SDK tool-result uses \"output\"; older versions used \"result\" — check both\n          const toolResult =\n            \"output\" in p ? p.output : \"result\" in p ? p.result : null;\n          const toolName = \"toolName\" in p ? (p.toolName as string) : \"\";\n          toolCallStates.delete(p.toolCallId as string);\n\n          // Check if this is a state update tool\n          if (\n            toolName === \"AGUISendStateSnapshot\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const snapshot = (toolResult as Record<string, unknown>).snapshot;\n            if (snapshot !== undefined) {\n              const stateSnapshotEvent: StateSnapshotEvent = {\n                type: EventType.STATE_SNAPSHOT,\n                snapshot,\n              };\n              for (const event of normalizeStateEvent(stateSnapshotEvent)) {\n                yield event;\n              }\n            }\n          } else if (\n            toolName === \"AGUISendStateDelta\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const delta = (toolResult as { delta?: StateDeltaEvent[\"delta\"] })\n              .delta;\n            if (delta !== undefined) {\n              const stateDeltaEvent: StateDeltaEvent = {\n                type: EventType.STATE_DELTA,\n                delta,\n              };\n              for (const event of normalizeStateEvent(stateDeltaEvent)) {\n                yield event;\n              }\n            }\n          }\n\n          // Always emit the tool result event for the LLM\n          let serializedResult: string;\n          try {\n            serializedResult = JSON.stringify(toolResult);\n          } catch {\n            serializedResult = `[Unserializable tool result from ${toolName || \"unknown tool\"}]`;\n          }\n          const resultEvent: ToolCallResultEvent = {\n            type: EventType.TOOL_CALL_RESULT,\n            role: \"tool\",\n            messageId: randomUUID(),\n            toolCallId: p.toolCallId as string,\n            content: serializedResult,\n          };\n          yield resultEvent;\n          break;\n        }\n\n        case \"finish\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"error\": {\n          if (abortSignal.aborted) {\n            return;\n          }\n          // Re-throw so the caller can emit RUN_ERROR\n          const err = p.error ?? p.message ?? p.cause;\n          if (err instanceof Error) throw err;\n          throw new Error(\n            typeof err === \"string\"\n              ? err\n              : `AI SDK stream error: ${JSON.stringify(p)}`,\n          );\n        }\n\n        default:\n          // Unknown event types are silently ignored\n          break;\n      }\n    }\n  } finally {\n    // Always close reasoning on exit (normal or exceptional)\n    yield* closeReasoningIfOpen();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAkCA,gBAAuB,mBACrB,YACA,aACA,mBACA,cAC2B;CAC3B,IAAI,gDAAwB;CAC5B,IAAI,yDAAiC;CACrC,IAAI,gBAAgB;CACpB,MAAM,sBAAsBA,+CAA2B,aAAa;CAEpE,MAAM,iCAAiB,IAAI,KAQxB;CAEH,MAAM,uBAAuB,eAAuB;EAClD,IAAI,QAAQ,eAAe,IAAI,WAAW;AAC1C,MAAI,CAAC,OAAO;AACV,WAAQ;IAAE,SAAS;IAAO,cAAc;IAAO,OAAO;IAAO;AAC7D,kBAAe,IAAI,YAAY,MAAM;;AAEvC,SAAO;;;;;;;;;CAUT,UAAU,uBAA6C;AACrD,MAAI,CAAC,cAAe;AACpB,kBAAgB;AAKhB,QAJkD;GAChD,MAAMC,wBAAU;GAChB,WAAW;GACZ;AAMD,QAJwC;GACtC,MAAMA,wBAAU;GAChB,WAAW;GACZ;;AAIH,KAAI;AACF,aAAW,MAAM,QAAQ,YAAY;GACnC,MAAM,IAAI;AAIV,OAAI,EAAE,SAAS,kBACb,QAAO,sBAAsB;AAG/B,WAAQ,EAAE,MAAV;IACE,KAAK,QAEH;IAGF,KAAK,mBAAmB;KAGtB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,0BACE,cAAc,eAAe,MACxB,iDACW;AAKlB,WAJiD;MAC/C,MAAMA,wBAAU;MAChB,WAAW;MACZ;AAOD,WAL0D;MACxD,MAAMA,wBAAU;MAChB,WAAW;MACX,MAAM;MACP;AAED,qBAAgB;AAChB;;IAGF,KAAK,mBAAmB;KACtB,MAAM,QAAS,EAAE,QAAmB;AACpC,SAAI,CAAC,MAAO;AAMZ,WAL0D;MACxD,MAAMA,wBAAU;MAChB,WAAW;MACX;MACD;AAED;;IAGF,KAAK,gBAGH;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAW,EAAE;AACnB,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAMA,wBAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAGH;;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,eAAe;AAMrB,WALqC;MACnC,MAAMA,wBAAU;MAChB;MACA,OAAO,EAAE;MACV;AAED;;IAGF,KAAK,iBAEH;IAGF,KAAK,cAAc;KAGjB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,iBACE,cAAc,eAAe,MACxB,iDACW;AAClB;;IAGF,KAAK,cAAc;KAEjB,MAAM,YAAY,UAAU,IAAK,EAAE,OAAkB;AAOrD,WANyC;MACvC,MAAMA,wBAAU;MAChB,MAAM;MACN;MACA,OAAO;MACR;AAED;;IAGF,KAAK,aAAa;KAChB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAY,EAAE,YAAuB,MAAM;AAEjD,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAMA,wBAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAIH,SAAI,CAAC,MAAM,gBAAgB,WAAW,KAAK,EAAE,UAAU,QAAW;MAChE,IAAI,kBAAkB;AACtB,UAAI,OAAO,EAAE,UAAU,SACrB,mBAAkB,EAAE;UAEpB,KAAI;AACF,yBAAkB,KAAK,UAAU,EAAE,MAAM;cACnC;AACN,yBAAkB,OAAO,EAAE,MAAM;;AAIrC,UAAI,gBAAgB,SAAS,GAAG;AAM9B,aALqC;QACnC,MAAMA,wBAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAIzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAMA,wBAAU;OAChB;OACD;;AAGH;;IAGF,KAAK,yBAAyB;KAK5B,MAAM,SAAU,EAAE,YAAY,EAAE;KAKhC,MAAM,aACH,EAAE,cAAqC,OAAO;AACjD,SAAI,CAAC,WACH,OAAM,IAAI,MACR,qDACD;KAEH,MAAM,QAAQ,oBAAoB,WAAW;KAC7C,MAAM,WAAW,MAAM,YAAY,OAAO;AAK1C,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAMA,wBAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,YAAY;OAC3B;;AAGH,SAAI,CAAC,MAAM,gBAAgB,OAAO,UAAU,QAAW;MACrD,IAAI,aAAa;AACjB,UAAI;AACF,oBACE,OAAO,OAAO,UAAU,WACpB,OAAO,QACP,KAAK,UAAU,OAAO,MAAM;cAC5B;AACN,oBAAa,OAAO,OAAO,MAAM;;AAEnC,UAAI,WAAW,SAAS,GAAG;AAMzB,aALqC;QACnC,MAAMA,wBAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAGzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAMA,wBAAU;OAChB;OACD;;AAIH,wBAAmB,KAAK;MACtB,IAAI;MACJ;MACA,QAAQ;MACR,SAAS,WAAW,YAAY,SAAS,MAAM;MAC/C,GAAI,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE;MAC/C,CAAC;AACF;;IAGF,KAAK,eAAe;KAElB,MAAM,aACJ,YAAY,IAAI,EAAE,SAAS,YAAY,IAAI,EAAE,SAAS;KACxD,MAAM,WAAW,cAAc,IAAK,EAAE,WAAsB;AAC5D,oBAAe,OAAO,EAAE,WAAqB;AAG7C,SACE,aAAa,2BACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,WAAY,WAAuC;AACzD,UAAI,aAAa,QAAW;OAC1B,MAAM,qBAAyC;QAC7C,MAAMA,wBAAU;QAChB;QACD;AACD,YAAK,MAAM,SAAS,oBAAoB,mBAAmB,CACzD,OAAM;;gBAIV,aAAa,wBACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,QAAS,WACZ;AACH,UAAI,UAAU,QAAW;OACvB,MAAM,kBAAmC;QACvC,MAAMA,wBAAU;QAChB;QACD;AACD,YAAK,MAAM,SAAS,oBAAoB,gBAAgB,CACtD,OAAM;;;KAMZ,IAAI;AACJ,SAAI;AACF,yBAAmB,KAAK,UAAU,WAAW;aACvC;AACN,yBAAmB,oCAAoC,YAAY,eAAe;;AASpF,WAPyC;MACvC,MAAMA,wBAAU;MAChB,MAAM;MACN,+CAAuB;MACvB,YAAY,EAAE;MACd,SAAS;MACV;AAED;;IAGF,KAAK,SAEH;IAGF,KAAK,SAAS;AACZ,SAAI,YAAY,QACd;KAGF,MAAM,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE;AACtC,SAAI,eAAe,MAAO,OAAM;AAChC,WAAM,IAAI,MACR,OAAO,QAAQ,WACX,MACA,wBAAwB,KAAK,UAAU,EAAE,GAC9C;;IAGH,QAEE;;;WAGE;AAER,SAAO,sBAAsB"}

package/dist/agent/converters/aisdk.d.cts

@@ -16,7 +16,7 @@ import { BaseEvent, Interrupt } from "@ag-ui/client";
  * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).
  * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
  */
-declare function convertAISDKStream(fullStream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[]): AsyncGenerator<BaseEvent>;
+declare function convertAISDKStream(fullStream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[], initialState?: unknown): AsyncGenerator<BaseEvent>;
 //#endregion
 export { convertAISDKStream };
 //# sourceMappingURL=aisdk.d.cts.map

package/dist/agent/converters/aisdk.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"aisdk.d.cts","names":[],"sources":["../../../src/agent/converters/aisdk.ts"],"mappings":";;;;;;AAiCA;;;;;;;;;;;;iBAAuB,kBAAA,CACrB,UAAA,EAAY,aAAA,WACZ,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,KACnB,cAAA,CAAe,SAAA"}
+{"version":3,"file":"aisdk.d.cts","names":[],"sources":["../../../src/agent/converters/aisdk.ts"],"mappings":";;;;;;AAkCA;;;;;;;;;;;;iBAAuB,kBAAA,CACrB,UAAA,EAAY,aAAA,WACZ,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,IACpB,YAAA,aACC,cAAA,CAAe,SAAA"}

package/dist/agent/converters/aisdk.d.mts

@@ -16,7 +16,7 @@ import { BaseEvent, Interrupt } from "@ag-ui/client";
  * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).
  * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
  */
-declare function convertAISDKStream(fullStream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[]): AsyncGenerator<BaseEvent>;
+declare function convertAISDKStream(fullStream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[], initialState?: unknown): AsyncGenerator<BaseEvent>;
 //#endregion
 export { convertAISDKStream };
 //# sourceMappingURL=aisdk.d.mts.map

package/dist/agent/converters/aisdk.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"aisdk.d.mts","names":[],"sources":["../../../src/agent/converters/aisdk.ts"],"mappings":";;;;;;AAiCA;;;;;;;;;;;;iBAAuB,kBAAA,CACrB,UAAA,EAAY,aAAA,WACZ,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,KACnB,cAAA,CAAe,SAAA"}
+{"version":3,"file":"aisdk.d.mts","names":[],"sources":["../../../src/agent/converters/aisdk.ts"],"mappings":";;;;;;AAkCA;;;;;;;;;;;;iBAAuB,kBAAA,CACrB,UAAA,EAAY,aAAA,WACZ,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,IACpB,YAAA,aACC,cAAA,CAAe,SAAA"}

package/dist/agent/converters/aisdk.mjs

@@ -1,4 +1,5 @@
 import "reflect-metadata";
+import { createStateEventNormalizer } from "../state-delta.mjs";
 import { randomUUID } from "@copilotkit/shared";
 import { EventType } from "@ag-ui/client";
 
@@ -17,10 +18,11 @@ import { EventType } from "@ag-ui/client";
 * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).
 * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
 */
-async function* convertAISDKStream(fullStream, abortSignal, pendingInterrupts) {
+async function* convertAISDKStream(fullStream, abortSignal, pendingInterrupts, initialState) {
 	let messageId = randomUUID();
 	let reasoningMessageId = randomUUID();
 	let isInReasoning = false;
+	const normalizeStateEvent = createStateEventNormalizer(initialState);
 	const toolCallStates = /* @__PURE__ */ new Map();
 	const ensureToolCallState = (toolCallId) => {
 		let state = toolCallStates.get(toolCallId);
@@ -219,16 +221,22 @@ async function* convertAISDKStream(fullStream, abortSignal, pendingInterrupts) {
 					toolCallStates.delete(p.toolCallId);
 					if (toolName === "AGUISendStateSnapshot" && toolResult && typeof toolResult === "object") {
 						const snapshot = toolResult.snapshot;
-						if (snapshot !== void 0) yield {
-							type: EventType.STATE_SNAPSHOT,
-							snapshot
-						};
+						if (snapshot !== void 0) {
+							const stateSnapshotEvent = {
+								type: EventType.STATE_SNAPSHOT,
+								snapshot
+							};
+							for (const event of normalizeStateEvent(stateSnapshotEvent)) yield event;
+						}
 					} else if (toolName === "AGUISendStateDelta" && toolResult && typeof toolResult === "object") {
 						const delta = toolResult.delta;
-						if (delta !== void 0) yield {
-							type: EventType.STATE_DELTA,
-							delta
-						};
+						if (delta !== void 0) {
+							const stateDeltaEvent = {
+								type: EventType.STATE_DELTA,
+								delta
+							};
+							for (const event of normalizeStateEvent(stateDeltaEvent)) yield event;
+						}
 					}
 					let serializedResult;
 					try {

package/dist/agent/converters/aisdk.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"aisdk.mjs","names":[],"sources":["../../../src/agent/converters/aisdk.ts"],"sourcesContent":["import type {\n  BaseEvent,\n  Interrupt,\n  ReasoningEndEvent,\n  ReasoningMessageContentEvent,\n  ReasoningMessageEndEvent,\n  ReasoningMessageStartEvent,\n  ReasoningStartEvent,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n  StateSnapshotEvent,\n  StateDeltaEvent,\n} from \"@ag-ui/client\";\nimport { EventType } from \"@ag-ui/client\";\nimport { randomUUID } from \"@copilotkit/shared\";\n\n/**\n * Converts an AI SDK `fullStream` into AG-UI `BaseEvent` objects.\n *\n * This is a pure converter — it does NOT emit lifecycle events\n * (RUN_STARTED / RUN_FINISHED / RUN_ERROR). The caller (Agent class)\n * is responsible for those.\n *\n * Terminal stream events (finish, error, abort) cause the generator to\n * return so the caller can handle lifecycle appropriately.\n *\n * `pendingInterrupts`, when provided, is filled with one AG-UI Interrupt per\n * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).\n * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.\n */\nexport async function* convertAISDKStream(\n  fullStream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n  pendingInterrupts?: Interrupt[],\n): AsyncGenerator<BaseEvent> {\n  let messageId = randomUUID();\n  let reasoningMessageId = randomUUID();\n  let isInReasoning = false;\n\n  const toolCallStates = new Map<\n    string,\n    {\n      started: boolean;\n      hasArgsDelta: boolean;\n      ended: boolean;\n      toolName?: string;\n    }\n  >();\n\n  const ensureToolCallState = (toolCallId: string) => {\n    let state = toolCallStates.get(toolCallId);\n    if (!state) {\n      state = { started: false, hasArgsDelta: false, ended: false };\n      toolCallStates.set(toolCallId, state);\n    }\n    return state;\n  };\n\n  /**\n   * Auto-close an open reasoning lifecycle.\n   * Some AI SDK providers (notably @ai-sdk/anthropic) never emit \"reasoning-end\",\n   * which leaves downstream state machines stuck. This helper emits the\n   * missing REASONING_MESSAGE_END + REASONING_END events so the stream\n   * can transition to text, tool-call, or finish phases.\n   */\n  function* closeReasoningIfOpen(): Generator<BaseEvent> {\n    if (!isInReasoning) return;\n    isInReasoning = false;\n    const reasoningMsgEnd: ReasoningMessageEndEvent = {\n      type: EventType.REASONING_MESSAGE_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningMsgEnd;\n    const reasoningEnd: ReasoningEndEvent = {\n      type: EventType.REASONING_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningEnd;\n  }\n\n  try {\n    for await (const part of fullStream) {\n      const p = part as Record<string, unknown>;\n\n      // Close any open reasoning lifecycle on every event except\n      // reasoning-delta, which arrives mid-block and must not interrupt it.\n      if (p.type !== \"reasoning-delta\") {\n        yield* closeReasoningIfOpen();\n      }\n\n      switch (p.type) {\n        case \"abort\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"reasoning-start\": {\n          // Use SDK-provided id, or generate a fresh UUID if id is falsy/\"0\"\n          // to prevent consecutive reasoning blocks from sharing a messageId\n          const providedId = \"id\" in p ? p.id : undefined;\n          reasoningMessageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          const reasoningStartEvent: ReasoningStartEvent = {\n            type: EventType.REASONING_START,\n            messageId: reasoningMessageId,\n          };\n          yield reasoningStartEvent;\n          const reasoningMessageStart: ReasoningMessageStartEvent = {\n            type: EventType.REASONING_MESSAGE_START,\n            messageId: reasoningMessageId,\n            role: \"reasoning\",\n          };\n          yield reasoningMessageStart;\n          isInReasoning = true;\n          break;\n        }\n\n        case \"reasoning-delta\": {\n          const delta = (p.text as string) ?? \"\";\n          if (!delta) break; // skip — @ag-ui/core schema requires delta to be non-empty\n          const reasoningDeltaEvent: ReasoningMessageContentEvent = {\n            type: EventType.REASONING_MESSAGE_CONTENT,\n            messageId: reasoningMessageId,\n            delta,\n          };\n          yield reasoningDeltaEvent;\n          break;\n        }\n\n        case \"reasoning-end\": {\n          // closeReasoningIfOpen() already called before the switch — no-op here\n          // if the SDK never emits this event (e.g. @ai-sdk/anthropic).\n          break;\n        }\n\n        case \"tool-input-start\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = p.toolName as string;\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n          break;\n        }\n\n        case \"tool-input-delta\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.hasArgsDelta = true;\n          const argsEvent: ToolCallArgsEvent = {\n            type: EventType.TOOL_CALL_ARGS,\n            toolCallId,\n            delta: p.delta as string,\n          };\n          yield argsEvent;\n          break;\n        }\n\n        case \"tool-input-end\": {\n          // No direct event – the subsequent \"tool-call\" part marks completion.\n          break;\n        }\n\n        case \"text-start\": {\n          // New text message starting - use the SDK-provided id\n          // Use randomUUID() if part.id is falsy or \"0\" to prevent message merging issues\n          const providedId = \"id\" in p ? p.id : undefined;\n          messageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          break;\n        }\n\n        case \"text-delta\": {\n          // AI SDK text-delta events use 'text' (not 'delta')\n          const textDelta = \"text\" in p ? (p.text as string) : \"\";\n          const textEvent: TextMessageChunkEvent = {\n            type: EventType.TEXT_MESSAGE_CHUNK,\n            role: \"assistant\",\n            messageId,\n            delta: textDelta,\n          };\n          yield textEvent;\n          break;\n        }\n\n        case \"tool-call\": {\n          const toolCallId = p.toolCallId as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = (p.toolName as string) ?? state.toolName;\n\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n\n          if (!state.hasArgsDelta && \"input\" in p && p.input !== undefined) {\n            let serializedInput = \"\";\n            if (typeof p.input === \"string\") {\n              serializedInput = p.input;\n            } else {\n              try {\n                serializedInput = JSON.stringify(p.input);\n              } catch {\n                serializedInput = String(p.input);\n              }\n            }\n\n            if (serializedInput.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serializedInput,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n          break;\n        }\n\n        case \"tool-approval-request\": {\n          // AI SDK native human-in-the-loop: a tool declared `needsApproval`\n          // was called. The preceding \"tool-call\" part already streamed the\n          // tool name + args; this part carries the toolCallId to pause on.\n          // (Some shapes nest the call under `toolCall` — read both.)\n          const nested = (p.toolCall ?? {}) as {\n            toolCallId?: string;\n            toolName?: string;\n            input?: unknown;\n          };\n          const toolCallId =\n            (p.toolCallId as string | undefined) ?? nested.toolCallId;\n          if (!toolCallId) {\n            throw new Error(\n              \"AI SDK tool-approval-request is missing toolCallId\",\n            );\n          }\n          const state = ensureToolCallState(toolCallId);\n          const toolName = state.toolName ?? nested.toolName;\n\n          // Defensive: emit the tool-call lifecycle if the \"tool-call\" part\n          // didn't precede this one, so the assistant tool-call is in history\n          // for the resume run (which keys the injected result by toolCallId).\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: toolName ?? \"\",\n            };\n            yield startEvent;\n          }\n          if (!state.hasArgsDelta && nested.input !== undefined) {\n            let serialized = \"\";\n            try {\n              serialized =\n                typeof nested.input === \"string\"\n                  ? nested.input\n                  : JSON.stringify(nested.input);\n            } catch {\n              serialized = String(nested.input);\n            }\n            if (serialized.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serialized,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n\n          pendingInterrupts?.push({\n            id: toolCallId,\n            toolCallId,\n            reason: \"tool_approval\",\n            message: toolName ? `Approve \"${toolName}\"?` : undefined,\n            ...(toolName ? { metadata: { toolName } } : {}),\n          });\n          break;\n        }\n\n        case \"tool-result\": {\n          // AI SDK tool-result uses \"output\"; older versions used \"result\" — check both\n          const toolResult =\n            \"output\" in p ? p.output : \"result\" in p ? p.result : null;\n          const toolName = \"toolName\" in p ? (p.toolName as string) : \"\";\n          toolCallStates.delete(p.toolCallId as string);\n\n          // Check if this is a state update tool\n          if (\n            toolName === \"AGUISendStateSnapshot\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const snapshot = (toolResult as Record<string, unknown>).snapshot;\n            if (snapshot !== undefined) {\n              const stateSnapshotEvent: StateSnapshotEvent = {\n                type: EventType.STATE_SNAPSHOT,\n                snapshot,\n              };\n              yield stateSnapshotEvent;\n            }\n          } else if (\n            toolName === \"AGUISendStateDelta\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const delta = (toolResult as { delta?: StateDeltaEvent[\"delta\"] })\n              .delta;\n            if (delta !== undefined) {\n              const stateDeltaEvent: StateDeltaEvent = {\n                type: EventType.STATE_DELTA,\n                delta,\n              };\n              yield stateDeltaEvent;\n            }\n          }\n\n          // Always emit the tool result event for the LLM\n          let serializedResult: string;\n          try {\n            serializedResult = JSON.stringify(toolResult);\n          } catch {\n            serializedResult = `[Unserializable tool result from ${toolName || \"unknown tool\"}]`;\n          }\n          const resultEvent: ToolCallResultEvent = {\n            type: EventType.TOOL_CALL_RESULT,\n            role: \"tool\",\n            messageId: randomUUID(),\n            toolCallId: p.toolCallId as string,\n            content: serializedResult,\n          };\n          yield resultEvent;\n          break;\n        }\n\n        case \"finish\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"error\": {\n          if (abortSignal.aborted) {\n            return;\n          }\n          // Re-throw so the caller can emit RUN_ERROR\n          const err = p.error ?? p.message ?? p.cause;\n          if (err instanceof Error) throw err;\n          throw new Error(\n            typeof err === \"string\"\n              ? err\n              : `AI SDK stream error: ${JSON.stringify(p)}`,\n          );\n        }\n\n        default:\n          // Unknown event types are silently ignored\n          break;\n      }\n    }\n  } finally {\n    // Always close reasoning on exit (normal or exceptional)\n    yield* closeReasoningIfOpen();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAiCA,gBAAuB,mBACrB,YACA,aACA,mBAC2B;CAC3B,IAAI,YAAY,YAAY;CAC5B,IAAI,qBAAqB,YAAY;CACrC,IAAI,gBAAgB;CAEpB,MAAM,iCAAiB,IAAI,KAQxB;CAEH,MAAM,uBAAuB,eAAuB;EAClD,IAAI,QAAQ,eAAe,IAAI,WAAW;AAC1C,MAAI,CAAC,OAAO;AACV,WAAQ;IAAE,SAAS;IAAO,cAAc;IAAO,OAAO;IAAO;AAC7D,kBAAe,IAAI,YAAY,MAAM;;AAEvC,SAAO;;;;;;;;;CAUT,UAAU,uBAA6C;AACrD,MAAI,CAAC,cAAe;AACpB,kBAAgB;AAKhB,QAJkD;GAChD,MAAM,UAAU;GAChB,WAAW;GACZ;AAMD,QAJwC;GACtC,MAAM,UAAU;GAChB,WAAW;GACZ;;AAIH,KAAI;AACF,aAAW,MAAM,QAAQ,YAAY;GACnC,MAAM,IAAI;AAIV,OAAI,EAAE,SAAS,kBACb,QAAO,sBAAsB;AAG/B,WAAQ,EAAE,MAAV;IACE,KAAK,QAEH;IAGF,KAAK,mBAAmB;KAGtB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,0BACE,cAAc,eAAe,MACxB,aACD,YAAY;AAKlB,WAJiD;MAC/C,MAAM,UAAU;MAChB,WAAW;MACZ;AAOD,WAL0D;MACxD,MAAM,UAAU;MAChB,WAAW;MACX,MAAM;MACP;AAED,qBAAgB;AAChB;;IAGF,KAAK,mBAAmB;KACtB,MAAM,QAAS,EAAE,QAAmB;AACpC,SAAI,CAAC,MAAO;AAMZ,WAL0D;MACxD,MAAM,UAAU;MAChB,WAAW;MACX;MACD;AAED;;IAGF,KAAK,gBAGH;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAW,EAAE;AACnB,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAM,UAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAGH;;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,eAAe;AAMrB,WALqC;MACnC,MAAM,UAAU;MAChB;MACA,OAAO,EAAE;MACV;AAED;;IAGF,KAAK,iBAEH;IAGF,KAAK,cAAc;KAGjB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,iBACE,cAAc,eAAe,MACxB,aACD,YAAY;AAClB;;IAGF,KAAK,cAAc;KAEjB,MAAM,YAAY,UAAU,IAAK,EAAE,OAAkB;AAOrD,WANyC;MACvC,MAAM,UAAU;MAChB,MAAM;MACN;MACA,OAAO;MACR;AAED;;IAGF,KAAK,aAAa;KAChB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAY,EAAE,YAAuB,MAAM;AAEjD,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAM,UAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAIH,SAAI,CAAC,MAAM,gBAAgB,WAAW,KAAK,EAAE,UAAU,QAAW;MAChE,IAAI,kBAAkB;AACtB,UAAI,OAAO,EAAE,UAAU,SACrB,mBAAkB,EAAE;UAEpB,KAAI;AACF,yBAAkB,KAAK,UAAU,EAAE,MAAM;cACnC;AACN,yBAAkB,OAAO,EAAE,MAAM;;AAIrC,UAAI,gBAAgB,SAAS,GAAG;AAM9B,aALqC;QACnC,MAAM,UAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAIzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAM,UAAU;OAChB;OACD;;AAGH;;IAGF,KAAK,yBAAyB;KAK5B,MAAM,SAAU,EAAE,YAAY,EAAE;KAKhC,MAAM,aACH,EAAE,cAAqC,OAAO;AACjD,SAAI,CAAC,WACH,OAAM,IAAI,MACR,qDACD;KAEH,MAAM,QAAQ,oBAAoB,WAAW;KAC7C,MAAM,WAAW,MAAM,YAAY,OAAO;AAK1C,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAM,UAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,YAAY;OAC3B;;AAGH,SAAI,CAAC,MAAM,gBAAgB,OAAO,UAAU,QAAW;MACrD,IAAI,aAAa;AACjB,UAAI;AACF,oBACE,OAAO,OAAO,UAAU,WACpB,OAAO,QACP,KAAK,UAAU,OAAO,MAAM;cAC5B;AACN,oBAAa,OAAO,OAAO,MAAM;;AAEnC,UAAI,WAAW,SAAS,GAAG;AAMzB,aALqC;QACnC,MAAM,UAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAGzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAM,UAAU;OAChB;OACD;;AAIH,wBAAmB,KAAK;MACtB,IAAI;MACJ;MACA,QAAQ;MACR,SAAS,WAAW,YAAY,SAAS,MAAM;MAC/C,GAAI,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE;MAC/C,CAAC;AACF;;IAGF,KAAK,eAAe;KAElB,MAAM,aACJ,YAAY,IAAI,EAAE,SAAS,YAAY,IAAI,EAAE,SAAS;KACxD,MAAM,WAAW,cAAc,IAAK,EAAE,WAAsB;AAC5D,oBAAe,OAAO,EAAE,WAAqB;AAG7C,SACE,aAAa,2BACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,WAAY,WAAuC;AACzD,UAAI,aAAa,OAKf,OAJ+C;OAC7C,MAAM,UAAU;OAChB;OACD;gBAIH,aAAa,wBACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,QAAS,WACZ;AACH,UAAI,UAAU,OAKZ,OAJyC;OACvC,MAAM,UAAU;OAChB;OACD;;KAML,IAAI;AACJ,SAAI;AACF,yBAAmB,KAAK,UAAU,WAAW;aACvC;AACN,yBAAmB,oCAAoC,YAAY,eAAe;;AASpF,WAPyC;MACvC,MAAM,UAAU;MAChB,MAAM;MACN,WAAW,YAAY;MACvB,YAAY,EAAE;MACd,SAAS;MACV;AAED;;IAGF,KAAK,SAEH;IAGF,KAAK,SAAS;AACZ,SAAI,YAAY,QACd;KAGF,MAAM,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE;AACtC,SAAI,eAAe,MAAO,OAAM;AAChC,WAAM,IAAI,MACR,OAAO,QAAQ,WACX,MACA,wBAAwB,KAAK,UAAU,EAAE,GAC9C;;IAGH,QAEE;;;WAGE;AAER,SAAO,sBAAsB"}
+{"version":3,"file":"aisdk.mjs","names":[],"sources":["../../../src/agent/converters/aisdk.ts"],"sourcesContent":["import type {\n  BaseEvent,\n  Interrupt,\n  ReasoningEndEvent,\n  ReasoningMessageContentEvent,\n  ReasoningMessageEndEvent,\n  ReasoningMessageStartEvent,\n  ReasoningStartEvent,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n  StateSnapshotEvent,\n  StateDeltaEvent,\n} from \"@ag-ui/client\";\nimport { EventType } from \"@ag-ui/client\";\nimport { randomUUID } from \"@copilotkit/shared\";\nimport { createStateEventNormalizer } from \"../state-delta\";\n\n/**\n * Converts an AI SDK `fullStream` into AG-UI `BaseEvent` objects.\n *\n * This is a pure converter — it does NOT emit lifecycle events\n * (RUN_STARTED / RUN_FINISHED / RUN_ERROR). The caller (Agent class)\n * is responsible for those.\n *\n * Terminal stream events (finish, error, abort) cause the generator to\n * return so the caller can handle lifecycle appropriately.\n *\n * `pendingInterrupts`, when provided, is filled with one AG-UI Interrupt per\n * AI SDK `tool-approval-request` part (a tool declared `needsApproval: true`).\n * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.\n */\nexport async function* convertAISDKStream(\n  fullStream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n  pendingInterrupts?: Interrupt[],\n  initialState?: unknown,\n): AsyncGenerator<BaseEvent> {\n  let messageId = randomUUID();\n  let reasoningMessageId = randomUUID();\n  let isInReasoning = false;\n  const normalizeStateEvent = createStateEventNormalizer(initialState);\n\n  const toolCallStates = new Map<\n    string,\n    {\n      started: boolean;\n      hasArgsDelta: boolean;\n      ended: boolean;\n      toolName?: string;\n    }\n  >();\n\n  const ensureToolCallState = (toolCallId: string) => {\n    let state = toolCallStates.get(toolCallId);\n    if (!state) {\n      state = { started: false, hasArgsDelta: false, ended: false };\n      toolCallStates.set(toolCallId, state);\n    }\n    return state;\n  };\n\n  /**\n   * Auto-close an open reasoning lifecycle.\n   * Some AI SDK providers (notably @ai-sdk/anthropic) never emit \"reasoning-end\",\n   * which leaves downstream state machines stuck. This helper emits the\n   * missing REASONING_MESSAGE_END + REASONING_END events so the stream\n   * can transition to text, tool-call, or finish phases.\n   */\n  function* closeReasoningIfOpen(): Generator<BaseEvent> {\n    if (!isInReasoning) return;\n    isInReasoning = false;\n    const reasoningMsgEnd: ReasoningMessageEndEvent = {\n      type: EventType.REASONING_MESSAGE_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningMsgEnd;\n    const reasoningEnd: ReasoningEndEvent = {\n      type: EventType.REASONING_END,\n      messageId: reasoningMessageId,\n    };\n    yield reasoningEnd;\n  }\n\n  try {\n    for await (const part of fullStream) {\n      const p = part as Record<string, unknown>;\n\n      // Close any open reasoning lifecycle on every event except\n      // reasoning-delta, which arrives mid-block and must not interrupt it.\n      if (p.type !== \"reasoning-delta\") {\n        yield* closeReasoningIfOpen();\n      }\n\n      switch (p.type) {\n        case \"abort\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"reasoning-start\": {\n          // Use SDK-provided id, or generate a fresh UUID if id is falsy/\"0\"\n          // to prevent consecutive reasoning blocks from sharing a messageId\n          const providedId = \"id\" in p ? p.id : undefined;\n          reasoningMessageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          const reasoningStartEvent: ReasoningStartEvent = {\n            type: EventType.REASONING_START,\n            messageId: reasoningMessageId,\n          };\n          yield reasoningStartEvent;\n          const reasoningMessageStart: ReasoningMessageStartEvent = {\n            type: EventType.REASONING_MESSAGE_START,\n            messageId: reasoningMessageId,\n            role: \"reasoning\",\n          };\n          yield reasoningMessageStart;\n          isInReasoning = true;\n          break;\n        }\n\n        case \"reasoning-delta\": {\n          const delta = (p.text as string) ?? \"\";\n          if (!delta) break; // skip — @ag-ui/core schema requires delta to be non-empty\n          const reasoningDeltaEvent: ReasoningMessageContentEvent = {\n            type: EventType.REASONING_MESSAGE_CONTENT,\n            messageId: reasoningMessageId,\n            delta,\n          };\n          yield reasoningDeltaEvent;\n          break;\n        }\n\n        case \"reasoning-end\": {\n          // closeReasoningIfOpen() already called before the switch — no-op here\n          // if the SDK never emits this event (e.g. @ai-sdk/anthropic).\n          break;\n        }\n\n        case \"tool-input-start\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = p.toolName as string;\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n          break;\n        }\n\n        case \"tool-input-delta\": {\n          const toolCallId = p.id as string;\n          const state = ensureToolCallState(toolCallId);\n          state.hasArgsDelta = true;\n          const argsEvent: ToolCallArgsEvent = {\n            type: EventType.TOOL_CALL_ARGS,\n            toolCallId,\n            delta: p.delta as string,\n          };\n          yield argsEvent;\n          break;\n        }\n\n        case \"tool-input-end\": {\n          // No direct event – the subsequent \"tool-call\" part marks completion.\n          break;\n        }\n\n        case \"text-start\": {\n          // New text message starting - use the SDK-provided id\n          // Use randomUUID() if part.id is falsy or \"0\" to prevent message merging issues\n          const providedId = \"id\" in p ? p.id : undefined;\n          messageId =\n            providedId && providedId !== \"0\"\n              ? (providedId as string)\n              : randomUUID();\n          break;\n        }\n\n        case \"text-delta\": {\n          // AI SDK text-delta events use 'text' (not 'delta')\n          const textDelta = \"text\" in p ? (p.text as string) : \"\";\n          const textEvent: TextMessageChunkEvent = {\n            type: EventType.TEXT_MESSAGE_CHUNK,\n            role: \"assistant\",\n            messageId,\n            delta: textDelta,\n          };\n          yield textEvent;\n          break;\n        }\n\n        case \"tool-call\": {\n          const toolCallId = p.toolCallId as string;\n          const state = ensureToolCallState(toolCallId);\n          state.toolName = (p.toolName as string) ?? state.toolName;\n\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: p.toolName as string,\n            };\n            yield startEvent;\n          }\n\n          if (!state.hasArgsDelta && \"input\" in p && p.input !== undefined) {\n            let serializedInput = \"\";\n            if (typeof p.input === \"string\") {\n              serializedInput = p.input;\n            } else {\n              try {\n                serializedInput = JSON.stringify(p.input);\n              } catch {\n                serializedInput = String(p.input);\n              }\n            }\n\n            if (serializedInput.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serializedInput,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n          break;\n        }\n\n        case \"tool-approval-request\": {\n          // AI SDK native human-in-the-loop: a tool declared `needsApproval`\n          // was called. The preceding \"tool-call\" part already streamed the\n          // tool name + args; this part carries the toolCallId to pause on.\n          // (Some shapes nest the call under `toolCall` — read both.)\n          const nested = (p.toolCall ?? {}) as {\n            toolCallId?: string;\n            toolName?: string;\n            input?: unknown;\n          };\n          const toolCallId =\n            (p.toolCallId as string | undefined) ?? nested.toolCallId;\n          if (!toolCallId) {\n            throw new Error(\n              \"AI SDK tool-approval-request is missing toolCallId\",\n            );\n          }\n          const state = ensureToolCallState(toolCallId);\n          const toolName = state.toolName ?? nested.toolName;\n\n          // Defensive: emit the tool-call lifecycle if the \"tool-call\" part\n          // didn't precede this one, so the assistant tool-call is in history\n          // for the resume run (which keys the injected result by toolCallId).\n          if (!state.started) {\n            state.started = true;\n            const startEvent: ToolCallStartEvent = {\n              type: EventType.TOOL_CALL_START,\n              parentMessageId: messageId,\n              toolCallId,\n              toolCallName: toolName ?? \"\",\n            };\n            yield startEvent;\n          }\n          if (!state.hasArgsDelta && nested.input !== undefined) {\n            let serialized = \"\";\n            try {\n              serialized =\n                typeof nested.input === \"string\"\n                  ? nested.input\n                  : JSON.stringify(nested.input);\n            } catch {\n              serialized = String(nested.input);\n            }\n            if (serialized.length > 0) {\n              const argsEvent: ToolCallArgsEvent = {\n                type: EventType.TOOL_CALL_ARGS,\n                toolCallId,\n                delta: serialized,\n              };\n              yield argsEvent;\n              state.hasArgsDelta = true;\n            }\n          }\n          if (!state.ended) {\n            state.ended = true;\n            const endEvent: ToolCallEndEvent = {\n              type: EventType.TOOL_CALL_END,\n              toolCallId,\n            };\n            yield endEvent;\n          }\n\n          pendingInterrupts?.push({\n            id: toolCallId,\n            toolCallId,\n            reason: \"tool_approval\",\n            message: toolName ? `Approve \"${toolName}\"?` : undefined,\n            ...(toolName ? { metadata: { toolName } } : {}),\n          });\n          break;\n        }\n\n        case \"tool-result\": {\n          // AI SDK tool-result uses \"output\"; older versions used \"result\" — check both\n          const toolResult =\n            \"output\" in p ? p.output : \"result\" in p ? p.result : null;\n          const toolName = \"toolName\" in p ? (p.toolName as string) : \"\";\n          toolCallStates.delete(p.toolCallId as string);\n\n          // Check if this is a state update tool\n          if (\n            toolName === \"AGUISendStateSnapshot\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const snapshot = (toolResult as Record<string, unknown>).snapshot;\n            if (snapshot !== undefined) {\n              const stateSnapshotEvent: StateSnapshotEvent = {\n                type: EventType.STATE_SNAPSHOT,\n                snapshot,\n              };\n              for (const event of normalizeStateEvent(stateSnapshotEvent)) {\n                yield event;\n              }\n            }\n          } else if (\n            toolName === \"AGUISendStateDelta\" &&\n            toolResult &&\n            typeof toolResult === \"object\"\n          ) {\n            const delta = (toolResult as { delta?: StateDeltaEvent[\"delta\"] })\n              .delta;\n            if (delta !== undefined) {\n              const stateDeltaEvent: StateDeltaEvent = {\n                type: EventType.STATE_DELTA,\n                delta,\n              };\n              for (const event of normalizeStateEvent(stateDeltaEvent)) {\n                yield event;\n              }\n            }\n          }\n\n          // Always emit the tool result event for the LLM\n          let serializedResult: string;\n          try {\n            serializedResult = JSON.stringify(toolResult);\n          } catch {\n            serializedResult = `[Unserializable tool result from ${toolName || \"unknown tool\"}]`;\n          }\n          const resultEvent: ToolCallResultEvent = {\n            type: EventType.TOOL_CALL_RESULT,\n            role: \"tool\",\n            messageId: randomUUID(),\n            toolCallId: p.toolCallId as string,\n            content: serializedResult,\n          };\n          yield resultEvent;\n          break;\n        }\n\n        case \"finish\": {\n          // Terminal — let the caller handle lifecycle\n          return;\n        }\n\n        case \"error\": {\n          if (abortSignal.aborted) {\n            return;\n          }\n          // Re-throw so the caller can emit RUN_ERROR\n          const err = p.error ?? p.message ?? p.cause;\n          if (err instanceof Error) throw err;\n          throw new Error(\n            typeof err === \"string\"\n              ? err\n              : `AI SDK stream error: ${JSON.stringify(p)}`,\n          );\n        }\n\n        default:\n          // Unknown event types are silently ignored\n          break;\n      }\n    }\n  } finally {\n    // Always close reasoning on exit (normal or exceptional)\n    yield* closeReasoningIfOpen();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAkCA,gBAAuB,mBACrB,YACA,aACA,mBACA,cAC2B;CAC3B,IAAI,YAAY,YAAY;CAC5B,IAAI,qBAAqB,YAAY;CACrC,IAAI,gBAAgB;CACpB,MAAM,sBAAsB,2BAA2B,aAAa;CAEpE,MAAM,iCAAiB,IAAI,KAQxB;CAEH,MAAM,uBAAuB,eAAuB;EAClD,IAAI,QAAQ,eAAe,IAAI,WAAW;AAC1C,MAAI,CAAC,OAAO;AACV,WAAQ;IAAE,SAAS;IAAO,cAAc;IAAO,OAAO;IAAO;AAC7D,kBAAe,IAAI,YAAY,MAAM;;AAEvC,SAAO;;;;;;;;;CAUT,UAAU,uBAA6C;AACrD,MAAI,CAAC,cAAe;AACpB,kBAAgB;AAKhB,QAJkD;GAChD,MAAM,UAAU;GAChB,WAAW;GACZ;AAMD,QAJwC;GACtC,MAAM,UAAU;GAChB,WAAW;GACZ;;AAIH,KAAI;AACF,aAAW,MAAM,QAAQ,YAAY;GACnC,MAAM,IAAI;AAIV,OAAI,EAAE,SAAS,kBACb,QAAO,sBAAsB;AAG/B,WAAQ,EAAE,MAAV;IACE,KAAK,QAEH;IAGF,KAAK,mBAAmB;KAGtB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,0BACE,cAAc,eAAe,MACxB,aACD,YAAY;AAKlB,WAJiD;MAC/C,MAAM,UAAU;MAChB,WAAW;MACZ;AAOD,WAL0D;MACxD,MAAM,UAAU;MAChB,WAAW;MACX,MAAM;MACP;AAED,qBAAgB;AAChB;;IAGF,KAAK,mBAAmB;KACtB,MAAM,QAAS,EAAE,QAAmB;AACpC,SAAI,CAAC,MAAO;AAMZ,WAL0D;MACxD,MAAM,UAAU;MAChB,WAAW;MACX;MACD;AAED;;IAGF,KAAK,gBAGH;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAW,EAAE;AACnB,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAM,UAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAGH;;IAGF,KAAK,oBAAoB;KACvB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,eAAe;AAMrB,WALqC;MACnC,MAAM,UAAU;MAChB;MACA,OAAO,EAAE;MACV;AAED;;IAGF,KAAK,iBAEH;IAGF,KAAK,cAAc;KAGjB,MAAM,aAAa,QAAQ,IAAI,EAAE,KAAK;AACtC,iBACE,cAAc,eAAe,MACxB,aACD,YAAY;AAClB;;IAGF,KAAK,cAAc;KAEjB,MAAM,YAAY,UAAU,IAAK,EAAE,OAAkB;AAOrD,WANyC;MACvC,MAAM,UAAU;MAChB,MAAM;MACN;MACA,OAAO;MACR;AAED;;IAGF,KAAK,aAAa;KAChB,MAAM,aAAa,EAAE;KACrB,MAAM,QAAQ,oBAAoB,WAAW;AAC7C,WAAM,WAAY,EAAE,YAAuB,MAAM;AAEjD,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAM,UAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,EAAE;OACjB;;AAIH,SAAI,CAAC,MAAM,gBAAgB,WAAW,KAAK,EAAE,UAAU,QAAW;MAChE,IAAI,kBAAkB;AACtB,UAAI,OAAO,EAAE,UAAU,SACrB,mBAAkB,EAAE;UAEpB,KAAI;AACF,yBAAkB,KAAK,UAAU,EAAE,MAAM;cACnC;AACN,yBAAkB,OAAO,EAAE,MAAM;;AAIrC,UAAI,gBAAgB,SAAS,GAAG;AAM9B,aALqC;QACnC,MAAM,UAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAIzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAM,UAAU;OAChB;OACD;;AAGH;;IAGF,KAAK,yBAAyB;KAK5B,MAAM,SAAU,EAAE,YAAY,EAAE;KAKhC,MAAM,aACH,EAAE,cAAqC,OAAO;AACjD,SAAI,CAAC,WACH,OAAM,IAAI,MACR,qDACD;KAEH,MAAM,QAAQ,oBAAoB,WAAW;KAC7C,MAAM,WAAW,MAAM,YAAY,OAAO;AAK1C,SAAI,CAAC,MAAM,SAAS;AAClB,YAAM,UAAU;AAOhB,YANuC;OACrC,MAAM,UAAU;OAChB,iBAAiB;OACjB;OACA,cAAc,YAAY;OAC3B;;AAGH,SAAI,CAAC,MAAM,gBAAgB,OAAO,UAAU,QAAW;MACrD,IAAI,aAAa;AACjB,UAAI;AACF,oBACE,OAAO,OAAO,UAAU,WACpB,OAAO,QACP,KAAK,UAAU,OAAO,MAAM;cAC5B;AACN,oBAAa,OAAO,OAAO,MAAM;;AAEnC,UAAI,WAAW,SAAS,GAAG;AAMzB,aALqC;QACnC,MAAM,UAAU;QAChB;QACA,OAAO;QACR;AAED,aAAM,eAAe;;;AAGzB,SAAI,CAAC,MAAM,OAAO;AAChB,YAAM,QAAQ;AAKd,YAJmC;OACjC,MAAM,UAAU;OAChB;OACD;;AAIH,wBAAmB,KAAK;MACtB,IAAI;MACJ;MACA,QAAQ;MACR,SAAS,WAAW,YAAY,SAAS,MAAM;MAC/C,GAAI,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE;MAC/C,CAAC;AACF;;IAGF,KAAK,eAAe;KAElB,MAAM,aACJ,YAAY,IAAI,EAAE,SAAS,YAAY,IAAI,EAAE,SAAS;KACxD,MAAM,WAAW,cAAc,IAAK,EAAE,WAAsB;AAC5D,oBAAe,OAAO,EAAE,WAAqB;AAG7C,SACE,aAAa,2BACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,WAAY,WAAuC;AACzD,UAAI,aAAa,QAAW;OAC1B,MAAM,qBAAyC;QAC7C,MAAM,UAAU;QAChB;QACD;AACD,YAAK,MAAM,SAAS,oBAAoB,mBAAmB,CACzD,OAAM;;gBAIV,aAAa,wBACb,cACA,OAAO,eAAe,UACtB;MACA,MAAM,QAAS,WACZ;AACH,UAAI,UAAU,QAAW;OACvB,MAAM,kBAAmC;QACvC,MAAM,UAAU;QAChB;QACD;AACD,YAAK,MAAM,SAAS,oBAAoB,gBAAgB,CACtD,OAAM;;;KAMZ,IAAI;AACJ,SAAI;AACF,yBAAmB,KAAK,UAAU,WAAW;aACvC;AACN,yBAAmB,oCAAoC,YAAY,eAAe;;AASpF,WAPyC;MACvC,MAAM,UAAU;MAChB,MAAM;MACN,WAAW,YAAY;MACvB,YAAY,EAAE;MACd,SAAS;MACV;AAED;;IAGF,KAAK,SAEH;IAGF,KAAK,SAAS;AACZ,SAAI,YAAY,QACd;KAGF,MAAM,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE;AACtC,SAAI,eAAe,MAAO,OAAM;AAChC,WAAM,IAAI,MACR,OAAO,QAAQ,WACX,MACA,wBAAwB,KAAK,UAAU,EAAE,GAC9C;;IAGH,QAEE;;;WAGE;AAER,SAAO,sBAAsB"}

package/dist/agent/converters/tanstack.cjs

@@ -1,5 +1,6 @@
 require("reflect-metadata");
 const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+const require_state_delta = require('../state-delta.cjs');
 let _copilotkit_shared = require("@copilotkit/shared");
 let _ag_ui_client = require("@ag-ui/client");
 
@@ -170,12 +171,13 @@ function convertInputToTanStackAI(input) {
 * CUSTOM "approval-requested" chunk (a tool declared `needsApproval: true`).
 * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
 */
-async function* convertTanStackStream(stream, abortSignal, pendingInterrupts) {
+async function* convertTanStackStream(stream, abortSignal, pendingInterrupts, initialState) {
 	const messageId = (0, _copilotkit_shared.randomUUID)();
 	const toolNamesById = /* @__PURE__ */ new Map();
 	let reasoningRunOpen = false;
 	let reasoningMessageOpen = false;
 	let reasoningMessageId = (0, _copilotkit_shared.randomUUID)();
+	const normalizeStateEvent = require_state_delta.createStateEventNormalizer(initialState);
 	function* closeReasoningIfOpen() {
 		if (reasoningMessageOpen) {
 			reasoningMessageOpen = false;
@@ -255,14 +257,20 @@ async function* convertTanStackStream(stream, abortSignal, pendingInterrupts) {
 			const toolName = toolNamesById.get(toolCallId);
 			const rawPayload = raw.content ?? raw.result;
 			const parsedContent = typeof rawPayload === "string" ? safeParse(rawPayload) : rawPayload;
-			if (toolName === "AGUISendStateSnapshot" && parsedContent && typeof parsedContent === "object" && "snapshot" in parsedContent) yield {
-				type: _ag_ui_client.EventType.STATE_SNAPSHOT,
-				snapshot: parsedContent.snapshot
-			};
-			if (toolName === "AGUISendStateDelta" && parsedContent && typeof parsedContent === "object" && "delta" in parsedContent) yield {
-				type: _ag_ui_client.EventType.STATE_DELTA,
-				delta: parsedContent.delta
-			};
+			if (toolName === "AGUISendStateSnapshot" && parsedContent && typeof parsedContent === "object" && "snapshot" in parsedContent) {
+				const stateSnapshotEvent = {
+					type: _ag_ui_client.EventType.STATE_SNAPSHOT,
+					snapshot: parsedContent.snapshot
+				};
+				for (const event of normalizeStateEvent(stateSnapshotEvent)) yield event;
+			}
+			if (toolName === "AGUISendStateDelta" && parsedContent && typeof parsedContent === "object" && "delta" in parsedContent) {
+				const stateDeltaEvent = {
+					type: _ag_ui_client.EventType.STATE_DELTA,
+					delta: parsedContent.delta
+				};
+				for (const event of normalizeStateEvent(stateDeltaEvent)) yield event;
+			}
 			let serializedContent;
 			if (typeof rawPayload === "string") serializedContent = rawPayload;
 			else try {

package/dist/agent/converters/tanstack.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"tanstack.cjs","names":["EventType"],"sources":["../../../src/agent/converters/tanstack.ts"],"sourcesContent":["import type {\n  BaseEvent,\n  Interrupt,\n  RunAgentInput,\n  Message,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n  StateSnapshotEvent,\n  StateDeltaEvent,\n  ReasoningStartEvent,\n  ReasoningMessageStartEvent,\n  ReasoningMessageContentEvent,\n  ReasoningMessageEndEvent,\n  ReasoningEndEvent,\n} from \"@ag-ui/client\";\nimport { EventType } from \"@ag-ui/client\";\nimport { randomUUID } from \"@copilotkit/shared\";\n\ntype ContentPartSource =\n  | { type: \"data\"; value: string; mimeType: string }\n  | { type: \"url\"; value: string; mimeType?: string };\n\n/**\n * A TanStack AI content part (text, image, audio, video, or document).\n */\nexport type TanStackContentPart =\n  | { type: \"text\"; content: string }\n  | { type: \"image\"; source: ContentPartSource }\n  | { type: \"audio\"; source: ContentPartSource }\n  | { type: \"video\"; source: ContentPartSource }\n  | { type: \"document\"; source: ContentPartSource };\n\n/**\n * Message format expected by TanStack AI's `chat()`.\n *\n * Content is typed as `any[]` for the multimodal case so messages are directly\n * passable to any adapter without casts — different adapters constrain which\n * modalities they accept (e.g. OpenAI only allows text + image).\n * Use `TanStackContentPart` to inspect individual parts if needed.\n */\nexport interface TanStackChatMessage {\n  role: \"user\" | \"assistant\" | \"tool\";\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  content: string | null | any[];\n  name?: string;\n  toolCalls?: Array<{\n    id: string;\n    type: \"function\";\n    function: { name: string; arguments: string };\n  }>;\n  toolCallId?: string;\n}\n\n/**\n * A TanStack AI client-side tool, derived from a frontend-provided AG-UI tool.\n *\n * Shaped to match `@tanstack/ai`'s `ClientTool` (`__toolSide: \"client\"`, no\n * `execute`): the model may CALL it, but TanStack does not run it — it pauses\n * the run and hands the call back to the AG-UI client (the CopilotKit frontend\n * / bot) to execute, mirroring CopilotKit's client-tool round-trip. `chat()`\n * accepts a JSON Schema directly as `inputSchema`, so the AG-UI tool's\n * `parameters` pass through unchanged.\n */\nexport interface TanStackClientTool {\n  __toolSide: \"client\";\n  name: string;\n  description: string;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  inputSchema: any;\n}\n\n/**\n * Result of converting RunAgentInput to TanStack AI format.\n */\nexport interface TanStackInputResult {\n  /** Chat messages (only user/assistant/tool roles; all others excluded) */\n  messages: TanStackChatMessage[];\n  /** System prompts extracted from system/developer messages, context, and state */\n  systemPrompts: string[];\n  /**\n   * Client-side tools derived from `input.tools` (the frontend-provided tools\n   * the CopilotKit client forwards on every run). Pass these into `chat()`\n   * alongside any server/provider tools so the model can call the frontend's\n   * generative-UI and human-in-the-loop tools; TanStack pauses the run on a\n   * client-tool call and the client executes it.\n   */\n  tools: TanStackClientTool[];\n}\n\n/**\n * Converts AG-UI user message content to TanStack AI format.\n * Handles plain strings, multimodal parts (image/audio/video/document),\n * and legacy BinaryInputContent for backward compatibility.\n */\nfunction convertUserContent(\n  content: unknown,\n): string | null | TanStackContentPart[] {\n  if (!content) return null;\n  if (typeof content === \"string\") return content;\n  if (!Array.isArray(content)) return null;\n  if (content.length === 0) return \"\";\n\n  const parts: TanStackContentPart[] = [];\n\n  for (const part of content) {\n    if (!part || typeof part !== \"object\" || !(\"type\" in part)) continue;\n\n    switch ((part as { type: string }).type) {\n      case \"text\": {\n        const text = (part as { text?: string }).text;\n        if (text != null) parts.push({ type: \"text\", content: text });\n        break;\n      }\n\n      case \"image\":\n      case \"audio\":\n      case \"video\":\n      case \"document\": {\n        const source = (part as { source?: any }).source;\n        if (!source) break;\n        const partType = (part as { type: string }).type as\n          | \"image\"\n          | \"audio\"\n          | \"video\"\n          | \"document\";\n        if (source.type === \"data\") {\n          parts.push({\n            type: partType,\n            source: {\n              type: \"data\",\n              value: source.value,\n              mimeType: source.mimeType,\n            },\n          });\n        } else if (source.type === \"url\") {\n          parts.push({\n            type: partType,\n            source: {\n              type: \"url\",\n              value: source.value,\n              ...(source.mimeType ? { mimeType: source.mimeType } : {}),\n            },\n          });\n        }\n        break;\n      }\n\n      // Legacy BinaryInputContent backward compatibility\n      case \"binary\": {\n        const legacy = part as {\n          mimeType?: string;\n          data?: string;\n          url?: string;\n        };\n        const mimeType = legacy.mimeType ?? \"application/octet-stream\";\n        const isImage = mimeType.startsWith(\"image/\");\n\n        if (legacy.data) {\n          const partType = isImage ? \"image\" : \"document\";\n          parts.push({\n            type: partType,\n            source: { type: \"data\", value: legacy.data, mimeType },\n          });\n        } else if (legacy.url) {\n          const partType = isImage ? \"image\" : \"document\";\n          parts.push({\n            type: partType,\n            source: { type: \"url\", value: legacy.url, mimeType },\n          });\n        }\n        break;\n      }\n    }\n  }\n\n  return parts.length > 0 ? parts : \"\";\n}\n\n/**\n * Recursively normalizes a frontend tool's JSON Schema so OpenAI accepts it as\n * a function-tool schema.\n *\n * Frontend tools are often authored with permissive Zod (`z.any()`,\n * `z.record(...)`, `.passthrough()`), which serialize to open objects —\n * `additionalProperties: {}` (an empty sub-schema) or `additionalProperties:\n * true`. OpenAI rejects both: strict mode requires `additionalProperties:\n * false`, and an empty `{}` sub-schema fails base validation (\"schema must\n * have a 'type' key\"). The classic (Vercel AI SDK) path sanitized these\n * implicitly via a Zod round-trip; the TanStack path forwards the raw schema,\n * so we close open objects here to match. (Models can't supply free-form extra\n * keys either way — same as the classic path.)\n */\nfunction sanitizeClientToolSchema(schema: unknown): unknown {\n  if (Array.isArray(schema)) {\n    return schema.map(sanitizeClientToolSchema);\n  }\n  if (!schema || typeof schema !== \"object\") {\n    return schema;\n  }\n  const node: Record<string, unknown> = {\n    ...(schema as Record<string, unknown>),\n  };\n\n  // Any `additionalProperties` (empty `{}`, `true`, or a sub-schema) becomes\n  // `false` — the only form OpenAI accepts for strict function tools.\n  if (\"additionalProperties\" in node) {\n    node.additionalProperties = false;\n  }\n\n  if (node.properties && typeof node.properties === \"object\") {\n    const props: Record<string, unknown> = {};\n    for (const [key, value] of Object.entries(\n      node.properties as Record<string, unknown>,\n    )) {\n      props[key] = sanitizeClientToolSchema(value);\n    }\n    node.properties = props;\n  }\n\n  if (\"items\" in node) {\n    node.items = sanitizeClientToolSchema(node.items);\n  }\n\n  for (const combinator of [\"anyOf\", \"allOf\", \"oneOf\"] as const) {\n    if (Array.isArray(node[combinator])) {\n      node[combinator] = (node[combinator] as unknown[]).map(\n        sanitizeClientToolSchema,\n      );\n    }\n  }\n\n  return node;\n}\n\n/**\n * Converts a RunAgentInput into the format expected by TanStack AI's `chat()`.\n *\n * - Keeps only user/assistant/tool messages (activity, reasoning, and other roles are also excluded)\n * - Extracts system/developer messages into `systemPrompts`\n * - Appends context entries and application state to `systemPrompts`\n * - Preserves tool calls on assistant messages and toolCallId on tool messages\n */\nexport function convertInputToTanStackAI(\n  input: RunAgentInput,\n): TanStackInputResult {\n  // Allowlist: only pass user/assistant/tool messages to TanStack.\n  // Other roles (system, developer, activity, reasoning) are either\n  // extracted into systemPrompts or not applicable.\n  const chatRoles = new Set([\"user\", \"assistant\", \"tool\"]);\n  const messages: TanStackChatMessage[] = input.messages\n    .filter((m: Message) => chatRoles.has(m.role))\n    .map((m: Message): TanStackChatMessage => {\n      const msg: TanStackChatMessage = {\n        role: m.role as \"user\" | \"assistant\" | \"tool\",\n        content:\n          m.role === \"user\"\n            ? convertUserContent(m.content)\n            : typeof m.content === \"string\"\n              ? m.content\n              : null,\n      };\n      if (m.role === \"assistant\" && \"toolCalls\" in m && m.toolCalls) {\n        msg.toolCalls = m.toolCalls.map((tc) => ({\n          id: tc.id,\n          type: \"function\" as const,\n          function: {\n            name: tc.function.name,\n            arguments: tc.function.arguments,\n          },\n        }));\n      }\n      if (m.role === \"tool\" && \"toolCallId\" in m) {\n        msg.toolCallId = (m as Record<string, unknown>).toolCallId as string;\n      }\n      return msg;\n    });\n\n  const systemPrompts: string[] = [];\n  for (const m of input.messages) {\n    if ((m.role === \"system\" || m.role === \"developer\") && m.content) {\n      systemPrompts.push(\n        typeof m.content === \"string\" ? m.content : JSON.stringify(m.content),\n      );\n    }\n  }\n\n  if (input.context?.length) {\n    for (const ctx of input.context) {\n      systemPrompts.push(`${ctx.description}:\\n${ctx.value}`);\n    }\n  }\n\n  if (\n    input.state !== undefined &&\n    input.state !== null &&\n    typeof input.state === \"object\" &&\n    Object.keys(input.state).length > 0\n  ) {\n    systemPrompts.push(\n      `Application State:\\n\\`\\`\\`json\\n${JSON.stringify(input.state, null, 2)}\\n\\`\\`\\``,\n    );\n  }\n\n  // Frontend-provided tools become client-side TanStack tools (no executor):\n  // the model can call them, TanStack pauses the run, and the AG-UI client\n  // executes them and resumes — the CopilotKit client-tool round-trip.\n  const tools: TanStackClientTool[] = (input.tools ?? []).map((t) => ({\n    __toolSide: \"client\",\n    name: t.name,\n    description: t.description,\n    inputSchema: sanitizeClientToolSchema(t.parameters),\n  }));\n\n  return { messages, systemPrompts, tools };\n}\n\n/**\n * Converts a TanStack AI stream into AG-UI `BaseEvent` objects.\n *\n * This is a pure converter — it does NOT emit lifecycle events\n * (RUN_STARTED / RUN_FINISHED / RUN_ERROR). The caller (Agent class)\n * is responsible for those.\n *\n * `pendingInterrupts`, when provided, is filled with one AG-UI Interrupt per\n * CUSTOM \"approval-requested\" chunk (a tool declared `needsApproval: true`).\n * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.\n */\nexport async function* convertTanStackStream(\n  stream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n  pendingInterrupts?: Interrupt[],\n): AsyncGenerator<BaseEvent> {\n  const messageId = randomUUID();\n  const toolNamesById = new Map<string, string>();\n  // Track the reasoning lifecycle at two granularities so closeReasoningIfOpen\n  // emits exactly the events still owed. A single boolean conflates the run\n  // (REASONING_START → REASONING_END) with the message\n  // (REASONING_MESSAGE_START → REASONING_MESSAGE_END) and produces a duplicate\n  // REASONING_MESSAGE_END when upstream emits MSG_END but not END before\n  // text/tools resume.\n  let reasoningRunOpen = false;\n  let reasoningMessageOpen = false;\n  let reasoningMessageId = randomUUID();\n\n  function* closeReasoningIfOpen(): Generator<BaseEvent> {\n    if (reasoningMessageOpen) {\n      reasoningMessageOpen = false;\n      const msgEnd: ReasoningMessageEndEvent = {\n        type: EventType.REASONING_MESSAGE_END,\n        messageId: reasoningMessageId,\n      };\n      yield msgEnd;\n    }\n    if (reasoningRunOpen) {\n      reasoningRunOpen = false;\n      const end: ReasoningEndEvent = {\n        type: EventType.REASONING_END,\n        messageId: reasoningMessageId,\n      };\n      yield end;\n    }\n  }\n\n  // TanStack's chat() engine runs a multi-turn agent loop and emits a\n  // RUN_STARTED / RUN_FINISHED pair PER model turn — not once for the whole\n  // run. When it executes a tool itself (an MCP server tool or a provider tool\n  // like web_search), it does so between turns and streams a TOOL_CALL_RESULT\n  // followed by the next turn's text. The overall run lifecycle is owned by the\n  // Agent wrapper (it emits exactly one outer RUN_STARTED / RUN_FINISHED), so\n  // we drop TanStack's per-turn lifecycle markers and convert every content\n  // event across all turns. (A previous version stopped converting at the first\n  // RUN_FINISHED — that truncated the run at the first tool turn and silently\n  // dropped both the tool result and the model's final answer.)\n  //\n  // chat() can re-announce a tool call when it re-prompts after executing it,\n  // so START / END are de-duplicated by toolCallId to avoid emitting a pair\n  // twice (which would violate the ag-ui verify middleware).\n  const startedToolCalls = new Set<string>();\n  const endedToolCalls = new Set<string>();\n\n  for await (const chunk of stream) {\n    if (abortSignal.aborted) break;\n\n    const raw = chunk as Record<string, unknown>;\n    const type = raw.type as string;\n\n    // TanStack native human-in-the-loop: a tool declared `needsApproval: true`\n    // emits a CUSTOM \"approval-requested\" chunk. These are built from the\n    // finish event and can arrive around lifecycle markers, so handle them\n    // before dropping TanStack's per-turn lifecycle events.\n    // The tool-call lifecycle was already streamed in the model pass.\n    if (type === \"CUSTOM\" && raw.name === \"approval-requested\") {\n      const value = (raw.value ?? {}) as {\n        toolCallId?: string;\n        toolName?: string;\n      };\n      const toolCallId = value.toolCallId;\n      if (toolCallId) {\n        pendingInterrupts?.push({\n          id: toolCallId,\n          toolCallId,\n          reason: \"tool_approval\",\n          message: value.toolName ? `Approve \"${value.toolName}\"?` : undefined,\n          ...(value.toolName ? { metadata: { toolName: value.toolName } } : {}),\n        });\n      }\n      continue;\n    }\n\n    // Per-turn lifecycle markers are owned by the Agent wrapper, not forwarded.\n    if (type === \"RUN_STARTED\" || type === \"RUN_FINISHED\") {\n      continue;\n    }\n\n    // Surface engine errors instead of dropping them: throw so the Agent\n    // wrapper emits a terminal RUN_ERROR. Without this a failed run (e.g. a\n    // provider 4xx) would finish empty with no indication of what went wrong.\n    if (type === \"RUN_ERROR\") {\n      throw new Error(\n        typeof raw.message === \"string\" ? raw.message : \"TanStack AI run error\",\n      );\n    }\n\n    if (type === \"TEXT_MESSAGE_CONTENT\" && raw.delta != null) {\n      yield* closeReasoningIfOpen();\n      const textEvent: TextMessageChunkEvent = {\n        type: EventType.TEXT_MESSAGE_CHUNK,\n        role: \"assistant\",\n        messageId,\n        delta: raw.delta as string,\n      };\n      yield textEvent;\n    } else if (type === \"TOOL_CALL_START\") {\n      const toolCallId = raw.toolCallId as string;\n      if (startedToolCalls.has(toolCallId)) continue;\n      startedToolCalls.add(toolCallId);\n      yield* closeReasoningIfOpen();\n      toolNamesById.set(toolCallId, raw.toolCallName as string);\n      const startEvent: ToolCallStartEvent = {\n        type: EventType.TOOL_CALL_START,\n        parentMessageId: messageId,\n        toolCallId,\n        toolCallName: raw.toolCallName as string,\n      };\n      yield startEvent;\n    } else if (type === \"TOOL_CALL_ARGS\") {\n      // Drop args re-announced after the call has ended (the re-prompt pass);\n      // forwarding them would corrupt the already-closed call's accumulated args.\n      if (endedToolCalls.has(raw.toolCallId as string)) continue;\n      yield* closeReasoningIfOpen();\n      const argsEvent: ToolCallArgsEvent = {\n        type: EventType.TOOL_CALL_ARGS,\n        toolCallId: raw.toolCallId as string,\n        delta: raw.delta as string,\n      };\n      yield argsEvent;\n    } else if (type === \"TOOL_CALL_END\") {\n      const toolCallId = raw.toolCallId as string;\n      if (endedToolCalls.has(toolCallId)) continue;\n      endedToolCalls.add(toolCallId);\n      yield* closeReasoningIfOpen();\n      const endEvent: ToolCallEndEvent = {\n        type: EventType.TOOL_CALL_END,\n        toolCallId,\n      };\n      yield endEvent;\n    } else if (type === \"TOOL_CALL_RESULT\") {\n      yield* closeReasoningIfOpen();\n      const toolCallId = raw.toolCallId as string;\n      const toolName = toolNamesById.get(toolCallId);\n      // Accept the payload from either `content` (canonical TanStack shape)\n      // or `result` (alternate shape used by some adapters / tests). Both\n      // state-tool detection and the final TOOL_CALL_RESULT serialization\n      // must read the same field, otherwise STATE_SNAPSHOT/STATE_DELTA can\n      // be silently dropped when upstream uses `result`.\n      const rawPayload = raw.content ?? raw.result;\n\n      const parsedContent =\n        typeof rawPayload === \"string\" ? safeParse(rawPayload) : rawPayload;\n\n      if (\n        toolName === \"AGUISendStateSnapshot\" &&\n        parsedContent &&\n        typeof parsedContent === \"object\" &&\n        \"snapshot\" in parsedContent\n      ) {\n        const stateSnapshotEvent: StateSnapshotEvent = {\n          type: EventType.STATE_SNAPSHOT,\n          snapshot: (parsedContent as Record<string, unknown>).snapshot,\n        };\n        yield stateSnapshotEvent;\n      }\n\n      if (\n        toolName === \"AGUISendStateDelta\" &&\n        parsedContent &&\n        typeof parsedContent === \"object\" &&\n        \"delta\" in parsedContent\n      ) {\n        const stateDeltaEvent: StateDeltaEvent = {\n          type: EventType.STATE_DELTA,\n          delta: (parsedContent as Record<string, unknown>).delta as never,\n        };\n        yield stateDeltaEvent;\n      }\n\n      let serializedContent: string;\n      if (typeof rawPayload === \"string\") {\n        serializedContent = rawPayload;\n      } else {\n        try {\n          serializedContent = JSON.stringify(rawPayload ?? null);\n        } catch {\n          serializedContent = \"[Unserializable tool result]\";\n        }\n      }\n\n      const resultEvent: ToolCallResultEvent = {\n        type: EventType.TOOL_CALL_RESULT,\n        role: \"tool\",\n        messageId: randomUUID(),\n        toolCallId,\n        content: serializedContent,\n      };\n      yield resultEvent;\n      toolNamesById.delete(toolCallId);\n    } else if (type === \"REASONING_START\") {\n      // If a prior reasoning run is still open (no REASONING_END before this\n      // new START), close it cleanly first so MSG_END / END pair correctly.\n      yield* closeReasoningIfOpen();\n      reasoningRunOpen = true;\n      reasoningMessageId = (raw.messageId as string) ?? randomUUID();\n      const startEvt: ReasoningStartEvent = {\n        type: EventType.REASONING_START,\n        messageId: reasoningMessageId,\n      };\n      yield startEvt;\n    } else if (type === \"REASONING_MESSAGE_START\") {\n      reasoningMessageOpen = true;\n      const evt: ReasoningMessageStartEvent = {\n        type: EventType.REASONING_MESSAGE_START,\n        messageId: reasoningMessageId,\n        role: \"reasoning\",\n      };\n      yield evt;\n    } else if (type === \"REASONING_MESSAGE_CONTENT\") {\n      const evt: ReasoningMessageContentEvent = {\n        type: EventType.REASONING_MESSAGE_CONTENT,\n        messageId: reasoningMessageId,\n        delta: raw.delta as string,\n      };\n      yield evt;\n    } else if (type === \"REASONING_MESSAGE_END\") {\n      reasoningMessageOpen = false;\n      const evt: ReasoningMessageEndEvent = {\n        type: EventType.REASONING_MESSAGE_END,\n        messageId: reasoningMessageId,\n      };\n      yield evt;\n    } else if (type === \"REASONING_END\") {\n      // If upstream sends REASONING_END while a message is still open, emit\n      // the missing REASONING_MESSAGE_END FIRST so the closing pair stays in\n      // order (MSG_END before END). Otherwise the next non-reasoning chunk\n      // would trigger closeReasoningIfOpen and emit MSG_END after END.\n      if (reasoningMessageOpen) {\n        reasoningMessageOpen = false;\n        const msgEnd: ReasoningMessageEndEvent = {\n          type: EventType.REASONING_MESSAGE_END,\n          messageId: reasoningMessageId,\n        };\n        yield msgEnd;\n      }\n      reasoningRunOpen = false;\n      const evt: ReasoningEndEvent = {\n        type: EventType.REASONING_END,\n        messageId: reasoningMessageId,\n      };\n      yield evt;\n    }\n  }\n\n  yield* closeReasoningIfOpen();\n}\n\nfunction safeParse(value: string): unknown {\n  try {\n    return JSON.parse(value);\n  } catch {\n    return value;\n  }\n}\n"],"mappings":";;;;;;;;;;;AAiGA,SAAS,mBACP,SACuC;AACvC,KAAI,CAAC,QAAS,QAAO;AACrB,KAAI,OAAO,YAAY,SAAU,QAAO;AACxC,KAAI,CAAC,MAAM,QAAQ,QAAQ,CAAE,QAAO;AACpC,KAAI,QAAQ,WAAW,EAAG,QAAO;CAEjC,MAAM,QAA+B,EAAE;AAEvC,MAAK,MAAM,QAAQ,SAAS;AAC1B,MAAI,CAAC,QAAQ,OAAO,SAAS,YAAY,EAAE,UAAU,MAAO;AAE5D,UAAS,KAA0B,MAAnC;GACE,KAAK,QAAQ;IACX,MAAM,OAAQ,KAA2B;AACzC,QAAI,QAAQ,KAAM,OAAM,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;AAC7D;;GAGF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK,YAAY;IACf,MAAM,SAAU,KAA0B;AAC1C,QAAI,CAAC,OAAQ;IACb,MAAM,WAAY,KAA0B;AAK5C,QAAI,OAAO,SAAS,OAClB,OAAM,KAAK;KACT,MAAM;KACN,QAAQ;MACN,MAAM;MACN,OAAO,OAAO;MACd,UAAU,OAAO;MAClB;KACF,CAAC;aACO,OAAO,SAAS,MACzB,OAAM,KAAK;KACT,MAAM;KACN,QAAQ;MACN,MAAM;MACN,OAAO,OAAO;MACd,GAAI,OAAO,WAAW,EAAE,UAAU,OAAO,UAAU,GAAG,EAAE;MACzD;KACF,CAAC;AAEJ;;GAIF,KAAK,UAAU;IACb,MAAM,SAAS;IAKf,MAAM,WAAW,OAAO,YAAY;IACpC,MAAM,UAAU,SAAS,WAAW,SAAS;AAE7C,QAAI,OAAO,MAAM;KACf,MAAM,WAAW,UAAU,UAAU;AACrC,WAAM,KAAK;MACT,MAAM;MACN,QAAQ;OAAE,MAAM;OAAQ,OAAO,OAAO;OAAM;OAAU;MACvD,CAAC;eACO,OAAO,KAAK;KACrB,MAAM,WAAW,UAAU,UAAU;AACrC,WAAM,KAAK;MACT,MAAM;MACN,QAAQ;OAAE,MAAM;OAAO,OAAO,OAAO;OAAK;OAAU;MACrD,CAAC;;AAEJ;;;;AAKN,QAAO,MAAM,SAAS,IAAI,QAAQ;;;;;;;;;;;;;;;;AAiBpC,SAAS,yBAAyB,QAA0B;AAC1D,KAAI,MAAM,QAAQ,OAAO,CACvB,QAAO,OAAO,IAAI,yBAAyB;AAE7C,KAAI,CAAC,UAAU,OAAO,WAAW,SAC/B,QAAO;CAET,MAAM,OAAgC,EACpC,GAAI,QACL;AAID,KAAI,0BAA0B,KAC5B,MAAK,uBAAuB;AAG9B,KAAI,KAAK,cAAc,OAAO,KAAK,eAAe,UAAU;EAC1D,MAAM,QAAiC,EAAE;AACzC,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAChC,KAAK,WACN,CACC,OAAM,OAAO,yBAAyB,MAAM;AAE9C,OAAK,aAAa;;AAGpB,KAAI,WAAW,KACb,MAAK,QAAQ,yBAAyB,KAAK,MAAM;AAGnD,MAAK,MAAM,cAAc;EAAC;EAAS;EAAS;EAAQ,CAClD,KAAI,MAAM,QAAQ,KAAK,YAAY,CACjC,MAAK,cAAe,KAAK,YAA0B,IACjD,yBACD;AAIL,QAAO;;;;;;;;;;AAWT,SAAgB,yBACd,OACqB;CAIrB,MAAM,YAAY,IAAI,IAAI;EAAC;EAAQ;EAAa;EAAO,CAAC;CACxD,MAAM,WAAkC,MAAM,SAC3C,QAAQ,MAAe,UAAU,IAAI,EAAE,KAAK,CAAC,CAC7C,KAAK,MAAoC;EACxC,MAAM,MAA2B;GAC/B,MAAM,EAAE;GACR,SACE,EAAE,SAAS,SACP,mBAAmB,EAAE,QAAQ,GAC7B,OAAO,EAAE,YAAY,WACnB,EAAE,UACF;GACT;AACD,MAAI,EAAE,SAAS,eAAe,eAAe,KAAK,EAAE,UAClD,KAAI,YAAY,EAAE,UAAU,KAAK,QAAQ;GACvC,IAAI,GAAG;GACP,MAAM;GACN,UAAU;IACR,MAAM,GAAG,SAAS;IAClB,WAAW,GAAG,SAAS;IACxB;GACF,EAAE;AAEL,MAAI,EAAE,SAAS,UAAU,gBAAgB,EACvC,KAAI,aAAc,EAA8B;AAElD,SAAO;GACP;CAEJ,MAAM,gBAA0B,EAAE;AAClC,MAAK,MAAM,KAAK,MAAM,SACpB,MAAK,EAAE,SAAS,YAAY,EAAE,SAAS,gBAAgB,EAAE,QACvD,eAAc,KACZ,OAAO,EAAE,YAAY,WAAW,EAAE,UAAU,KAAK,UAAU,EAAE,QAAQ,CACtE;AAIL,KAAI,MAAM,SAAS,OACjB,MAAK,MAAM,OAAO,MAAM,QACtB,eAAc,KAAK,GAAG,IAAI,YAAY,KAAK,IAAI,QAAQ;AAI3D,KACE,MAAM,UAAU,UAChB,MAAM,UAAU,QAChB,OAAO,MAAM,UAAU,YACvB,OAAO,KAAK,MAAM,MAAM,CAAC,SAAS,EAElC,eAAc,KACZ,mCAAmC,KAAK,UAAU,MAAM,OAAO,MAAM,EAAE,CAAC,UACzE;AAaH,QAAO;EAAE;EAAU;EAAe,QAPG,MAAM,SAAS,EAAE,EAAE,KAAK,OAAO;GAClE,YAAY;GACZ,MAAM,EAAE;GACR,aAAa,EAAE;GACf,aAAa,yBAAyB,EAAE,WAAW;GACpD,EAAE;EAEsC;;;;;;;;;;;;;AAc3C,gBAAuB,sBACrB,QACA,aACA,mBAC2B;CAC3B,MAAM,gDAAwB;CAC9B,MAAM,gCAAgB,IAAI,KAAqB;CAO/C,IAAI,mBAAmB;CACvB,IAAI,uBAAuB;CAC3B,IAAI,yDAAiC;CAErC,UAAU,uBAA6C;AACrD,MAAI,sBAAsB;AACxB,0BAAuB;AAKvB,SAJyC;IACvC,MAAMA,wBAAU;IAChB,WAAW;IACZ;;AAGH,MAAI,kBAAkB;AACpB,sBAAmB;AAKnB,SAJ+B;IAC7B,MAAMA,wBAAU;IAChB,WAAW;IACZ;;;CAmBL,MAAM,mCAAmB,IAAI,KAAa;CAC1C,MAAM,iCAAiB,IAAI,KAAa;AAExC,YAAW,MAAM,SAAS,QAAQ;AAChC,MAAI,YAAY,QAAS;EAEzB,MAAM,MAAM;EACZ,MAAM,OAAO,IAAI;AAOjB,MAAI,SAAS,YAAY,IAAI,SAAS,sBAAsB;GAC1D,MAAM,QAAS,IAAI,SAAS,EAAE;GAI9B,MAAM,aAAa,MAAM;AACzB,OAAI,WACF,oBAAmB,KAAK;IACtB,IAAI;IACJ;IACA,QAAQ;IACR,SAAS,MAAM,WAAW,YAAY,MAAM,SAAS,MAAM;IAC3D,GAAI,MAAM,WAAW,EAAE,UAAU,EAAE,UAAU,MAAM,UAAU,EAAE,GAAG,EAAE;IACrE,CAAC;AAEJ;;AAIF,MAAI,SAAS,iBAAiB,SAAS,eACrC;AAMF,MAAI,SAAS,YACX,OAAM,IAAI,MACR,OAAO,IAAI,YAAY,WAAW,IAAI,UAAU,wBACjD;AAGH,MAAI,SAAS,0BAA0B,IAAI,SAAS,MAAM;AACxD,UAAO,sBAAsB;AAO7B,SANyC;IACvC,MAAMA,wBAAU;IAChB,MAAM;IACN;IACA,OAAO,IAAI;IACZ;aAEQ,SAAS,mBAAmB;GACrC,MAAM,aAAa,IAAI;AACvB,OAAI,iBAAiB,IAAI,WAAW,CAAE;AACtC,oBAAiB,IAAI,WAAW;AAChC,UAAO,sBAAsB;AAC7B,iBAAc,IAAI,YAAY,IAAI,aAAuB;AAOzD,SANuC;IACrC,MAAMA,wBAAU;IAChB,iBAAiB;IACjB;IACA,cAAc,IAAI;IACnB;aAEQ,SAAS,kBAAkB;AAGpC,OAAI,eAAe,IAAI,IAAI,WAAqB,CAAE;AAClD,UAAO,sBAAsB;AAM7B,SALqC;IACnC,MAAMA,wBAAU;IAChB,YAAY,IAAI;IAChB,OAAO,IAAI;IACZ;aAEQ,SAAS,iBAAiB;GACnC,MAAM,aAAa,IAAI;AACvB,OAAI,eAAe,IAAI,WAAW,CAAE;AACpC,kBAAe,IAAI,WAAW;AAC9B,UAAO,sBAAsB;AAK7B,SAJmC;IACjC,MAAMA,wBAAU;IAChB;IACD;aAEQ,SAAS,oBAAoB;AACtC,UAAO,sBAAsB;GAC7B,MAAM,aAAa,IAAI;GACvB,MAAM,WAAW,cAAc,IAAI,WAAW;GAM9C,MAAM,aAAa,IAAI,WAAW,IAAI;GAEtC,MAAM,gBACJ,OAAO,eAAe,WAAW,UAAU,WAAW,GAAG;AAE3D,OACE,aAAa,2BACb,iBACA,OAAO,kBAAkB,YACzB,cAAc,cAMd,OAJ+C;IAC7C,MAAMA,wBAAU;IAChB,UAAW,cAA0C;IACtD;AAIH,OACE,aAAa,wBACb,iBACA,OAAO,kBAAkB,YACzB,WAAW,cAMX,OAJyC;IACvC,MAAMA,wBAAU;IAChB,OAAQ,cAA0C;IACnD;GAIH,IAAI;AACJ,OAAI,OAAO,eAAe,SACxB,qBAAoB;OAEpB,KAAI;AACF,wBAAoB,KAAK,UAAU,cAAc,KAAK;WAChD;AACN,wBAAoB;;AAWxB,SAPyC;IACvC,MAAMA,wBAAU;IAChB,MAAM;IACN,+CAAuB;IACvB;IACA,SAAS;IACV;AAED,iBAAc,OAAO,WAAW;aACvB,SAAS,mBAAmB;AAGrC,UAAO,sBAAsB;AAC7B,sBAAmB;AACnB,wBAAsB,IAAI,iDAAoC;AAK9D,SAJsC;IACpC,MAAMA,wBAAU;IAChB,WAAW;IACZ;aAEQ,SAAS,2BAA2B;AAC7C,0BAAuB;AAMvB,SALwC;IACtC,MAAMA,wBAAU;IAChB,WAAW;IACX,MAAM;IACP;aAEQ,SAAS,4BAMlB,OAL0C;GACxC,MAAMA,wBAAU;GAChB,WAAW;GACX,OAAO,IAAI;GACZ;WAEQ,SAAS,yBAAyB;AAC3C,0BAAuB;AAKvB,SAJsC;IACpC,MAAMA,wBAAU;IAChB,WAAW;IACZ;aAEQ,SAAS,iBAAiB;AAKnC,OAAI,sBAAsB;AACxB,2BAAuB;AAKvB,UAJyC;KACvC,MAAMA,wBAAU;KAChB,WAAW;KACZ;;AAGH,sBAAmB;AAKnB,SAJ+B;IAC7B,MAAMA,wBAAU;IAChB,WAAW;IACZ;;;AAKL,QAAO,sBAAsB;;AAG/B,SAAS,UAAU,OAAwB;AACzC,KAAI;AACF,SAAO,KAAK,MAAM,MAAM;SAClB;AACN,SAAO"}
+{"version":3,"file":"tanstack.cjs","names":["createStateEventNormalizer","EventType"],"sources":["../../../src/agent/converters/tanstack.ts"],"sourcesContent":["import type {\n  BaseEvent,\n  Interrupt,\n  RunAgentInput,\n  Message,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n  StateSnapshotEvent,\n  StateDeltaEvent,\n  ReasoningStartEvent,\n  ReasoningMessageStartEvent,\n  ReasoningMessageContentEvent,\n  ReasoningMessageEndEvent,\n  ReasoningEndEvent,\n} from \"@ag-ui/client\";\nimport { EventType } from \"@ag-ui/client\";\nimport { randomUUID } from \"@copilotkit/shared\";\nimport { createStateEventNormalizer } from \"../state-delta\";\n\ntype ContentPartSource =\n  | { type: \"data\"; value: string; mimeType: string }\n  | { type: \"url\"; value: string; mimeType?: string };\n\n/**\n * A TanStack AI content part (text, image, audio, video, or document).\n */\nexport type TanStackContentPart =\n  | { type: \"text\"; content: string }\n  | { type: \"image\"; source: ContentPartSource }\n  | { type: \"audio\"; source: ContentPartSource }\n  | { type: \"video\"; source: ContentPartSource }\n  | { type: \"document\"; source: ContentPartSource };\n\n/**\n * Message format expected by TanStack AI's `chat()`.\n *\n * Content is typed as `any[]` for the multimodal case so messages are directly\n * passable to any adapter without casts — different adapters constrain which\n * modalities they accept (e.g. OpenAI only allows text + image).\n * Use `TanStackContentPart` to inspect individual parts if needed.\n */\nexport interface TanStackChatMessage {\n  role: \"user\" | \"assistant\" | \"tool\";\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  content: string | null | any[];\n  name?: string;\n  toolCalls?: Array<{\n    id: string;\n    type: \"function\";\n    function: { name: string; arguments: string };\n  }>;\n  toolCallId?: string;\n}\n\n/**\n * A TanStack AI client-side tool, derived from a frontend-provided AG-UI tool.\n *\n * Shaped to match `@tanstack/ai`'s `ClientTool` (`__toolSide: \"client\"`, no\n * `execute`): the model may CALL it, but TanStack does not run it — it pauses\n * the run and hands the call back to the AG-UI client (the CopilotKit frontend\n * / bot) to execute, mirroring CopilotKit's client-tool round-trip. `chat()`\n * accepts a JSON Schema directly as `inputSchema`, so the AG-UI tool's\n * `parameters` pass through unchanged.\n */\nexport interface TanStackClientTool {\n  __toolSide: \"client\";\n  name: string;\n  description: string;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  inputSchema: any;\n}\n\n/**\n * Result of converting RunAgentInput to TanStack AI format.\n */\nexport interface TanStackInputResult {\n  /** Chat messages (only user/assistant/tool roles; all others excluded) */\n  messages: TanStackChatMessage[];\n  /** System prompts extracted from system/developer messages, context, and state */\n  systemPrompts: string[];\n  /**\n   * Client-side tools derived from `input.tools` (the frontend-provided tools\n   * the CopilotKit client forwards on every run). Pass these into `chat()`\n   * alongside any server/provider tools so the model can call the frontend's\n   * generative-UI and human-in-the-loop tools; TanStack pauses the run on a\n   * client-tool call and the client executes it.\n   */\n  tools: TanStackClientTool[];\n}\n\n/**\n * Converts AG-UI user message content to TanStack AI format.\n * Handles plain strings, multimodal parts (image/audio/video/document),\n * and legacy BinaryInputContent for backward compatibility.\n */\nfunction convertUserContent(\n  content: unknown,\n): string | null | TanStackContentPart[] {\n  if (!content) return null;\n  if (typeof content === \"string\") return content;\n  if (!Array.isArray(content)) return null;\n  if (content.length === 0) return \"\";\n\n  const parts: TanStackContentPart[] = [];\n\n  for (const part of content) {\n    if (!part || typeof part !== \"object\" || !(\"type\" in part)) continue;\n\n    switch ((part as { type: string }).type) {\n      case \"text\": {\n        const text = (part as { text?: string }).text;\n        if (text != null) parts.push({ type: \"text\", content: text });\n        break;\n      }\n\n      case \"image\":\n      case \"audio\":\n      case \"video\":\n      case \"document\": {\n        const source = (part as { source?: any }).source;\n        if (!source) break;\n        const partType = (part as { type: string }).type as\n          | \"image\"\n          | \"audio\"\n          | \"video\"\n          | \"document\";\n        if (source.type === \"data\") {\n          parts.push({\n            type: partType,\n            source: {\n              type: \"data\",\n              value: source.value,\n              mimeType: source.mimeType,\n            },\n          });\n        } else if (source.type === \"url\") {\n          parts.push({\n            type: partType,\n            source: {\n              type: \"url\",\n              value: source.value,\n              ...(source.mimeType ? { mimeType: source.mimeType } : {}),\n            },\n          });\n        }\n        break;\n      }\n\n      // Legacy BinaryInputContent backward compatibility\n      case \"binary\": {\n        const legacy = part as {\n          mimeType?: string;\n          data?: string;\n          url?: string;\n        };\n        const mimeType = legacy.mimeType ?? \"application/octet-stream\";\n        const isImage = mimeType.startsWith(\"image/\");\n\n        if (legacy.data) {\n          const partType = isImage ? \"image\" : \"document\";\n          parts.push({\n            type: partType,\n            source: { type: \"data\", value: legacy.data, mimeType },\n          });\n        } else if (legacy.url) {\n          const partType = isImage ? \"image\" : \"document\";\n          parts.push({\n            type: partType,\n            source: { type: \"url\", value: legacy.url, mimeType },\n          });\n        }\n        break;\n      }\n    }\n  }\n\n  return parts.length > 0 ? parts : \"\";\n}\n\n/**\n * Recursively normalizes a frontend tool's JSON Schema so OpenAI accepts it as\n * a function-tool schema.\n *\n * Frontend tools are often authored with permissive Zod (`z.any()`,\n * `z.record(...)`, `.passthrough()`), which serialize to open objects —\n * `additionalProperties: {}` (an empty sub-schema) or `additionalProperties:\n * true`. OpenAI rejects both: strict mode requires `additionalProperties:\n * false`, and an empty `{}` sub-schema fails base validation (\"schema must\n * have a 'type' key\"). The classic (Vercel AI SDK) path sanitized these\n * implicitly via a Zod round-trip; the TanStack path forwards the raw schema,\n * so we close open objects here to match. (Models can't supply free-form extra\n * keys either way — same as the classic path.)\n */\nfunction sanitizeClientToolSchema(schema: unknown): unknown {\n  if (Array.isArray(schema)) {\n    return schema.map(sanitizeClientToolSchema);\n  }\n  if (!schema || typeof schema !== \"object\") {\n    return schema;\n  }\n  const node: Record<string, unknown> = {\n    ...(schema as Record<string, unknown>),\n  };\n\n  // Any `additionalProperties` (empty `{}`, `true`, or a sub-schema) becomes\n  // `false` — the only form OpenAI accepts for strict function tools.\n  if (\"additionalProperties\" in node) {\n    node.additionalProperties = false;\n  }\n\n  if (node.properties && typeof node.properties === \"object\") {\n    const props: Record<string, unknown> = {};\n    for (const [key, value] of Object.entries(\n      node.properties as Record<string, unknown>,\n    )) {\n      props[key] = sanitizeClientToolSchema(value);\n    }\n    node.properties = props;\n  }\n\n  if (\"items\" in node) {\n    node.items = sanitizeClientToolSchema(node.items);\n  }\n\n  for (const combinator of [\"anyOf\", \"allOf\", \"oneOf\"] as const) {\n    if (Array.isArray(node[combinator])) {\n      node[combinator] = (node[combinator] as unknown[]).map(\n        sanitizeClientToolSchema,\n      );\n    }\n  }\n\n  return node;\n}\n\n/**\n * Converts a RunAgentInput into the format expected by TanStack AI's `chat()`.\n *\n * - Keeps only user/assistant/tool messages (activity, reasoning, and other roles are also excluded)\n * - Extracts system/developer messages into `systemPrompts`\n * - Appends context entries and application state to `systemPrompts`\n * - Preserves tool calls on assistant messages and toolCallId on tool messages\n */\nexport function convertInputToTanStackAI(\n  input: RunAgentInput,\n): TanStackInputResult {\n  // Allowlist: only pass user/assistant/tool messages to TanStack.\n  // Other roles (system, developer, activity, reasoning) are either\n  // extracted into systemPrompts or not applicable.\n  const chatRoles = new Set([\"user\", \"assistant\", \"tool\"]);\n  const messages: TanStackChatMessage[] = input.messages\n    .filter((m: Message) => chatRoles.has(m.role))\n    .map((m: Message): TanStackChatMessage => {\n      const msg: TanStackChatMessage = {\n        role: m.role as \"user\" | \"assistant\" | \"tool\",\n        content:\n          m.role === \"user\"\n            ? convertUserContent(m.content)\n            : typeof m.content === \"string\"\n              ? m.content\n              : null,\n      };\n      if (m.role === \"assistant\" && \"toolCalls\" in m && m.toolCalls) {\n        msg.toolCalls = m.toolCalls.map((tc) => ({\n          id: tc.id,\n          type: \"function\" as const,\n          function: {\n            name: tc.function.name,\n            arguments: tc.function.arguments,\n          },\n        }));\n      }\n      if (m.role === \"tool\" && \"toolCallId\" in m) {\n        msg.toolCallId = (m as Record<string, unknown>).toolCallId as string;\n      }\n      return msg;\n    });\n\n  const systemPrompts: string[] = [];\n  for (const m of input.messages) {\n    if ((m.role === \"system\" || m.role === \"developer\") && m.content) {\n      systemPrompts.push(\n        typeof m.content === \"string\" ? m.content : JSON.stringify(m.content),\n      );\n    }\n  }\n\n  if (input.context?.length) {\n    for (const ctx of input.context) {\n      systemPrompts.push(`${ctx.description}:\\n${ctx.value}`);\n    }\n  }\n\n  if (\n    input.state !== undefined &&\n    input.state !== null &&\n    typeof input.state === \"object\" &&\n    Object.keys(input.state).length > 0\n  ) {\n    systemPrompts.push(\n      `Application State:\\n\\`\\`\\`json\\n${JSON.stringify(input.state, null, 2)}\\n\\`\\`\\``,\n    );\n  }\n\n  // Frontend-provided tools become client-side TanStack tools (no executor):\n  // the model can call them, TanStack pauses the run, and the AG-UI client\n  // executes them and resumes — the CopilotKit client-tool round-trip.\n  const tools: TanStackClientTool[] = (input.tools ?? []).map((t) => ({\n    __toolSide: \"client\",\n    name: t.name,\n    description: t.description,\n    inputSchema: sanitizeClientToolSchema(t.parameters),\n  }));\n\n  return { messages, systemPrompts, tools };\n}\n\n/**\n * Converts a TanStack AI stream into AG-UI `BaseEvent` objects.\n *\n * This is a pure converter — it does NOT emit lifecycle events\n * (RUN_STARTED / RUN_FINISHED / RUN_ERROR). The caller (Agent class)\n * is responsible for those.\n *\n * `pendingInterrupts`, when provided, is filled with one AG-UI Interrupt per\n * CUSTOM \"approval-requested\" chunk (a tool declared `needsApproval: true`).\n * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.\n */\nexport async function* convertTanStackStream(\n  stream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n  pendingInterrupts?: Interrupt[],\n  initialState?: unknown,\n): AsyncGenerator<BaseEvent> {\n  const messageId = randomUUID();\n  const toolNamesById = new Map<string, string>();\n  // Track the reasoning lifecycle at two granularities so closeReasoningIfOpen\n  // emits exactly the events still owed. A single boolean conflates the run\n  // (REASONING_START → REASONING_END) with the message\n  // (REASONING_MESSAGE_START → REASONING_MESSAGE_END) and produces a duplicate\n  // REASONING_MESSAGE_END when upstream emits MSG_END but not END before\n  // text/tools resume.\n  let reasoningRunOpen = false;\n  let reasoningMessageOpen = false;\n  let reasoningMessageId = randomUUID();\n  const normalizeStateEvent = createStateEventNormalizer(initialState);\n\n  function* closeReasoningIfOpen(): Generator<BaseEvent> {\n    if (reasoningMessageOpen) {\n      reasoningMessageOpen = false;\n      const msgEnd: ReasoningMessageEndEvent = {\n        type: EventType.REASONING_MESSAGE_END,\n        messageId: reasoningMessageId,\n      };\n      yield msgEnd;\n    }\n    if (reasoningRunOpen) {\n      reasoningRunOpen = false;\n      const end: ReasoningEndEvent = {\n        type: EventType.REASONING_END,\n        messageId: reasoningMessageId,\n      };\n      yield end;\n    }\n  }\n\n  // TanStack's chat() engine runs a multi-turn agent loop and emits a\n  // RUN_STARTED / RUN_FINISHED pair PER model turn — not once for the whole\n  // run. When it executes a tool itself (an MCP server tool or a provider tool\n  // like web_search), it does so between turns and streams a TOOL_CALL_RESULT\n  // followed by the next turn's text. The overall run lifecycle is owned by the\n  // Agent wrapper (it emits exactly one outer RUN_STARTED / RUN_FINISHED), so\n  // we drop TanStack's per-turn lifecycle markers and convert every content\n  // event across all turns. (A previous version stopped converting at the first\n  // RUN_FINISHED — that truncated the run at the first tool turn and silently\n  // dropped both the tool result and the model's final answer.)\n  //\n  // chat() can re-announce a tool call when it re-prompts after executing it,\n  // so START / END are de-duplicated by toolCallId to avoid emitting a pair\n  // twice (which would violate the ag-ui verify middleware).\n  const startedToolCalls = new Set<string>();\n  const endedToolCalls = new Set<string>();\n\n  for await (const chunk of stream) {\n    if (abortSignal.aborted) break;\n\n    const raw = chunk as Record<string, unknown>;\n    const type = raw.type as string;\n\n    // TanStack native human-in-the-loop: a tool declared `needsApproval: true`\n    // emits a CUSTOM \"approval-requested\" chunk. These are built from the\n    // finish event and can arrive around lifecycle markers, so handle them\n    // before dropping TanStack's per-turn lifecycle events.\n    // The tool-call lifecycle was already streamed in the model pass.\n    if (type === \"CUSTOM\" && raw.name === \"approval-requested\") {\n      const value = (raw.value ?? {}) as {\n        toolCallId?: string;\n        toolName?: string;\n      };\n      const toolCallId = value.toolCallId;\n      if (toolCallId) {\n        pendingInterrupts?.push({\n          id: toolCallId,\n          toolCallId,\n          reason: \"tool_approval\",\n          message: value.toolName ? `Approve \"${value.toolName}\"?` : undefined,\n          ...(value.toolName ? { metadata: { toolName: value.toolName } } : {}),\n        });\n      }\n      continue;\n    }\n\n    // Per-turn lifecycle markers are owned by the Agent wrapper, not forwarded.\n    if (type === \"RUN_STARTED\" || type === \"RUN_FINISHED\") {\n      continue;\n    }\n\n    // Surface engine errors instead of dropping them: throw so the Agent\n    // wrapper emits a terminal RUN_ERROR. Without this a failed run (e.g. a\n    // provider 4xx) would finish empty with no indication of what went wrong.\n    if (type === \"RUN_ERROR\") {\n      throw new Error(\n        typeof raw.message === \"string\" ? raw.message : \"TanStack AI run error\",\n      );\n    }\n\n    if (type === \"TEXT_MESSAGE_CONTENT\" && raw.delta != null) {\n      yield* closeReasoningIfOpen();\n      const textEvent: TextMessageChunkEvent = {\n        type: EventType.TEXT_MESSAGE_CHUNK,\n        role: \"assistant\",\n        messageId,\n        delta: raw.delta as string,\n      };\n      yield textEvent;\n    } else if (type === \"TOOL_CALL_START\") {\n      const toolCallId = raw.toolCallId as string;\n      if (startedToolCalls.has(toolCallId)) continue;\n      startedToolCalls.add(toolCallId);\n      yield* closeReasoningIfOpen();\n      toolNamesById.set(toolCallId, raw.toolCallName as string);\n      const startEvent: ToolCallStartEvent = {\n        type: EventType.TOOL_CALL_START,\n        parentMessageId: messageId,\n        toolCallId,\n        toolCallName: raw.toolCallName as string,\n      };\n      yield startEvent;\n    } else if (type === \"TOOL_CALL_ARGS\") {\n      // Drop args re-announced after the call has ended (the re-prompt pass);\n      // forwarding them would corrupt the already-closed call's accumulated args.\n      if (endedToolCalls.has(raw.toolCallId as string)) continue;\n      yield* closeReasoningIfOpen();\n      const argsEvent: ToolCallArgsEvent = {\n        type: EventType.TOOL_CALL_ARGS,\n        toolCallId: raw.toolCallId as string,\n        delta: raw.delta as string,\n      };\n      yield argsEvent;\n    } else if (type === \"TOOL_CALL_END\") {\n      const toolCallId = raw.toolCallId as string;\n      if (endedToolCalls.has(toolCallId)) continue;\n      endedToolCalls.add(toolCallId);\n      yield* closeReasoningIfOpen();\n      const endEvent: ToolCallEndEvent = {\n        type: EventType.TOOL_CALL_END,\n        toolCallId,\n      };\n      yield endEvent;\n    } else if (type === \"TOOL_CALL_RESULT\") {\n      yield* closeReasoningIfOpen();\n      const toolCallId = raw.toolCallId as string;\n      const toolName = toolNamesById.get(toolCallId);\n      // Accept the payload from either `content` (canonical TanStack shape)\n      // or `result` (alternate shape used by some adapters / tests). Both\n      // state-tool detection and the final TOOL_CALL_RESULT serialization\n      // must read the same field, otherwise STATE_SNAPSHOT/STATE_DELTA can\n      // be silently dropped when upstream uses `result`.\n      const rawPayload = raw.content ?? raw.result;\n\n      const parsedContent =\n        typeof rawPayload === \"string\" ? safeParse(rawPayload) : rawPayload;\n\n      if (\n        toolName === \"AGUISendStateSnapshot\" &&\n        parsedContent &&\n        typeof parsedContent === \"object\" &&\n        \"snapshot\" in parsedContent\n      ) {\n        const stateSnapshotEvent: StateSnapshotEvent = {\n          type: EventType.STATE_SNAPSHOT,\n          snapshot: (parsedContent as Record<string, unknown>).snapshot,\n        };\n        for (const event of normalizeStateEvent(stateSnapshotEvent)) {\n          yield event;\n        }\n      }\n\n      if (\n        toolName === \"AGUISendStateDelta\" &&\n        parsedContent &&\n        typeof parsedContent === \"object\" &&\n        \"delta\" in parsedContent\n      ) {\n        const stateDeltaEvent: StateDeltaEvent = {\n          type: EventType.STATE_DELTA,\n          delta: (parsedContent as Record<string, unknown>).delta as never,\n        };\n        for (const event of normalizeStateEvent(stateDeltaEvent)) {\n          yield event;\n        }\n      }\n\n      let serializedContent: string;\n      if (typeof rawPayload === \"string\") {\n        serializedContent = rawPayload;\n      } else {\n        try {\n          serializedContent = JSON.stringify(rawPayload ?? null);\n        } catch {\n          serializedContent = \"[Unserializable tool result]\";\n        }\n      }\n\n      const resultEvent: ToolCallResultEvent = {\n        type: EventType.TOOL_CALL_RESULT,\n        role: \"tool\",\n        messageId: randomUUID(),\n        toolCallId,\n        content: serializedContent,\n      };\n      yield resultEvent;\n      toolNamesById.delete(toolCallId);\n    } else if (type === \"REASONING_START\") {\n      // If a prior reasoning run is still open (no REASONING_END before this\n      // new START), close it cleanly first so MSG_END / END pair correctly.\n      yield* closeReasoningIfOpen();\n      reasoningRunOpen = true;\n      reasoningMessageId = (raw.messageId as string) ?? randomUUID();\n      const startEvt: ReasoningStartEvent = {\n        type: EventType.REASONING_START,\n        messageId: reasoningMessageId,\n      };\n      yield startEvt;\n    } else if (type === \"REASONING_MESSAGE_START\") {\n      reasoningMessageOpen = true;\n      const evt: ReasoningMessageStartEvent = {\n        type: EventType.REASONING_MESSAGE_START,\n        messageId: reasoningMessageId,\n        role: \"reasoning\",\n      };\n      yield evt;\n    } else if (type === \"REASONING_MESSAGE_CONTENT\") {\n      const evt: ReasoningMessageContentEvent = {\n        type: EventType.REASONING_MESSAGE_CONTENT,\n        messageId: reasoningMessageId,\n        delta: raw.delta as string,\n      };\n      yield evt;\n    } else if (type === \"REASONING_MESSAGE_END\") {\n      reasoningMessageOpen = false;\n      const evt: ReasoningMessageEndEvent = {\n        type: EventType.REASONING_MESSAGE_END,\n        messageId: reasoningMessageId,\n      };\n      yield evt;\n    } else if (type === \"REASONING_END\") {\n      // If upstream sends REASONING_END while a message is still open, emit\n      // the missing REASONING_MESSAGE_END FIRST so the closing pair stays in\n      // order (MSG_END before END). Otherwise the next non-reasoning chunk\n      // would trigger closeReasoningIfOpen and emit MSG_END after END.\n      if (reasoningMessageOpen) {\n        reasoningMessageOpen = false;\n        const msgEnd: ReasoningMessageEndEvent = {\n          type: EventType.REASONING_MESSAGE_END,\n          messageId: reasoningMessageId,\n        };\n        yield msgEnd;\n      }\n      reasoningRunOpen = false;\n      const evt: ReasoningEndEvent = {\n        type: EventType.REASONING_END,\n        messageId: reasoningMessageId,\n      };\n      yield evt;\n    }\n  }\n\n  yield* closeReasoningIfOpen();\n}\n\nfunction safeParse(value: string): unknown {\n  try {\n    return JSON.parse(value);\n  } catch {\n    return value;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAkGA,SAAS,mBACP,SACuC;AACvC,KAAI,CAAC,QAAS,QAAO;AACrB,KAAI,OAAO,YAAY,SAAU,QAAO;AACxC,KAAI,CAAC,MAAM,QAAQ,QAAQ,CAAE,QAAO;AACpC,KAAI,QAAQ,WAAW,EAAG,QAAO;CAEjC,MAAM,QAA+B,EAAE;AAEvC,MAAK,MAAM,QAAQ,SAAS;AAC1B,MAAI,CAAC,QAAQ,OAAO,SAAS,YAAY,EAAE,UAAU,MAAO;AAE5D,UAAS,KAA0B,MAAnC;GACE,KAAK,QAAQ;IACX,MAAM,OAAQ,KAA2B;AACzC,QAAI,QAAQ,KAAM,OAAM,KAAK;KAAE,MAAM;KAAQ,SAAS;KAAM,CAAC;AAC7D;;GAGF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK,YAAY;IACf,MAAM,SAAU,KAA0B;AAC1C,QAAI,CAAC,OAAQ;IACb,MAAM,WAAY,KAA0B;AAK5C,QAAI,OAAO,SAAS,OAClB,OAAM,KAAK;KACT,MAAM;KACN,QAAQ;MACN,MAAM;MACN,OAAO,OAAO;MACd,UAAU,OAAO;MAClB;KACF,CAAC;aACO,OAAO,SAAS,MACzB,OAAM,KAAK;KACT,MAAM;KACN,QAAQ;MACN,MAAM;MACN,OAAO,OAAO;MACd,GAAI,OAAO,WAAW,EAAE,UAAU,OAAO,UAAU,GAAG,EAAE;MACzD;KACF,CAAC;AAEJ;;GAIF,KAAK,UAAU;IACb,MAAM,SAAS;IAKf,MAAM,WAAW,OAAO,YAAY;IACpC,MAAM,UAAU,SAAS,WAAW,SAAS;AAE7C,QAAI,OAAO,MAAM;KACf,MAAM,WAAW,UAAU,UAAU;AACrC,WAAM,KAAK;MACT,MAAM;MACN,QAAQ;OAAE,MAAM;OAAQ,OAAO,OAAO;OAAM;OAAU;MACvD,CAAC;eACO,OAAO,KAAK;KACrB,MAAM,WAAW,UAAU,UAAU;AACrC,WAAM,KAAK;MACT,MAAM;MACN,QAAQ;OAAE,MAAM;OAAO,OAAO,OAAO;OAAK;OAAU;MACrD,CAAC;;AAEJ;;;;AAKN,QAAO,MAAM,SAAS,IAAI,QAAQ;;;;;;;;;;;;;;;;AAiBpC,SAAS,yBAAyB,QAA0B;AAC1D,KAAI,MAAM,QAAQ,OAAO,CACvB,QAAO,OAAO,IAAI,yBAAyB;AAE7C,KAAI,CAAC,UAAU,OAAO,WAAW,SAC/B,QAAO;CAET,MAAM,OAAgC,EACpC,GAAI,QACL;AAID,KAAI,0BAA0B,KAC5B,MAAK,uBAAuB;AAG9B,KAAI,KAAK,cAAc,OAAO,KAAK,eAAe,UAAU;EAC1D,MAAM,QAAiC,EAAE;AACzC,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAChC,KAAK,WACN,CACC,OAAM,OAAO,yBAAyB,MAAM;AAE9C,OAAK,aAAa;;AAGpB,KAAI,WAAW,KACb,MAAK,QAAQ,yBAAyB,KAAK,MAAM;AAGnD,MAAK,MAAM,cAAc;EAAC;EAAS;EAAS;EAAQ,CAClD,KAAI,MAAM,QAAQ,KAAK,YAAY,CACjC,MAAK,cAAe,KAAK,YAA0B,IACjD,yBACD;AAIL,QAAO;;;;;;;;;;AAWT,SAAgB,yBACd,OACqB;CAIrB,MAAM,YAAY,IAAI,IAAI;EAAC;EAAQ;EAAa;EAAO,CAAC;CACxD,MAAM,WAAkC,MAAM,SAC3C,QAAQ,MAAe,UAAU,IAAI,EAAE,KAAK,CAAC,CAC7C,KAAK,MAAoC;EACxC,MAAM,MAA2B;GAC/B,MAAM,EAAE;GACR,SACE,EAAE,SAAS,SACP,mBAAmB,EAAE,QAAQ,GAC7B,OAAO,EAAE,YAAY,WACnB,EAAE,UACF;GACT;AACD,MAAI,EAAE,SAAS,eAAe,eAAe,KAAK,EAAE,UAClD,KAAI,YAAY,EAAE,UAAU,KAAK,QAAQ;GACvC,IAAI,GAAG;GACP,MAAM;GACN,UAAU;IACR,MAAM,GAAG,SAAS;IAClB,WAAW,GAAG,SAAS;IACxB;GACF,EAAE;AAEL,MAAI,EAAE,SAAS,UAAU,gBAAgB,EACvC,KAAI,aAAc,EAA8B;AAElD,SAAO;GACP;CAEJ,MAAM,gBAA0B,EAAE;AAClC,MAAK,MAAM,KAAK,MAAM,SACpB,MAAK,EAAE,SAAS,YAAY,EAAE,SAAS,gBAAgB,EAAE,QACvD,eAAc,KACZ,OAAO,EAAE,YAAY,WAAW,EAAE,UAAU,KAAK,UAAU,EAAE,QAAQ,CACtE;AAIL,KAAI,MAAM,SAAS,OACjB,MAAK,MAAM,OAAO,MAAM,QACtB,eAAc,KAAK,GAAG,IAAI,YAAY,KAAK,IAAI,QAAQ;AAI3D,KACE,MAAM,UAAU,UAChB,MAAM,UAAU,QAChB,OAAO,MAAM,UAAU,YACvB,OAAO,KAAK,MAAM,MAAM,CAAC,SAAS,EAElC,eAAc,KACZ,mCAAmC,KAAK,UAAU,MAAM,OAAO,MAAM,EAAE,CAAC,UACzE;AAaH,QAAO;EAAE;EAAU;EAAe,QAPG,MAAM,SAAS,EAAE,EAAE,KAAK,OAAO;GAClE,YAAY;GACZ,MAAM,EAAE;GACR,aAAa,EAAE;GACf,aAAa,yBAAyB,EAAE,WAAW;GACpD,EAAE;EAEsC;;;;;;;;;;;;;AAc3C,gBAAuB,sBACrB,QACA,aACA,mBACA,cAC2B;CAC3B,MAAM,gDAAwB;CAC9B,MAAM,gCAAgB,IAAI,KAAqB;CAO/C,IAAI,mBAAmB;CACvB,IAAI,uBAAuB;CAC3B,IAAI,yDAAiC;CACrC,MAAM,sBAAsBA,+CAA2B,aAAa;CAEpE,UAAU,uBAA6C;AACrD,MAAI,sBAAsB;AACxB,0BAAuB;AAKvB,SAJyC;IACvC,MAAMC,wBAAU;IAChB,WAAW;IACZ;;AAGH,MAAI,kBAAkB;AACpB,sBAAmB;AAKnB,SAJ+B;IAC7B,MAAMA,wBAAU;IAChB,WAAW;IACZ;;;CAmBL,MAAM,mCAAmB,IAAI,KAAa;CAC1C,MAAM,iCAAiB,IAAI,KAAa;AAExC,YAAW,MAAM,SAAS,QAAQ;AAChC,MAAI,YAAY,QAAS;EAEzB,MAAM,MAAM;EACZ,MAAM,OAAO,IAAI;AAOjB,MAAI,SAAS,YAAY,IAAI,SAAS,sBAAsB;GAC1D,MAAM,QAAS,IAAI,SAAS,EAAE;GAI9B,MAAM,aAAa,MAAM;AACzB,OAAI,WACF,oBAAmB,KAAK;IACtB,IAAI;IACJ;IACA,QAAQ;IACR,SAAS,MAAM,WAAW,YAAY,MAAM,SAAS,MAAM;IAC3D,GAAI,MAAM,WAAW,EAAE,UAAU,EAAE,UAAU,MAAM,UAAU,EAAE,GAAG,EAAE;IACrE,CAAC;AAEJ;;AAIF,MAAI,SAAS,iBAAiB,SAAS,eACrC;AAMF,MAAI,SAAS,YACX,OAAM,IAAI,MACR,OAAO,IAAI,YAAY,WAAW,IAAI,UAAU,wBACjD;AAGH,MAAI,SAAS,0BAA0B,IAAI,SAAS,MAAM;AACxD,UAAO,sBAAsB;AAO7B,SANyC;IACvC,MAAMA,wBAAU;IAChB,MAAM;IACN;IACA,OAAO,IAAI;IACZ;aAEQ,SAAS,mBAAmB;GACrC,MAAM,aAAa,IAAI;AACvB,OAAI,iBAAiB,IAAI,WAAW,CAAE;AACtC,oBAAiB,IAAI,WAAW;AAChC,UAAO,sBAAsB;AAC7B,iBAAc,IAAI,YAAY,IAAI,aAAuB;AAOzD,SANuC;IACrC,MAAMA,wBAAU;IAChB,iBAAiB;IACjB;IACA,cAAc,IAAI;IACnB;aAEQ,SAAS,kBAAkB;AAGpC,OAAI,eAAe,IAAI,IAAI,WAAqB,CAAE;AAClD,UAAO,sBAAsB;AAM7B,SALqC;IACnC,MAAMA,wBAAU;IAChB,YAAY,IAAI;IAChB,OAAO,IAAI;IACZ;aAEQ,SAAS,iBAAiB;GACnC,MAAM,aAAa,IAAI;AACvB,OAAI,eAAe,IAAI,WAAW,CAAE;AACpC,kBAAe,IAAI,WAAW;AAC9B,UAAO,sBAAsB;AAK7B,SAJmC;IACjC,MAAMA,wBAAU;IAChB;IACD;aAEQ,SAAS,oBAAoB;AACtC,UAAO,sBAAsB;GAC7B,MAAM,aAAa,IAAI;GACvB,MAAM,WAAW,cAAc,IAAI,WAAW;GAM9C,MAAM,aAAa,IAAI,WAAW,IAAI;GAEtC,MAAM,gBACJ,OAAO,eAAe,WAAW,UAAU,WAAW,GAAG;AAE3D,OACE,aAAa,2BACb,iBACA,OAAO,kBAAkB,YACzB,cAAc,eACd;IACA,MAAM,qBAAyC;KAC7C,MAAMA,wBAAU;KAChB,UAAW,cAA0C;KACtD;AACD,SAAK,MAAM,SAAS,oBAAoB,mBAAmB,CACzD,OAAM;;AAIV,OACE,aAAa,wBACb,iBACA,OAAO,kBAAkB,YACzB,WAAW,eACX;IACA,MAAM,kBAAmC;KACvC,MAAMA,wBAAU;KAChB,OAAQ,cAA0C;KACnD;AACD,SAAK,MAAM,SAAS,oBAAoB,gBAAgB,CACtD,OAAM;;GAIV,IAAI;AACJ,OAAI,OAAO,eAAe,SACxB,qBAAoB;OAEpB,KAAI;AACF,wBAAoB,KAAK,UAAU,cAAc,KAAK;WAChD;AACN,wBAAoB;;AAWxB,SAPyC;IACvC,MAAMA,wBAAU;IAChB,MAAM;IACN,+CAAuB;IACvB;IACA,SAAS;IACV;AAED,iBAAc,OAAO,WAAW;aACvB,SAAS,mBAAmB;AAGrC,UAAO,sBAAsB;AAC7B,sBAAmB;AACnB,wBAAsB,IAAI,iDAAoC;AAK9D,SAJsC;IACpC,MAAMA,wBAAU;IAChB,WAAW;IACZ;aAEQ,SAAS,2BAA2B;AAC7C,0BAAuB;AAMvB,SALwC;IACtC,MAAMA,wBAAU;IAChB,WAAW;IACX,MAAM;IACP;aAEQ,SAAS,4BAMlB,OAL0C;GACxC,MAAMA,wBAAU;GAChB,WAAW;GACX,OAAO,IAAI;GACZ;WAEQ,SAAS,yBAAyB;AAC3C,0BAAuB;AAKvB,SAJsC;IACpC,MAAMA,wBAAU;IAChB,WAAW;IACZ;aAEQ,SAAS,iBAAiB;AAKnC,OAAI,sBAAsB;AACxB,2BAAuB;AAKvB,UAJyC;KACvC,MAAMA,wBAAU;KAChB,WAAW;KACZ;;AAGH,sBAAmB;AAKnB,SAJ+B;IAC7B,MAAMA,wBAAU;IAChB,WAAW;IACZ;;;AAKL,QAAO,sBAAsB;;AAG/B,SAAS,UAAU,OAAwB;AACzC,KAAI;AACF,SAAO,KAAK,MAAM,MAAM;SAClB;AACN,SAAO"}

package/dist/agent/converters/tanstack.d.cts

@@ -77,7 +77,7 @@ declare function convertInputToTanStackAI(input: RunAgentInput): TanStackInputRe
  * CUSTOM "approval-requested" chunk (a tool declared `needsApproval: true`).
  * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
  */
-declare function convertTanStackStream(stream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[]): AsyncGenerator<BaseEvent>;
+declare function convertTanStackStream(stream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[], initialState?: unknown): AsyncGenerator<BaseEvent>;
 //#endregion
 export { TanStackChatMessage, TanStackInputResult, convertInputToTanStackAI, convertTanStackStream };
 //# sourceMappingURL=tanstack.d.cts.map

package/dist/agent/converters/tanstack.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"tanstack.d.cts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;;AA6EA;;;;;;;UAlCiB,mBAAA;EACf,IAAA;EAEA,OAAA;EACA,IAAA;EACA,SAAA,GAAY,KAAA;IACV,EAAA;IACA,IAAA;IACA,QAAA;MAAY,IAAA;MAAc,SAAA;IAAA;EAAA;EAE5B,UAAA;AAAA;;;;;;;;;;;UAae,kBAAA;EACf,UAAA;EACA,IAAA;EACA,WAAA;EAEA,WAAA;AAAA;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;;;;;;;;EAQA,KAAA,EAAO,kBAAA;AAAA;;;;;;;;;iBA4JO,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;;;;;iBAmFoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,KACnB,cAAA,CAAe,SAAA"}
+{"version":3,"file":"tanstack.d.cts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;;AA8EA;;;;;;;UAlCiB,mBAAA;EACf,IAAA;EAEA,OAAA;EACA,IAAA;EACA,SAAA,GAAY,KAAA;IACV,EAAA;IACA,IAAA;IACA,QAAA;MAAY,IAAA;MAAc,SAAA;IAAA;EAAA;EAE5B,UAAA;AAAA;;;;;;;;;;;UAae,kBAAA;EACf,UAAA;EACA,IAAA;EACA,WAAA;EAEA,WAAA;AAAA;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;;;;;;;;EAQA,KAAA,EAAO,kBAAA;AAAA;;;;;;;;;iBA4JO,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;;;;;iBAmFoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,IACpB,YAAA,aACC,cAAA,CAAe,SAAA"}

package/dist/agent/converters/tanstack.d.mts

@@ -77,7 +77,7 @@ declare function convertInputToTanStackAI(input: RunAgentInput): TanStackInputRe
  * CUSTOM "approval-requested" chunk (a tool declared `needsApproval: true`).
  * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
  */
-declare function convertTanStackStream(stream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[]): AsyncGenerator<BaseEvent>;
+declare function convertTanStackStream(stream: AsyncIterable<unknown>, abortSignal: AbortSignal, pendingInterrupts?: Interrupt[], initialState?: unknown): AsyncGenerator<BaseEvent>;
 //#endregion
 export { TanStackChatMessage, TanStackInputResult, convertInputToTanStackAI, convertTanStackStream };
 //# sourceMappingURL=tanstack.d.mts.map

package/dist/agent/converters/tanstack.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"tanstack.d.mts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;;AA6EA;;;;;;;UAlCiB,mBAAA;EACf,IAAA;EAEA,OAAA;EACA,IAAA;EACA,SAAA,GAAY,KAAA;IACV,EAAA;IACA,IAAA;IACA,QAAA;MAAY,IAAA;MAAc,SAAA;IAAA;EAAA;EAE5B,UAAA;AAAA;;;;;;;;;;;UAae,kBAAA;EACf,UAAA;EACA,IAAA;EACA,WAAA;EAEA,WAAA;AAAA;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;;;;;;;;EAQA,KAAA,EAAO,kBAAA;AAAA;;;;;;;;;iBA4JO,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;;;;;iBAmFoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,KACnB,cAAA,CAAe,SAAA"}
+{"version":3,"file":"tanstack.d.mts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;;AA8EA;;;;;;;UAlCiB,mBAAA;EACf,IAAA;EAEA,OAAA;EACA,IAAA;EACA,SAAA,GAAY,KAAA;IACV,EAAA;IACA,IAAA;IACA,QAAA;MAAY,IAAA;MAAc,SAAA;IAAA;EAAA;EAE5B,UAAA;AAAA;;;;;;;;;;;UAae,kBAAA;EACf,UAAA;EACA,IAAA;EACA,WAAA;EAEA,WAAA;AAAA;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;;;;;;;;EAQA,KAAA,EAAO,kBAAA;AAAA;;;;;;;;;iBA4JO,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;;;;;iBAmFoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,EACb,iBAAA,GAAoB,SAAA,IACpB,YAAA,aACC,cAAA,CAAe,SAAA"}

package/dist/agent/converters/tanstack.mjs

@@ -1,4 +1,5 @@
 import "reflect-metadata";
+import { createStateEventNormalizer } from "../state-delta.mjs";
 import { randomUUID } from "@copilotkit/shared";
 import { EventType } from "@ag-ui/client";
 
@@ -169,12 +170,13 @@ function convertInputToTanStackAI(input) {
 * CUSTOM "approval-requested" chunk (a tool declared `needsApproval: true`).
 * The caller turns a non-empty array into a RUN_FINISHED `outcome:interrupt`.
 */
-async function* convertTanStackStream(stream, abortSignal, pendingInterrupts) {
+async function* convertTanStackStream(stream, abortSignal, pendingInterrupts, initialState) {
 	const messageId = randomUUID();
 	const toolNamesById = /* @__PURE__ */ new Map();
 	let reasoningRunOpen = false;
 	let reasoningMessageOpen = false;
 	let reasoningMessageId = randomUUID();
+	const normalizeStateEvent = createStateEventNormalizer(initialState);
 	function* closeReasoningIfOpen() {
 		if (reasoningMessageOpen) {
 			reasoningMessageOpen = false;
@@ -254,14 +256,20 @@ async function* convertTanStackStream(stream, abortSignal, pendingInterrupts) {
 			const toolName = toolNamesById.get(toolCallId);
 			const rawPayload = raw.content ?? raw.result;
 			const parsedContent = typeof rawPayload === "string" ? safeParse(rawPayload) : rawPayload;
-			if (toolName === "AGUISendStateSnapshot" && parsedContent && typeof parsedContent === "object" && "snapshot" in parsedContent) yield {
-				type: EventType.STATE_SNAPSHOT,
-				snapshot: parsedContent.snapshot
-			};
-			if (toolName === "AGUISendStateDelta" && parsedContent && typeof parsedContent === "object" && "delta" in parsedContent) yield {
-				type: EventType.STATE_DELTA,
-				delta: parsedContent.delta
-			};
+			if (toolName === "AGUISendStateSnapshot" && parsedContent && typeof parsedContent === "object" && "snapshot" in parsedContent) {
+				const stateSnapshotEvent = {
+					type: EventType.STATE_SNAPSHOT,
+					snapshot: parsedContent.snapshot
+				};
+				for (const event of normalizeStateEvent(stateSnapshotEvent)) yield event;
+			}
+			if (toolName === "AGUISendStateDelta" && parsedContent && typeof parsedContent === "object" && "delta" in parsedContent) {
+				const stateDeltaEvent = {
+					type: EventType.STATE_DELTA,
+					delta: parsedContent.delta
+				};
+				for (const event of normalizeStateEvent(stateDeltaEvent)) yield event;
+			}
 			let serializedContent;
 			if (typeof rawPayload === "string") serializedContent = rawPayload;
 			else try {