@assistant-ui/react-ai-sdk 1.1.18 → 1.1.20

package/dist/frontendTools.js

@@ -1,15 +1,9 @@
-// src/frontendTools.ts
 import { jsonSchema } from "ai";
-var frontendTools = (tools) => Object.fromEntries(
-  Object.entries(tools).map(([name, tool]) => [
+export const frontendTools = (tools) => Object.fromEntries(Object.entries(tools).map(([name, tool]) => [
     name,
     {
-      ...tool.description ? { description: tool.description } : void 0,
-      inputSchema: jsonSchema(tool.parameters)
-    }
-  ])
-);
-export {
-  frontendTools
-};
+        ...(tool.description ? { description: tool.description } : undefined),
+        inputSchema: jsonSchema(tool.parameters),
+    },
+]));
 //# sourceMappingURL=frontendTools.js.map

package/dist/frontendTools.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/frontendTools.ts"],"sourcesContent":["import { jsonSchema } from \"ai\";\nimport type { JSONSchema7 } from \"json-schema\";\n\nexport const frontendTools = (\n  tools: Record<string, { description?: string; parameters: JSONSchema7 }>,\n) =>\n  Object.fromEntries(\n    Object.entries(tools).map(([name, tool]) => [\n      name,\n      {\n        ...(tool.description ? { description: tool.description } : undefined),\n        inputSchema: jsonSchema(tool.parameters),\n      },\n    ]),\n  );\n"],"mappings":";AAAA,SAAS,kBAAkB;AAGpB,IAAM,gBAAgB,CAC3B,UAEA,OAAO;AAAA,EACL,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAAA,IAC1C;AAAA,IACA;AAAA,MACE,GAAI,KAAK,cAAc,EAAE,aAAa,KAAK,YAAY,IAAI;AAAA,MAC3D,aAAa,WAAW,KAAK,UAAU;AAAA,IACzC;AAAA,EACF,CAAC;AACH;","names":[]}
+{"version":3,"file":"frontendTools.js","sourceRoot":"","sources":["../src/frontendTools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAGhC,MAAM,CAAC,MAAM,aAAa,GAAG,CAC3B,KAAwE,EACxE,EAAE,CACF,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;IAC1C,IAAI;IACJ;QACE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACrE,WAAW,EAAE,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC;KACzC;CACF,CAAC,CACH,CAAC"}

package/dist/index.d.ts

@@ -1,3 +1,3 @@
-export * from "./ui";
-export { frontendTools } from "./frontendTools";
+export * from "./ui/index.js";
+export { frontendTools } from "./frontendTools.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,MAAM,CAAC;AACrB,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,8BAAqB;AACrB,OAAO,EAAE,aAAa,EAAE,2BAAwB"}

package/dist/index.js

@@ -1,7 +1,3 @@
-// src/index.ts
 export * from "./ui/index.js";
-import { frontendTools } from "./frontendTools.js";
-export {
-  frontendTools
-};
+export { frontendTools } from "./frontendTools.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"sourcesContent":["export * from \"./ui\";\nexport { frontendTools } from \"./frontendTools\";\n"],"mappings":";AAAA,cAAc;AACd,SAAS,qBAAqB;","names":[]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,8BAAqB;AACrB,OAAO,EAAE,aAAa,EAAE,2BAAwB"}

package/dist/ui/adapters/aiSDKFormatAdapter.js

