@assistant-ui/react-a2a 0.1.2 → 0.1.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useA2ARuntime.d.ts","sourceRoot":"","sources":["../src/useA2ARuntime.ts"],"names":[],"mappings":"AACA,OAAO,EACL,UAAU,EAEV,WAAW,EACX,YAAY,EACZ,oBAAoB,EACpB,iBAAiB,EACjB,yBAAyB,EACzB,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,EAC1B,qBAAqB,EACtB,MAAM,SAAS,CAAC;AASjB,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAGxD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AA4E7D,eAAO,MAAM,eAAe,gCAG3B,CAAC;AAEF,eAAO,MAAM,eAAe,qBAG3B,CAAC;AAEF,eAAO,MAAM,UAAU,mBA5BJ,UAAU,EAAE,UAAU,oBAAoB,KAAK,OAAO,CAAC,IAAI,CA+B7E,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,mLAS3B;IACD;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,MAAM,EAAE,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACtC;;OAEG;IACH,mBAAmB,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACjD,gBAAgB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;QAChD,QAAQ,EAAE,UAAU,EAAE,CAAC;QACvB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;KAC3B,CAAC,CAAC;IACH,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,CAAC;QAChC,MAAM,CAAC,EAAE,sBAAsB,CAAC;QAChC,QAAQ,CAAC,EAAE,eAAe,CAAC;KAC5B,GACD,SAAS,CAAC;IACd;;OAEG;IACH,aAAa,CAAC,EACV;QACE;;WAEG;QACH,YAAY,CAAC,EAAE,yBAAyB,CAAC;QACzC;;WAEG;QACH,WAAW,CAAC,EAAE,wBAAwB,CAAC;QACvC;;WAEG;QACH,OAAO,CAAC,EAAE,oBAAoB,CAAC;QAC/B;;WAEG;QACH,aAAa,CAAC,EAAE,0BAA0B,CAAC;QAC3C;;WAEG;QACH,aAAa,CAAC,EAAE,qBAAqB,CAAC;KACvC,GACD,SAAS,CAAC;CACf,
|
|
1
|
+
{"version":3,"file":"useA2ARuntime.d.ts","sourceRoot":"","sources":["../src/useA2ARuntime.ts"],"names":[],"mappings":"AACA,OAAO,EACL,UAAU,EAEV,WAAW,EACX,YAAY,EACZ,oBAAoB,EACpB,iBAAiB,EACjB,yBAAyB,EACzB,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,EAC1B,qBAAqB,EACtB,MAAM,SAAS,CAAC;AASjB,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAGxD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AA4E7D,eAAO,MAAM,eAAe,gCAG3B,CAAC;AAEF,eAAO,MAAM,eAAe,qBAG3B,CAAC;AAEF,eAAO,MAAM,UAAU,mBA5BJ,UAAU,EAAE,UAAU,oBAAoB,KAAK,OAAO,CAAC,IAAI,CA+B7E,CAAC;AAEF,eAAO,MAAM,aAAa,GAAI,mLAS3B;IACD;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD,MAAM,EAAE,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACtC;;OAEG;IACH,mBAAmB,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACjD,gBAAgB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;QAChD,QAAQ,EAAE,UAAU,EAAE,CAAC;QACvB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;KAC3B,CAAC,CAAC;IACH,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,CAAC;QAChC,MAAM,CAAC,EAAE,sBAAsB,CAAC;QAChC,QAAQ,CAAC,EAAE,eAAe,CAAC;KAC5B,GACD,SAAS,CAAC;IACd;;OAEG;IACH,aAAa,CAAC,EACV;QACE;;WAEG;QACH,YAAY,CAAC,EAAE,yBAAyB,CAAC;QACzC;;WAEG;QACH,WAAW,CAAC,EAAE,wBAAwB,CAAC;QACvC;;WAEG;QACH,OAAO,CAAC,EAAE,oBAAoB,CAAC;QAC/B;;WAEG;QACH,aAAa,CAAC,EAAE,0BAA0B,CAAC;QAC3C;;WAEG;QACH,aAAa,CAAC,EAAE,qBAAqB,CAAC;KACvC,GACD,SAAS,CAAC;CACf,mDAyJA,CAAC"}
|
package/dist/useA2ARuntime.js
CHANGED
|
@@ -49,7 +49,7 @@ var getMessageContent = (msg) => {
|
|
|
49
49
|
}
|
|
50
50
|
return content;
|
|
51
51
|
};
|
|
52
|
-
var symbolA2ARuntimeExtras = Symbol("a2a-runtime-extras");
|
|
52
|
+
var symbolA2ARuntimeExtras = /* @__PURE__ */ Symbol("a2a-runtime-extras");
|
|
53
53
|
var asA2ARuntimeExtras = (extras) => {
|
|
54
54
|
if (typeof extras !== "object" || extras == null || !(symbolA2ARuntimeExtras in extras))
|
|
55
55
|
throw new Error(
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\nimport {\n A2AMessage,\n A2AToolCall,\n A2AArtifact,\n A2ATaskState,\n A2ASendMessageConfig,\n A2AStreamCallback,\n OnTaskUpdateEventCallback,\n OnArtifactsEventCallback,\n OnErrorEventCallback,\n OnStateUpdateEventCallback,\n OnCustomEventCallback,\n} from \"./types\";\nimport {\n useExternalMessageConverter,\n useExternalStoreRuntime,\n useThread,\n useThreadListItemRuntime,\n} from \"@assistant-ui/react\";\nimport { convertA2AMessage } from \"./convertA2AMessages\";\nimport { useA2AMessages } from \"./useA2AMessages\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\nimport { FeedbackAdapter } from \"@assistant-ui/react\";\nimport { SpeechSynthesisAdapter } from \"@assistant-ui/react\";\nimport { appendA2AChunk } from \"./appendA2AChunk\";\n\nconst getPendingToolCalls = (messages: A2AMessage[]) => {\n const pendingToolCalls = new Map<string, A2AToolCall>();\n for (const message of messages) {\n if (message.role === \"assistant\") {\n for (const toolCall of message.tool_calls ?? []) {\n pendingToolCalls.set(toolCall.id, toolCall);\n }\n }\n if (message.role === \"tool\") {\n pendingToolCalls.delete(message.tool_call_id!);\n }\n }\n\n return [...pendingToolCalls.values()];\n};\n\nconst getMessageContent = (msg: AppendMessage) => {\n const allContent = [\n ...msg.content,\n ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n ];\n const content = allContent.map((part) => {\n const type = part.type;\n switch (type) {\n case \"text\":\n return { type: \"text\" as const, text: part.text };\n case \"image\":\n return { type: \"image_url\" as const, image_url: { url: part.image } };\n\n case \"tool-call\":\n throw new Error(\"Tool call appends are not supported.\");\n\n default:\n const _exhaustiveCheck:\n | \"reasoning\"\n | \"source\"\n | \"file\"\n | \"audio\"\n | \"data\" = type;\n throw new Error(\n `Unsupported append message part type: ${_exhaustiveCheck}`,\n );\n }\n });\n\n if (content.length === 1 && content[0]?.type === \"text\") {\n return content[0].text ?? \"\";\n }\n\n return content;\n};\n\nconst symbolA2ARuntimeExtras = Symbol(\"a2a-runtime-extras\");\ntype A2ARuntimeExtras = {\n [symbolA2ARuntimeExtras]: true;\n send: (messages: A2AMessage[], config: A2ASendMessageConfig) => Promise<void>;\n taskState: A2ATaskState | undefined;\n artifacts: A2AArtifact[];\n};\n\nconst asA2ARuntimeExtras = (extras: unknown): A2ARuntimeExtras => {\n if (\n typeof extras !== \"object\" ||\n extras == null ||\n !(symbolA2ARuntimeExtras in extras)\n )\n throw new Error(\n \"This method can only be called when you are using useA2ARuntime\",\n );\n\n return extras as A2ARuntimeExtras;\n};\n\nexport const useA2ATaskState = () => {\n const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));\n return taskState;\n};\n\nexport const useA2AArtifacts = () => {\n const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));\n return artifacts;\n};\n\nexport const useA2ASend = () => {\n const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));\n return send;\n};\n\nexport const useA2ARuntime = ({\n autoCancelPendingToolCalls,\n adapters: { attachments, feedback, speech } = {},\n unstable_allowCancellation,\n stream,\n contextId,\n onSwitchToNewThread,\n onSwitchToThread,\n eventHandlers,\n}: {\n /**\n * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n */\n contextId?: string | undefined;\n autoCancelPendingToolCalls?: boolean | undefined;\n unstable_allowCancellation?: boolean | undefined;\n stream: A2AStreamCallback<A2AMessage>;\n /**\n * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n */\n onSwitchToNewThread?: () => Promise<void> | void;\n onSwitchToThread?: (contextId: string) => Promise<{\n messages: A2AMessage[];\n artifacts?: A2AArtifact[];\n }>;\n adapters?:\n | {\n attachments?: AttachmentAdapter;\n speech?: SpeechSynthesisAdapter;\n feedback?: FeedbackAdapter;\n }\n | undefined;\n /**\n * Event handlers for various A2A stream events\n */\n eventHandlers?:\n | {\n /**\n * Called when task updates are received from the A2A stream\n */\n onTaskUpdate?: OnTaskUpdateEventCallback;\n /**\n * Called when artifacts are received from the A2A stream\n */\n onArtifacts?: OnArtifactsEventCallback;\n /**\n * Called when errors occur during A2A stream processing\n */\n onError?: OnErrorEventCallback;\n /**\n * Called when state updates are received from the A2A stream\n */\n onStateUpdate?: OnStateUpdateEventCallback;\n /**\n * Called when custom events are received from the A2A stream\n */\n onCustomEvent?: OnCustomEventCallback;\n }\n | undefined;\n}) => {\n const {\n taskState,\n artifacts,\n setArtifacts,\n messages,\n sendMessage,\n cancel,\n setMessages,\n } = useA2AMessages({\n appendMessage: appendA2AChunk,\n stream,\n ...(eventHandlers && { eventHandlers }),\n });\n\n const [isRunning, setIsRunning] = useState(false);\n const handleSendMessage = async (\n messages: A2AMessage[],\n config: A2ASendMessageConfig,\n ) => {\n try {\n setIsRunning(true);\n await sendMessage(messages, config);\n } catch (error) {\n console.error(\"Error streaming A2A messages:\", error);\n } finally {\n setIsRunning(false);\n }\n };\n\n const threadMessages = useExternalMessageConverter({\n callback: convertA2AMessage,\n messages,\n isRunning,\n });\n\n const switchToThread = !onSwitchToThread\n ? undefined\n : async (externalId: string) => {\n const { messages, artifacts } = await onSwitchToThread(externalId);\n setMessages(messages);\n if (artifacts) {\n setArtifacts(artifacts);\n }\n };\n\n const threadList: NonNullable<\n ExternalStoreAdapter[\"adapters\"]\n >[\"threadList\"] = {\n threadId: contextId,\n onSwitchToNewThread: !onSwitchToNewThread\n ? undefined\n : async () => {\n await onSwitchToNewThread();\n setMessages([]);\n setArtifacts([]);\n },\n onSwitchToThread: switchToThread,\n };\n\n const loadingRef = useRef(false);\n const threadListItemRuntime = useThreadListItemRuntime({ optional: true });\n useEffect(() => {\n if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;\n\n const externalId = threadListItemRuntime.getState().externalId;\n if (externalId) {\n loadingRef.current = true;\n switchToThread(externalId).finally(() => {\n loadingRef.current = false;\n });\n }\n // eslint-disable-next-line react-hooks/exhaustive-deps\n }, []);\n\n return useExternalStoreRuntime({\n isRunning,\n messages: threadMessages,\n adapters: {\n attachments,\n feedback,\n speech,\n threadList,\n },\n extras: {\n [symbolA2ARuntimeExtras]: true,\n taskState,\n artifacts,\n send: handleSendMessage,\n } satisfies A2ARuntimeExtras,\n onNew: (msg) => {\n const cancellations =\n autoCancelPendingToolCalls !== false\n ? getPendingToolCalls(messages).map(\n (t) =>\n ({\n role: \"tool\",\n tool_call_id: t.id,\n content: JSON.stringify({ cancelled: true }),\n status: {\n type: \"incomplete\",\n reason: \"cancelled\",\n },\n }) satisfies A2AMessage & { role: \"tool\" },\n )\n : [];\n\n const config: A2ASendMessageConfig = {};\n if (contextId !== undefined) config.contextId = contextId;\n if (msg.runConfig !== undefined) config.runConfig = msg.runConfig;\n return handleSendMessage(\n [\n ...cancellations,\n {\n role: \"user\",\n content: getMessageContent(msg),\n },\n ],\n config,\n );\n },\n onAddToolResult: async ({\n toolCallId,\n toolName: _toolName,\n result,\n isError,\n artifact,\n }) => {\n // TODO parallel human in the loop calls\n const message: A2AMessage = {\n role: \"tool\",\n tool_call_id: toolCallId,\n content: JSON.stringify(result),\n status: isError\n ? { type: \"incomplete\", reason: \"error\" }\n : { type: \"complete\", reason: \"stop\" },\n };\n if (artifact) {\n message.artifacts = [artifact] as A2AArtifact[];\n }\n const config: A2ASendMessageConfig = {};\n if (contextId !== undefined) config.contextId = contextId;\n await handleSendMessage(\n [message],\n // TODO reuse runconfig here!\n config,\n );\n },\n onCancel: unstable_allowCancellation\n ? async () => {\n cancel();\n }\n : undefined,\n });\n};\n"],"mappings":";AAAA,SAAS,WAAW,QAAQ,gBAAgB;AAc5C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,yBAAyB;AAClC,SAAS,sBAAsB;AAM/B,SAAS,sBAAsB;AAE/B,IAAM,sBAAsB,CAAC,aAA2B;AACtD,QAAM,mBAAmB,oBAAI,IAAyB;AACtD,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,aAAa;AAChC,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAa;AAAA,IAC/C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAI,IAAI,aAAa,QAAQ,CAAC,MAAM,EAAE,OAAO,KAAK,CAAC;AAAA,EACrD;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MAExD;AACE,cAAM,mBAKO;AACb,cAAM,IAAI;AAAA,UACR,yCAAyC,gBAAgB;AAAA,QAC3D;AAAA,IACJ;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEA,IAAM,yBAAyB,OAAO,oBAAoB;AAQ1D,IAAM,qBAAqB,CAAC,WAAsC;AAChE,MACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,0BAA0B;AAE5B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO;AACT;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,EAAE,UAAU,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AACnE,SAAO;AACT;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,EAAE,UAAU,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AACnE,SAAO;AACT;AAEO,IAAM,aAAa,MAAM;AAC9B,QAAM,EAAE,KAAK,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AAC9D,SAAO;AACT;AAEO,IAAM,gBAAgB,CAAC;AAAA,EAC5B;AAAA,EACA,UAAU,EAAE,aAAa,UAAU,OAAO,IAAI,CAAC;AAAA,EAC/C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAkDM;AACJ,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI,eAAe;AAAA,IACjB,eAAe;AAAA,IACf;AAAA,IACA,GAAI,iBAAiB,EAAE,cAAc;AAAA,EACvC,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,KAAK;AAChD,QAAM,oBAAoB,OACxBA,WACA,WACG;AACH,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,WAAU,MAAM;AAAA,IACpC,SAAS,OAAO;AACd,cAAQ,MAAM,iCAAiC,KAAK;AAAA,IACtD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,iBAAiB,4BAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,QAAM,iBAAiB,CAAC,mBACpB,SACA,OAAO,eAAuB;AAC5B,UAAM,EAAE,UAAAA,WAAU,WAAAC,WAAU,IAAI,MAAM,iBAAiB,UAAU;AACjE,gBAAYD,SAAQ;AACpB,QAAIC,YAAW;AACb,mBAAaA,UAAS;AAAA,IACxB;AAAA,EACF;AAEJ,QAAM,aAEY;AAAA,IAChB,UAAU;AAAA,IACV,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AACd,mBAAa,CAAC,CAAC;AAAA,IACjB;AAAA,IACJ,kBAAkB;AAAA,EACpB;AAEA,QAAM,aAAa,OAAO,KAAK;AAC/B,QAAM,wBAAwB,yBAAyB,EAAE,UAAU,KAAK,CAAC;AACzE,YAAU,MAAM;AACd,QAAI,CAAC,yBAAyB,CAAC,kBAAkB,WAAW,QAAS;AAErE,UAAM,aAAa,sBAAsB,SAAS,EAAE;AACpD,QAAI,YAAY;AACd,iBAAW,UAAU;AACrB,qBAAe,UAAU,EAAE,QAAQ,MAAM;AACvC,mBAAW,UAAU;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EAEF,GAAG,CAAC,CAAC;AAEL,SAAO,wBAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,QAAQ;AAAA,MACN,CAAC,sBAAsB,GAAG;AAAA,MAC1B;AAAA,MACA;AAAA,MACA,MAAM;AAAA,IACR;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,UAC3C,QAAQ;AAAA,YACN,MAAM;AAAA,YACN,QAAQ;AAAA,UACV;AAAA,QACF;AAAA,MACJ,IACA,CAAC;AAEP,YAAM,SAA+B,CAAC;AACtC,UAAI,cAAc,OAAW,QAAO,YAAY;AAChD,UAAI,IAAI,cAAc,OAAW,QAAO,YAAY,IAAI;AACxD,aAAO;AAAA,QACL;AAAA,UACE,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS,kBAAkB,GAAG;AAAA,UAChC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,IACA,iBAAiB,OAAO;AAAA,MACtB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA;AAAA,MACA;AAAA,IACF,MAAM;AAEJ,YAAM,UAAsB;AAAA,QAC1B,MAAM;AAAA,QACN,cAAc;AAAA,QACd,SAAS,KAAK,UAAU,MAAM;AAAA,QAC9B,QAAQ,UACJ,EAAE,MAAM,cAAc,QAAQ,QAAQ,IACtC,EAAE,MAAM,YAAY,QAAQ,OAAO;AAAA,MACzC;AACA,UAAI,UAAU;AACZ,gBAAQ,YAAY,CAAC,QAAQ;AAAA,MAC/B;AACA,YAAM,SAA+B,CAAC;AACtC,UAAI,cAAc,OAAW,QAAO,YAAY;AAChD,YAAM;AAAA,QACJ,CAAC,OAAO;AAAA;AAAA,QAER;AAAA,MACF;AAAA,IACF;AAAA,IACA,UAAU,6BACN,YAAY;AACV,aAAO;AAAA,IACT,IACA;AAAA,EACN,CAAC;AACH;","names":["messages","artifacts"]}
|
|
1
|
+
{"version":3,"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\nimport {\n A2AMessage,\n A2AToolCall,\n A2AArtifact,\n A2ATaskState,\n A2ASendMessageConfig,\n A2AStreamCallback,\n OnTaskUpdateEventCallback,\n OnArtifactsEventCallback,\n OnErrorEventCallback,\n OnStateUpdateEventCallback,\n OnCustomEventCallback,\n} from \"./types\";\nimport {\n useExternalMessageConverter,\n useExternalStoreRuntime,\n useThread,\n useThreadListItemRuntime,\n} from \"@assistant-ui/react\";\nimport { convertA2AMessage } from \"./convertA2AMessages\";\nimport { useA2AMessages } from \"./useA2AMessages\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\nimport { FeedbackAdapter } from \"@assistant-ui/react\";\nimport { SpeechSynthesisAdapter } from \"@assistant-ui/react\";\nimport { appendA2AChunk } from \"./appendA2AChunk\";\n\nconst getPendingToolCalls = (messages: A2AMessage[]) => {\n const pendingToolCalls = new Map<string, A2AToolCall>();\n for (const message of messages) {\n if (message.role === \"assistant\") {\n for (const toolCall of message.tool_calls ?? []) {\n pendingToolCalls.set(toolCall.id, toolCall);\n }\n }\n if (message.role === \"tool\") {\n pendingToolCalls.delete(message.tool_call_id!);\n }\n }\n\n return [...pendingToolCalls.values()];\n};\n\nconst getMessageContent = (msg: AppendMessage) => {\n const allContent = [\n ...msg.content,\n ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n ];\n const content = allContent.map((part) => {\n const type = part.type;\n switch (type) {\n case \"text\":\n return { type: \"text\" as const, text: part.text };\n case \"image\":\n return { type: \"image_url\" as const, image_url: { url: part.image } };\n\n case \"tool-call\":\n throw new Error(\"Tool call appends are not supported.\");\n\n default:\n const _exhaustiveCheck:\n | \"reasoning\"\n | \"source\"\n | \"file\"\n | \"audio\"\n | \"data\" = type;\n throw new Error(\n `Unsupported append message part type: ${_exhaustiveCheck}`,\n );\n }\n });\n\n if (content.length === 1 && content[0]?.type === \"text\") {\n return content[0].text ?? \"\";\n }\n\n return content;\n};\n\nconst symbolA2ARuntimeExtras = Symbol(\"a2a-runtime-extras\");\ntype A2ARuntimeExtras = {\n [symbolA2ARuntimeExtras]: true;\n send: (messages: A2AMessage[], config: A2ASendMessageConfig) => Promise<void>;\n taskState: A2ATaskState | undefined;\n artifacts: A2AArtifact[];\n};\n\nconst asA2ARuntimeExtras = (extras: unknown): A2ARuntimeExtras => {\n if (\n typeof extras !== \"object\" ||\n extras == null ||\n !(symbolA2ARuntimeExtras in extras)\n )\n throw new Error(\n \"This method can only be called when you are using useA2ARuntime\",\n );\n\n return extras as A2ARuntimeExtras;\n};\n\nexport const useA2ATaskState = () => {\n const { taskState } = useThread((t) => asA2ARuntimeExtras(t.extras));\n return taskState;\n};\n\nexport const useA2AArtifacts = () => {\n const { artifacts } = useThread((t) => asA2ARuntimeExtras(t.extras));\n return artifacts;\n};\n\nexport const useA2ASend = () => {\n const { send } = useThread((t) => asA2ARuntimeExtras(t.extras));\n return send;\n};\n\nexport const useA2ARuntime = ({\n autoCancelPendingToolCalls,\n adapters: { attachments, feedback, speech } = {},\n unstable_allowCancellation,\n stream,\n contextId,\n onSwitchToNewThread,\n onSwitchToThread,\n eventHandlers,\n}: {\n /**\n * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n */\n contextId?: string | undefined;\n autoCancelPendingToolCalls?: boolean | undefined;\n unstable_allowCancellation?: boolean | undefined;\n stream: A2AStreamCallback<A2AMessage>;\n /**\n * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n */\n onSwitchToNewThread?: () => Promise<void> | void;\n onSwitchToThread?: (contextId: string) => Promise<{\n messages: A2AMessage[];\n artifacts?: A2AArtifact[];\n }>;\n adapters?:\n | {\n attachments?: AttachmentAdapter;\n speech?: SpeechSynthesisAdapter;\n feedback?: FeedbackAdapter;\n }\n | undefined;\n /**\n * Event handlers for various A2A stream events\n */\n eventHandlers?:\n | {\n /**\n * Called when task updates are received from the A2A stream\n */\n onTaskUpdate?: OnTaskUpdateEventCallback;\n /**\n * Called when artifacts are received from the A2A stream\n */\n onArtifacts?: OnArtifactsEventCallback;\n /**\n * Called when errors occur during A2A stream processing\n */\n onError?: OnErrorEventCallback;\n /**\n * Called when state updates are received from the A2A stream\n */\n onStateUpdate?: OnStateUpdateEventCallback;\n /**\n * Called when custom events are received from the A2A stream\n */\n onCustomEvent?: OnCustomEventCallback;\n }\n | undefined;\n}) => {\n const {\n taskState,\n artifacts,\n setArtifacts,\n messages,\n sendMessage,\n cancel,\n setMessages,\n } = useA2AMessages({\n appendMessage: appendA2AChunk,\n stream,\n ...(eventHandlers && { eventHandlers }),\n });\n\n const [isRunning, setIsRunning] = useState(false);\n const handleSendMessage = async (\n messages: A2AMessage[],\n config: A2ASendMessageConfig,\n ) => {\n try {\n setIsRunning(true);\n await sendMessage(messages, config);\n } catch (error) {\n console.error(\"Error streaming A2A messages:\", error);\n } finally {\n setIsRunning(false);\n }\n };\n\n const threadMessages = useExternalMessageConverter({\n callback: convertA2AMessage,\n messages,\n isRunning,\n });\n\n const switchToThread = !onSwitchToThread\n ? undefined\n : async (externalId: string) => {\n const { messages, artifacts } = await onSwitchToThread(externalId);\n setMessages(messages);\n if (artifacts) {\n setArtifacts(artifacts);\n }\n };\n\n const threadList: NonNullable<\n ExternalStoreAdapter[\"adapters\"]\n >[\"threadList\"] = {\n threadId: contextId,\n onSwitchToNewThread: !onSwitchToNewThread\n ? undefined\n : async () => {\n await onSwitchToNewThread();\n setMessages([]);\n setArtifacts([]);\n },\n onSwitchToThread: switchToThread,\n };\n\n const loadingRef = useRef(false);\n const threadListItemRuntime = useThreadListItemRuntime({ optional: true });\n useEffect(() => {\n if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;\n\n const externalId = threadListItemRuntime.getState().externalId;\n if (externalId) {\n loadingRef.current = true;\n switchToThread(externalId).finally(() => {\n loadingRef.current = false;\n });\n }\n }, []);\n\n return useExternalStoreRuntime({\n isRunning,\n messages: threadMessages,\n adapters: {\n attachments,\n feedback,\n speech,\n threadList,\n },\n extras: {\n [symbolA2ARuntimeExtras]: true,\n taskState,\n artifacts,\n send: handleSendMessage,\n } satisfies A2ARuntimeExtras,\n onNew: (msg) => {\n const cancellations =\n autoCancelPendingToolCalls !== false\n ? getPendingToolCalls(messages).map(\n (t) =>\n ({\n role: \"tool\",\n tool_call_id: t.id,\n content: JSON.stringify({ cancelled: true }),\n status: {\n type: \"incomplete\",\n reason: \"cancelled\",\n },\n }) satisfies A2AMessage & { role: \"tool\" },\n )\n : [];\n\n const config: A2ASendMessageConfig = {};\n if (contextId !== undefined) config.contextId = contextId;\n if (msg.runConfig !== undefined) config.runConfig = msg.runConfig;\n return handleSendMessage(\n [\n ...cancellations,\n {\n role: \"user\",\n content: getMessageContent(msg),\n },\n ],\n config,\n );\n },\n onAddToolResult: async ({\n toolCallId,\n toolName: _toolName,\n result,\n isError,\n artifact,\n }) => {\n // TODO parallel human in the loop calls\n const message: A2AMessage = {\n role: \"tool\",\n tool_call_id: toolCallId,\n content: JSON.stringify(result),\n status: isError\n ? { type: \"incomplete\", reason: \"error\" }\n : { type: \"complete\", reason: \"stop\" },\n };\n if (artifact) {\n message.artifacts = [artifact] as A2AArtifact[];\n }\n const config: A2ASendMessageConfig = {};\n if (contextId !== undefined) config.contextId = contextId;\n await handleSendMessage(\n [message],\n // TODO reuse runconfig here!\n config,\n );\n },\n onCancel: unstable_allowCancellation\n ? async () => {\n cancel();\n }\n : undefined,\n });\n};\n"],"mappings":";AAAA,SAAS,WAAW,QAAQ,gBAAgB;AAc5C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,yBAAyB;AAClC,SAAS,sBAAsB;AAM/B,SAAS,sBAAsB;AAE/B,IAAM,sBAAsB,CAAC,aAA2B;AACtD,QAAM,mBAAmB,oBAAI,IAAyB;AACtD,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,aAAa;AAChC,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAa;AAAA,IAC/C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAI,IAAI,aAAa,QAAQ,CAAC,MAAM,EAAE,OAAO,KAAK,CAAC;AAAA,EACrD;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MAExD;AACE,cAAM,mBAKO;AACb,cAAM,IAAI;AAAA,UACR,yCAAyC,gBAAgB;AAAA,QAC3D;AAAA,IACJ;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEA,IAAM,yBAAyB,uBAAO,oBAAoB;AAQ1D,IAAM,qBAAqB,CAAC,WAAsC;AAChE,MACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,0BAA0B;AAE5B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO;AACT;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,EAAE,UAAU,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AACnE,SAAO;AACT;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,EAAE,UAAU,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AACnE,SAAO;AACT;AAEO,IAAM,aAAa,MAAM;AAC9B,QAAM,EAAE,KAAK,IAAI,UAAU,CAAC,MAAM,mBAAmB,EAAE,MAAM,CAAC;AAC9D,SAAO;AACT;AAEO,IAAM,gBAAgB,CAAC;AAAA,EAC5B;AAAA,EACA,UAAU,EAAE,aAAa,UAAU,OAAO,IAAI,CAAC;AAAA,EAC/C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAkDM;AACJ,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI,eAAe;AAAA,IACjB,eAAe;AAAA,IACf;AAAA,IACA,GAAI,iBAAiB,EAAE,cAAc;AAAA,EACvC,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS,KAAK;AAChD,QAAM,oBAAoB,OACxBA,WACA,WACG;AACH,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,WAAU,MAAM;AAAA,IACpC,SAAS,OAAO;AACd,cAAQ,MAAM,iCAAiC,KAAK;AAAA,IACtD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,iBAAiB,4BAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,QAAM,iBAAiB,CAAC,mBACpB,SACA,OAAO,eAAuB;AAC5B,UAAM,EAAE,UAAAA,WAAU,WAAAC,WAAU,IAAI,MAAM,iBAAiB,UAAU;AACjE,gBAAYD,SAAQ;AACpB,QAAIC,YAAW;AACb,mBAAaA,UAAS;AAAA,IACxB;AAAA,EACF;AAEJ,QAAM,aAEY;AAAA,IAChB,UAAU;AAAA,IACV,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AACd,mBAAa,CAAC,CAAC;AAAA,IACjB;AAAA,IACJ,kBAAkB;AAAA,EACpB;AAEA,QAAM,aAAa,OAAO,KAAK;AAC/B,QAAM,wBAAwB,yBAAyB,EAAE,UAAU,KAAK,CAAC;AACzE,YAAU,MAAM;AACd,QAAI,CAAC,yBAAyB,CAAC,kBAAkB,WAAW,QAAS;AAErE,UAAM,aAAa,sBAAsB,SAAS,EAAE;AACpD,QAAI,YAAY;AACd,iBAAW,UAAU;AACrB,qBAAe,UAAU,EAAE,QAAQ,MAAM;AACvC,mBAAW,UAAU;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,SAAO,wBAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,QAAQ;AAAA,MACN,CAAC,sBAAsB,GAAG;AAAA,MAC1B;AAAA,MACA;AAAA,MACA,MAAM;AAAA,IACR;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,UAC3C,QAAQ;AAAA,YACN,MAAM;AAAA,YACN,QAAQ;AAAA,UACV;AAAA,QACF;AAAA,MACJ,IACA,CAAC;AAEP,YAAM,SAA+B,CAAC;AACtC,UAAI,cAAc,OAAW,QAAO,YAAY;AAChD,UAAI,IAAI,cAAc,OAAW,QAAO,YAAY,IAAI;AACxD,aAAO;AAAA,QACL;AAAA,UACE,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS,kBAAkB,GAAG;AAAA,UAChC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,IACA,iBAAiB,OAAO;AAAA,MACtB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA;AAAA,MACA;AAAA,IACF,MAAM;AAEJ,YAAM,UAAsB;AAAA,QAC1B,MAAM;AAAA,QACN,cAAc;AAAA,QACd,SAAS,KAAK,UAAU,MAAM;AAAA,QAC9B,QAAQ,UACJ,EAAE,MAAM,cAAc,QAAQ,QAAQ,IACtC,EAAE,MAAM,YAAY,QAAQ,OAAO;AAAA,MACzC;AACA,UAAI,UAAU;AACZ,gBAAQ,YAAY,CAAC,QAAQ;AAAA,MAC/B;AACA,YAAM,SAA+B,CAAC;AACtC,UAAI,cAAc,OAAW,QAAO,YAAY;AAChD,YAAM;AAAA,QACJ,CAAC,OAAO;AAAA;AAAA,QAER;AAAA,MACF;AAAA,IACF;AAAA,IACA,UAAU,6BACN,YAAY;AACV,aAAO;AAAA,IACT,IACA;AAAA,EACN,CAAC;AACH;","names":["messages","artifacts"]}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@assistant-ui/react-a2a",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.3",
|
|
4
4
|
"license": "MIT",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"exports": {
|
|
@@ -18,12 +18,12 @@
|
|
|
18
18
|
],
|
|
19
19
|
"sideEffects": false,
|
|
20
20
|
"dependencies": {
|
|
21
|
-
"assistant-stream": "^0.2.
|
|
21
|
+
"assistant-stream": "^0.2.44",
|
|
22
22
|
"uuid": "^13.0.0",
|
|
23
23
|
"zod": "^4.1.13"
|
|
24
24
|
},
|
|
25
25
|
"peerDependencies": {
|
|
26
|
-
"@assistant-ui/react": "^0.11.
|
|
26
|
+
"@assistant-ui/react": "^0.11.49",
|
|
27
27
|
"@types/react": "*",
|
|
28
28
|
"react": "^18 || ^19 || ^19.0.0-rc"
|
|
29
29
|
},
|
|
@@ -38,13 +38,11 @@
|
|
|
38
38
|
"@types/node": "^24.10.1",
|
|
39
39
|
"@types/react": "^19.2.7",
|
|
40
40
|
"@types/uuid": "^11.0.0",
|
|
41
|
-
"eslint": "^9",
|
|
42
|
-
"eslint-config-next": "16.0.4",
|
|
43
41
|
"jsdom": "^27.2.0",
|
|
44
|
-
"react": "19.2.
|
|
45
|
-
"tsx": "^4.
|
|
46
|
-
"vitest": "^4.0.
|
|
47
|
-
"@assistant-ui/react": "0.11.
|
|
42
|
+
"react": "19.2.1",
|
|
43
|
+
"tsx": "^4.21.0",
|
|
44
|
+
"vitest": "^4.0.15",
|
|
45
|
+
"@assistant-ui/react": "0.11.49",
|
|
48
46
|
"@assistant-ui/x-buildutils": "0.0.1"
|
|
49
47
|
},
|
|
50
48
|
"publishConfig": {
|
|
@@ -61,7 +59,6 @@
|
|
|
61
59
|
},
|
|
62
60
|
"scripts": {
|
|
63
61
|
"build": "tsx scripts/build.mts",
|
|
64
|
-
"lint": "eslint .",
|
|
65
62
|
"test": "vitest run --passWithNoTests",
|
|
66
63
|
"test:watch": "vitest"
|
|
67
64
|
}
|