@assistant-ui/react-a2a 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 AgentbaseAI Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,3 @@
+# `@assistant-ui/react-a2a`
+
+A2A integration for `@assistant-ui/react`.

package/dist/A2AMessageAccumulator.d.ts

@@ -0,0 +1,16 @@
+export type A2AStateAccumulatorConfig<TMessage> = {
+    initialMessages?: TMessage[];
+    appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;
+};
+export declare class A2AMessageAccumulator<TMessage extends {
+    id?: string;
+}> {
+    private messagesMap;
+    private appendMessage;
+    constructor({ initialMessages, appendMessage, }?: A2AStateAccumulatorConfig<TMessage>);
+    private ensureMessageId;
+    addMessages(newMessages: TMessage[]): TMessage[];
+    getMessages(): TMessage[];
+    clear(): void;
+}
+//# sourceMappingURL=A2AMessageAccumulator.d.ts.map

package/dist/A2AMessageAccumulator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"A2AMessageAccumulator.d.ts","sourceRoot":"","sources":["../src/A2AMessageAccumulator.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,yBAAyB,CAAC,QAAQ,IAAI;IAChD,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,IAAI,EAAE,QAAQ,KAAK,QAAQ,CAAC;CAC1E,CAAC;AAEF,qBAAa,qBAAqB,CAAC,QAAQ,SAAS;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE;IACjE,OAAO,CAAC,WAAW,CAA+B;IAClD,OAAO,CAAC,aAAa,CAGP;gBAEF,EACV,eAAoB,EACpB,aAGa,GACd,GAAE,yBAAyB,CAAC,QAAQ,CAAM;IAK3C,OAAO,CAAC,eAAe;IAIhB,WAAW,CAAC,WAAW,EAAE,QAAQ,EAAE;IAWnC,WAAW,IAAI,QAAQ,EAAE;IAIzB,KAAK;CAGb"}

package/dist/A2AMessageAccumulator.js

@@ -0,0 +1,35 @@
+// src/A2AMessageAccumulator.ts
+import { v4 as uuidv4 } from "uuid";
+var A2AMessageAccumulator = class {
+  messagesMap = /* @__PURE__ */ new Map();
+  appendMessage;
+  constructor({
+    initialMessages = [],
+    appendMessage = ((_, curr) => curr)
+  } = {}) {
+    this.appendMessage = appendMessage;
+    this.addMessages(initialMessages);
+  }
+  ensureMessageId(message) {
+    return message.id ? message : { ...message, id: uuidv4() };
+  }
+  addMessages(newMessages) {
+    if (newMessages.length === 0) return this.getMessages();
+    for (const message of newMessages.map(this.ensureMessageId)) {
+      const messageId = message.id;
+      const previous = this.messagesMap.get(messageId);
+      this.messagesMap.set(messageId, this.appendMessage(previous, message));
+    }
+    return this.getMessages();
+  }
+  getMessages() {
+    return [...this.messagesMap.values()];
+  }
+  clear() {
+    this.messagesMap.clear();
+  }
+};
+export {
+  A2AMessageAccumulator
+};
+//# sourceMappingURL=A2AMessageAccumulator.js.map

package/dist/A2AMessageAccumulator.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/A2AMessageAccumulator.ts"],"sourcesContent":["import { v4 as uuidv4 } from \"uuid\";\n\nexport type A2AStateAccumulatorConfig<TMessage> = {\n  initialMessages?: TMessage[];\n  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;\n};\n\nexport class A2AMessageAccumulator<TMessage extends { id?: string }> {\n  private messagesMap = new Map<string, TMessage>();\n  private appendMessage: (\n    prev: TMessage | undefined,\n    curr: TMessage,\n  ) => TMessage;\n\n  constructor({\n    initialMessages = [],\n    appendMessage = ((_: TMessage | undefined, curr: TMessage) => curr) as (\n      prev: TMessage | undefined,\n      curr: TMessage,\n    ) => TMessage,\n  }: A2AStateAccumulatorConfig<TMessage> = {}) {\n    this.appendMessage = appendMessage;\n    this.addMessages(initialMessages);\n  }\n\n  private ensureMessageId(message: TMessage): TMessage {\n    return message.id ? message : { ...message, id: uuidv4() };\n  }\n\n  public addMessages(newMessages: TMessage[]) {\n    if (newMessages.length === 0) return this.getMessages();\n\n    for (const message of newMessages.map(this.ensureMessageId)) {\n      const messageId = message.id!; // ensureMessageId guarantees id exists\n      const previous = this.messagesMap.get(messageId);\n      this.messagesMap.set(messageId, this.appendMessage(previous, message));\n    }\n    return this.getMessages();\n  }\n\n  public getMessages(): TMessage[] {\n    return [...this.messagesMap.values()];\n  }\n\n  public clear() {\n    this.messagesMap.clear();\n  }\n}\n"],"mappings":";AAAA,SAAS,MAAM,cAAc;AAOtB,IAAM,wBAAN,MAA8D;AAAA,EAC3D,cAAc,oBAAI,IAAsB;AAAA,EACxC;AAAA,EAKR,YAAY;AAAA,IACV,kBAAkB,CAAC;AAAA,IACnB,iBAAiB,CAAC,GAAyB,SAAmB;AAAA,EAIhE,IAAyC,CAAC,GAAG;AAC3C,SAAK,gBAAgB;AACrB,SAAK,YAAY,eAAe;AAAA,EAClC;AAAA,EAEQ,gBAAgB,SAA6B;AACnD,WAAO,QAAQ,KAAK,UAAU,EAAE,GAAG,SAAS,IAAI,OAAO,EAAE;AAAA,EAC3D;AAAA,EAEO,YAAY,aAAyB;AAC1C,QAAI,YAAY,WAAW,EAAG,QAAO,KAAK,YAAY;AAEtD,eAAW,WAAW,YAAY,IAAI,KAAK,eAAe,GAAG;AAC3D,YAAM,YAAY,QAAQ;AAC1B,YAAM,WAAW,KAAK,YAAY,IAAI,SAAS;AAC/C,WAAK,YAAY,IAAI,WAAW,KAAK,cAAc,UAAU,OAAO,CAAC;AAAA,IACvE;AACA,WAAO,KAAK,YAAY;AAAA,EAC1B;AAAA,EAEO,cAA0B;AAC/B,WAAO,CAAC,GAAG,KAAK,YAAY,OAAO,CAAC;AAAA,EACtC;AAAA,EAEO,QAAQ;AACb,SAAK,YAAY,MAAM;AAAA,EACzB;AACF;","names":[]}