@@ -1,28 +1,23 @@
-// src/ui/adapters/aiSDKFormatAdapter.ts
-var aiSDKV5FormatAdapter = {
-  format: "ai-sdk/v5",
-  encode({
-    message: { id, parts, ...message }
-  }) {
-    return {
-      ...message,
-      parts: parts.filter((part) => part.type !== "file")
-    };
-  },
-  decode(stored) {
-    return {
-      parentId: stored.parent_id,
-      message: {
-        id: stored.id,
-        ...stored.content
-      }
-    };
-  },
-  getId(message) {
-    return message.id;
-  }
-};
-export {
-  aiSDKV5FormatAdapter
+export const aiSDKV5FormatAdapter = {
+    format: "ai-sdk/v5",
+    encode({ message: { id, parts, ...message }, }) {
+        // Filter out FileContentParts until they are supported
+        return {
+            ...message,
+            parts: parts.filter((part) => part.type !== "file"),
+        };
+    },
+    decode(stored) {
+        return {
+            parentId: stored.parent_id,
+            message: {
+                id: stored.id,
+                ...stored.content,
+            },
+        };
+    },
+    getId(message) {
+        return message.id;
+    },
 };
 //# sourceMappingURL=aiSDKFormatAdapter.js.map

package/dist/ui/adapters/aiSDKFormatAdapter.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/adapters/aiSDKFormatAdapter.ts"],"sourcesContent":["import { UIMessage } from \"ai\";\nimport {\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageStorageEntry,\n} from \"@assistant-ui/react\";\n\n// Storage format for AI SDK messages - just the UIMessage\nexport type AISDKStorageFormat = Omit<UIMessage, \"id\">;\n\nexport const aiSDKV5FormatAdapter: MessageFormatAdapter<\n  UIMessage,\n  AISDKStorageFormat\n> = {\n  format: \"ai-sdk/v5\",\n\n  encode({\n    message: { id, parts, ...message },\n  }: MessageFormatItem<UIMessage>): AISDKStorageFormat {\n    // Filter out FileContentParts until they are supported\n    return {\n      ...message,\n      parts: parts.filter((part) => part.type !== \"file\"),\n    };\n  },\n\n  decode(\n    stored: MessageStorageEntry<AISDKStorageFormat>,\n  ): MessageFormatItem<UIMessage> {\n    return {\n      parentId: stored.parent_id,\n      message: {\n        id: stored.id,\n        ...stored.content,\n      },\n    };\n  },\n\n  getId(message: UIMessage): string {\n    return message.id;\n  },\n};\n"],"mappings":";AAUO,IAAM,uBAGT;AAAA,EACF,QAAQ;AAAA,EAER,OAAO;AAAA,IACL,SAAS,EAAE,IAAI,OAAO,GAAG,QAAQ;AAAA,EACnC,GAAqD;AAEnD,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,MAAM,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM;AAAA,IACpD;AAAA,EACF;AAAA,EAEA,OACE,QAC8B;AAC9B,WAAO;AAAA,MACL,UAAU,OAAO;AAAA,MACjB,SAAS;AAAA,QACP,IAAI,OAAO;AAAA,QACX,GAAG,OAAO;AAAA,MACZ;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,SAA4B;AAChC,WAAO,QAAQ;AAAA,EACjB;AACF;","names":[]}
+{"version":3,"file":"aiSDKFormatAdapter.js","sourceRoot":"","sources":["../../../src/ui/adapters/aiSDKFormatAdapter.ts"],"names":[],"mappings":"AAUA,MAAM,CAAC,MAAM,oBAAoB,GAG7B;IACF,MAAM,EAAE,WAAW;IAEnB,MAAM,CAAC,EACL,OAAO,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GACL;QAC7B,uDAAuD;QACvD,OAAO;YACL,GAAG,OAAO;YACV,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SACpD,CAAC;IACJ,CAAC;IAED,MAAM,CACJ,MAA+C;QAE/C,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,SAAS;YAC1B,OAAO,EAAE;gBACP,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,GAAG,MAAM,CAAC,OAAO;aAClB;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAkB;QACtB,OAAO,OAAO,CAAC,EAAE,CAAC;IACpB,CAAC;CACF,CAAC"}

package/dist/ui/getVercelAIMessages.js

