@copilotkit/runtime 1.56.4 → 1.56.5-canary.1777664617

package/dist/agent/converters/tanstack.cjs

@@ -130,36 +130,84 @@ function convertInputToTanStackAI(input) {
 */
 async function* convertTanStackStream(stream, abortSignal) {
 	const messageId = (0, _copilotkit_shared.randomUUID)();
+	const toolNamesById = /* @__PURE__ */ new Map();
+	let reasoningRunOpen = false;
+	let reasoningMessageOpen = false;
+	let reasoningMessageId = (0, _copilotkit_shared.randomUUID)();
+	function* closeReasoningIfOpen() {
+		if (reasoningMessageOpen) {
+			reasoningMessageOpen = false;
+			yield {
+				type: _ag_ui_client.EventType.REASONING_MESSAGE_END,
+				messageId: reasoningMessageId
+			};
+		}
+		if (reasoningRunOpen) {
+			reasoningRunOpen = false;
+			yield {
+				type: _ag_ui_client.EventType.REASONING_END,
+				messageId: reasoningMessageId
+			};
+		}
+	}
+	let runFinished = false;
 	for await (const chunk of stream) {
 		if (abortSignal.aborted) break;
 		const raw = chunk;
 		const type = raw.type;
-		if (type === "TEXT_MESSAGE_CONTENT" && raw.delta) yield {
-			type: _ag_ui_client.EventType.TEXT_MESSAGE_CHUNK,
-			role: "assistant",
-			messageId,
-			delta: raw.delta
-		};
-		else if (type === "TOOL_CALL_START") yield {
-			type: _ag_ui_client.EventType.TOOL_CALL_START,
-			parentMessageId: messageId,
-			toolCallId: raw.toolCallId,
-			toolCallName: raw.toolCallName
-		};
-		else if (type === "TOOL_CALL_ARGS") yield {
-			type: _ag_ui_client.EventType.TOOL_CALL_ARGS,
-			toolCallId: raw.toolCallId,
-			delta: raw.delta
-		};
-		else if (type === "TOOL_CALL_END") yield {
-			type: _ag_ui_client.EventType.TOOL_CALL_END,
-			toolCallId: raw.toolCallId
-		};
-		else if (type === "TOOL_CALL_RESULT") {
+		if (type === "RUN_FINISHED") {
+			runFinished = true;
+			continue;
+		}
+		if (runFinished) continue;
+		if (type === "TEXT_MESSAGE_CONTENT" && raw.delta != null) {
+			yield* closeReasoningIfOpen();
+			yield {
+				type: _ag_ui_client.EventType.TEXT_MESSAGE_CHUNK,
+				role: "assistant",
+				messageId,
+				delta: raw.delta
+			};
+		} else if (type === "TOOL_CALL_START") {
+			yield* closeReasoningIfOpen();
+			toolNamesById.set(raw.toolCallId, raw.toolCallName);
+			yield {
+				type: _ag_ui_client.EventType.TOOL_CALL_START,
+				parentMessageId: messageId,
+				toolCallId: raw.toolCallId,
+				toolCallName: raw.toolCallName
+			};
+		} else if (type === "TOOL_CALL_ARGS") {
+			yield* closeReasoningIfOpen();
+			yield {
+				type: _ag_ui_client.EventType.TOOL_CALL_ARGS,
+				toolCallId: raw.toolCallId,
+				delta: raw.delta
+			};
+		} else if (type === "TOOL_CALL_END") {
+			yield* closeReasoningIfOpen();
+			yield {
+				type: _ag_ui_client.EventType.TOOL_CALL_END,
+				toolCallId: raw.toolCallId
+			};
+		} else if (type === "TOOL_CALL_RESULT") {
+			yield* closeReasoningIfOpen();
+			const toolCallId = raw.toolCallId;
+			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
+			};
 			let serializedContent;
-			if (typeof raw.content === "string") serializedContent = raw.content;
+			if (typeof rawPayload === "string") serializedContent = rawPayload;
 			else try {
-				serializedContent = JSON.stringify(raw.content ?? raw.result ?? null);
+				serializedContent = JSON.stringify(rawPayload ?? null);
 			} catch {
 				serializedContent = "[Unserializable tool result]";
 			}
@@ -167,11 +215,59 @@ async function* convertTanStackStream(stream, abortSignal) {
 				type: _ag_ui_client.EventType.TOOL_CALL_RESULT,
 				role: "tool",
 				messageId: (0, _copilotkit_shared.randomUUID)(),
-				toolCallId: raw.toolCallId,
+				toolCallId,
 				content: serializedContent
 			};
+			toolNamesById.delete(toolCallId);
+		} else if (type === "REASONING_START") {
+			yield* closeReasoningIfOpen();
+			reasoningRunOpen = true;
+			reasoningMessageId = raw.messageId ?? (0, _copilotkit_shared.randomUUID)();
+			yield {
+				type: _ag_ui_client.EventType.REASONING_START,
+				messageId: reasoningMessageId
+			};
+		} else if (type === "REASONING_MESSAGE_START") {
+			reasoningMessageOpen = true;
+			yield {
+				type: _ag_ui_client.EventType.REASONING_MESSAGE_START,
+				messageId: reasoningMessageId,
+				role: "reasoning"
+			};
+		} else if (type === "REASONING_MESSAGE_CONTENT") yield {
+			type: _ag_ui_client.EventType.REASONING_MESSAGE_CONTENT,
+			messageId: reasoningMessageId,
+			delta: raw.delta
+		};
+		else if (type === "REASONING_MESSAGE_END") {
+			reasoningMessageOpen = false;
+			yield {
+				type: _ag_ui_client.EventType.REASONING_MESSAGE_END,
+				messageId: reasoningMessageId
+			};
+		} else if (type === "REASONING_END") {
+			if (reasoningMessageOpen) {
+				reasoningMessageOpen = false;
+				yield {
+					type: _ag_ui_client.EventType.REASONING_MESSAGE_END,
+					messageId: reasoningMessageId
+				};
+			}
+			reasoningRunOpen = false;
+			yield {
+				type: _ag_ui_client.EventType.REASONING_END,
+				messageId: reasoningMessageId
+			};
 		}
 	}
+	yield* closeReasoningIfOpen();
+}
+function safeParse(value) {
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
 }
 
 //#endregion

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

@@ -1 +1 @@
-{"version":3,"file":"tanstack.cjs","names":["EventType"],"sources":["../../../src/agent/converters/tanstack.ts"],"sourcesContent":["import {\n  BaseEvent,\n  EventType,\n  RunAgentInput,\n  Message,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n} 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 * 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\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 * 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  return { messages, systemPrompts };\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 */\nexport async function* convertTanStackStream(\n  stream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n): AsyncGenerator<BaseEvent> {\n  const messageId = randomUUID();\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    if (type === \"TEXT_MESSAGE_CONTENT\" && raw.delta) {\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 startEvent: ToolCallStartEvent = {\n        type: EventType.TOOL_CALL_START,\n        parentMessageId: messageId,\n        toolCallId: raw.toolCallId as string,\n        toolCallName: raw.toolCallName as string,\n      };\n      yield startEvent;\n    } else if (type === \"TOOL_CALL_ARGS\") {\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 endEvent: ToolCallEndEvent = {\n        type: EventType.TOOL_CALL_END,\n        toolCallId: raw.toolCallId as string,\n      };\n      yield endEvent;\n    } else if (type === \"TOOL_CALL_RESULT\") {\n      let serializedContent: string;\n      if (typeof raw.content === \"string\") {\n        serializedContent = raw.content;\n      } else {\n        try {\n          serializedContent = JSON.stringify(raw.content ?? raw.result ?? null);\n        } catch {\n          serializedContent = \"[Unserializable tool result]\";\n        }\n      }\n      const resultEvent: ToolCallResultEvent = {\n        type: EventType.TOOL_CALL_RESULT,\n        role: \"tool\",\n        messageId: randomUUID(),\n        toolCallId: raw.toolCallId as string,\n        content: serializedContent,\n      };\n      yield resultEvent;\n    }\n    // Unhandled chunk types are silently ignored.\n    // Known gaps: STATE_SNAPSHOT, STATE_DELTA, and REASONING events are not\n    // converted from TanStack streams. Shared state and reasoning will not\n    // surface when using the TanStack backend. Use the AI SDK backend if these\n    // features are required.\n  }\n}\n"],"mappings":";;;;;;;;;;;AA+DA,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;;;;;;;;;;AAWpC,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;AAGH,QAAO;EAAE;EAAU;EAAe;;;;;;;;;AAUpC,gBAAuB,sBACrB,QACA,aAC2B;CAC3B,MAAM,gDAAwB;AAE9B,YAAW,MAAM,SAAS,QAAQ;AAChC,MAAI,YAAY,QAAS;EAEzB,MAAM,MAAM;EACZ,MAAM,OAAO,IAAI;AAEjB,MAAI,SAAS,0BAA0B,IAAI,MAOzC,OANyC;GACvC,MAAMA,wBAAU;GAChB,MAAM;GACN;GACA,OAAO,IAAI;GACZ;WAEQ,SAAS,kBAOlB,OANuC;GACrC,MAAMA,wBAAU;GAChB,iBAAiB;GACjB,YAAY,IAAI;GAChB,cAAc,IAAI;GACnB;WAEQ,SAAS,iBAMlB,OALqC;GACnC,MAAMA,wBAAU;GAChB,YAAY,IAAI;GAChB,OAAO,IAAI;GACZ;WAEQ,SAAS,gBAKlB,OAJmC;GACjC,MAAMA,wBAAU;GAChB,YAAY,IAAI;GACjB;WAEQ,SAAS,oBAAoB;GACtC,IAAI;AACJ,OAAI,OAAO,IAAI,YAAY,SACzB,qBAAoB,IAAI;OAExB,KAAI;AACF,wBAAoB,KAAK,UAAU,IAAI,WAAW,IAAI,UAAU,KAAK;WAC/D;AACN,wBAAoB;;AAUxB,SAPyC;IACvC,MAAMA,wBAAU;IAChB,MAAM;IACN,+CAAuB;IACvB,YAAY,IAAI;IAChB,SAAS;IACV"}
+{"version":3,"file":"tanstack.cjs","names":["EventType"],"sources":["../../../src/agent/converters/tanstack.ts"],"sourcesContent":["import {\n  BaseEvent,\n  EventType,\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 { 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 * 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\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 * 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  return { messages, systemPrompts };\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 */\nexport async function* convertTanStackStream(\n  stream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\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: after the model\n  // returns tool calls, the engine tries to execute them and re-prompt. This\n  // produces a second round of TOOL_CALL_START / TOOL_CALL_END events that\n  // duplicate the ones from the first streaming pass. The CopilotKit runtime\n  // handles tool execution externally (via the frontend SDK), so we must stop\n  // converting events once the TanStack adapter signals the first turn is\n  // complete with RUN_FINISHED.\n  let runFinished = false;\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    // Stop converting after the first RUN_FINISHED — any subsequent events\n    // come from TanStack's internal tool-execution loop and would produce\n    // duplicate TOOL_CALL_END events that violate the ag-ui verify middleware.\n    if (type === \"RUN_FINISHED\") {\n      runFinished = true;\n      continue;\n    }\n    if (runFinished) continue;\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      yield* closeReasoningIfOpen();\n      toolNamesById.set(raw.toolCallId as string, raw.toolCallName as string);\n      const startEvent: ToolCallStartEvent = {\n        type: EventType.TOOL_CALL_START,\n        parentMessageId: messageId,\n        toolCallId: raw.toolCallId as string,\n        toolCallName: raw.toolCallName as string,\n      };\n      yield startEvent;\n    } else if (type === \"TOOL_CALL_ARGS\") {\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      yield* closeReasoningIfOpen();\n      const endEvent: ToolCallEndEvent = {\n        type: EventType.TOOL_CALL_END,\n        toolCallId: raw.toolCallId as string,\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":";;;;;;;;;;;AAsEA,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;;;;;;;;;;AAWpC,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;AAGH,QAAO;EAAE;EAAU;EAAe;;;;;;;;;AAUpC,gBAAuB,sBACrB,QACA,aAC2B;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;;;CAYL,IAAI,cAAc;AAElB,YAAW,MAAM,SAAS,QAAQ;AAChC,MAAI,YAAY,QAAS;EAEzB,MAAM,MAAM;EACZ,MAAM,OAAO,IAAI;AAKjB,MAAI,SAAS,gBAAgB;AAC3B,iBAAc;AACd;;AAEF,MAAI,YAAa;AAEjB,MAAI,SAAS,0BAA0B,IAAI,SAAS,MAAM;AACxD,UAAO,sBAAsB;AAO7B,SANyC;IACvC,MAAMA,wBAAU;IAChB,MAAM;IACN;IACA,OAAO,IAAI;IACZ;aAEQ,SAAS,mBAAmB;AACrC,UAAO,sBAAsB;AAC7B,iBAAc,IAAI,IAAI,YAAsB,IAAI,aAAuB;AAOvE,SANuC;IACrC,MAAMA,wBAAU;IAChB,iBAAiB;IACjB,YAAY,IAAI;IAChB,cAAc,IAAI;IACnB;aAEQ,SAAS,kBAAkB;AACpC,UAAO,sBAAsB;AAM7B,SALqC;IACnC,MAAMA,wBAAU;IAChB,YAAY,IAAI;IAChB,OAAO,IAAI;IACZ;aAEQ,SAAS,iBAAiB;AACnC,UAAO,sBAAsB;AAK7B,SAJmC;IACjC,MAAMA,wBAAU;IAChB,YAAY,IAAI;IACjB;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"}

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

@@ -1 +1 @@
-{"version":3,"file":"tanstack.d.cts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;AA2JA;;;;;;;;AAAA,UAxHiB,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;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;AAAA;;;;;;;;;iBAoGc,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;iBAqEoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,GACZ,cAAA,CAAe,SAAA"}
+{"version":3,"file":"tanstack.d.cts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;AAkKA;;;;;;;;AAAA,UAxHiB,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;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;AAAA;;;;;;;;;iBAoGc,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;iBAqEoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,GACZ,cAAA,CAAe,SAAA"}

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

@@ -1 +1 @@
-{"version":3,"file":"tanstack.d.mts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;AA2JA;;;;;;;;AAAA,UAxHiB,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;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;AAAA;;;;;;;;;iBAoGc,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;iBAqEoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,GACZ,cAAA,CAAe,SAAA"}
+{"version":3,"file":"tanstack.d.mts","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"mappings":";;;;AAkKA;;;;;;;;AAAA,UAxHiB,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;;;;UAMe,mBAAA;;EAEf,QAAA,EAAU,mBAAA;;EAEV,aAAA;AAAA;;;;;;;;;iBAoGc,wBAAA,CACd,KAAA,EAAO,aAAA,GACN,mBAAA;;;;;;;;iBAqEoB,qBAAA,CACrB,MAAA,EAAQ,aAAA,WACR,WAAA,EAAa,WAAA,GACZ,cAAA,CAAe,SAAA"}

package/dist/agent/converters/tanstack.mjs

@@ -129,36 +129,84 @@ function convertInputToTanStackAI(input) {
 */
 async function* convertTanStackStream(stream, abortSignal) {
 	const messageId = randomUUID();
+	const toolNamesById = /* @__PURE__ */ new Map();
+	let reasoningRunOpen = false;
+	let reasoningMessageOpen = false;
+	let reasoningMessageId = randomUUID();
+	function* closeReasoningIfOpen() {
+		if (reasoningMessageOpen) {
+			reasoningMessageOpen = false;
+			yield {
+				type: EventType.REASONING_MESSAGE_END,
+				messageId: reasoningMessageId
+			};
+		}
+		if (reasoningRunOpen) {
+			reasoningRunOpen = false;
+			yield {
+				type: EventType.REASONING_END,
+				messageId: reasoningMessageId
+			};
+		}
+	}
+	let runFinished = false;
 	for await (const chunk of stream) {
 		if (abortSignal.aborted) break;
 		const raw = chunk;
 		const type = raw.type;
-		if (type === "TEXT_MESSAGE_CONTENT" && raw.delta) yield {
-			type: EventType.TEXT_MESSAGE_CHUNK,
-			role: "assistant",
-			messageId,
-			delta: raw.delta
-		};
-		else if (type === "TOOL_CALL_START") yield {
-			type: EventType.TOOL_CALL_START,
-			parentMessageId: messageId,
-			toolCallId: raw.toolCallId,
-			toolCallName: raw.toolCallName
-		};
-		else if (type === "TOOL_CALL_ARGS") yield {
-			type: EventType.TOOL_CALL_ARGS,
-			toolCallId: raw.toolCallId,
-			delta: raw.delta
-		};
-		else if (type === "TOOL_CALL_END") yield {
-			type: EventType.TOOL_CALL_END,
-			toolCallId: raw.toolCallId
-		};
-		else if (type === "TOOL_CALL_RESULT") {
+		if (type === "RUN_FINISHED") {
+			runFinished = true;
+			continue;
+		}
+		if (runFinished) continue;
+		if (type === "TEXT_MESSAGE_CONTENT" && raw.delta != null) {
+			yield* closeReasoningIfOpen();
+			yield {
+				type: EventType.TEXT_MESSAGE_CHUNK,
+				role: "assistant",
+				messageId,
+				delta: raw.delta
+			};
+		} else if (type === "TOOL_CALL_START") {
+			yield* closeReasoningIfOpen();
+			toolNamesById.set(raw.toolCallId, raw.toolCallName);
+			yield {
+				type: EventType.TOOL_CALL_START,
+				parentMessageId: messageId,
+				toolCallId: raw.toolCallId,
+				toolCallName: raw.toolCallName
+			};
+		} else if (type === "TOOL_CALL_ARGS") {
+			yield* closeReasoningIfOpen();
+			yield {
+				type: EventType.TOOL_CALL_ARGS,
+				toolCallId: raw.toolCallId,
+				delta: raw.delta
+			};
+		} else if (type === "TOOL_CALL_END") {
+			yield* closeReasoningIfOpen();
+			yield {
+				type: EventType.TOOL_CALL_END,
+				toolCallId: raw.toolCallId
+			};
+		} else if (type === "TOOL_CALL_RESULT") {
+			yield* closeReasoningIfOpen();
+			const toolCallId = raw.toolCallId;
+			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
+			};
 			let serializedContent;
-			if (typeof raw.content === "string") serializedContent = raw.content;
+			if (typeof rawPayload === "string") serializedContent = rawPayload;
 			else try {
-				serializedContent = JSON.stringify(raw.content ?? raw.result ?? null);
+				serializedContent = JSON.stringify(rawPayload ?? null);
 			} catch {
 				serializedContent = "[Unserializable tool result]";
 			}
@@ -166,11 +214,59 @@ async function* convertTanStackStream(stream, abortSignal) {
 				type: EventType.TOOL_CALL_RESULT,
 				role: "tool",
 				messageId: randomUUID(),
-				toolCallId: raw.toolCallId,
+				toolCallId,
 				content: serializedContent
 			};
+			toolNamesById.delete(toolCallId);
+		} else if (type === "REASONING_START") {
+			yield* closeReasoningIfOpen();
+			reasoningRunOpen = true;
+			reasoningMessageId = raw.messageId ?? randomUUID();
+			yield {
+				type: EventType.REASONING_START,
+				messageId: reasoningMessageId
+			};
+		} else if (type === "REASONING_MESSAGE_START") {
+			reasoningMessageOpen = true;
+			yield {
+				type: EventType.REASONING_MESSAGE_START,
+				messageId: reasoningMessageId,
+				role: "reasoning"
+			};
+		} else if (type === "REASONING_MESSAGE_CONTENT") yield {
+			type: EventType.REASONING_MESSAGE_CONTENT,
+			messageId: reasoningMessageId,
+			delta: raw.delta
+		};
+		else if (type === "REASONING_MESSAGE_END") {
+			reasoningMessageOpen = false;
+			yield {
+				type: EventType.REASONING_MESSAGE_END,
+				messageId: reasoningMessageId
+			};
+		} else if (type === "REASONING_END") {
+			if (reasoningMessageOpen) {
+				reasoningMessageOpen = false;
+				yield {
+					type: EventType.REASONING_MESSAGE_END,
+					messageId: reasoningMessageId
+				};
+			}
+			reasoningRunOpen = false;
+			yield {
+				type: EventType.REASONING_END,
+				messageId: reasoningMessageId
+			};
 		}
 	}
+	yield* closeReasoningIfOpen();
+}
+function safeParse(value) {
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
 }
 
 //#endregion

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

@@ -1 +1 @@
-{"version":3,"file":"tanstack.mjs","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"sourcesContent":["import {\n  BaseEvent,\n  EventType,\n  RunAgentInput,\n  Message,\n  TextMessageChunkEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n  ToolCallResultEvent,\n} 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 * 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\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 * 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  return { messages, systemPrompts };\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 */\nexport async function* convertTanStackStream(\n  stream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\n): AsyncGenerator<BaseEvent> {\n  const messageId = randomUUID();\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    if (type === \"TEXT_MESSAGE_CONTENT\" && raw.delta) {\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 startEvent: ToolCallStartEvent = {\n        type: EventType.TOOL_CALL_START,\n        parentMessageId: messageId,\n        toolCallId: raw.toolCallId as string,\n        toolCallName: raw.toolCallName as string,\n      };\n      yield startEvent;\n    } else if (type === \"TOOL_CALL_ARGS\") {\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 endEvent: ToolCallEndEvent = {\n        type: EventType.TOOL_CALL_END,\n        toolCallId: raw.toolCallId as string,\n      };\n      yield endEvent;\n    } else if (type === \"TOOL_CALL_RESULT\") {\n      let serializedContent: string;\n      if (typeof raw.content === \"string\") {\n        serializedContent = raw.content;\n      } else {\n        try {\n          serializedContent = JSON.stringify(raw.content ?? raw.result ?? null);\n        } catch {\n          serializedContent = \"[Unserializable tool result]\";\n        }\n      }\n      const resultEvent: ToolCallResultEvent = {\n        type: EventType.TOOL_CALL_RESULT,\n        role: \"tool\",\n        messageId: randomUUID(),\n        toolCallId: raw.toolCallId as string,\n        content: serializedContent,\n      };\n      yield resultEvent;\n    }\n    // Unhandled chunk types are silently ignored.\n    // Known gaps: STATE_SNAPSHOT, STATE_DELTA, and REASONING events are not\n    // converted from TanStack streams. Shared state and reasoning will not\n    // surface when using the TanStack backend. Use the AI SDK backend if these\n    // features are required.\n  }\n}\n"],"mappings":";;;;;;;;;;AA+DA,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;;;;;;;;;;AAWpC,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;AAGH,QAAO;EAAE;EAAU;EAAe;;;;;;;;;AAUpC,gBAAuB,sBACrB,QACA,aAC2B;CAC3B,MAAM,YAAY,YAAY;AAE9B,YAAW,MAAM,SAAS,QAAQ;AAChC,MAAI,YAAY,QAAS;EAEzB,MAAM,MAAM;EACZ,MAAM,OAAO,IAAI;AAEjB,MAAI,SAAS,0BAA0B,IAAI,MAOzC,OANyC;GACvC,MAAM,UAAU;GAChB,MAAM;GACN;GACA,OAAO,IAAI;GACZ;WAEQ,SAAS,kBAOlB,OANuC;GACrC,MAAM,UAAU;GAChB,iBAAiB;GACjB,YAAY,IAAI;GAChB,cAAc,IAAI;GACnB;WAEQ,SAAS,iBAMlB,OALqC;GACnC,MAAM,UAAU;GAChB,YAAY,IAAI;GAChB,OAAO,IAAI;GACZ;WAEQ,SAAS,gBAKlB,OAJmC;GACjC,MAAM,UAAU;GAChB,YAAY,IAAI;GACjB;WAEQ,SAAS,oBAAoB;GACtC,IAAI;AACJ,OAAI,OAAO,IAAI,YAAY,SACzB,qBAAoB,IAAI;OAExB,KAAI;AACF,wBAAoB,KAAK,UAAU,IAAI,WAAW,IAAI,UAAU,KAAK;WAC/D;AACN,wBAAoB;;AAUxB,SAPyC;IACvC,MAAM,UAAU;IAChB,MAAM;IACN,WAAW,YAAY;IACvB,YAAY,IAAI;IAChB,SAAS;IACV"}
+{"version":3,"file":"tanstack.mjs","names":[],"sources":["../../../src/agent/converters/tanstack.ts"],"sourcesContent":["import {\n  BaseEvent,\n  EventType,\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 { 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 * 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\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 * 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  return { messages, systemPrompts };\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 */\nexport async function* convertTanStackStream(\n  stream: AsyncIterable<unknown>,\n  abortSignal: AbortSignal,\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: after the model\n  // returns tool calls, the engine tries to execute them and re-prompt. This\n  // produces a second round of TOOL_CALL_START / TOOL_CALL_END events that\n  // duplicate the ones from the first streaming pass. The CopilotKit runtime\n  // handles tool execution externally (via the frontend SDK), so we must stop\n  // converting events once the TanStack adapter signals the first turn is\n  // complete with RUN_FINISHED.\n  let runFinished = false;\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    // Stop converting after the first RUN_FINISHED — any subsequent events\n    // come from TanStack's internal tool-execution loop and would produce\n    // duplicate TOOL_CALL_END events that violate the ag-ui verify middleware.\n    if (type === \"RUN_FINISHED\") {\n      runFinished = true;\n      continue;\n    }\n    if (runFinished) continue;\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      yield* closeReasoningIfOpen();\n      toolNamesById.set(raw.toolCallId as string, raw.toolCallName as string);\n      const startEvent: ToolCallStartEvent = {\n        type: EventType.TOOL_CALL_START,\n        parentMessageId: messageId,\n        toolCallId: raw.toolCallId as string,\n        toolCallName: raw.toolCallName as string,\n      };\n      yield startEvent;\n    } else if (type === \"TOOL_CALL_ARGS\") {\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      yield* closeReasoningIfOpen();\n      const endEvent: ToolCallEndEvent = {\n        type: EventType.TOOL_CALL_END,\n        toolCallId: raw.toolCallId as string,\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":";;;;;;;;;;AAsEA,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;;;;;;;;;;AAWpC,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;AAGH,QAAO;EAAE;EAAU;EAAe;;;;;;;;;AAUpC,gBAAuB,sBACrB,QACA,aAC2B;CAC3B,MAAM,YAAY,YAAY;CAC9B,MAAM,gCAAgB,IAAI,KAAqB;CAO/C,IAAI,mBAAmB;CACvB,IAAI,uBAAuB;CAC3B,IAAI,qBAAqB,YAAY;CAErC,UAAU,uBAA6C;AACrD,MAAI,sBAAsB;AACxB,0BAAuB;AAKvB,SAJyC;IACvC,MAAM,UAAU;IAChB,WAAW;IACZ;;AAGH,MAAI,kBAAkB;AACpB,sBAAmB;AAKnB,SAJ+B;IAC7B,MAAM,UAAU;IAChB,WAAW;IACZ;;;CAYL,IAAI,cAAc;AAElB,YAAW,MAAM,SAAS,QAAQ;AAChC,MAAI,YAAY,QAAS;EAEzB,MAAM,MAAM;EACZ,MAAM,OAAO,IAAI;AAKjB,MAAI,SAAS,gBAAgB;AAC3B,iBAAc;AACd;;AAEF,MAAI,YAAa;AAEjB,MAAI,SAAS,0BAA0B,IAAI,SAAS,MAAM;AACxD,UAAO,sBAAsB;AAO7B,SANyC;IACvC,MAAM,UAAU;IAChB,MAAM;IACN;IACA,OAAO,IAAI;IACZ;aAEQ,SAAS,mBAAmB;AACrC,UAAO,sBAAsB;AAC7B,iBAAc,IAAI,IAAI,YAAsB,IAAI,aAAuB;AAOvE,SANuC;IACrC,MAAM,UAAU;IAChB,iBAAiB;IACjB,YAAY,IAAI;IAChB,cAAc,IAAI;IACnB;aAEQ,SAAS,kBAAkB;AACpC,UAAO,sBAAsB;AAM7B,SALqC;IACnC,MAAM,UAAU;IAChB,YAAY,IAAI;IAChB,OAAO,IAAI;IACZ;aAEQ,SAAS,iBAAiB;AACnC,UAAO,sBAAsB;AAK7B,SAJmC;IACjC,MAAM,UAAU;IAChB,YAAY,IAAI;IACjB;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,MAAM,UAAU;IAChB,UAAW,cAA0C;IACtD;AAIH,OACE,aAAa,wBACb,iBACA,OAAO,kBAAkB,YACzB,WAAW,cAMX,OAJyC;IACvC,MAAM,UAAU;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,MAAM,UAAU;IAChB,MAAM;IACN,WAAW,YAAY;IACvB;IACA,SAAS;IACV;AAED,iBAAc,OAAO,WAAW;aACvB,SAAS,mBAAmB;AAGrC,UAAO,sBAAsB;AAC7B,sBAAmB;AACnB,wBAAsB,IAAI,aAAwB,YAAY;AAK9D,SAJsC;IACpC,MAAM,UAAU;IAChB,WAAW;IACZ;aAEQ,SAAS,2BAA2B;AAC7C,0BAAuB;AAMvB,SALwC;IACtC,MAAM,UAAU;IAChB,WAAW;IACX,MAAM;IACP;aAEQ,SAAS,4BAMlB,OAL0C;GACxC,MAAM,UAAU;GAChB,WAAW;GACX,OAAO,IAAI;GACZ;WAEQ,SAAS,yBAAyB;AAC3C,0BAAuB;AAKvB,SAJsC;IACpC,MAAM,UAAU;IAChB,WAAW;IACZ;aAEQ,SAAS,iBAAiB;AAKnC,OAAI,sBAAsB;AACxB,2BAAuB;AAKvB,UAJyC;KACvC,MAAM,UAAU;KAChB,WAAW;KACZ;;AAGH,sBAAmB;AAKnB,SAJ+B;IAC7B,MAAM,UAAU;IAChB,WAAW;IACZ;;;AAKL,QAAO,sBAAsB;;AAG/B,SAAS,UAAU,OAAwB;AACzC,KAAI;AACF,SAAO,KAAK,MAAM,MAAM;SAClB;AACN,SAAO"}

package/dist/lib/runtime/agent-integrations/langgraph/agent.cjs

@@ -93,7 +93,14 @@ var LangGraphAgent = class extends _ag_ui_langgraph.LangGraphAgent {
 		return true;
 	}
 	run(input) {
-		return super.run(input).pipe((0, rxjs.map)((processedEvent) => {
+		const enrichedInput = {
+			...input,
+			forwardedProps: {
+				...input.forwardedProps,
+				streamSubgraphs: input.forwardedProps?.streamSubgraphs ?? true
+			}
+		};
+		return super.run(enrichedInput).pipe((0, rxjs.map)((processedEvent) => {
 			if (processedEvent.type === _ag_ui_client.EventType.RAW) {
 				const event = processedEvent.event ?? processedEvent.rawEvent;
 				const eventType = event.event;

package/dist/lib/runtime/agent-integrations/langgraph/agent.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"agent.cjs","names":["AGUILangGraphAgent","EventType","CustomEventNames","LangGraphEventTypes"],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"sourcesContent":["import { map, Observable } from \"rxjs\";\nimport { LangGraphEventTypes } from \"../../../../agents/langgraph/events\";\nimport { BaseEvent, RawEvent } from \"@ag-ui/core\";\nimport {\n  LangGraphAgent as AGUILangGraphAgent,\n  LangGraphHttpAgent,\n  type LangGraphAgentConfig,\n  ProcessedEvents,\n  SchemaKeys,\n  type State,\n  StateEnrichment,\n} from \"@ag-ui/langgraph\";\nimport { Message as LangGraphMessage } from \"@langchain/langgraph-sdk/dist/types.messages\";\nimport { ThreadState } from \"@langchain/langgraph-sdk\";\n\ninterface CopilotKitStateEnrichment {\n  copilotkit: {\n    actions: StateEnrichment[\"ag-ui\"][\"tools\"];\n    context: StateEnrichment[\"ag-ui\"][\"context\"];\n  };\n}\n\nimport { RunAgentInput, EventType, CustomEvent } from \"@ag-ui/client\";\n\n// Import and re-export from separate file to maintain API compatibility\nimport {\n  CustomEventNames,\n  TextMessageEvents,\n  ToolCallEvents,\n  PredictStateTool,\n} from \"./consts\";\nexport { CustomEventNames };\n\nexport class LangGraphAgent extends AGUILangGraphAgent {\n  constructor(config: LangGraphAgentConfig) {\n    super(config);\n  }\n\n  dispatchEvent(event: ProcessedEvents) {\n    if (event.type === EventType.CUSTOM) {\n      // const event = processedEvent as unknown as CustomEvent;\n      const customEvent = event as unknown as CustomEvent;\n\n      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitMessage) {\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_START,\n          role: \"assistant\",\n          messageId: customEvent.value.message_id,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_CONTENT,\n          messageId: customEvent.value.message_id,\n          delta: customEvent.value.message,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_END,\n          messageId: customEvent.value.message_id,\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (\n        customEvent.name === CustomEventNames.CopilotKitManuallyEmitToolCall\n      ) {\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_START,\n          toolCallId: customEvent.value.id,\n          toolCallName: customEvent.value.name,\n          parentMessageId: customEvent.value.id,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_ARGS,\n          toolCallId: customEvent.value.id,\n          delta: customEvent.value.args,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_END,\n          toolCallId: customEvent.value.id,\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (\n        customEvent.name ===\n        CustomEventNames.CopilotKitManuallyEmitIntermediateState\n      ) {\n        this.activeRun.manuallyEmittedState = customEvent.value;\n        this.dispatchEvent({\n          type: EventType.STATE_SNAPSHOT,\n          snapshot: this.getStateSnapshot({\n            values: this.activeRun.manuallyEmittedState,\n          } as ThreadState<State>),\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (customEvent.name === CustomEventNames.CopilotKitExit) {\n        this.subscriber.next({\n          type: EventType.CUSTOM,\n          name: \"Exit\",\n          value: true,\n        });\n        return true;\n      }\n    }\n\n    // Intercept all text message and tool call events and check if should disable\n    const rawEvent = (event as ToolCallEvents | TextMessageEvents).rawEvent;\n    if (!rawEvent) {\n      this.subscriber.next(event);\n      return true;\n    }\n\n    const isMessageEvent =\n      event.type === EventType.TEXT_MESSAGE_START ||\n      event.type === EventType.TEXT_MESSAGE_CONTENT ||\n      event.type === EventType.TEXT_MESSAGE_END;\n    const isToolEvent =\n      event.type === EventType.TOOL_CALL_START ||\n      event.type === EventType.TOOL_CALL_ARGS ||\n      event.type === EventType.TOOL_CALL_END;\n    if (\"copilotkit:emit-tool-calls\" in (rawEvent.metadata || {})) {\n      if (\n        rawEvent.metadata[\"copilotkit:emit-tool-calls\"] === false &&\n        isToolEvent\n      ) {\n        return false;\n      }\n    }\n    if (\"copilotkit:emit-messages\" in (rawEvent.metadata || {})) {\n      if (\n        rawEvent.metadata[\"copilotkit:emit-messages\"] === false &&\n        isMessageEvent\n      ) {\n        // Clean up tracked message state to prevent stale records from\n        // leaking into subsequent nodes that have emit-messages enabled.\n        if (this.activeRun?.id) {\n          this.messagesInProcess[this.activeRun.id] = null;\n        }\n        return false;\n      }\n    }\n\n    this.subscriber.next(event);\n    return true;\n  }\n\n  // @ts-ignore\n  run(input: RunAgentInput): Observable<BaseEvent> {\n    return super.run(input).pipe(\n      map((processedEvent) => {\n        // Turn raw event into emit state snapshot from tool call event\n        if (processedEvent.type === EventType.RAW) {\n          // Get the LangGraph event from the AGUI event.\n          const event =\n            (processedEvent as RawEvent).event ??\n            (processedEvent as RawEvent).rawEvent;\n\n          const eventType = event.event;\n          const toolCallData = event.data?.chunk?.tool_call_chunks?.[0];\n          const toolCallUsedToPredictState = event.metadata?.[\n            \"copilotkit:emit-intermediate-state\"\n          ]?.some(\n            (predictStateTool: PredictStateTool) =>\n              predictStateTool.tool === toolCallData?.name,\n          );\n\n          if (\n            eventType === LangGraphEventTypes.OnChatModelStream &&\n            toolCallUsedToPredictState\n          ) {\n            return {\n              type: EventType.CUSTOM,\n              name: \"PredictState\",\n              value: event.metadata[\"copilotkit:emit-intermediate-state\"],\n            };\n          }\n        }\n\n        return processedEvent;\n      }),\n    );\n  }\n\n  langGraphDefaultMergeState(\n    state: State,\n    messages: LangGraphMessage[],\n    input: RunAgentInput,\n  ): State<StateEnrichment & CopilotKitStateEnrichment> {\n    const aguiMergedState = super.langGraphDefaultMergeState(\n      state,\n      messages,\n      input,\n    );\n    const { tools: returnedTools, \"ag-ui\": agui } = aguiMergedState;\n    // tolerate undefined and de-duplicate by stable key (id | name | key)\n    const rawCombinedTools = [\n      ...((returnedTools as any[]) ?? []),\n      ...((agui?.tools as any[]) ?? []),\n    ];\n    const combinedTools = Array.from(\n      new Map(\n        rawCombinedTools.map((t: any) => [\n          t?.id ?? t?.name ?? t?.key ?? JSON.stringify(t),\n          t,\n        ]),\n      ).values(),\n    );\n\n    return {\n      ...aguiMergedState,\n      copilotkit: {\n        actions: combinedTools,\n        context: agui?.context ?? [],\n      },\n    };\n  }\n\n  async getSchemaKeys(): Promise<SchemaKeys> {\n    const CONSTANT_KEYS = [\"copilotkit\"];\n    const schemaKeys = await super.getSchemaKeys();\n    return {\n      config: schemaKeys.config,\n      input: schemaKeys.input ? [...schemaKeys.input, ...CONSTANT_KEYS] : null,\n      output: schemaKeys.output\n        ? [...schemaKeys.output, ...CONSTANT_KEYS]\n        : null,\n      context: schemaKeys.context\n        ? [...schemaKeys.context, ...CONSTANT_KEYS]\n        : null,\n    };\n  }\n}\n\nexport { LangGraphHttpAgent };\n"],"mappings":";;;;;;;;;AAiCA,IAAa,iBAAb,cAAoCA,gCAAmB;CACrD,YAAY,QAA8B;AACxC,QAAM,OAAO;;CAGf,cAAc,OAAwB;AACpC,MAAI,MAAM,SAASC,wBAAU,QAAQ;GAEnC,MAAM,cAAc;AAEpB,OAAI,YAAY,SAASC,gCAAiB,+BAA+B;AACvE,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SAASC,gCAAiB,gCACtC;AACA,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,cAAc,YAAY,MAAM;KAChC,iBAAiB,YAAY,MAAM;KACnC,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SACZC,gCAAiB,yCACjB;AACA,SAAK,UAAU,uBAAuB,YAAY;AAClD,SAAK,cAAc;KACjB,MAAMD,wBAAU;KAChB,UAAU,KAAK,iBAAiB,EAC9B,QAAQ,KAAK,UAAU,sBACxB,CAAuB;KACxB,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OAAI,YAAY,SAASC,gCAAiB,gBAAgB;AACxD,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,OAAO;KACR,CAAC;AACF,WAAO;;;EAKX,MAAM,WAAY,MAA6C;AAC/D,MAAI,CAAC,UAAU;AACb,QAAK,WAAW,KAAK,MAAM;AAC3B,UAAO;;EAGT,MAAM,iBACJ,MAAM,SAASA,wBAAU,sBACzB,MAAM,SAASA,wBAAU,wBACzB,MAAM,SAASA,wBAAU;EAC3B,MAAM,cACJ,MAAM,SAASA,wBAAU,mBACzB,MAAM,SAASA,wBAAU,kBACzB,MAAM,SAASA,wBAAU;AAC3B,MAAI,iCAAiC,SAAS,YAAY,EAAE,GAC1D;OACE,SAAS,SAAS,kCAAkC,SACpD,YAEA,QAAO;;AAGX,MAAI,+BAA+B,SAAS,YAAY,EAAE,GACxD;OACE,SAAS,SAAS,gCAAgC,SAClD,gBACA;AAGA,QAAI,KAAK,WAAW,GAClB,MAAK,kBAAkB,KAAK,UAAU,MAAM;AAE9C,WAAO;;;AAIX,OAAK,WAAW,KAAK,MAAM;AAC3B,SAAO;;CAIT,IAAI,OAA6C;AAC/C,SAAO,MAAM,IAAI,MAAM,CAAC,oBACjB,mBAAmB;AAEtB,OAAI,eAAe,SAASA,wBAAU,KAAK;IAEzC,MAAM,QACH,eAA4B,SAC5B,eAA4B;IAE/B,MAAM,YAAY,MAAM;IACxB,MAAM,eAAe,MAAM,MAAM,OAAO,mBAAmB;IAC3D,MAAM,6BAA6B,MAAM,WACvC,uCACC,MACA,qBACC,iBAAiB,SAAS,cAAc,KAC3C;AAED,QACE,cAAcE,mCAAoB,qBAClC,2BAEA,QAAO;KACL,MAAMF,wBAAU;KAChB,MAAM;KACN,OAAO,MAAM,SAAS;KACvB;;AAIL,UAAO;IACP,CACH;;CAGH,2BACE,OACA,UACA,OACoD;EACpD,MAAM,kBAAkB,MAAM,2BAC5B,OACA,UACA,MACD;EACD,MAAM,EAAE,OAAO,eAAe,SAAS,SAAS;EAEhD,MAAM,mBAAmB,CACvB,GAAK,iBAA2B,EAAE,EAClC,GAAK,MAAM,SAAmB,EAAE,CACjC;EACD,MAAM,gBAAgB,MAAM,KAC1B,IAAI,IACF,iBAAiB,KAAK,MAAW,CAC/B,GAAG,MAAM,GAAG,QAAQ,GAAG,OAAO,KAAK,UAAU,EAAE,EAC/C,EACD,CAAC,CACH,CAAC,QAAQ,CACX;AAED,SAAO;GACL,GAAG;GACH,YAAY;IACV,SAAS;IACT,SAAS,MAAM,WAAW,EAAE;IAC7B;GACF;;CAGH,MAAM,gBAAqC;EACzC,MAAM,gBAAgB,CAAC,aAAa;EACpC,MAAM,aAAa,MAAM,MAAM,eAAe;AAC9C,SAAO;GACL,QAAQ,WAAW;GACnB,OAAO,WAAW,QAAQ,CAAC,GAAG,WAAW,OAAO,GAAG,cAAc,GAAG;GACpE,QAAQ,WAAW,SACf,CAAC,GAAG,WAAW,QAAQ,GAAG,cAAc,GACxC;GACJ,SAAS,WAAW,UAChB,CAAC,GAAG,WAAW,SAAS,GAAG,cAAc,GACzC;GACL"}
+{"version":3,"file":"agent.cjs","names":["AGUILangGraphAgent","EventType","CustomEventNames","LangGraphEventTypes"],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"sourcesContent":["import { map, Observable } from \"rxjs\";\nimport { LangGraphEventTypes } from \"../../../../agents/langgraph/events\";\nimport { BaseEvent, RawEvent } from \"@ag-ui/core\";\nimport {\n  LangGraphAgent as AGUILangGraphAgent,\n  LangGraphHttpAgent,\n  type LangGraphAgentConfig,\n  ProcessedEvents,\n  SchemaKeys,\n  type State,\n  StateEnrichment,\n} from \"@ag-ui/langgraph\";\nimport { Message as LangGraphMessage } from \"@langchain/langgraph-sdk/dist/types.messages\";\nimport { ThreadState } from \"@langchain/langgraph-sdk\";\n\ninterface CopilotKitStateEnrichment {\n  copilotkit: {\n    actions: StateEnrichment[\"ag-ui\"][\"tools\"];\n    context: StateEnrichment[\"ag-ui\"][\"context\"];\n  };\n}\n\nimport { RunAgentInput, EventType, CustomEvent } from \"@ag-ui/client\";\n\n// Import and re-export from separate file to maintain API compatibility\nimport {\n  CustomEventNames,\n  TextMessageEvents,\n  ToolCallEvents,\n  PredictStateTool,\n} from \"./consts\";\nexport { CustomEventNames };\n\nexport class LangGraphAgent extends AGUILangGraphAgent {\n  constructor(config: LangGraphAgentConfig) {\n    super(config);\n  }\n\n  dispatchEvent(event: ProcessedEvents) {\n    if (event.type === EventType.CUSTOM) {\n      // const event = processedEvent as unknown as CustomEvent;\n      const customEvent = event as unknown as CustomEvent;\n\n      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitMessage) {\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_START,\n          role: \"assistant\",\n          messageId: customEvent.value.message_id,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_CONTENT,\n          messageId: customEvent.value.message_id,\n          delta: customEvent.value.message,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_END,\n          messageId: customEvent.value.message_id,\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (\n        customEvent.name === CustomEventNames.CopilotKitManuallyEmitToolCall\n      ) {\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_START,\n          toolCallId: customEvent.value.id,\n          toolCallName: customEvent.value.name,\n          parentMessageId: customEvent.value.id,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_ARGS,\n          toolCallId: customEvent.value.id,\n          delta: customEvent.value.args,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_END,\n          toolCallId: customEvent.value.id,\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (\n        customEvent.name ===\n        CustomEventNames.CopilotKitManuallyEmitIntermediateState\n      ) {\n        this.activeRun.manuallyEmittedState = customEvent.value;\n        this.dispatchEvent({\n          type: EventType.STATE_SNAPSHOT,\n          snapshot: this.getStateSnapshot({\n            values: this.activeRun.manuallyEmittedState,\n          } as ThreadState<State>),\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (customEvent.name === CustomEventNames.CopilotKitExit) {\n        this.subscriber.next({\n          type: EventType.CUSTOM,\n          name: \"Exit\",\n          value: true,\n        });\n        return true;\n      }\n    }\n\n    // Intercept all text message and tool call events and check if should disable\n    const rawEvent = (event as ToolCallEvents | TextMessageEvents).rawEvent;\n    if (!rawEvent) {\n      this.subscriber.next(event);\n      return true;\n    }\n\n    const isMessageEvent =\n      event.type === EventType.TEXT_MESSAGE_START ||\n      event.type === EventType.TEXT_MESSAGE_CONTENT ||\n      event.type === EventType.TEXT_MESSAGE_END;\n    const isToolEvent =\n      event.type === EventType.TOOL_CALL_START ||\n      event.type === EventType.TOOL_CALL_ARGS ||\n      event.type === EventType.TOOL_CALL_END;\n    if (\"copilotkit:emit-tool-calls\" in (rawEvent.metadata || {})) {\n      if (\n        rawEvent.metadata[\"copilotkit:emit-tool-calls\"] === false &&\n        isToolEvent\n      ) {\n        return false;\n      }\n    }\n    if (\"copilotkit:emit-messages\" in (rawEvent.metadata || {})) {\n      if (\n        rawEvent.metadata[\"copilotkit:emit-messages\"] === false &&\n        isMessageEvent\n      ) {\n        // Clean up tracked message state to prevent stale records from\n        // leaking into subsequent nodes that have emit-messages enabled.\n        if (this.activeRun?.id) {\n          this.messagesInProcess[this.activeRun.id] = null;\n        }\n        return false;\n      }\n    }\n\n    this.subscriber.next(event);\n    return true;\n  }\n\n  // @ts-ignore\n  run(input: RunAgentInput): Observable<BaseEvent> {\n    const enrichedInput = {\n      ...input,\n      forwardedProps: {\n        ...input.forwardedProps,\n        streamSubgraphs: input.forwardedProps?.streamSubgraphs ?? true,\n      },\n    };\n    return super.run(enrichedInput).pipe(\n      map((processedEvent) => {\n        // Turn raw event into emit state snapshot from tool call event\n        if (processedEvent.type === EventType.RAW) {\n          // Get the LangGraph event from the AGUI event.\n          const event =\n            (processedEvent as RawEvent).event ??\n            (processedEvent as RawEvent).rawEvent;\n\n          const eventType = event.event;\n          const toolCallData = event.data?.chunk?.tool_call_chunks?.[0];\n          const toolCallUsedToPredictState = event.metadata?.[\n            \"copilotkit:emit-intermediate-state\"\n          ]?.some(\n            (predictStateTool: PredictStateTool) =>\n              predictStateTool.tool === toolCallData?.name,\n          );\n\n          if (\n            eventType === LangGraphEventTypes.OnChatModelStream &&\n            toolCallUsedToPredictState\n          ) {\n            return {\n              type: EventType.CUSTOM,\n              name: \"PredictState\",\n              value: event.metadata[\"copilotkit:emit-intermediate-state\"],\n            };\n          }\n        }\n\n        return processedEvent;\n      }),\n    );\n  }\n\n  langGraphDefaultMergeState(\n    state: State,\n    messages: LangGraphMessage[],\n    input: RunAgentInput,\n  ): State<StateEnrichment & CopilotKitStateEnrichment> {\n    const aguiMergedState = super.langGraphDefaultMergeState(\n      state,\n      messages,\n      input,\n    );\n    const { tools: returnedTools, \"ag-ui\": agui } = aguiMergedState;\n    // tolerate undefined and de-duplicate by stable key (id | name | key)\n    const rawCombinedTools = [\n      ...((returnedTools as any[]) ?? []),\n      ...((agui?.tools as any[]) ?? []),\n    ];\n    const combinedTools = Array.from(\n      new Map(\n        rawCombinedTools.map((t: any) => [\n          t?.id ?? t?.name ?? t?.key ?? JSON.stringify(t),\n          t,\n        ]),\n      ).values(),\n    );\n\n    return {\n      ...aguiMergedState,\n      copilotkit: {\n        actions: combinedTools,\n        context: agui?.context ?? [],\n      },\n    };\n  }\n\n  async getSchemaKeys(): Promise<SchemaKeys> {\n    const CONSTANT_KEYS = [\"copilotkit\"];\n    const schemaKeys = await super.getSchemaKeys();\n    return {\n      config: schemaKeys.config,\n      input: schemaKeys.input ? [...schemaKeys.input, ...CONSTANT_KEYS] : null,\n      output: schemaKeys.output\n        ? [...schemaKeys.output, ...CONSTANT_KEYS]\n        : null,\n      context: schemaKeys.context\n        ? [...schemaKeys.context, ...CONSTANT_KEYS]\n        : null,\n    };\n  }\n}\n\nexport { LangGraphHttpAgent };\n"],"mappings":";;;;;;;;;AAiCA,IAAa,iBAAb,cAAoCA,gCAAmB;CACrD,YAAY,QAA8B;AACxC,QAAM,OAAO;;CAGf,cAAc,OAAwB;AACpC,MAAI,MAAM,SAASC,wBAAU,QAAQ;GAEnC,MAAM,cAAc;AAEpB,OAAI,YAAY,SAASC,gCAAiB,+BAA+B;AACvE,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SAASC,gCAAiB,gCACtC;AACA,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,cAAc,YAAY,MAAM;KAChC,iBAAiB,YAAY,MAAM;KACnC,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SACZC,gCAAiB,yCACjB;AACA,SAAK,UAAU,uBAAuB,YAAY;AAClD,SAAK,cAAc;KACjB,MAAMD,wBAAU;KAChB,UAAU,KAAK,iBAAiB,EAC9B,QAAQ,KAAK,UAAU,sBACxB,CAAuB;KACxB,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OAAI,YAAY,SAASC,gCAAiB,gBAAgB;AACxD,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,OAAO;KACR,CAAC;AACF,WAAO;;;EAKX,MAAM,WAAY,MAA6C;AAC/D,MAAI,CAAC,UAAU;AACb,QAAK,WAAW,KAAK,MAAM;AAC3B,UAAO;;EAGT,MAAM,iBACJ,MAAM,SAASA,wBAAU,sBACzB,MAAM,SAASA,wBAAU,wBACzB,MAAM,SAASA,wBAAU;EAC3B,MAAM,cACJ,MAAM,SAASA,wBAAU,mBACzB,MAAM,SAASA,wBAAU,kBACzB,MAAM,SAASA,wBAAU;AAC3B,MAAI,iCAAiC,SAAS,YAAY,EAAE,GAC1D;OACE,SAAS,SAAS,kCAAkC,SACpD,YAEA,QAAO;;AAGX,MAAI,+BAA+B,SAAS,YAAY,EAAE,GACxD;OACE,SAAS,SAAS,gCAAgC,SAClD,gBACA;AAGA,QAAI,KAAK,WAAW,GAClB,MAAK,kBAAkB,KAAK,UAAU,MAAM;AAE9C,WAAO;;;AAIX,OAAK,WAAW,KAAK,MAAM;AAC3B,SAAO;;CAIT,IAAI,OAA6C;EAC/C,MAAM,gBAAgB;GACpB,GAAG;GACH,gBAAgB;IACd,GAAG,MAAM;IACT,iBAAiB,MAAM,gBAAgB,mBAAmB;IAC3D;GACF;AACD,SAAO,MAAM,IAAI,cAAc,CAAC,oBACzB,mBAAmB;AAEtB,OAAI,eAAe,SAASA,wBAAU,KAAK;IAEzC,MAAM,QACH,eAA4B,SAC5B,eAA4B;IAE/B,MAAM,YAAY,MAAM;IACxB,MAAM,eAAe,MAAM,MAAM,OAAO,mBAAmB;IAC3D,MAAM,6BAA6B,MAAM,WACvC,uCACC,MACA,qBACC,iBAAiB,SAAS,cAAc,KAC3C;AAED,QACE,cAAcE,mCAAoB,qBAClC,2BAEA,QAAO;KACL,MAAMF,wBAAU;KAChB,MAAM;KACN,OAAO,MAAM,SAAS;KACvB;;AAIL,UAAO;IACP,CACH;;CAGH,2BACE,OACA,UACA,OACoD;EACpD,MAAM,kBAAkB,MAAM,2BAC5B,OACA,UACA,MACD;EACD,MAAM,EAAE,OAAO,eAAe,SAAS,SAAS;EAEhD,MAAM,mBAAmB,CACvB,GAAK,iBAA2B,EAAE,EAClC,GAAK,MAAM,SAAmB,EAAE,CACjC;EACD,MAAM,gBAAgB,MAAM,KAC1B,IAAI,IACF,iBAAiB,KAAK,MAAW,CAC/B,GAAG,MAAM,GAAG,QAAQ,GAAG,OAAO,KAAK,UAAU,EAAE,EAC/C,EACD,CAAC,CACH,CAAC,QAAQ,CACX;AAED,SAAO;GACL,GAAG;GACH,YAAY;IACV,SAAS;IACT,SAAS,MAAM,WAAW,EAAE;IAC7B;GACF;;CAGH,MAAM,gBAAqC;EACzC,MAAM,gBAAgB,CAAC,aAAa;EACpC,MAAM,aAAa,MAAM,MAAM,eAAe;AAC9C,SAAO;GACL,QAAQ,WAAW;GACnB,OAAO,WAAW,QAAQ,CAAC,GAAG,WAAW,OAAO,GAAG,cAAc,GAAG;GACpE,QAAQ,WAAW,SACf,CAAC,GAAG,WAAW,QAAQ,GAAG,cAAc,GACxC;GACJ,SAAS,WAAW,UAChB,CAAC,GAAG,WAAW,SAAS,GAAG,cAAc,GACzC;GACL"}

package/dist/lib/runtime/agent-integrations/langgraph/agent.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.cts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;EAoCtC,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,SAAA,IACV,KAAA,EAAO,aAAA,GACN,KAAA,CAAM,eAAA,GAAkB,yBAAA;EA8BrB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;AAAA"}
+{"version":3,"file":"agent.d.cts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;EA2CtC,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,SAAA,IACV,KAAA,EAAO,aAAA,GACN,KAAA,CAAM,eAAA,GAAkB,yBAAA;EA8BrB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;AAAA"}

package/dist/lib/runtime/agent-integrations/langgraph/agent.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.mts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;EAoCtC,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,SAAA,IACV,KAAA,EAAO,aAAA,GACN,KAAA,CAAM,eAAA,GAAkB,yBAAA;EA8BrB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;AAAA"}
+{"version":3,"file":"agent.d.mts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;EA2CtC,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,SAAA,IACV,KAAA,EAAO,aAAA,GACN,KAAA,CAAM,eAAA,GAAkB,yBAAA;EA8BrB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;AAAA"}

package/dist/lib/runtime/agent-integrations/langgraph/agent.mjs

@@ -92,7 +92,14 @@ var LangGraphAgent$1 = class extends LangGraphAgent {
 		return true;
 	}
 	run(input) {
-		return super.run(input).pipe(map((processedEvent) => {
+		const enrichedInput = {
+			...input,
+			forwardedProps: {
+				...input.forwardedProps,
+				streamSubgraphs: input.forwardedProps?.streamSubgraphs ?? true
+			}
+		};
+		return super.run(enrichedInput).pipe(map((processedEvent) => {
 			if (processedEvent.type === EventType.RAW) {
 				const event = processedEvent.event ?? processedEvent.rawEvent;
 				const eventType = event.event;