package/dist/appendA2AChunk.d.ts

@@ -0,0 +1,3 @@
+import { A2AMessage } from "./types";
+export declare const appendA2AChunk: (prev: A2AMessage | undefined, curr: A2AMessage) => A2AMessage;
+//# sourceMappingURL=appendA2AChunk.d.ts.map

package/dist/appendA2AChunk.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"appendA2AChunk.d.ts","sourceRoot":"","sources":["../src/appendA2AChunk.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAGrC,eAAO,MAAM,cAAc,GACzB,MAAM,UAAU,GAAG,SAAS,EAC5B,MAAM,UAAU,KACf,UAkHF,CAAC"}

package/dist/appendA2AChunk.js

@@ -0,0 +1,90 @@
+// src/appendA2AChunk.ts
+import { parsePartialJsonObject } from "assistant-stream/utils";
+var appendA2AChunk = (prev, curr) => {
+  if (!prev || prev.role !== curr.role || prev.id !== curr.id) {
+    return curr;
+  }
+  if (curr.role === "assistant") {
+    const newContent = Array.isArray(prev.content) ? [...prev.content] : typeof prev.content === "string" ? [{ type: "text", text: prev.content }] : [];
+    if (typeof curr.content === "string") {
+      const lastIndex = newContent.length - 1;
+      const lastPart = newContent[lastIndex];
+      if (lastPart?.type === "text") {
+        lastPart.text += curr.content;
+      } else {
+        newContent.push({ type: "text", text: curr.content });
+      }
+    } else if (Array.isArray(curr.content)) {
+      for (const contentPart of curr.content) {
+        const lastIndex = newContent.length - 1;
+        const lastPart = newContent[lastIndex];
+        if (contentPart.type === "text" && lastPart?.type === "text") {
+          lastPart.text += contentPart.text;
+        } else {
+          newContent.push(contentPart);
+        }
+      }
+    }
+    const newToolCalls = [...prev.tool_calls ?? []];
+    if (curr.tool_calls) {
+      for (const toolCall of curr.tool_calls) {
+        const existingIndex = newToolCalls.findIndex(
+          (tc) => tc.id === toolCall.id
+        );
+        if (existingIndex >= 0) {
+          const existing = newToolCalls[existingIndex];
+          newToolCalls[existingIndex] = {
+            ...existing,
+            ...toolCall,
+            // If argsText is provided in chunks, concatenate it
+            argsText: (existing.argsText || "") + (toolCall.argsText || ""),
+            // Try to parse merged args, fallback to existing or new args
+            args: parsePartialJsonObject(
+              (existing.argsText || "") + (toolCall.argsText || "")
+            ) || toolCall.args || existing.args
+          };
+        } else {
+          newToolCalls.push(toolCall);
+        }
+      }
+    }
+    const newArtifacts = [...prev.artifacts ?? []];
+    if (curr.artifacts) {
+      for (const artifact of curr.artifacts) {
+        const existingIndex = newArtifacts.findIndex(
+          (a) => a.name === artifact.name
+        );
+        if (existingIndex >= 0) {
+          const existingArtifact = newArtifacts[existingIndex];
+          newArtifacts[existingIndex] = {
+            name: existingArtifact.name,
+            parts: [...existingArtifact.parts, ...artifact.parts]
+          };
+        } else {
+          newArtifacts.push(artifact);
+        }
+      }
+    }
+    const result2 = {
+      ...prev,
+      content: newContent
+    };
+    const newStatus = curr.status || prev.status;
+    if (newStatus) result2.status = newStatus;
+    if (newToolCalls.length > 0) result2.tool_calls = newToolCalls;
+    if (newArtifacts.length > 0) result2.artifacts = newArtifacts;
+    return result2;
+  }
+  const result = {
+    ...prev,
+    ...curr
+  };
+  if (curr.artifacts || prev.artifacts) {
+    result.artifacts = [...prev.artifacts ?? [], ...curr.artifacts ?? []];
+  }
+  return result;
+};
+export {
+  appendA2AChunk
+};
+//# sourceMappingURL=appendA2AChunk.js.map