@@ -1,11 +1,5 @@
-// src/ui/getVercelAIMessages.tsx
-import {
-  getExternalStoreMessages
-} from "@assistant-ui/react";
-var getVercelAIMessages = (message) => {
-  return getExternalStoreMessages(message);
-};
-export {
-  getVercelAIMessages
+import { getExternalStoreMessages, } from "@assistant-ui/react";
+export const getVercelAIMessages = (message) => {
+    return getExternalStoreMessages(message);
 };
 //# sourceMappingURL=getVercelAIMessages.js.map

package/dist/ui/getVercelAIMessages.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/getVercelAIMessages.tsx"],"sourcesContent":["import {\n  getExternalStoreMessages,\n  type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { UIMessage } from \"ai\";\n\nexport const getVercelAIMessages = <UI_MESSAGE extends UIMessage = UIMessage>(\n  message: ThreadMessage,\n) => {\n  return getExternalStoreMessages(message) as UI_MESSAGE[];\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAEK;AAGA,IAAM,sBAAsB,CACjC,YACG;AACH,SAAO,yBAAyB,OAAO;AACzC;","names":[]}
+{"version":3,"file":"getVercelAIMessages.js","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,wBAAwB,GAEzB,MAAM,qBAAqB,CAAC;AAG7B,MAAM,CAAC,MAAM,mBAAmB,GAAG,CACjC,OAAsB,EACtB,EAAE;IACF,OAAO,wBAAwB,CAAC,OAAO,CAAiB,CAAC;AAC3D,CAAC,CAAC"}

package/dist/ui/index.d.ts

@@ -1,5 +1,5 @@
-export { useAISDKRuntime } from "./use-chat/useAISDKRuntime";
-export { useChatRuntime } from "./use-chat/useChatRuntime";
-export type { UseChatRuntimeOptions } from "./use-chat/useChatRuntime";
-export { AssistantChatTransport } from "./use-chat/AssistantChatTransport";
+export { useAISDKRuntime } from "./use-chat/useAISDKRuntime.js";
+export { useChatRuntime } from "./use-chat/useChatRuntime.js";
+export type { UseChatRuntimeOptions } from "./use-chat/useChatRuntime.js";
+export { AssistantChatTransport } from "./use-chat/AssistantChatTransport.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/ui/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,YAAY,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,sBAAsB,EAAE,MAAM,mCAAmC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,sCAAmC;AAC7D,OAAO,EAAE,cAAc,EAAE,qCAAkC;AAC3D,YAAY,EAAE,qBAAqB,EAAE,qCAAkC;AACvE,OAAO,EAAE,sBAAsB,EAAE,6CAA0C"}

package/dist/ui/index.js

@@ -1,10 +1,4 @@
-// src/ui/index.ts
-import { useAISDKRuntime } from "./use-chat/useAISDKRuntime.js";
-import { useChatRuntime } from "./use-chat/useChatRuntime.js";
-import { AssistantChatTransport } from "./use-chat/AssistantChatTransport.js";
-export {
-  AssistantChatTransport,
-  useAISDKRuntime,
-  useChatRuntime
-};
+export { useAISDKRuntime } from "./use-chat/useAISDKRuntime.js";
+export { useChatRuntime } from "./use-chat/useChatRuntime.js";
+export { AssistantChatTransport } from "./use-chat/AssistantChatTransport.js";
 //# sourceMappingURL=index.js.map

package/dist/ui/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/index.ts"],"sourcesContent":["export { useAISDKRuntime } from \"./use-chat/useAISDKRuntime\";\nexport { useChatRuntime } from \"./use-chat/useChatRuntime\";\nexport type { UseChatRuntimeOptions } from \"./use-chat/useChatRuntime\";\nexport { AssistantChatTransport } from \"./use-chat/AssistantChatTransport\";\n"],"mappings":";AAAA,SAAS,uBAAuB;AAChC,SAAS,sBAAsB;AAE/B,SAAS,8BAA8B;","names":[]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,sCAAmC;AAC7D,OAAO,EAAE,cAAc,EAAE,qCAAkC;AAE3D,OAAO,EAAE,sBAAsB,EAAE,6CAA0C"}

package/dist/ui/use-chat/AssistantChatTransport.js

@@ -1,63 +1,54 @@
-// src/ui/use-chat/AssistantChatTransport.tsx
-import {
-  DefaultChatTransport
-} from "ai";
+import { DefaultChatTransport, } from "ai";
 import z from "zod";