package/dist/appendA2AChunk.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/appendA2AChunk.ts"],"sourcesContent":["import { A2AMessage } from \"./types\";\nimport { parsePartialJsonObject } from \"assistant-stream/utils\";\n\nexport const appendA2AChunk = (\n  prev: A2AMessage | undefined,\n  curr: A2AMessage,\n): A2AMessage => {\n  // If no previous message or different message type, return current as-is\n  if (!prev || prev.role !== curr.role || prev.id !== curr.id) {\n    return curr;\n  }\n\n  // For assistant messages, we need to handle streaming content and tool calls\n  if (curr.role === \"assistant\") {\n    const newContent = Array.isArray(prev.content)\n      ? [...prev.content]\n      : typeof prev.content === \"string\"\n        ? [{ type: \"text\" as const, text: prev.content }]\n        : [];\n\n    // Append new content chunks\n    if (typeof curr.content === \"string\") {\n      const lastIndex = newContent.length - 1;\n      const lastPart = newContent[lastIndex];\n\n      if (lastPart?.type === \"text\") {\n        // Append to existing text part\n        (lastPart as { type: \"text\"; text: string }).text += curr.content;\n      } else {\n        // Create new text part\n        newContent.push({ type: \"text\", text: curr.content });\n      }\n    } else if (Array.isArray(curr.content)) {\n      for (const contentPart of curr.content) {\n        const lastIndex = newContent.length - 1;\n        const lastPart = newContent[lastIndex];\n\n        if (contentPart.type === \"text\" && lastPart?.type === \"text\") {\n          // Append to existing text part\n          (lastPart as { type: \"text\"; text: string }).text += contentPart.text;\n        } else {\n          // Add new content part\n          newContent.push(contentPart);\n        }\n      }\n    }\n\n    // Merge tool calls - A2A typically sends complete tool calls rather than chunks\n    const newToolCalls = [...(prev.tool_calls ?? [])];\n    if (curr.tool_calls) {\n      for (const toolCall of curr.tool_calls) {\n        const existingIndex = newToolCalls.findIndex(\n          (tc) => tc.id === toolCall.id,\n        );\n        if (existingIndex >= 0) {\n          // Update existing tool call (merge args if needed)\n          const existing = newToolCalls[existingIndex]!;\n          newToolCalls[existingIndex] = {\n            ...existing,\n            ...toolCall,\n            // If argsText is provided in chunks, concatenate it\n            argsText: (existing.argsText || \"\") + (toolCall.argsText || \"\"),\n            // Try to parse merged args, fallback to existing or new args\n            args:\n              parsePartialJsonObject(\n                (existing.argsText || \"\") + (toolCall.argsText || \"\"),\n              ) ||\n              toolCall.args ||\n              existing.args,\n          };\n        } else {\n          // Add new tool call\n          newToolCalls.push(toolCall);\n        }\n      }\n    }\n\n    // Merge artifacts\n    const newArtifacts = [...(prev.artifacts ?? [])];\n    if (curr.artifacts) {\n      for (const artifact of curr.artifacts) {\n        const existingIndex = newArtifacts.findIndex(\n          (a) => a.name === artifact.name,\n        );\n        if (existingIndex >= 0) {\n          // Merge artifact parts\n          const existingArtifact = newArtifacts[existingIndex]!;\n          newArtifacts[existingIndex] = {\n            name: existingArtifact.name,\n            parts: [...existingArtifact.parts, ...artifact.parts],\n          };\n        } else {\n          // Add new artifact\n          newArtifacts.push(artifact);\n        }\n      }\n    }\n\n    const result: A2AMessage = {\n      ...prev,\n      content: newContent,\n    };\n    const newStatus = curr.status || prev.status;\n    if (newStatus) result.status = newStatus;\n    if (newToolCalls.length > 0) result.tool_calls = newToolCalls;\n    if (newArtifacts.length > 0) result.artifacts = newArtifacts;\n    return result;\n  }\n\n  // For other message types (user, system, tool), just return the current message\n  // as they typically don't stream in chunks\n  const result: A2AMessage = {\n    ...prev,\n    ...curr,\n  };\n  // Preserve any existing artifacts and merge with new ones\n  if (curr.artifacts || prev.artifacts) {\n    result.artifacts = [...(prev.artifacts ?? []), ...(curr.artifacts ?? [])];\n  }\n  return result;\n};\n"],"mappings":";AACA,SAAS,8BAA8B;AAEhC,IAAM,iBAAiB,CAC5B,MACA,SACe;AAEf,MAAI,CAAC,QAAQ,KAAK,SAAS,KAAK,QAAQ,KAAK,OAAO,KAAK,IAAI;AAC3D,WAAO;AAAA,EACT;AAGA,MAAI,KAAK,SAAS,aAAa;AAC7B,UAAM,aAAa,MAAM,QAAQ,KAAK,OAAO,IACzC,CAAC,GAAG,KAAK,OAAO,IAChB,OAAO,KAAK,YAAY,WACtB,CAAC,EAAE,MAAM,QAAiB,MAAM,KAAK,QAAQ,CAAC,IAC9C,CAAC;AAGP,QAAI,OAAO,KAAK,YAAY,UAAU;AACpC,YAAM,YAAY,WAAW,SAAS;AACtC,YAAM,WAAW,WAAW,SAAS;AAErC,UAAI,UAAU,SAAS,QAAQ;AAE7B,QAAC,SAA4C,QAAQ,KAAK;AAAA,MAC5D,OAAO;AAEL,mBAAW,KAAK,EAAE,MAAM,QAAQ,MAAM,KAAK,QAAQ,CAAC;AAAA,MACtD;AAAA,IACF,WAAW,MAAM,QAAQ,KAAK,OAAO,GAAG;AACtC,iBAAW,eAAe,KAAK,SAAS;AACtC,cAAM,YAAY,WAAW,SAAS;AACtC,cAAM,WAAW,WAAW,SAAS;AAErC,YAAI,YAAY,SAAS,UAAU,UAAU,SAAS,QAAQ;AAE5D,UAAC,SAA4C,QAAQ,YAAY;AAAA,QACnE,OAAO;AAEL,qBAAW,KAAK,WAAW;AAAA,QAC7B;AAAA,MACF;AAAA,IACF;AAGA,UAAM,eAAe,CAAC,GAAI,KAAK,cAAc,CAAC,CAAE;AAChD,QAAI,KAAK,YAAY;AACnB,iBAAW,YAAY,KAAK,YAAY;AACtC,cAAM,gBAAgB,aAAa;AAAA,UACjC,CAAC,OAAO,GAAG,OAAO,SAAS;AAAA,QAC7B;AACA,YAAI,iBAAiB,GAAG;AAEtB,gBAAM,WAAW,aAAa,aAAa;AAC3C,uBAAa,aAAa,IAAI;AAAA,YAC5B,GAAG;AAAA,YACH,GAAG;AAAA;AAAA,YAEH,WAAW,SAAS,YAAY,OAAO,SAAS,YAAY;AAAA;AAAA,YAE5D,MACE;AAAA,eACG,SAAS,YAAY,OAAO,SAAS,YAAY;AAAA,YACpD,KACA,SAAS,QACT,SAAS;AAAA,UACb;AAAA,QACF,OAAO;AAEL,uBAAa,KAAK,QAAQ;AAAA,QAC5B;AAAA,MACF;AAAA,IACF;AAGA,UAAM,eAAe,CAAC,GAAI,KAAK,aAAa,CAAC,CAAE;AAC/C,QAAI,KAAK,WAAW;AAClB,iBAAW,YAAY,KAAK,WAAW;AACrC,cAAM,gBAAgB,aAAa;AAAA,UACjC,CAAC,MAAM,EAAE,SAAS,SAAS;AAAA,QAC7B;AACA,YAAI,iBAAiB,GAAG;AAEtB,gBAAM,mBAAmB,aAAa,aAAa;AACnD,uBAAa,aAAa,IAAI;AAAA,YAC5B,MAAM,iBAAiB;AAAA,YACvB,OAAO,CAAC,GAAG,iBAAiB,OAAO,GAAG,SAAS,KAAK;AAAA,UACtD;AAAA,QACF,OAAO;AAEL,uBAAa,KAAK,QAAQ;AAAA,QAC5B;AAAA,MACF;AAAA,IACF;AAEA,UAAMA,UAAqB;AAAA,MACzB,GAAG;AAAA,MACH,SAAS;AAAA,IACX;AACA,UAAM,YAAY,KAAK,UAAU,KAAK;AACtC,QAAI,UAAW,CAAAA,QAAO,SAAS;AAC/B,QAAI,aAAa,SAAS,EAAG,CAAAA,QAAO,aAAa;AACjD,QAAI,aAAa,SAAS,EAAG,CAAAA,QAAO,YAAY;AAChD,WAAOA;AAAA,EACT;AAIA,QAAM,SAAqB;AAAA,IACzB,GAAG;AAAA,IACH,GAAG;AAAA,EACL;AAEA,MAAI,KAAK,aAAa,KAAK,WAAW;AACpC,WAAO,YAAY,CAAC,GAAI,KAAK,aAAa,CAAC,GAAI,GAAI,KAAK,aAAa,CAAC,CAAE;AAAA,EAC1E;AACA,SAAO;AACT;","names":["result"]}

package/dist/convertA2AMessages.d.ts

@@ -0,0 +1,64 @@
+import { A2AMessage } from "./types";
+import { ToolCallMessagePart } from "@assistant-ui/react";
+export declare const convertA2AMessage: (message: A2AMessage) => {
+    id: string | undefined;
+    role: "system";
+    content: {
+        type: "text";
+        text: string;
+    }[];
+    status?: never;
+} | {
+    id: string | undefined;
+    role: "user";
+    content: NonNullable<import("@assistant-ui/react").ThreadUserMessagePart | null>[];
+    status?: never;
+} | {
+    id: string | undefined;
+    role: "assistant";
+    content: (NonNullable<import("@assistant-ui/react").ThreadUserMessagePart | null> | ToolCallMessagePart)[];
+    status: import("@assistant-ui/react").MessageStatus | undefined;
+} | {
+    id: string | undefined;
+    role: "user";
+    content: {
+        type: "tool-call";
+        toolCallId: string;
+        toolName: string;
+        result: any;
+        isError: boolean;
+    }[];
+    status?: never;
+};
+export declare const convertA2AMessages: (messages: A2AMessage[]) => ({
+    id: string | undefined;
+    role: "system";
+    content: {
+        type: "text";
+        text: string;
+    }[];
+    status?: never;
+} | {
+    id: string | undefined;
+    role: "user";
+    content: NonNullable<import("@assistant-ui/react").ThreadUserMessagePart | null>[];
+    status?: never;
+} | {
+    id: string | undefined;
+    role: "assistant";
+    content: (NonNullable<import("@assistant-ui/react").ThreadUserMessagePart | null> | ToolCallMessagePart)[];
+    status: import("@assistant-ui/react").MessageStatus | undefined;
+} | {
+    id: string | undefined;
+    role: "user";
+    content: {
+        type: "tool-call";
+        toolCallId: string;
+        toolName: string;
+        result: any;
+        isError: boolean;
+    }[];
+    status?: never;
+})[];
+export declare const useA2AMessageConverter: (messages: A2AMessage[], isRunning: boolean) => import("@assistant-ui/react").ThreadMessage[];
+//# sourceMappingURL=convertA2AMessages.d.ts.map

package/dist/convertA2AMessages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"convertA2AMessages.d.ts","sourceRoot":"","sources":["../src/convertA2AMessages.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAkC1D,eAAO,MAAM,iBAAiB,GAAI,SAAS,UAAU;;;;;cAOe,MAAM;;;;;;;;;;;;;;;;;;;;;;;;CAiDzE,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,UAAU,UAAU,EAAE;;;;;cAnDW,MAAM;;;;;;;;;;;;;;;;;;;;;;;;IAoDzC,CAAC;AAElC,eAAO,MAAM,sBAAsB,GACjC,UAAU,UAAU,EAAE,EACtB,WAAW,OAAO,kDAMhB,CAAC"}

package/dist/convertA2AMessages.js

@@ -0,0 +1,93 @@
+"use client";
+
+// src/convertA2AMessages.ts
+import { useExternalMessageConverter } from "@assistant-ui/react";
+var contentToParts = (content) => {
+  if (typeof content === "string")
+    return [{ type: "text", text: content }];
+  return content.map((part) => {
+    const type = part.type;
+    switch (type) {
+      case "text":
+        return { type: "text", text: part.text };
+      case "image_url":
+        if (typeof part.image_url === "string") {
+          return { type: "image", image: part.image_url };
+        } else {
+          return {
+            type: "image",
+            image: part.image_url.url
+          };
+        }
+      case "data":
+        return {
+          type: "text",
+          text: `[Data: ${JSON.stringify(part.data)}]`
+        };
+      default:
+        return null;
+    }
+  }).filter((part) => part !== null);
+};
+var convertA2AMessage = (message) => {
+  const role = message.role;
+  switch (role) {
+    case "system":
+      return {
+        id: message.id,
+        role: "system",
+        content: [{ type: "text", text: message.content }]
+      };
+    case "user":
+      return {
+        id: message.id,
+        role: "user",
+        content: contentToParts(message.content)
+      };
+    case "assistant": {
+      const toolCallParts = message.tool_calls?.map((toolCall) => ({
+        type: "tool-call",
+        toolCallId: toolCall.id,
+        toolName: toolCall.name,
+        args: toolCall.args,
+        argsText: toolCall.argsText ?? JSON.stringify(toolCall.args)
+      })) ?? [];
+      return {
+        id: message.id,
+        role: "assistant",
+        content: [...contentToParts(message.content), ...toolCallParts],
+        status: message.status
+      };
+    }
+    case "tool":
+      return {
+        id: message.id,
+        role: "user",
+        content: [
+          {
+            type: "tool-call",
+            toolCallId: message.tool_call_id,
+            toolName: "",
+            // A2A doesn't store tool name in tool messages
+            result: JSON.parse(message.content),
+            isError: message.status?.type === "incomplete" && message.status?.reason === "error"
+          }
+        ]
+      };
+    default:
+      const _exhaustiveCheck = role;
+      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);
+  }
+};
+var convertA2AMessages = (messages) => messages.map(convertA2AMessage);
+var useA2AMessageConverter = (messages, isRunning) => useExternalMessageConverter({
+  callback: convertA2AMessage,
+  messages,
+  isRunning
+});
+export {
+  convertA2AMessage,
+  convertA2AMessages,
+  useA2AMessageConverter
+};
+//# sourceMappingURL=convertA2AMessages.js.map