-var toAISDKTools = (tools) => {
-  return Object.fromEntries(
-    Object.entries(tools).map(([name, tool]) => [
-      name,
-      {
-        ...tool.description ? { description: tool.description } : void 0,
-        parameters: tool.parameters instanceof z.ZodType ? z.toJSONSchema(tool.parameters) : tool.parameters
-      }
-    ])
-  );
+const toAISDKTools = (tools) => {
+    return Object.fromEntries(Object.entries(tools).map(([name, tool]) => [
+        name,
+        {
+            ...(tool.description ? { description: tool.description } : undefined),
+            parameters: (tool.parameters instanceof z.ZodType
+                ? z.toJSONSchema(tool.parameters)
+                : tool.parameters),
+        },
+    ]));
 };
-var getEnabledTools = (tools) => {
-  return Object.fromEntries(
-    Object.entries(tools).filter(
-      ([, tool]) => !tool.disabled && tool.type !== "backend"
-    )
-  );
-};
-var AssistantChatTransport = class extends DefaultChatTransport {
-  runtime;
-  constructor(initOptions) {
-    super({
-      ...initOptions,
-      prepareSendMessagesRequest: async (options) => {
-        const context = this.runtime?.thread.getModelContext();
-        const id = (await this.runtime?.threads.mainItem.initialize())?.remoteId ?? options.id;
-        const optionsEx = {
-          ...options,
-          body: {
-            callSettings: context?.callSettings,
-            system: context?.system,
-            tools: toAISDKTools(getEnabledTools(context?.tools ?? {})),
-            ...options?.body
-          }
-        };
-        const preparedRequest = await initOptions?.prepareSendMessagesRequest?.(optionsEx);
-        return {
-          ...preparedRequest,
-          body: preparedRequest?.body ?? {
-            ...optionsEx.body,
-            id,
-            messages: options.messages,
-            trigger: options.trigger,
-            messageId: options.messageId,
-            metadata: options.requestMetadata
-          }
-        };
-      }
-    });
-  }
-  setRuntime(runtime) {
-    this.runtime = runtime;
-  }
-};
-export {
-  AssistantChatTransport
+const getEnabledTools = (tools) => {
+    return Object.fromEntries(Object.entries(tools).filter(([, tool]) => !tool.disabled && tool.type !== "backend"));
 };
+export class AssistantChatTransport extends DefaultChatTransport {
+    runtime;
+    constructor(initOptions) {
+        super({
+            ...initOptions,
+            prepareSendMessagesRequest: async (options) => {
+                const context = this.runtime?.thread.getModelContext();
+                const id = (await this.runtime?.threads.mainItem.initialize())?.remoteId ??
+                    options.id;
+                const optionsEx = {
+                    ...options,
+                    body: {
+                        callSettings: context?.callSettings,
+                        system: context?.system,
+                        tools: toAISDKTools(getEnabledTools(context?.tools ?? {})),
+                        ...options?.body,
+                    },
+                };
+                const preparedRequest = await initOptions?.prepareSendMessagesRequest?.(optionsEx);
+                return {
+                    ...preparedRequest,
+                    body: preparedRequest?.body ?? {
+                        ...optionsEx.body,
+                        id,
+                        messages: options.messages,
+                        trigger: options.trigger,
+                        messageId: options.messageId,
+                        metadata: options.requestMetadata,
+                    },
+                };
+            },
+        });
+    }
+    setRuntime(runtime) {
+        this.runtime = runtime;
+    }
+}
 //# sourceMappingURL=AssistantChatTransport.js.map

package/dist/ui/use-chat/AssistantChatTransport.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/ui/use-chat/AssistantChatTransport.tsx"],"sourcesContent":["import { AssistantRuntime, Tool } from \"@assistant-ui/react\";\nimport {\n  DefaultChatTransport,\n  HttpChatTransportInitOptions,\n  JSONSchema7,\n  UIMessage,\n} from \"ai\";\nimport z from \"zod\";\n\nconst toAISDKTools = (tools: Record<string, Tool>) => {\n  return Object.fromEntries(\n    Object.entries(tools).map(([name, tool]) => [\n      name,\n      {\n        ...(tool.description ? { description: tool.description } : undefined),\n        parameters: (tool.parameters instanceof z.ZodType\n          ? z.toJSONSchema(tool.parameters)\n          : tool.parameters) as JSONSchema7,\n      },\n    ]),\n  );\n};\n\nconst getEnabledTools = (tools: Record<string, Tool>) => {\n  return Object.fromEntries(\n    Object.entries(tools).filter(\n      ([, tool]) => !tool.disabled && tool.type !== \"backend\",\n    ),\n  );\n};\n\nexport class AssistantChatTransport<\n  UI_MESSAGE extends UIMessage,\n> extends DefaultChatTransport<UI_MESSAGE> {\n  private runtime: AssistantRuntime | undefined;\n  constructor(initOptions?: HttpChatTransportInitOptions<UI_MESSAGE>) {\n    super({\n      ...initOptions,\n      prepareSendMessagesRequest: async (options) => {\n        const context = this.runtime?.thread.getModelContext();\n        const id =\n          (await this.runtime?.threads.mainItem.initialize())?.remoteId ??\n          options.id;\n\n        const optionsEx = {\n          ...options,\n          body: {\n            callSettings: context?.callSettings,\n            system: context?.system,\n            tools: toAISDKTools(getEnabledTools(context?.tools ?? {})),\n            ...options?.body,\n          },\n        };\n        const preparedRequest =\n          await initOptions?.prepareSendMessagesRequest?.(optionsEx);\n\n        return {\n          ...preparedRequest,\n          body: preparedRequest?.body ?? {\n            ...optionsEx.body,\n            id,\n            messages: options.messages,\n            trigger: options.trigger,\n            messageId: options.messageId,\n            metadata: options.requestMetadata,\n          },\n        };\n      },\n    });\n  }\n\n  setRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n}\n"],"mappings":";AACA;AAAA,EACE;AAAA,OAIK;AACP,OAAO,OAAO;AAEd,IAAM,eAAe,CAAC,UAAgC;AACpD,SAAO,OAAO;AAAA,IACZ,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAAA,MAC1C;AAAA,MACA;AAAA,QACE,GAAI,KAAK,cAAc,EAAE,aAAa,KAAK,YAAY,IAAI;AAAA,QAC3D,YAAa,KAAK,sBAAsB,EAAE,UACtC,EAAE,aAAa,KAAK,UAAU,IAC9B,KAAK;AAAA,MACX;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEA,IAAM,kBAAkB,CAAC,UAAgC;AACvD,SAAO,OAAO;AAAA,IACZ,OAAO,QAAQ,KAAK,EAAE;AAAA,MACpB,CAAC,CAAC,EAAE,IAAI,MAAM,CAAC,KAAK,YAAY,KAAK,SAAS;AAAA,IAChD;AAAA,EACF;AACF;AAEO,IAAM,yBAAN,cAEG,qBAAiC;AAAA,EACjC;AAAA,EACR,YAAY,aAAwD;AAClE,UAAM;AAAA,MACJ,GAAG;AAAA,MACH,4BAA4B,OAAO,YAAY;AAC7C,cAAM,UAAU,KAAK,SAAS,OAAO,gBAAgB;AACrD,cAAM,MACH,MAAM,KAAK,SAAS,QAAQ,SAAS,WAAW,IAAI,YACrD,QAAQ;AAEV,cAAM,YAAY;AAAA,UAChB,GAAG;AAAA,UACH,MAAM;AAAA,YACJ,cAAc,SAAS;AAAA,YACvB,QAAQ,SAAS;AAAA,YACjB,OAAO,aAAa,gBAAgB,SAAS,SAAS,CAAC,CAAC,CAAC;AAAA,YACzD,GAAG,SAAS;AAAA,UACd;AAAA,QACF;AACA,cAAM,kBACJ,MAAM,aAAa,6BAA6B,SAAS;AAE3D,eAAO;AAAA,UACL,GAAG;AAAA,UACH,MAAM,iBAAiB,QAAQ;AAAA,YAC7B,GAAG,UAAU;AAAA,YACb;AAAA,YACA,UAAU,QAAQ;AAAA,YAClB,SAAS,QAAQ;AAAA,YACjB,WAAW,QAAQ;AAAA,YACnB,UAAU,QAAQ;AAAA,UACpB;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,WAAW,SAA2B;AACpC,SAAK,UAAU;AAAA,EACjB;AACF;","names":[]}
+{"version":3,"file":"AssistantChatTransport.js","sourceRoot":"","sources":["../../../src/ui/use-chat/AssistantChatTransport.tsx"],"names":[],"mappings":"AACA,OAAO,EACL,oBAAoB,GAIrB,MAAM,IAAI,CAAC;AACZ,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,MAAM,YAAY,GAAG,CAAC,KAA2B,EAAE,EAAE;IACnD,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QAC1C,IAAI;QACJ;YACE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACrE,UAAU,EAAE,CAAC,IAAI,CAAC,UAAU,YAAY,CAAC,CAAC,OAAO;gBAC/C,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC;gBACjC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAgB;SACpC;KACF,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,KAA2B,EAAE,EAAE;IACtD,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,CAC1B,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,CACxD,CACF,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,OAAO,sBAEX,SAAQ,oBAAgC;IAChC,OAAO,CAA+B;IAC9C,YAAY,WAAsD;QAChE,KAAK,CAAC;YACJ,GAAG,WAAW;YACd,0BAA0B,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;gBACvD,MAAM,EAAE,GACN,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,EAAE,QAAQ;oBAC7D,OAAO,CAAC,EAAE,CAAC;gBAEb,MAAM,SAAS,GAAG;oBAChB,GAAG,OAAO;oBACV,IAAI,EAAE;wBACJ,YAAY,EAAE,OAAO,EAAE,YAAY;wBACnC,MAAM,EAAE,OAAO,EAAE,MAAM;wBACvB,KAAK,EAAE,YAAY,CAAC,eAAe,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;wBAC1D,GAAG,OAAO,EAAE,IAAI;qBACjB;iBACF,CAAC;gBACF,MAAM,eAAe,GACnB,MAAM,WAAW,EAAE,0BAA0B,EAAE,CAAC,SAAS,CAAC,CAAC;gBAE7D,OAAO;oBACL,GAAG,eAAe;oBAClB,IAAI,EAAE,eAAe,EAAE,IAAI,IAAI;wBAC7B,GAAG,SAAS,CAAC,IAAI;wBACjB,EAAE;wBACF,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,QAAQ,EAAE,OAAO,CAAC,eAAe;qBAClC;iBACF,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAAyB;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF"}