package/dist/convertA2AMessages.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/convertA2AMessages.ts"],"sourcesContent":["\"use client\";\n\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { A2AMessage } from \"./types\";\nimport { ToolCallMessagePart } from \"@assistant-ui/react\";\nimport { ThreadUserMessage } from \"@assistant-ui/react\";\n\nconst contentToParts = (content: A2AMessage[\"content\"]) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n  return content\n    .map((part): ThreadUserMessage[\"content\"][number] | null => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"image_url\":\n          if (typeof part.image_url === \"string\") {\n            return { type: \"image\", image: part.image_url };\n          } else {\n            return {\n              type: \"image\",\n              image: part.image_url.url,\n            };\n          }\n        case \"data\":\n          // Convert data parts to text representation for display\n          return {\n            type: \"text\",\n            text: `[Data: ${JSON.stringify(part.data)}]`,\n          };\n        default:\n          return null;\n      }\n    })\n    .filter((part): part is NonNullable<typeof part> => part !== null);\n};\n\nexport const convertA2AMessage = (message: A2AMessage) => {\n  const role = message.role;\n  switch (role) {\n    case \"system\":\n      return {\n        id: message.id,\n        role: \"system\" as const,\n        content: [{ type: \"text\" as const, text: message.content as string }],\n      };\n\n    case \"user\":\n      return {\n        id: message.id,\n        role: \"user\" as const,\n        content: contentToParts(message.content),\n      };\n\n    case \"assistant\": {\n      const toolCallParts: ToolCallMessagePart[] =\n        message.tool_calls?.map((toolCall) => ({\n          type: \"tool-call\",\n          toolCallId: toolCall.id,\n          toolName: toolCall.name,\n          args: toolCall.args,\n          argsText: toolCall.argsText ?? JSON.stringify(toolCall.args),\n        })) ?? [];\n\n      return {\n        id: message.id,\n        role: \"assistant\" as const,\n        content: [...contentToParts(message.content), ...toolCallParts],\n        status: message.status,\n      };\n    }\n\n    case \"tool\":\n      return {\n        id: message.id,\n        role: \"user\" as const,\n        content: [\n          {\n            type: \"tool-call\" as const,\n            toolCallId: message.tool_call_id!,\n            toolName: \"\", // A2A doesn't store tool name in tool messages\n            result: JSON.parse(message.content as string),\n            isError:\n              message.status?.type === \"incomplete\" &&\n              message.status?.reason === \"error\",\n          },\n        ],\n      };\n\n    default:\n      const _exhaustiveCheck: never = role;\n      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);\n  }\n};\n\nexport const convertA2AMessages = (messages: A2AMessage[]) =>\n  messages.map(convertA2AMessage);\n\nexport const useA2AMessageConverter = (\n  messages: A2AMessage[],\n  isRunning: boolean,\n) =>\n  useExternalMessageConverter({\n    callback: convertA2AMessage,\n    messages,\n    isRunning,\n  });\n"],"mappings":";;;AAEA,SAAS,mCAAmC;AAK5C,IAAM,iBAAiB,CAAC,YAAmC;AACzD,MAAI,OAAO,YAAY;AACrB,WAAO,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,CAAC;AAClD,SAAO,QACJ,IAAI,CAAC,SAAsD;AAC1D,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,MACzC,KAAK;AACH,YAAI,OAAO,KAAK,cAAc,UAAU;AACtC,iBAAO,EAAE,MAAM,SAAS,OAAO,KAAK,UAAU;AAAA,QAChD,OAAO;AACL,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,OAAO,KAAK,UAAU;AAAA,UACxB;AAAA,QACF;AAAA,MACF,KAAK;AAEH,eAAO;AAAA,UACL,MAAM;AAAA,UACN,MAAM,UAAU,KAAK,UAAU,KAAK,IAAI,CAAC;AAAA,QAC3C;AAAA,MACF;AACE,eAAO;AAAA,IACX;AAAA,EACF,CAAC,EACA,OAAO,CAAC,SAA2C,SAAS,IAAI;AACrE;AAEO,IAAM,oBAAoB,CAAC,YAAwB;AACxD,QAAM,OAAO,QAAQ;AACrB,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO;AAAA,QACL,IAAI,QAAQ;AAAA,QACZ,MAAM;AAAA,QACN,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,QAAkB,CAAC;AAAA,MACtE;AAAA,IAEF,KAAK;AACH,aAAO;AAAA,QACL,IAAI,QAAQ;AAAA,QACZ,MAAM;AAAA,QACN,SAAS,eAAe,QAAQ,OAAO;AAAA,MACzC;AAAA,IAEF,KAAK,aAAa;AAChB,YAAM,gBACJ,QAAQ,YAAY,IAAI,CAAC,cAAc;AAAA,QACrC,MAAM;AAAA,QACN,YAAY,SAAS;AAAA,QACrB,UAAU,SAAS;AAAA,QACnB,MAAM,SAAS;AAAA,QACf,UAAU,SAAS,YAAY,KAAK,UAAU,SAAS,IAAI;AAAA,MAC7D,EAAE,KAAK,CAAC;AAEV,aAAO;AAAA,QACL,IAAI,QAAQ;AAAA,QACZ,MAAM;AAAA,QACN,SAAS,CAAC,GAAG,eAAe,QAAQ,OAAO,GAAG,GAAG,aAAa;AAAA,QAC9D,QAAQ,QAAQ;AAAA,MAClB;AAAA,IACF;AAAA,IAEA,KAAK;AACH,aAAO;AAAA,QACL,IAAI,QAAQ;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,UACP;AAAA,YACE,MAAM;AAAA,YACN,YAAY,QAAQ;AAAA,YACpB,UAAU;AAAA;AAAA,YACV,QAAQ,KAAK,MAAM,QAAQ,OAAiB;AAAA,YAC5C,SACE,QAAQ,QAAQ,SAAS,gBACzB,QAAQ,QAAQ,WAAW;AAAA,UAC/B;AAAA,QACF;AAAA,MACF;AAAA,IAEF;AACE,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,yBAAyB,gBAAgB,EAAE;AAAA,EAC/D;AACF;AAEO,IAAM,qBAAqB,CAAC,aACjC,SAAS,IAAI,iBAAiB;AAEzB,IAAM,yBAAyB,CACpC,UACA,cAEA,4BAA4B;AAAA,EAC1B,UAAU;AAAA,EACV;AAAA,EACA;AACF,CAAC;","names":[]}

package/dist/index.d.ts

@@ -0,0 +1,7 @@
+export * from "./useA2ARuntime";
+export * from "./useA2AMessages";
+export * from "./convertA2AMessages";
+export * from "./types";
+export * from "./A2AMessageAccumulator";
+export * from "./appendA2AChunk";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,SAAS,CAAC;AACxB,cAAc,yBAAyB,CAAC;AACxC,cAAc,kBAAkB,CAAC"}

package/dist/index.js

@@ -0,0 +1,8 @@
+// src/index.ts
+export * from "./useA2ARuntime.js";
+export * from "./useA2AMessages.js";
+export * from "./convertA2AMessages.js";
+export * from "./types.js";
+export * from "./A2AMessageAccumulator.js";
+export * from "./appendA2AChunk.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export * from \"./useA2ARuntime\";\nexport * from \"./useA2AMessages\";\nexport * from \"./convertA2AMessages\";\nexport * from \"./types\";\nexport * from \"./A2AMessageAccumulator\";\nexport * from \"./appendA2AChunk\";\n"],"mappings":";AAAA,cAAc;AACd,cAAc;AACd,cAAc;AACd,cAAc;AACd,cAAc;AACd,cAAc;","names":[]}

package/dist/testUtils.d.ts

@@ -0,0 +1,4 @@
+import { A2AMessage } from "./types";
+import { A2AMessagesEvent } from "./useA2AMessages";
+export declare const mockStreamCallbackFactory: (events: Array<A2AMessagesEvent<A2AMessage>>) => () => AsyncGenerator<import("./types").A2AEvent, void, unknown>;
+//# sourceMappingURL=testUtils.d.ts.map

package/dist/testUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"testUtils.d.ts","sourceRoot":"","sources":["../src/testUtils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpD,eAAO,MAAM,yBAAyB,GACpC,QAAQ,KAAK,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,oEAM1C,CAAC"}

package/dist/testUtils.js

@@ -0,0 +1,10 @@
+// src/testUtils.ts
+var mockStreamCallbackFactory = (events) => async function* () {
+  for (const event of events) {
+    yield event;
+  }
+};
+export {
+  mockStreamCallbackFactory
+};
+//# sourceMappingURL=testUtils.js.map

package/dist/testUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/testUtils.ts"],"sourcesContent":["import { A2AMessage } from \"./types\";\nimport { A2AMessagesEvent } from \"./useA2AMessages\";\n\nexport const mockStreamCallbackFactory = (\n  events: Array<A2AMessagesEvent<A2AMessage>>,\n) =>\n  async function* () {\n    for (const event of events) {\n      yield event;\n    }\n  };\n"],"mappings":";AAGO,IAAM,4BAA4B,CACvC,WAEA,mBAAmB;AACjB,aAAW,SAAS,QAAQ;AAC1B,UAAM;AAAA,EACR;AACF;","names":[]}

package/dist/types.d.ts