package/dist/ui/use-chat/useAISDKRuntime.js

@@ -1,161 +1,142 @@
 "use client";
-
-// src/ui/use-chat/useAISDKRuntime.tsx
 import { useState, useMemo } from "react";
 import { isToolUIPart } from "ai";
-import {
-  useExternalStoreRuntime,
-  useRuntimeAdapters,
-  INTERNAL
-} from "@assistant-ui/react";
+import { useExternalStoreRuntime, useRuntimeAdapters, INTERNAL, } from "@assistant-ui/react";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil.js";
 import { toCreateMessage } from "../utils/toCreateMessage.js";
 import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter.js";
 import { getVercelAIMessages } from "../getVercelAIMessages.js";
 import { AISDKMessageConverter } from "../utils/convertMessage.js";
-import {
-  aiSDKV5FormatAdapter
-} from "../adapters/aiSDKFormatAdapter.js";
+import { aiSDKV5FormatAdapter, } from "../adapters/aiSDKFormatAdapter.js";
 import { useExternalHistory } from "./useExternalHistory.js";
-var useAISDKRuntime = (chatHelpers, {
-  adapters,
-  toCreateMessage: customToCreateMessage,
-  cancelPendingToolCallsOnSend = true
-} = {}) => {
-  const contextAdapters = useRuntimeAdapters();
-  const [toolStatuses, setToolStatuses] = useState({});
-  const hasExecutingTools = Object.values(toolStatuses).some(
-    (s) => s?.type === "executing"
-  );
-  const isRunning = chatHelpers.status === "submitted" || chatHelpers.status === "streaming" || hasExecutingTools;
-  const messages = AISDKMessageConverter.useThreadMessages({
-    isRunning,
-    messages: chatHelpers.messages,
-    metadata: useMemo(
-      () => ({
-        toolStatuses,
-        ...chatHelpers.error && { error: chatHelpers.error.message }
-      }),
-      [toolStatuses, chatHelpers.error]
-    )
-  });
-  const [runtimeRef] = useState(() => ({
-    get current() {
-      return runtime;
-    }
-  }));
-  const toolInvocations = INTERNAL.useToolInvocations({
-    state: {
-      messages,
-      isRunning
-    },
-    getTools: () => runtimeRef.current.thread.getModelContext().tools,
-    onResult: (command) => {
-      if (command.type === "add-tool-result") {
-        chatHelpers.addToolResult({
-          tool: command.toolName,
-          toolCallId: command.toolCallId,
-          output: command.result
-        });
-      }
-    },
-    setToolStatuses
-  });
-  const isLoading = useExternalHistory(
-    runtimeRef,
-    adapters?.history ?? contextAdapters?.history,
-    AISDKMessageConverter.toThreadMessages,
-    aiSDKV5FormatAdapter,
-    (messages2) => {
-      chatHelpers.setMessages(messages2);
-    }
-  );
-  const completePendingToolCalls = async () => {
-    if (!cancelPendingToolCallsOnSend) return;
-    await toolInvocations.abort();
-    chatHelpers.setMessages((messages2) => {
-      const lastMessage = messages2.at(-1);
-      if (lastMessage?.role !== "assistant") return messages2;
-      let hasChanges = false;
-      const parts = lastMessage.parts?.map((part) => {
-        if (!isToolUIPart(part)) return part;
-        if (part.state === "output-available" || part.state === "output-error")
-          return part;
-        hasChanges = true;
-        return {
-          ...part,
-          state: "output-error",
-          errorText: "User cancelled tool call by sending a new message."
-        };
-      });
-      if (!hasChanges) return messages2;
-      return [...messages2.slice(0, -1), { ...lastMessage, parts }];
+export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, } = {}) => {
+    const contextAdapters = useRuntimeAdapters();
+    const [toolStatuses, setToolStatuses] = useState({});
+    const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
+    const isRunning = chatHelpers.status === "submitted" ||
+        chatHelpers.status === "streaming" ||
+        hasExecutingTools;
+    const messages = AISDKMessageConverter.useThreadMessages({
+        isRunning,
+        messages: chatHelpers.messages,
+        metadata: useMemo(() => ({
+            toolStatuses,
+            ...(chatHelpers.error && { error: chatHelpers.error.message }),
+        }), [toolStatuses, chatHelpers.error]),
     });
-  };
-  const runtime = useExternalStoreRuntime({
-    isRunning,
-    messages,
-    setMessages: (messages2) => chatHelpers.setMessages(
-      messages2.map(getVercelAIMessages).filter(Boolean).flat()
-    ),
-    onImport: (messages2) => chatHelpers.setMessages(
-      messages2.map(getVercelAIMessages).filter(Boolean).flat()
-    ),
-    onCancel: async () => {
-      chatHelpers.stop();
-      await toolInvocations.abort();
-    },
-    onNew: async (message) => {
-      await completePendingToolCalls();
-      const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
-      await chatHelpers.sendMessage(createMessage, {
-        metadata: message.runConfig
-      });
-    },
-    onEdit: async (message) => {
-      const newMessages = sliceMessagesUntil(
-        chatHelpers.messages,
-        message.parentId
-      );
-      chatHelpers.setMessages(newMessages);
-      const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
-      await chatHelpers.sendMessage(createMessage, {
-        metadata: message.runConfig
-      });
-    },
-    onReload: async (parentId, config) => {
-      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
-      chatHelpers.setMessages(newMessages);
-      await chatHelpers.regenerate({ metadata: config.runConfig });
-    },
-    onAddToolResult: ({ toolCallId, result, isError }) => {
-      if (isError) {
-        chatHelpers.addToolOutput({
-          state: "output-error",
-          tool: toolCallId,
-          toolCallId,
-          errorText: typeof result === "string" ? result : JSON.stringify(result)
-        });
-      } else {
-        chatHelpers.addToolOutput({
-          state: "output-available",
-          tool: toolCallId,
-          toolCallId,
-          output: result
+    const [runtimeRef] = useState(() => ({
+        get current() {
+            return runtime;
+        },
+    }));
+    const toolInvocations = INTERNAL.useToolInvocations({
+        state: {
+            messages,
+            isRunning,
+        },
+        getTools: () => runtimeRef.current.thread.getModelContext().tools,
+        onResult: (command) => {
+            if (command.type === "add-tool-result") {
+                chatHelpers.addToolResult({
+                    tool: command.toolName,
+                    toolCallId: command.toolCallId,
+                    output: command.result,
+                });
+            }
+        },
+        setToolStatuses,
+    });
+    const isLoading = useExternalHistory(runtimeRef, adapters?.history ?? contextAdapters?.history, AISDKMessageConverter.toThreadMessages, aiSDKV5FormatAdapter, (messages) => {
+        chatHelpers.setMessages(messages);
+    });
+    const completePendingToolCalls = async () => {
+        if (!cancelPendingToolCallsOnSend)
+            return;
+        await toolInvocations.abort();
+        // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)
+        chatHelpers.setMessages((messages) => {
+            const lastMessage = messages.at(-1);
+            if (lastMessage?.role !== "assistant")
+                return messages;
+            let hasChanges = false;
+            const parts = lastMessage.parts?.map((part) => {
+                if (!isToolUIPart(part))
+                    return part;
+                if (part.state === "output-available" || part.state === "output-error")
+                    return part;
+                hasChanges = true;
+                return {
+                    ...part,
+                    state: "output-error",
+                    errorText: "User cancelled tool call by sending a new message.",
+                };
+            });
+            if (!hasChanges)
+                return messages;
+            return [...messages.slice(0, -1), { ...lastMessage, parts }];
         });
-      }
-    },
-    onResumeToolCall: (options) => toolInvocations.resume(options.toolCallId, options.payload),
-    adapters: {
-      attachments: vercelAttachmentAdapter,
-      ...contextAdapters,
-      ...adapters
-    },
-    isLoading
-  });
-  return runtime;
-};
-export {
-  useAISDKRuntime
+    };
+    const runtime = useExternalStoreRuntime({
+        isRunning,
+        messages,
+        setMessages: (messages) => chatHelpers.setMessages(messages
+            .map((getVercelAIMessages))
+            .filter(Boolean)
+            .flat()),
+        onImport: (messages) => chatHelpers.setMessages(messages
+            .map((getVercelAIMessages))
+            .filter(Boolean)
+            .flat()),
+        onCancel: async () => {
+            chatHelpers.stop();
+            await toolInvocations.abort();
+        },
+        onNew: async (message) => {
+            await completePendingToolCalls();
+            const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
+            await chatHelpers.sendMessage(createMessage, {
+                metadata: message.runConfig,
+            });
+        },
+        onEdit: async (message) => {
+            const newMessages = sliceMessagesUntil(chatHelpers.messages, message.parentId);
+            chatHelpers.setMessages(newMessages);
+            const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
+            await chatHelpers.sendMessage(createMessage, {
+                metadata: message.runConfig,
+            });
+        },
+        onReload: async (parentId, config) => {
+            const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
+            chatHelpers.setMessages(newMessages);
+            await chatHelpers.regenerate({ metadata: config.runConfig });
+        },
+        onAddToolResult: ({ toolCallId, result, isError }) => {
+            if (isError) {
+                chatHelpers.addToolOutput({
+                    state: "output-error",
+                    tool: toolCallId,
+                    toolCallId,
+                    errorText: typeof result === "string" ? result : JSON.stringify(result),
+                });
+            }
+            else {
+                chatHelpers.addToolOutput({
+                    state: "output-available",
+                    tool: toolCallId,
+                    toolCallId,
+                    output: result,
+                });
+            }
+        },
+        onResumeToolCall: (options) => toolInvocations.resume(options.toolCallId, options.payload),
+        adapters: {
+            attachments: vercelAttachmentAdapter,
+            ...contextAdapters,
+            ...adapters,
+        },
+        isLoading,
+    });
+    return runtime;
 };
 //# sourceMappingURL=useAISDKRuntime.js.map