@@ -0,0 +1,94 @@
+import { MessageStatus } from "@assistant-ui/react";
+import { ReadonlyJSONObject } from "assistant-stream/utils";
+export type A2AMessage = {
+    id?: string;
+    role: "user" | "assistant" | "system" | "tool";
+    content: string | A2AMessageContent[];
+    tool_calls?: A2AToolCall[];
+    tool_call_id?: string;
+    artifacts?: A2AArtifact[];
+    status?: MessageStatus;
+};
+export type A2AMessageContent = {
+    type: "text";
+    text: string;
+} | {
+    type: "image_url";
+    image_url: string | {
+        url: string;
+    };
+} | {
+    type: "data";
+    data: any;
+};
+export type A2AToolCall = {
+    id: string;
+    name: string;
+    args: ReadonlyJSONObject;
+    argsText?: string;
+};
+export type A2AArtifact = {
+    name: string;
+    parts: A2AArtifactPart[];
+};
+export type A2AArtifactPart = {
+    kind: "text" | "data" | "file";
+    data?: any;
+    text?: string;
+    metadata?: Record<string, any>;
+};
+export type A2AEvent = {
+    event: A2AKnownEventTypes | string;
+    data: any;
+};
+export declare enum A2AKnownEventTypes {
+    TaskUpdate = "task-update",
+    TaskComplete = "task-complete",
+    TaskFailed = "task-failed",
+    Artifacts = "artifacts",
+    StateUpdate = "state-update",
+    Error = "error"
+}
+export type A2ATaskState = {
+    id: string;
+    state: "pending" | "working" | "completed" | "failed";
+    progress?: number;
+    message?: string;
+};
+export type A2ATaskResult = {
+    id: string;
+    status: {
+        state: "pending" | "working" | "completed" | "failed";
+        message?: string;
+    };
+    artifacts?: A2AArtifact[];
+    history?: Array<{
+        messageId: string;
+        role: string;
+        parts?: Array<{
+            kind: string;
+            text?: string;
+            data?: any;
+        }>;
+    }>;
+};
+export type A2AConfig = {
+    contextId?: string;
+    runConfig?: Record<string, any>;
+};
+export type A2ASendMessageConfig = A2AConfig & {
+    command?: A2ACommand;
+};
+export type A2ACommand = {
+    resume?: string;
+    interrupt?: string;
+};
+export type A2AStreamCallback<TMessage> = (messages: TMessage[], config: A2ASendMessageConfig & {
+    abortSignal: AbortSignal;
+}) => Promise<AsyncGenerator<A2AEvent>> | AsyncGenerator<A2AEvent>;
+export type OnTaskUpdateEventCallback = (data: A2ATaskState) => void | Promise<void>;
+export type OnArtifactsEventCallback = (artifacts: A2AArtifact[]) => void | Promise<void>;
+export type OnErrorEventCallback = (error: unknown) => void | Promise<void>;
+export type OnStateUpdateEventCallback = (state: unknown) => void | Promise<void>;
+export type OnCustomEventCallback = (type: string, data: unknown) => void | Promise<void>;
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAG5D,MAAM,MAAM,UAAU,GAAG;IACvB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,GAAG,iBAAiB,EAAE,CAAC;IACtC,UAAU,CAAC,EAAE,WAAW,EAAE,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;IAC1B,MAAM,CAAC,EAAE,aAAa,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GACzB;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GAC1D;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,GAAG,CAAA;CAAE,CAAC;AAEhC,MAAM,MAAM,WAAW,GAAG;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,kBAAkB,CAAC;IACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,eAAe,EAAE,CAAC;CAC1B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAC/B,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC,CAAC;AAGF,MAAM,MAAM,QAAQ,GAAG;IACrB,KAAK,EAAE,kBAAkB,GAAG,MAAM,CAAC;IACnC,IAAI,EAAE,GAAG,CAAC;CACX,CAAC;AAEF,oBAAY,kBAAkB;IAC5B,UAAU,gBAAgB;IAC1B,YAAY,kBAAkB;IAC9B,UAAU,gBAAgB;IAC1B,SAAS,cAAc;IACvB,WAAW,iBAAiB;IAC5B,KAAK,UAAU;CAChB;AAGD,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAGF,MAAM,MAAM,aAAa,GAAG;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE;QACN,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;QACtD,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;IAC1B,OAAO,CAAC,EAAE,KAAK,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,IAAI,CAAC,EAAE,MAAM,CAAC;YAAC,IAAI,CAAC,EAAE,GAAG,CAAA;SAAE,CAAC,CAAC;KAC5D,CAAC,CAAC;CACJ,CAAC;AAGF,MAAM,MAAM,SAAS,GAAG;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACjC,CAAC;AAGF,MAAM,MAAM,oBAAoB,GAAG,SAAS,GAAG;IAC7C,OAAO,CAAC,EAAE,UAAU,CAAC;CACtB,CAAC;AAGF,MAAM,MAAM,UAAU,GAAG;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAGF,MAAM,MAAM,iBAAiB,CAAC,QAAQ,IAAI,CACxC,QAAQ,EAAE,QAAQ,EAAE,EACpB,MAAM,EAAE,oBAAoB,GAAG;IAAE,WAAW,EAAE,WAAW,CAAA;CAAE,KACxD,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;AAGlE,MAAM,MAAM,yBAAyB,GAAG,CACtC,IAAI,EAAE,YAAY,KACf,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAC1B,MAAM,MAAM,wBAAwB,GAAG,CACrC,SAAS,EAAE,WAAW,EAAE,KACrB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAC1B,MAAM,MAAM,oBAAoB,GAAG,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAC5E,MAAM,MAAM,0BAA0B,GAAG,CACvC,KAAK,EAAE,OAAO,KACX,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAC1B,MAAM,MAAM,qBAAqB,GAAG,CAClC,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,OAAO,KACV,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC"}

package/dist/types.js

@@ -0,0 +1,14 @@
+// src/types.ts
+var A2AKnownEventTypes = /* @__PURE__ */ ((A2AKnownEventTypes2) => {
+  A2AKnownEventTypes2["TaskUpdate"] = "task-update";
+  A2AKnownEventTypes2["TaskComplete"] = "task-complete";
+  A2AKnownEventTypes2["TaskFailed"] = "task-failed";
+  A2AKnownEventTypes2["Artifacts"] = "artifacts";
+  A2AKnownEventTypes2["StateUpdate"] = "state-update";
+  A2AKnownEventTypes2["Error"] = "error";
+  return A2AKnownEventTypes2;
+})(A2AKnownEventTypes || {});
+export {
+  A2AKnownEventTypes
+};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/types.ts"],"sourcesContent":["import { MessageStatus } from \"@assistant-ui/react\";\nimport { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\n// A2A Message Types\nexport type A2AMessage = {\n  id?: string;\n  role: \"user\" | \"assistant\" | \"system\" | \"tool\";\n  content: string | A2AMessageContent[];\n  tool_calls?: A2AToolCall[];\n  tool_call_id?: string;\n  artifacts?: A2AArtifact[];\n  status?: MessageStatus;\n};\n\nexport type A2AMessageContent =\n  | { type: \"text\"; text: string }\n  | { type: \"image_url\"; image_url: string | { url: string } }\n  | { type: \"data\"; data: any };\n\nexport type A2AToolCall = {\n  id: string;\n  name: string;\n  args: ReadonlyJSONObject;\n  argsText?: string;\n};\n\nexport type A2AArtifact = {\n  name: string;\n  parts: A2AArtifactPart[];\n};\n\nexport type A2AArtifactPart = {\n  kind: \"text\" | \"data\" | \"file\";\n  data?: any;\n  text?: string;\n  metadata?: Record<string, any>;\n};\n\n// A2A Events (similar to LangGraph events)\nexport type A2AEvent = {\n  event: A2AKnownEventTypes | string;\n  data: any;\n};\n\nexport enum A2AKnownEventTypes {\n  TaskUpdate = \"task-update\",\n  TaskComplete = \"task-complete\",\n  TaskFailed = \"task-failed\",\n  Artifacts = \"artifacts\",\n  StateUpdate = \"state-update\",\n  Error = \"error\",\n}\n\n// A2A Task State\nexport type A2ATaskState = {\n  id: string;\n  state: \"pending\" | \"working\" | \"completed\" | \"failed\";\n  progress?: number;\n  message?: string;\n};\n\n// A2A Task Result\nexport type A2ATaskResult = {\n  id: string;\n  status: {\n    state: \"pending\" | \"working\" | \"completed\" | \"failed\";\n    message?: string;\n  };\n  artifacts?: A2AArtifact[];\n  history?: Array<{\n    messageId: string;\n    role: string;\n    parts?: Array<{ kind: string; text?: string; data?: any }>;\n  }>;\n};\n\n// A2A Configuration\nexport type A2AConfig = {\n  contextId?: string;\n  runConfig?: Record<string, any>;\n};\n\n// A2A Send Message Configuration\nexport type A2ASendMessageConfig = A2AConfig & {\n  command?: A2ACommand;\n};\n\n// A2A Commands (for interrupts/resume)\nexport type A2ACommand = {\n  resume?: string;\n  interrupt?: string;\n};\n\n// A2A Stream Callback\nexport type A2AStreamCallback<TMessage> = (\n  messages: TMessage[],\n  config: A2ASendMessageConfig & { abortSignal: AbortSignal },\n) => Promise<AsyncGenerator<A2AEvent>> | AsyncGenerator<A2AEvent>;\n\n// Event handler callback types\nexport type OnTaskUpdateEventCallback = (\n  data: A2ATaskState,\n) => void | Promise<void>;\nexport type OnArtifactsEventCallback = (\n  artifacts: A2AArtifact[],\n) => void | Promise<void>;\nexport type OnErrorEventCallback = (error: unknown) => void | Promise<void>;\nexport type OnStateUpdateEventCallback = (\n  state: unknown,\n) => void | Promise<void>;\nexport type OnCustomEventCallback = (\n  type: string,\n  data: unknown,\n) => void | Promise<void>;\n"],"mappings":";AA4CO,IAAK,qBAAL,kBAAKA,wBAAL;AACL,EAAAA,oBAAA,gBAAa;AACb,EAAAA,oBAAA,kBAAe;AACf,EAAAA,oBAAA,gBAAa;AACb,EAAAA,oBAAA,eAAY;AACZ,EAAAA,oBAAA,iBAAc;AACd,EAAAA,oBAAA,WAAQ;AANE,SAAAA;AAAA,GAAA;","names":["A2AKnownEventTypes"]}

package/dist/useA2AMessages.d.ts

@@ -0,0 +1,25 @@
+import { A2AEvent, A2ATaskState, A2AArtifact, A2ASendMessageConfig, A2AStreamCallback, OnTaskUpdateEventCallback, OnArtifactsEventCallback, OnErrorEventCallback, OnStateUpdateEventCallback, OnCustomEventCallback } from "./types";
+export type A2AMessagesEvent<_TMessage> = A2AEvent;
+export declare const useA2AMessages: <TMessage extends {
+    id?: string;
+}>({ stream, appendMessage, eventHandlers, }: {
+    stream: A2AStreamCallback<TMessage>;
+    appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;
+    eventHandlers?: {
+        onTaskUpdate?: OnTaskUpdateEventCallback;
+        onArtifacts?: OnArtifactsEventCallback;
+        onError?: OnErrorEventCallback;
+        onStateUpdate?: OnStateUpdateEventCallback;
+        onCustomEvent?: OnCustomEventCallback;
+    };
+}) => {
+    messages: TMessage[];
+    artifacts: A2AArtifact[];
+    taskState: A2ATaskState | undefined;
+    sendMessage: (newMessages: TMessage[], config: A2ASendMessageConfig) => Promise<void>;
+    cancel: () => void;
+    setMessages: import("react").Dispatch<import("react").SetStateAction<TMessage[]>>;
+    setArtifacts: import("react").Dispatch<import("react").SetStateAction<A2AArtifact[]>>;
+    setTaskState: import("react").Dispatch<import("react").SetStateAction<A2ATaskState | undefined>>;
+};
+//# sourceMappingURL=useA2AMessages.d.ts.map

package/dist/useA2AMessages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useA2AMessages.d.ts","sourceRoot":"","sources":["../src/useA2AMessages.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,QAAQ,EAER,YAAY,EACZ,WAAW,EACX,oBAAoB,EACpB,iBAAiB,EACjB,yBAAyB,EACzB,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,EAC1B,qBAAqB,EACtB,MAAM,SAAS,CAAC;AAEjB,MAAM,MAAM,gBAAgB,CAAC,SAAS,IAAI,QAAQ,CAAC;AAOnD,eAAO,MAAM,cAAc,GAAI,QAAQ,SAAS;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE,EAAE,2CAI9D;IACD,MAAM,EAAE,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IACpC,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,IAAI,EAAE,QAAQ,KAAK,QAAQ,CAAC;IACzE,aAAa,CAAC,EAAE;QACd,YAAY,CAAC,EAAE,yBAAyB,CAAC;QACzC,WAAW,CAAC,EAAE,wBAAwB,CAAC;QACvC,OAAO,CAAC,EAAE,oBAAoB,CAAC;QAC/B,aAAa,CAAC,EAAE,0BAA0B,CAAC;QAC3C,aAAa,CAAC,EAAE,qBAAqB,CAAC;KACvC,CAAC;CACH;;;;+BAUuB,QAAQ,EAAE,UAAU,oBAAoB;;;;;CAmI/D,CAAC"}