@ai-sdk/vue 2.0.0-alpha.5 → 2.0.0-alpha.6

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @ai-sdk/vue
 
+## 2.0.0-alpha.6
+
+### Major Changes
+
+- 8cbbad6: chore (ai): refactor and use chatstore in vue
+
+### Patch Changes
+
+- Updated dependencies [0d2c085]
+- Updated dependencies [48a7606]
+  - ai@5.0.0-alpha.6
+  - @ai-sdk/provider-utils@3.0.0-alpha.6
+
 ## 2.0.0-alpha.5
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,17 +1,24 @@
-import { UIMessage, CreateUIMessage, ChatRequestOptions, FileUIPart, OriginalUseChatOptions, JSONValue, CompletionRequestOptions, UseCompletionOptions } from 'ai';
+import { UIDataPartSchemas, UIMessage, InferUIDataParts, CreateUIMessage, ChatRequestOptions, FileUIPart, ChatStatus, UseChatOptions, CompletionRequestOptions, UseCompletionOptions } from 'ai';
 export { CreateUIMessage, UIMessage, UseChatOptions, UseCompletionOptions } from 'ai';
 import { Ref } from 'vue';
 
-type UseChatHelpers<MESSAGE_METADATA> = {
+type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = {
+    /**
+     * The id of the chat.
+     */
+    readonly chatId: string;
     /** Current messages in the chat */
-    messages: Ref<UIMessage<MESSAGE_METADATA>[]>;
+    readonly messages: Ref<UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]>;
     /** The error object of the API request */
-    error: Ref<undefined | Error>;
+    readonly error: Ref<Error | undefined>;
     /**
      * Append a user message to the chat list. This triggers the API call to fetch
      * the assistant's response.
+     *
+     * @param message The message to append
+     * @param options Additional options to pass to the API call
      */
-    append: (message: UIMessage<MESSAGE_METADATA> | CreateUIMessage<MESSAGE_METADATA>, chatRequestOptions?: ChatRequestOptions) => Promise<string | null | undefined>;
+    append: (message: CreateUIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>, options?: ChatRequestOptions) => Promise<void>;
     /**
      * Reload the last AI chat response for the given chat history. If the last
      * message isn't from the assistant, it will request the API to generate a
@@ -27,7 +34,7 @@ type UseChatHelpers<MESSAGE_METADATA> = {
      * edit the messages on the client, and then trigger the `reload` method
      * manually to regenerate the AI response.
      */
-    setMessages: (messages: UIMessage<MESSAGE_METADATA>[] | ((messages: UIMessage<MESSAGE_METADATA>[]) => UIMessage<MESSAGE_METADATA>[])) => void;
+    setMessages: (messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[] | ((messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]) => UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[])) => void;
     /** The current value of the input */
     input: Ref<string>;
     /** Form submission handler to automatically reset input and append a user message  */
@@ -44,32 +51,13 @@ type UseChatHelpers<MESSAGE_METADATA> = {
      * - `ready`: The full response has been received and processed; a new user message can be submitted.
      * - `error`: An error occurred during the API request, preventing successful completion.
      */
-    status: Ref<'submitted' | 'streaming' | 'ready' | 'error'>;
+    status: Ref<ChatStatus>;
     addToolResult: ({ toolCallId, result, }: {
         toolCallId: string;
         result: any;
     }) => void;
-    /** The id of the chat */
-    chatId: string;
 };
-declare function useChat<MESSAGE_METADATA = unknown>({ api, chatId, initialMessages, initialInput, streamProtocol, onFinish, onError, credentials, headers: metadataHeaders, body: metadataBody, generateId, onToolCall, fetch, maxSteps, experimental_prepareRequestBody, messageMetadataSchema, }?: OriginalUseChatOptions<MESSAGE_METADATA> & {
-    /**
-     * Experimental (Vue only). When a function is provided, it will be used
-     * to prepare the request body for the chat API. This can be useful for
-     * customizing the request body based on the messages and data in the chat.
-     *
-     * @param id The chat ID
-     * @param messages The current messages in the chat
-     * @param requestData The data object passed in the chat request
-     * @param requestBody The request body object passed in the chat request
-     */
-    experimental_prepareRequestBody?: (options: {
-        chatId: string;
-        messages: UIMessage<MESSAGE_METADATA>[];
-        requestData?: JSONValue;
-        requestBody?: object;
-    }) => unknown;
-}): UseChatHelpers<MESSAGE_METADATA>;
+declare function useChat<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>({ chatId, initialInput, onFinish, onError, generateId, onToolCall, chatStore: chatStoreArg, }?: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
 
 type UseCompletionHelpers = {
     /** The current completion result */

package/dist/index.d.ts

@@ -1,17 +1,24 @@
-import { UIMessage, CreateUIMessage, ChatRequestOptions, FileUIPart, OriginalUseChatOptions, JSONValue, CompletionRequestOptions, UseCompletionOptions } from 'ai';
+import { UIDataPartSchemas, UIMessage, InferUIDataParts, CreateUIMessage, ChatRequestOptions, FileUIPart, ChatStatus, UseChatOptions, CompletionRequestOptions, UseCompletionOptions } from 'ai';
 export { CreateUIMessage, UIMessage, UseChatOptions, UseCompletionOptions } from 'ai';
 import { Ref } from 'vue';
 
-type UseChatHelpers<MESSAGE_METADATA> = {
+type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = {
+    /**
+     * The id of the chat.
+     */
+    readonly chatId: string;
     /** Current messages in the chat */
-    messages: Ref<UIMessage<MESSAGE_METADATA>[]>;
+    readonly messages: Ref<UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]>;
     /** The error object of the API request */
-    error: Ref<undefined | Error>;
+    readonly error: Ref<Error | undefined>;
     /**
      * Append a user message to the chat list. This triggers the API call to fetch
      * the assistant's response.
+     *
+     * @param message The message to append
+     * @param options Additional options to pass to the API call
      */
-    append: (message: UIMessage<MESSAGE_METADATA> | CreateUIMessage<MESSAGE_METADATA>, chatRequestOptions?: ChatRequestOptions) => Promise<string | null | undefined>;
+    append: (message: CreateUIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>, options?: ChatRequestOptions) => Promise<void>;
     /**
      * Reload the last AI chat response for the given chat history. If the last
      * message isn't from the assistant, it will request the API to generate a
@@ -27,7 +34,7 @@ type UseChatHelpers<MESSAGE_METADATA> = {
      * edit the messages on the client, and then trigger the `reload` method
      * manually to regenerate the AI response.
      */
-    setMessages: (messages: UIMessage<MESSAGE_METADATA>[] | ((messages: UIMessage<MESSAGE_METADATA>[]) => UIMessage<MESSAGE_METADATA>[])) => void;
+    setMessages: (messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[] | ((messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]) => UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[])) => void;
     /** The current value of the input */
     input: Ref<string>;
     /** Form submission handler to automatically reset input and append a user message  */
@@ -44,32 +51,13 @@ type UseChatHelpers<MESSAGE_METADATA> = {
      * - `ready`: The full response has been received and processed; a new user message can be submitted.
      * - `error`: An error occurred during the API request, preventing successful completion.
      */
-    status: Ref<'submitted' | 'streaming' | 'ready' | 'error'>;
+    status: Ref<ChatStatus>;
     addToolResult: ({ toolCallId, result, }: {
         toolCallId: string;
         result: any;
     }) => void;
-    /** The id of the chat */
-    chatId: string;
 };
-declare function useChat<MESSAGE_METADATA = unknown>({ api, chatId, initialMessages, initialInput, streamProtocol, onFinish, onError, credentials, headers: metadataHeaders, body: metadataBody, generateId, onToolCall, fetch, maxSteps, experimental_prepareRequestBody, messageMetadataSchema, }?: OriginalUseChatOptions<MESSAGE_METADATA> & {
-    /**
-     * Experimental (Vue only). When a function is provided, it will be used
-     * to prepare the request body for the chat API. This can be useful for
-     * customizing the request body based on the messages and data in the chat.
-     *
-     * @param id The chat ID
-     * @param messages The current messages in the chat
-     * @param requestData The data object passed in the chat request
-     * @param requestBody The request body object passed in the chat request
-     */
-    experimental_prepareRequestBody?: (options: {
-        chatId: string;
-        messages: UIMessage<MESSAGE_METADATA>[];
-        requestData?: JSONValue;
-        requestBody?: object;
-    }) => unknown;
-}): UseChatHelpers<MESSAGE_METADATA>;
+declare function useChat<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>({ chatId, initialInput, onFinish, onError, generateId, onToolCall, chatStore: chatStoreArg, }?: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
 
 type UseCompletionHelpers = {
     /** The current completion result */

package/dist/index.js

@@ -36,190 +36,132 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // src/use-chat.ts
+var import_ai2 = require("ai");
+var import_vue2 = require("vue");
+
+// src/chat-store.ts
 var import_ai = require("ai");
-var import_swrv = __toESM(require("swrv"));
 var import_vue = require("vue");
-var useSWRV = import_swrv.default.default || import_swrv.default;
-var store = {};
-var statusStore = {};
+var VueChat = class {
+  constructor(messages) {
+    this.statusRef = (0, import_vue.ref)("ready");
+    this.errorRef = (0, import_vue.ref)(void 0);
+    this.activeResponse = void 0;
+    this.jobExecutor = new import_ai.SerialJobExecutor();
+    this.setStatus = (status) => {
+      this.statusRef.value = status;
+    };
+    this.setError = (error) => {
+      this.errorRef.value = error;
+    };
+    this.setActiveResponse = (activeResponse) => {
+      this.activeResponse = activeResponse;
+    };
+    this.setMessages = (messages) => {
+      this.messagesRef.value = messages;
+    };
+    this.pushMessage = (message) => {
+      this.messagesRef.value.push(message);
+    };
+    this.popMessage = () => {
+      this.messagesRef.value.pop();
+    };
+    this.replaceMessage = (index, message) => {
+      this.messagesRef.value[index] = { ...message };
+    };
+    this.messagesRef = (0, import_vue.ref)(messages != null ? messages : []);
+  }
+  get messages() {
+    return this.messagesRef.value;
+  }
+  get status() {
+    return this.statusRef.value;
+  }
+  get error() {
+    return this.errorRef.value;
+  }
+};
+function createChatStore(options) {
+  return new import_ai.ChatStore({
+    ...options,
+    createChat: (options2) => new VueChat(
+      options2.messages
+    )
+  });
+}
+
+// src/use-chat.ts
 function useChat({
-  api = "/api/chat",
   chatId,
-  initialMessages = [],
   initialInput = "",
-  streamProtocol = "ui-message",
   onFinish,
   onError,
-  credentials,
-  headers: metadataHeaders,
-  body: metadataBody,
-  generateId = import_ai.generateId,
+  generateId = import_ai2.generateId,
   onToolCall,
-  fetch,
-  maxSteps = 1,
-  experimental_prepareRequestBody,
-  messageMetadataSchema
-} = {
-  maxSteps: 1
-}) {
-  var _a, _b;
+  chatStore: chatStoreArg
+} = {}) {
   const stableChatId = chatId != null ? chatId : generateId();
-  const key = `${api}|${stableChatId}`;
-  const { data: messagesData, mutate: originalMutate } = useSWRV(key, () => {
-    var _a2;
-    return (_a2 = store[key]) != null ? _a2 : initialMessages;
+  const chatStore = chatStoreArg == null ? createChatStore(
+    (0, import_ai2.defaultChatStoreOptions)({
+      api: "/api/chat",
+      generateId
+    })()
+  ) : typeof chatStoreArg === "function" ? createChatStore(chatStoreArg()) : chatStoreArg;
+  const messages = (0, import_vue2.computed)(() => chatStore.getMessages(stableChatId));
+  const status = (0, import_vue2.computed)(() => chatStore.getStatus(stableChatId));
+  const error = (0, import_vue2.computed)(() => chatStore.getError(stableChatId));
+  const append = async (message, { headers, body } = {}) => chatStore.submitMessage({
+    chatId: stableChatId,
+    message,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
+  });
+  const reload = async ({ headers, body } = {}) => chatStore.resubmitLastUserMessage({
+    chatId: stableChatId,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
   });
-  const status = (_a = statusStore[stableChatId]) != null ? _a : statusStore[stableChatId] = (0, import_vue.ref)("ready");
-  (_b = messagesData.value) != null ? _b : messagesData.value = initialMessages;
-  const mutate = (data) => {
-    store[key] = data;
-    return originalMutate();
-  };
-  const messages = messagesData;
-  const error = (0, import_vue.ref)(void 0);
-  let abortController = null;
-  async function triggerRequest(messagesSnapshot, { headers, body } = {}) {
-    var _a2;
-    error.value = void 0;
-    status.value = "submitted";
-    const messageCount = messages.value.length;
-    const lastMessage = messages.value.at(-1);
-    const maxStep = lastMessage != null ? (0, import_ai.extractMaxToolInvocationStep)((0, import_ai.getToolInvocations)(lastMessage)) : 0;
-    try {
-      abortController = new AbortController();
-      mutate(messagesSnapshot);
-      await (0, import_ai.callChatApi)({
-        api,
-        body: (_a2 = experimental_prepareRequestBody == null ? void 0 : experimental_prepareRequestBody({
-          chatId: stableChatId,
-          messages: messagesSnapshot,
-          requestBody: body
-        })) != null ? _a2 : {
-          chatId: stableChatId,
-          messages: messagesSnapshot,
-          ...(0, import_vue.unref)(metadataBody),
-          // Use unref to unwrap the ref value
-          ...body
-        },
-        streamProtocol,
-        headers: {
-          ...metadataHeaders,
-          ...headers
-        },
-        abortController: () => abortController,
-        credentials,
-        onUpdate({ message }) {
-          status.value = "streaming";
-          const replaceLastMessage = message.id === messagesSnapshot[messagesSnapshot.length - 1].id;
-          mutate([
-            ...replaceLastMessage ? messagesSnapshot.slice(0, messagesSnapshot.length - 1) : messagesSnapshot,
-            message
-          ]);
-        },
-        onFinish,
-        generateId,
-        onToolCall,
-        fetch,
-        // enabled use of structured clone in processChatResponse:
-        lastMessage: recursiveToRaw(
-          messagesSnapshot[messagesSnapshot.length - 1]
-        ),
-        messageMetadataSchema
-      });
-      status.value = "ready";
-    } catch (err) {
-      if (err.name === "AbortError") {
-        abortController = null;
-        status.value = "ready";
-        return null;
-      }
-      if (onError && err instanceof Error) {
-        onError(err);
-      }
-      error.value = err;
-      status.value = "error";
-    } finally {
-      abortController = null;
-    }
-    if ((0, import_ai.shouldResubmitMessages)({
-      originalMaxToolInvocationStep: maxStep,
-      originalMessageCount: messageCount,
-      maxSteps,
-      messages: messages.value
-    })) {
-      await triggerRequest(messages.value);
-    }
-  }
-  const append = async (message, options) => {
-    var _a2;
-    return triggerRequest(
-      messages.value.concat({
-        ...message,
-        id: (_a2 = message.id) != null ? _a2 : generateId(),
-        parts: message.parts
-      }),
-      options
-    );
-  };
-  const reload = async (options) => {
-    const messagesSnapshot = messages.value;
-    if (messagesSnapshot.length === 0)
-      return null;
-    const lastMessage = messagesSnapshot[messagesSnapshot.length - 1];
-    if (lastMessage.role === "assistant") {
-      return triggerRequest(messagesSnapshot.slice(0, -1), options);
-    }
-    return triggerRequest(messagesSnapshot, options);
-  };
   const stop = () => {
-    if (abortController) {
-      abortController.abort();
-      abortController = null;
-    }
+    chatStore.stopStream({ chatId: stableChatId });
   };
-  const setMessages = (messagesArg) => {
-    if (typeof messagesArg === "function") {
-      messagesArg = messagesArg(messages.value);
+  const setMessages = (messagesParam) => {
+    if (typeof messagesParam === "function") {
+      messagesParam = messagesParam(chatStore.getMessages(stableChatId));
     }
-    mutate(messagesArg);
+    chatStore.setMessages({
+      id: stableChatId,
+      messages: messagesParam
+    });
   };
-  const input = (0, import_vue.ref)(initialInput);
+  const input = (0, import_vue2.ref)(initialInput);
   const handleSubmit = async (event, options = {}) => {
-    var _a2;
-    (_a2 = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a2.call(event);
+    var _a;
+    (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
     const inputValue = input.value;
-    const fileParts = Array.isArray(options == null ? void 0 : options.files) ? options.files : await (0, import_ai.convertFileListToFileUIParts)(options == null ? void 0 : options.files);
+    const fileParts = Array.isArray(options == null ? void 0 : options.files) ? options.files : await (0, import_ai2.convertFileListToFileUIParts)(options == null ? void 0 : options.files);
     if (!inputValue && fileParts.length === 0)
       return;
-    triggerRequest(
-      messages.value.concat({
+    await append(
+      {
         id: generateId(),
         role: "user",
+        metadata: void 0,
         parts: [...fileParts, { type: "text", text: inputValue }]
-      }),
-      options
+      },
+      {
+        headers: options.headers,
+        body: options.body
+      }
     );
     input.value = "";
   };
-  const addToolResult = ({
-    toolCallId,
-    result
-  }) => {
-    const currentMessages = messages.value;
-    (0, import_ai.updateToolCallResult)({
-      messages: currentMessages,
-      toolCallId,
-      toolResult: result
-    });
-    mutate(currentMessages);
-    if (status.value === "submitted" || status.value === "streaming") {
-      return;
-    }
-    const lastMessage = currentMessages[currentMessages.length - 1];
-    if ((0, import_ai.isAssistantMessageWithCompletedToolCalls)(lastMessage)) {
-      triggerRequest(currentMessages);
-    }
-  };
+  const addToolResult = (options) => chatStore.addToolResult({ chatId: stableChatId, ...options });
   return {
     chatId: stableChatId,
     messages,
@@ -234,27 +176,14 @@ function useChat({
     addToolResult
   };
 }
-function recursiveToRaw(inputValue) {
-  if (Array.isArray(inputValue)) {
-    return [...inputValue.map(recursiveToRaw)];
-  } else if (typeof inputValue === "object" && inputValue !== null) {
-    const clone = {};
-    for (const [key, value] of Object.entries(inputValue)) {
-      clone[key] = recursiveToRaw(value);
-    }
-    return clone;
-  } else {
-    return inputValue;
-  }
-}
 
 // src/use-completion.ts
-var import_ai2 = require("ai");
-var import_swrv2 = __toESM(require("swrv"));
-var import_vue2 = require("vue");
+var import_ai3 = require("ai");
+var import_swrv = __toESM(require("swrv"));
+var import_vue3 = require("vue");
 var uniqueId = 0;
-var useSWRV2 = import_swrv2.default.default || import_swrv2.default;
-var store2 = {};
+var useSWRV = import_swrv.default.default || import_swrv.default;
+var store = {};
 function useCompletion({
   api = "/api/completion",
   id,
@@ -271,25 +200,25 @@ function useCompletion({
   var _a;
   const completionId = id || `completion-${uniqueId++}`;
   const key = `${api}|${completionId}`;
-  const { data, mutate: originalMutate } = useSWRV2(
+  const { data, mutate: originalMutate } = useSWRV(
     key,
-    () => store2[key] || initialCompletion
+    () => store[key] || initialCompletion
   );
-  const { data: isLoading, mutate: mutateLoading } = useSWRV2(
+  const { data: isLoading, mutate: mutateLoading } = useSWRV(
     `${completionId}-loading`,
     null
   );
   (_a = isLoading.value) != null ? _a : isLoading.value = false;
   data.value || (data.value = initialCompletion);
   const mutate = (data2) => {
-    store2[key] = data2;
+    store[key] = data2;
     return originalMutate();
   };
   const completion = data;
-  const error = (0, import_vue2.ref)(void 0);
+  const error = (0, import_vue3.ref)(void 0);
   let abortController = null;
   async function triggerRequest(prompt, options) {
-    return (0, import_ai2.callCompletionApi)({
+    return (0, import_ai3.callCompletionApi)({
       api,
       prompt,
       credentials,
@@ -298,7 +227,7 @@ function useCompletion({
         ...options == null ? void 0 : options.headers
       },
       body: {
-        ...(0, import_vue2.unref)(body),
+        ...(0, import_vue3.unref)(body),
         ...options == null ? void 0 : options.body
       },
       streamProtocol,
@@ -327,7 +256,7 @@ function useCompletion({
   const setCompletion = (completion2) => {
     mutate(completion2);
   };
-  const input = (0, import_vue2.ref)(initialInput);
+  const input = (0, import_vue3.ref)(initialInput);
   const handleSubmit = (event) => {
     var _a2;
     (_a2 = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a2.call(event);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/use-chat.ts","../src/use-completion.ts"],"sourcesContent":["export * from './use-chat';\nexport * from './use-completion';\n","import type {\n  ChatRequestOptions,\n  CreateUIMessage,\n  FileUIPart,\n  JSONValue,\n  OriginalUseChatOptions,\n  UIMessage,\n  UseChatOptions,\n} from 'ai';\nimport {\n  callChatApi,\n  convertFileListToFileUIParts,\n  extractMaxToolInvocationStep,\n  generateId as generateIdFunc,\n  getToolInvocations,\n  isAssistantMessageWithCompletedToolCalls,\n  shouldResubmitMessages,\n  updateToolCallResult,\n} from 'ai';\nimport swrv from 'swrv';\nimport type { Ref } from 'vue';\nimport { ref, unref } from 'vue';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<MESSAGE_METADATA> = {\n  /** Current messages in the chat */\n  messages: Ref<UIMessage<MESSAGE_METADATA>[]>;\n  /** The error object of the API request */\n  error: Ref<undefined | Error>;\n  /**\n   * Append a user message to the chat list. This triggers the API call to fetch\n   * the assistant's response.\n   */\n  append: (\n    message: UIMessage<MESSAGE_METADATA> | CreateUIMessage<MESSAGE_METADATA>,\n    chatRequestOptions?: ChatRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Reload the last AI chat response for the given chat history. If the last\n   * message isn't from the assistant, it will request the API to generate a\n   * new response.\n   */\n  reload: (\n    chatRequestOptions?: ChatRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Abort the current request immediately, keep the generated tokens if any.\n   */\n  stop: () => void;\n  /**\n   * Update the `messages` state locally. This is useful when you want to\n   * edit the messages on the client, and then trigger the `reload` method\n   * manually to regenerate the AI response.\n   */\n  setMessages: (\n    messages:\n      | UIMessage<MESSAGE_METADATA>[]\n      | ((\n          messages: UIMessage<MESSAGE_METADATA>[],\n        ) => UIMessage<MESSAGE_METADATA>[]),\n  ) => void;\n  /** The current value of the input */\n  input: Ref<string>;\n  /** Form submission handler to automatically reset input and append a user message  */\n  handleSubmit: (\n    event?: { preventDefault?: () => void },\n    chatRequestOptions?: ChatRequestOptions & {\n      files?: FileList | FileUIPart[];\n    },\n  ) => void;\n\n  /**\n   * Hook status:\n   *\n   * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n   * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n   * - `ready`: The full response has been received and processed; a new user message can be submitted.\n   * - `error`: An error occurred during the API request, preventing successful completion.\n   */\n  status: Ref<'submitted' | 'streaming' | 'ready' | 'error'>;\n\n  addToolResult: ({\n    toolCallId,\n    result,\n  }: {\n    toolCallId: string;\n    result: any;\n  }) => void;\n\n  /** The id of the chat */\n  chatId: string;\n};\n\n// @ts-expect-error - some issues with the default export of useSWRV\nconst useSWRV = (swrv.default as (typeof import('swrv'))['default']) || swrv;\nconst store: Record<string, UIMessage<any>[] | undefined> = {};\nconst statusStore: Record<\n  string,\n  Ref<'submitted' | 'streaming' | 'ready' | 'error'>\n> = {};\n\nexport function useChat<MESSAGE_METADATA = unknown>(\n  {\n    api = '/api/chat',\n    chatId,\n    initialMessages = [],\n    initialInput = '',\n    streamProtocol = 'ui-message',\n    onFinish,\n    onError,\n    credentials,\n    headers: metadataHeaders,\n    body: metadataBody,\n    generateId = generateIdFunc,\n    onToolCall,\n    fetch,\n    maxSteps = 1,\n    experimental_prepareRequestBody,\n    messageMetadataSchema,\n  }: OriginalUseChatOptions<MESSAGE_METADATA> & {\n    /**\n     * Experimental (Vue only). When a function is provided, it will be used\n     * to prepare the request body for the chat API. This can be useful for\n     * customizing the request body based on the messages and data in the chat.\n     *\n     * @param id The chat ID\n     * @param messages The current messages in the chat\n     * @param requestData The data object passed in the chat request\n     * @param requestBody The request body object passed in the chat request\n     */\n    experimental_prepareRequestBody?: (options: {\n      chatId: string;\n      messages: UIMessage<MESSAGE_METADATA>[];\n      requestData?: JSONValue;\n      requestBody?: object;\n    }) => unknown;\n  } = {\n    maxSteps: 1,\n  },\n): UseChatHelpers<MESSAGE_METADATA> {\n  // Generate a unique ID for the chat if not provided.\n  const stableChatId = chatId ?? generateId();\n\n  const key = `${api}|${stableChatId}`;\n  const { data: messagesData, mutate: originalMutate } = useSWRV<\n    UIMessage<MESSAGE_METADATA>[]\n  >(key, () => store[key] ?? initialMessages);\n\n  const status =\n    statusStore[stableChatId] ??\n    (statusStore[stableChatId] = ref<\n      'submitted' | 'streaming' | 'ready' | 'error'\n    >('ready'));\n\n  // Force the `data` to be `initialMessages` if it's `undefined`.\n  messagesData.value ??= initialMessages;\n\n  const mutate = (data?: UIMessage<MESSAGE_METADATA>[]) => {\n    store[key] = data;\n    return originalMutate();\n  };\n\n  // Because of the `initialData` option, the `data` will never be `undefined`.\n  const messages = messagesData as Ref<UIMessage<MESSAGE_METADATA>[]>;\n\n  const error = ref<undefined | Error>(undefined);\n\n  let abortController: AbortController | null = null;\n\n  async function triggerRequest(\n    messagesSnapshot: UIMessage<MESSAGE_METADATA>[],\n    { headers, body }: ChatRequestOptions = {},\n  ) {\n    error.value = undefined;\n    status.value = 'submitted';\n\n    const messageCount = messages.value.length;\n    const lastMessage = messages.value.at(-1);\n    const maxStep =\n      lastMessage != null\n        ? extractMaxToolInvocationStep(getToolInvocations(lastMessage))\n        : 0;\n\n    try {\n      abortController = new AbortController();\n\n      // Do an optimistic update to show the updated messages immediately:\n      mutate(messagesSnapshot);\n\n      await callChatApi({\n        api,\n        body: experimental_prepareRequestBody?.({\n          chatId: stableChatId,\n          messages: messagesSnapshot,\n          requestBody: body,\n        }) ?? {\n          chatId: stableChatId,\n          messages: messagesSnapshot,\n          ...unref(metadataBody), // Use unref to unwrap the ref value\n          ...body,\n        },\n        streamProtocol,\n        headers: {\n          ...metadataHeaders,\n          ...headers,\n        },\n        abortController: () => abortController,\n        credentials,\n        onUpdate({ message }) {\n          status.value = 'streaming';\n\n          const replaceLastMessage =\n            message.id === messagesSnapshot[messagesSnapshot.length - 1].id;\n\n          mutate([\n            ...(replaceLastMessage\n              ? messagesSnapshot.slice(0, messagesSnapshot.length - 1)\n              : messagesSnapshot),\n            message,\n          ]);\n        },\n        onFinish,\n        generateId,\n        onToolCall,\n        fetch,\n        // enabled use of structured clone in processChatResponse:\n        lastMessage: recursiveToRaw(\n          messagesSnapshot[messagesSnapshot.length - 1],\n        ),\n        messageMetadataSchema,\n      });\n\n      status.value = 'ready';\n    } catch (err) {\n      // Ignore abort errors as they are expected.\n      if ((err as any).name === 'AbortError') {\n        abortController = null;\n        status.value = 'ready';\n        return null;\n      }\n\n      if (onError && err instanceof Error) {\n        onError(err);\n      }\n\n      error.value = err as Error;\n      status.value = 'error';\n    } finally {\n      abortController = null;\n    }\n\n    // auto-submit when all tool calls in the last assistant message have results:\n    if (\n      shouldResubmitMessages({\n        originalMaxToolInvocationStep: maxStep,\n        originalMessageCount: messageCount,\n        maxSteps,\n        messages: messages.value,\n      })\n    ) {\n      await triggerRequest(messages.value);\n    }\n  }\n\n  const append: UseChatHelpers<MESSAGE_METADATA>['append'] = async (\n    message,\n    options,\n  ) => {\n    return triggerRequest(\n      messages.value.concat({\n        ...message,\n        id: message.id ?? generateId(),\n        parts: message.parts,\n      }),\n      options,\n    );\n  };\n\n  const reload: UseChatHelpers<MESSAGE_METADATA>['reload'] = async options => {\n    const messagesSnapshot = messages.value;\n    if (messagesSnapshot.length === 0) return null;\n\n    const lastMessage = messagesSnapshot[messagesSnapshot.length - 1];\n    if (lastMessage.role === 'assistant') {\n      return triggerRequest(messagesSnapshot.slice(0, -1), options);\n    }\n\n    return triggerRequest(messagesSnapshot, options);\n  };\n\n  const stop = () => {\n    if (abortController) {\n      abortController.abort();\n      abortController = null;\n    }\n  };\n\n  const setMessages = (\n    messagesArg:\n      | UIMessage<MESSAGE_METADATA>[]\n      | ((\n          messages: UIMessage<MESSAGE_METADATA>[],\n        ) => UIMessage<MESSAGE_METADATA>[]),\n  ) => {\n    if (typeof messagesArg === 'function') {\n      messagesArg = messagesArg(messages.value);\n    }\n\n    mutate(messagesArg);\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = async (\n    event?: { preventDefault?: () => void },\n    options: ChatRequestOptions & { files?: FileList | FileUIPart[] } = {},\n  ) => {\n    event?.preventDefault?.();\n\n    const inputValue = input.value;\n\n    const fileParts = Array.isArray(options?.files)\n      ? options.files\n      : await convertFileListToFileUIParts(options?.files);\n\n    if (!inputValue && fileParts.length === 0) return;\n\n    triggerRequest(\n      messages.value.concat({\n        id: generateId(),\n        role: 'user',\n        parts: [...fileParts, { type: 'text', text: inputValue }],\n      }),\n      options,\n    );\n\n    input.value = '';\n  };\n\n  const addToolResult = ({\n    toolCallId,\n    result,\n  }: {\n    toolCallId: string;\n    result: unknown;\n  }) => {\n    const currentMessages = messages.value;\n\n    updateToolCallResult({\n      messages: currentMessages,\n      toolCallId,\n      toolResult: result,\n    });\n\n    mutate(currentMessages);\n\n    // when the request is ongoing, the auto-submit will be triggered after the request is finished\n    if (status.value === 'submitted' || status.value === 'streaming') {\n      return;\n    }\n\n    // auto-submit when all tool calls in the last assistant message have results:\n    const lastMessage = currentMessages[currentMessages.length - 1];\n    if (isAssistantMessageWithCompletedToolCalls(lastMessage)) {\n      triggerRequest(currentMessages);\n    }\n  };\n\n  return {\n    chatId: stableChatId,\n    messages,\n    append,\n    error,\n    reload,\n    stop,\n    setMessages,\n    input,\n    handleSubmit,\n    status: status as Ref<'submitted' | 'streaming' | 'ready' | 'error'>,\n    addToolResult,\n  };\n}\n\n// required for use of structured clone\nfunction recursiveToRaw<T>(inputValue: T): T {\n  if (Array.isArray(inputValue)) {\n    return [...inputValue.map(recursiveToRaw)] as T;\n  } else if (typeof inputValue === 'object' && inputValue !== null) {\n    const clone: any = {};\n    for (const [key, value] of Object.entries(inputValue)) {\n      clone[key] = recursiveToRaw(value);\n    }\n    return clone;\n  } else {\n    return inputValue;\n  }\n}\n","import type { CompletionRequestOptions, UseCompletionOptions } from 'ai';\nimport { callCompletionApi } from 'ai';\nimport swrv from 'swrv';\nimport type { Ref } from 'vue';\nimport { ref, unref } from 'vue';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n  /** The current completion result */\n  completion: Ref<string>;\n  /** The error object of the API request */\n  error: Ref<undefined | Error>;\n  /**\n   * Send a new prompt to the API endpoint and update the completion state.\n   */\n  complete: (\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Abort the current API request but keep the generated tokens.\n   */\n  stop: () => void;\n  /**\n   * Update the `completion` state locally.\n   */\n  setCompletion: (completion: string) => void;\n  /** The current value of the input */\n  input: Ref<string>;\n  /**\n   * Form submission handler to automatically reset input and append a user message\n   * @example\n   * ```jsx\n   * <form @submit=\"handleSubmit\">\n   *  <input @change=\"handleInputChange\" v-model=\"input\" />\n   * </form>\n   * ```\n   */\n  handleSubmit: (event?: { preventDefault?: () => void }) => void;\n  /** Whether the API request is in progress */\n  isLoading: Ref<boolean | undefined>;\n};\n\nlet uniqueId = 0;\n\n// @ts-expect-error - some issues with the default export of useSWRV\nconst useSWRV = (swrv.default as (typeof import('swrv'))['default']) || swrv;\nconst store: Record<string, any> = {};\n\nexport function useCompletion({\n  api = '/api/completion',\n  id,\n  initialCompletion = '',\n  initialInput = '',\n  credentials,\n  headers,\n  body,\n  streamProtocol,\n  onFinish,\n  onError,\n  fetch,\n}: UseCompletionOptions = {}): UseCompletionHelpers {\n  // Generate an unique id for the completion if not provided.\n  const completionId = id || `completion-${uniqueId++}`;\n\n  const key = `${api}|${completionId}`;\n  const { data, mutate: originalMutate } = useSWRV<string>(\n    key,\n    () => store[key] || initialCompletion,\n  );\n\n  const { data: isLoading, mutate: mutateLoading } = useSWRV<boolean>(\n    `${completionId}-loading`,\n    null,\n  );\n\n  isLoading.value ??= false;\n\n  // Force the `data` to be `initialCompletion` if it's `undefined`.\n  data.value ||= initialCompletion;\n\n  const mutate = (data: string) => {\n    store[key] = data;\n    return originalMutate();\n  };\n\n  // Because of the `initialData` option, the `data` will never be `undefined`.\n  const completion = data as Ref<string>;\n\n  const error = ref<undefined | Error>(undefined);\n\n  let abortController: AbortController | null = null;\n\n  async function triggerRequest(\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) {\n    return callCompletionApi({\n      api,\n      prompt,\n      credentials,\n      headers: {\n        ...headers,\n        ...options?.headers,\n      },\n      body: {\n        ...unref(body),\n        ...options?.body,\n      },\n      streamProtocol,\n      setCompletion: mutate,\n      setLoading: loading => mutateLoading(() => loading),\n      setError: err => {\n        error.value = err;\n      },\n      setAbortController: controller => {\n        abortController = controller;\n      },\n      onFinish,\n      onError,\n      fetch,\n    });\n  }\n\n  const complete: UseCompletionHelpers['complete'] = async (\n    prompt,\n    options,\n  ) => {\n    return triggerRequest(prompt, options);\n  };\n\n  const stop = () => {\n    if (abortController) {\n      abortController.abort();\n      abortController = null;\n    }\n  };\n\n  const setCompletion = (completion: string) => {\n    mutate(completion);\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = (event?: { preventDefault?: () => void }) => {\n    event?.preventDefault?.();\n    const inputValue = input.value;\n    return inputValue ? complete(inputValue) : undefined;\n  };\n\n  return {\n    completion,\n    complete,\n    error,\n    stop,\n    setCompletion,\n    input,\n    handleSubmit,\n    isLoading,\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACSA,gBASO;AACP,kBAAiB;AAEjB,iBAA2B;AA0E3B,IAAM,UAAW,YAAAA,QAAK,WAAkD,YAAAA;AACxE,IAAM,QAAsD,CAAC;AAC7D,IAAM,cAGF,CAAC;AAEE,SAAS,QACd;AAAA,EACE,MAAM;AAAA,EACN;AAAA,EACA,kBAAkB,CAAC;AAAA,EACnB,eAAe;AAAA,EACf,iBAAiB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAAS;AAAA,EACT,MAAM;AAAA,EACN,aAAa,UAAAC;AAAA,EACb;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA;AACF,IAiBI;AAAA,EACF,UAAU;AACZ,GACkC;AA5IpC;AA8IE,QAAM,eAAe,0BAAU,WAAW;AAE1C,QAAM,MAAM,GAAG,GAAG,IAAI,YAAY;AAClC,QAAM,EAAE,MAAM,cAAc,QAAQ,eAAe,IAAI,QAErD,KAAK,MAAG;AAnJZ,QAAAC;AAmJe,YAAAA,MAAA,MAAM,GAAG,MAAT,OAAAA,MAAc;AAAA,GAAe;AAE1C,QAAM,UACJ,iBAAY,YAAY,MAAxB,YACC,YAAY,YAAY,QAAI,gBAE3B,OAAO;AAGX,qBAAa,UAAb,yBAAa,QAAU;AAEvB,QAAM,SAAS,CAAC,SAAyC;AACvD,UAAM,GAAG,IAAI;AACb,WAAO,eAAe;AAAA,EACxB;AAGA,QAAM,WAAW;AAEjB,QAAM,YAAQ,gBAAuB,MAAS;AAE9C,MAAI,kBAA0C;AAE9C,iBAAe,eACb,kBACA,EAAE,SAAS,KAAK,IAAwB,CAAC,GACzC;AA7KJ,QAAAA;AA8KI,UAAM,QAAQ;AACd,WAAO,QAAQ;AAEf,UAAM,eAAe,SAAS,MAAM;AACpC,UAAM,cAAc,SAAS,MAAM,GAAG,EAAE;AACxC,UAAM,UACJ,eAAe,WACX,4CAA6B,8BAAmB,WAAW,CAAC,IAC5D;AAEN,QAAI;AACF,wBAAkB,IAAI,gBAAgB;AAGtC,aAAO,gBAAgB;AAEvB,gBAAM,uBAAY;AAAA,QAChB;AAAA,QACA,OAAMA,MAAA,mFAAkC;AAAA,UACtC,QAAQ;AAAA,UACR,UAAU;AAAA,UACV,aAAa;AAAA,QACf,OAJM,OAAAA,MAIA;AAAA,UACJ,QAAQ;AAAA,UACR,UAAU;AAAA,UACV,OAAG,kBAAM,YAAY;AAAA;AAAA,UACrB,GAAG;AAAA,QACL;AAAA,QACA;AAAA,QACA,SAAS;AAAA,UACP,GAAG;AAAA,UACH,GAAG;AAAA,QACL;AAAA,QACA,iBAAiB,MAAM;AAAA,QACvB;AAAA,QACA,SAAS,EAAE,QAAQ,GAAG;AACpB,iBAAO,QAAQ;AAEf,gBAAM,qBACJ,QAAQ,OAAO,iBAAiB,iBAAiB,SAAS,CAAC,EAAE;AAE/D,iBAAO;AAAA,YACL,GAAI,qBACA,iBAAiB,MAAM,GAAG,iBAAiB,SAAS,CAAC,IACrD;AAAA,YACJ;AAAA,UACF,CAAC;AAAA,QACH;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA;AAAA,QAEA,aAAa;AAAA,UACX,iBAAiB,iBAAiB,SAAS,CAAC;AAAA,QAC9C;AAAA,QACA;AAAA,MACF,CAAC;AAED,aAAO,QAAQ;AAAA,IACjB,SAAS,KAAK;AAEZ,UAAK,IAAY,SAAS,cAAc;AACtC,0BAAkB;AAClB,eAAO,QAAQ;AACf,eAAO;AAAA,MACT;AAEA,UAAI,WAAW,eAAe,OAAO;AACnC,gBAAQ,GAAG;AAAA,MACb;AAEA,YAAM,QAAQ;AACd,aAAO,QAAQ;AAAA,IACjB,UAAE;AACA,wBAAkB;AAAA,IACpB;AAGA,YACE,kCAAuB;AAAA,MACrB,+BAA+B;AAAA,MAC/B,sBAAsB;AAAA,MACtB;AAAA,MACA,UAAU,SAAS;AAAA,IACrB,CAAC,GACD;AACA,YAAM,eAAe,SAAS,KAAK;AAAA,IACrC;AAAA,EACF;AAEA,QAAM,SAAqD,OACzD,SACA,YACG;AA5QP,QAAAA;AA6QI,WAAO;AAAA,MACL,SAAS,MAAM,OAAO;AAAA,QACpB,GAAG;AAAA,QACH,KAAIA,MAAA,QAAQ,OAAR,OAAAA,MAAc,WAAW;AAAA,QAC7B,OAAO,QAAQ;AAAA,MACjB,CAAC;AAAA,MACD;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAqD,OAAM,YAAW;AAC1E,UAAM,mBAAmB,SAAS;AAClC,QAAI,iBAAiB,WAAW;AAAG,aAAO;AAE1C,UAAM,cAAc,iBAAiB,iBAAiB,SAAS,CAAC;AAChE,QAAI,YAAY,SAAS,aAAa;AACpC,aAAO,eAAe,iBAAiB,MAAM,GAAG,EAAE,GAAG,OAAO;AAAA,IAC9D;AAEA,WAAO,eAAe,kBAAkB,OAAO;AAAA,EACjD;AAEA,QAAM,OAAO,MAAM;AACjB,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,wBAAkB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,cAAc,CAClB,gBAKG;AACH,QAAI,OAAO,gBAAgB,YAAY;AACrC,oBAAc,YAAY,SAAS,KAAK;AAAA,IAC1C;AAEA,WAAO,WAAW;AAAA,EACpB;AAEA,QAAM,YAAQ,gBAAI,YAAY;AAE9B,QAAM,eAAe,OACnB,OACA,UAAoE,CAAC,MAClE;AA7TP,QAAAA;AA8TI,KAAAA,MAAA,+BAAO,mBAAP,gBAAAA,IAAA;AAEA,UAAM,aAAa,MAAM;AAEzB,UAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,UAAM,wCAA6B,mCAAS,KAAK;AAErD,QAAI,CAAC,cAAc,UAAU,WAAW;AAAG;AAE3C;AAAA,MACE,SAAS,MAAM,OAAO;AAAA,QACpB,IAAI,WAAW;AAAA,QACf,MAAM;AAAA,QACN,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,WAAW,CAAC;AAAA,MAC1D,CAAC;AAAA,MACD;AAAA,IACF;AAEA,UAAM,QAAQ;AAAA,EAChB;AAEA,QAAM,gBAAgB,CAAC;AAAA,IACrB;AAAA,IACA;AAAA,EACF,MAGM;AACJ,UAAM,kBAAkB,SAAS;AAEjC,wCAAqB;AAAA,MACnB,UAAU;AAAA,MACV;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,WAAO,eAAe;AAGtB,QAAI,OAAO,UAAU,eAAe,OAAO,UAAU,aAAa;AAChE;AAAA,IACF;AAGA,UAAM,cAAc,gBAAgB,gBAAgB,SAAS,CAAC;AAC9D,YAAI,oDAAyC,WAAW,GAAG;AACzD,qBAAe,eAAe;AAAA,IAChC;AAAA,EACF;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAGA,SAAS,eAAkB,YAAkB;AAC3C,MAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,WAAO,CAAC,GAAG,WAAW,IAAI,cAAc,CAAC;AAAA,EAC3C,WAAW,OAAO,eAAe,YAAY,eAAe,MAAM;AAChE,UAAM,QAAa,CAAC;AACpB,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,YAAM,GAAG,IAAI,eAAe,KAAK;AAAA,IACnC;AACA,WAAO;AAAA,EACT,OAAO;AACL,WAAO;AAAA,EACT;AACF;;;AC5YA,IAAAC,aAAkC;AAClC,IAAAC,eAAiB;AAEjB,IAAAC,cAA2B;AAwC3B,IAAI,WAAW;AAGf,IAAMC,WAAW,aAAAC,QAAK,WAAkD,aAAAA;AACxE,IAAMC,SAA6B,CAAC;AAE7B,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,IAA0B,CAAC,GAAyB;AA9DpD;AAgEE,QAAM,eAAe,MAAM,cAAc,UAAU;AAEnD,QAAM,MAAM,GAAG,GAAG,IAAI,YAAY;AAClC,QAAM,EAAE,MAAM,QAAQ,eAAe,IAAIF;AAAA,IACvC;AAAA,IACA,MAAME,OAAM,GAAG,KAAK;AAAA,EACtB;AAEA,QAAM,EAAE,MAAM,WAAW,QAAQ,cAAc,IAAIF;AAAA,IACjD,GAAG,YAAY;AAAA,IACf;AAAA,EACF;AAEA,kBAAU,UAAV,sBAAU,QAAU;AAGpB,OAAK,UAAL,KAAK,QAAU;AAEf,QAAM,SAAS,CAACG,UAAiB;AAC/B,IAAAD,OAAM,GAAG,IAAIC;AACb,WAAO,eAAe;AAAA,EACxB;AAGA,QAAM,aAAa;AAEnB,QAAM,YAAQ,iBAAuB,MAAS;AAE9C,MAAI,kBAA0C;AAE9C,iBAAe,eACb,QACA,SACA;AACA,eAAO,8BAAkB;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,GAAG;AAAA,QACH,GAAG,mCAAS;AAAA,MACd;AAAA,MACA,MAAM;AAAA,QACJ,OAAG,mBAAM,IAAI;AAAA,QACb,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf,YAAY,aAAW,cAAc,MAAM,OAAO;AAAA,MAClD,UAAU,SAAO;AACf,cAAM,QAAQ;AAAA,MAChB;AAAA,MACA,oBAAoB,gBAAc;AAChC,0BAAkB;AAAA,MACpB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,WAA6C,OACjD,QACA,YACG;AACH,WAAO,eAAe,QAAQ,OAAO;AAAA,EACvC;AAEA,QAAM,OAAO,MAAM;AACjB,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,wBAAkB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,gBAAgB,CAACC,gBAAuB;AAC5C,WAAOA,WAAU;AAAA,EACnB;AAEA,QAAM,YAAQ,iBAAI,YAAY;AAE9B,QAAM,eAAe,CAAC,UAA4C;AAjJpE,QAAAC;AAkJI,KAAAA,MAAA,+BAAO,mBAAP,gBAAAA,IAAA;AACA,UAAM,aAAa,MAAM;AACzB,WAAO,aAAa,SAAS,UAAU,IAAI;AAAA,EAC7C;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["swrv","generateIdFunc","_a","import_ai","import_swrv","import_vue","useSWRV","swrv","store","data","completion","_a"]}
+{"version":3,"sources":["../src/index.ts","../src/use-chat.ts","../src/chat-store.ts","../src/use-completion.ts"],"sourcesContent":["export * from './use-chat';\nexport * from './use-completion';\n","import type {\n  ChatRequestOptions,\n  ChatStatus,\n  ChatStore,\n  CreateUIMessage,\n  FileUIPart,\n  InferUIDataParts,\n  UIDataPartSchemas,\n  UIMessage,\n  UseChatOptions,\n} from 'ai';\nimport {\n  convertFileListToFileUIParts,\n  defaultChatStoreOptions,\n  generateId as generateIdFunc,\n} from 'ai';\nimport type { Ref } from 'vue';\nimport { computed, ref } from 'vue';\nimport { createChatStore } from './chat-store';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<\n  MESSAGE_METADATA = unknown,\n  DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> = {\n  /**\n   * The id of the chat.\n   */\n  readonly chatId: string;\n\n  /** Current messages in the chat */\n  readonly messages: Ref<\n    UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n  >;\n\n  /** The error object of the API request */\n  readonly error: Ref<Error | undefined>;\n\n  /**\n   * Append a user message to the chat list. This triggers the API call to fetch\n   * the assistant's response.\n   *\n   * @param message The message to append\n   * @param options Additional options to pass to the API call\n   */\n  append: (\n    message: CreateUIMessage<\n      MESSAGE_METADATA,\n      InferUIDataParts<DATA_PART_SCHEMAS>\n    >,\n    options?: ChatRequestOptions,\n  ) => Promise<void>;\n\n  /**\n   * Reload the last AI chat response for the given chat history. If the last\n   * message isn't from the assistant, it will request the API to generate a\n   * new response.\n   */\n  reload: (\n    chatRequestOptions?: ChatRequestOptions,\n  ) => Promise<string | null | undefined>;\n\n  /**\n   * Abort the current request immediately, keep the generated tokens if any.\n   */\n  stop: () => void;\n\n  /**\n   * Update the `messages` state locally. This is useful when you want to\n   * edit the messages on the client, and then trigger the `reload` method\n   * manually to regenerate the AI response.\n   */\n  setMessages: (\n    messages:\n      | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n      | ((\n          messages: UIMessage<\n            MESSAGE_METADATA,\n            InferUIDataParts<DATA_PART_SCHEMAS>\n          >[],\n        ) => UIMessage<\n          MESSAGE_METADATA,\n          InferUIDataParts<DATA_PART_SCHEMAS>\n        >[]),\n  ) => void;\n\n  /** The current value of the input */\n  input: Ref<string>;\n\n  /** Form submission handler to automatically reset input and append a user message  */\n  handleSubmit: (\n    event?: { preventDefault?: () => void },\n    chatRequestOptions?: ChatRequestOptions & {\n      files?: FileList | FileUIPart[];\n    },\n  ) => void;\n\n  /**\n   * Hook status:\n   *\n   * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n   * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n   * - `ready`: The full response has been received and processed; a new user message can be submitted.\n   * - `error`: An error occurred during the API request, preventing successful completion.\n   */\n  status: Ref<ChatStatus>;\n\n  addToolResult: ({\n    toolCallId,\n    result,\n  }: {\n    toolCallId: string;\n    result: any;\n  }) => void;\n};\n\nexport function useChat<\n  MESSAGE_METADATA = unknown,\n  DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>({\n  chatId,\n  initialInput = '',\n  onFinish,\n  onError,\n  generateId = generateIdFunc,\n  onToolCall,\n  chatStore: chatStoreArg,\n}: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> = {}): UseChatHelpers<\n  MESSAGE_METADATA,\n  DATA_PART_SCHEMAS\n> {\n  const stableChatId = chatId ?? generateId();\n\n  const chatStore =\n    chatStoreArg == null\n      ? createChatStore(\n          defaultChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n            api: '/api/chat',\n            generateId,\n          })(),\n        )\n      : typeof chatStoreArg === 'function'\n        ? createChatStore(chatStoreArg())\n        : chatStoreArg;\n\n  const messages = computed(() => chatStore.getMessages(stableChatId));\n  const status = computed(() => chatStore.getStatus(stableChatId));\n  const error = computed(() => chatStore.getError(stableChatId));\n\n  const append = async (\n    message: CreateUIMessage<\n      MESSAGE_METADATA,\n      InferUIDataParts<DATA_PART_SCHEMAS>\n    >,\n    { headers, body }: ChatRequestOptions = {},\n  ) =>\n    chatStore.submitMessage({\n      chatId: stableChatId,\n      message,\n      headers,\n      body,\n      onError,\n      onToolCall,\n      onFinish,\n    });\n\n  const reload = async ({ headers, body }: ChatRequestOptions = {}) =>\n    chatStore.resubmitLastUserMessage({\n      chatId: stableChatId,\n      headers,\n      body,\n      onError,\n      onToolCall,\n      onFinish,\n    });\n\n  const stop = () => {\n    chatStore.stopStream({ chatId: stableChatId });\n  };\n\n  const setMessages = (\n    messagesParam:\n      | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n      | ((\n          messages: UIMessage<\n            MESSAGE_METADATA,\n            InferUIDataParts<DATA_PART_SCHEMAS>\n          >[],\n        ) => UIMessage<\n          MESSAGE_METADATA,\n          InferUIDataParts<DATA_PART_SCHEMAS>\n        >[]),\n  ) => {\n    if (typeof messagesParam === 'function') {\n      messagesParam = messagesParam(chatStore.getMessages(stableChatId));\n    }\n\n    chatStore.setMessages({\n      id: stableChatId,\n      messages: messagesParam,\n    });\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = async (\n    event?: { preventDefault?: () => void },\n    options: ChatRequestOptions & { files?: FileList | FileUIPart[] } = {},\n  ) => {\n    event?.preventDefault?.();\n\n    const inputValue = input.value;\n\n    const fileParts = Array.isArray(options?.files)\n      ? options.files\n      : await convertFileListToFileUIParts(options?.files);\n\n    if (!inputValue && fileParts.length === 0) return;\n\n    await append(\n      {\n        id: generateId(),\n        role: 'user',\n        metadata: undefined,\n        parts: [...fileParts, { type: 'text', text: inputValue }],\n      },\n      {\n        headers: options.headers,\n        body: options.body,\n      },\n    );\n\n    input.value = '';\n  };\n\n  const addToolResult = (\n    options: Omit<\n      Parameters<\n        ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>['addToolResult']\n      >[0],\n      'chatId'\n    >,\n  ) => chatStore.addToolResult({ chatId: stableChatId, ...options });\n\n  return {\n    chatId: stableChatId,\n    messages,\n    append,\n    error,\n    reload,\n    stop,\n    setMessages,\n    input,\n    handleSubmit,\n    status,\n    addToolResult,\n  };\n}\n","import {\n  ChatStore,\n  SerialJobExecutor,\n  type ActiveResponse,\n  type Chat,\n  type ChatStatus,\n  type ChatStoreOptions,\n  type InferUIDataParts,\n  type UIDataPartSchemas,\n  type UIDataTypes,\n  type UIMessage,\n} from 'ai';\nimport { Ref, ref } from 'vue';\n\nclass VueChat<MESSAGE_METADATA, DATA_TYPES extends UIDataTypes>\n  implements Chat<MESSAGE_METADATA, DATA_TYPES>\n{\n  private messagesRef: Ref<UIMessage<MESSAGE_METADATA, DATA_TYPES>[]>;\n  private statusRef = ref<ChatStatus>('ready');\n  private errorRef = ref<Error | undefined>(undefined);\n\n  activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined = undefined;\n  jobExecutor = new SerialJobExecutor();\n\n  constructor(messages?: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) {\n    this.messagesRef = ref(messages ?? []) as Ref<\n      UIMessage<MESSAGE_METADATA, DATA_TYPES>[]\n    >;\n  }\n\n  get messages(): UIMessage<MESSAGE_METADATA, DATA_TYPES>[] {\n    return this.messagesRef.value;\n  }\n\n  get status(): ChatStatus {\n    return this.statusRef.value;\n  }\n\n  get error(): Error | undefined {\n    return this.errorRef.value;\n  }\n\n  setStatus = (status: ChatStatus) => {\n    this.statusRef.value = status;\n  };\n\n  setError = (error: Error | undefined) => {\n    this.errorRef.value = error;\n  };\n\n  setActiveResponse = (\n    activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined,\n  ) => {\n    this.activeResponse = activeResponse;\n  };\n\n  setMessages = (messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) => {\n    this.messagesRef.value = messages;\n  };\n\n  pushMessage = (message: UIMessage<MESSAGE_METADATA, DATA_TYPES>) => {\n    this.messagesRef.value.push(message);\n  };\n\n  popMessage = () => {\n    this.messagesRef.value.pop();\n  };\n\n  replaceMessage = (\n    index: number,\n    message: UIMessage<MESSAGE_METADATA, DATA_TYPES>,\n  ) => {\n    // message is cloned here because vue's deep reactivity shows unexpected behavior, particularly when updating tool invocation parts\n    this.messagesRef.value[index] = { ...message };\n  };\n}\n\nexport function createChatStore<\n  MESSAGE_METADATA = unknown,\n  DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>(\n  options: ChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>,\n): ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS> {\n  return new ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n    ...options,\n    createChat: options =>\n      new VueChat<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>(\n        options.messages,\n      ),\n  });\n}\n","import type { CompletionRequestOptions, UseCompletionOptions } from 'ai';\nimport { callCompletionApi } from 'ai';\nimport swrv from 'swrv';\nimport type { Ref } from 'vue';\nimport { ref, unref } from 'vue';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n  /** The current completion result */\n  completion: Ref<string>;\n  /** The error object of the API request */\n  error: Ref<undefined | Error>;\n  /**\n   * Send a new prompt to the API endpoint and update the completion state.\n   */\n  complete: (\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Abort the current API request but keep the generated tokens.\n   */\n  stop: () => void;\n  /**\n   * Update the `completion` state locally.\n   */\n  setCompletion: (completion: string) => void;\n  /** The current value of the input */\n  input: Ref<string>;\n  /**\n   * Form submission handler to automatically reset input and append a user message\n   * @example\n   * ```jsx\n   * <form @submit=\"handleSubmit\">\n   *  <input @change=\"handleInputChange\" v-model=\"input\" />\n   * </form>\n   * ```\n   */\n  handleSubmit: (event?: { preventDefault?: () => void }) => void;\n  /** Whether the API request is in progress */\n  isLoading: Ref<boolean | undefined>;\n};\n\nlet uniqueId = 0;\n\n// @ts-expect-error - some issues with the default export of useSWRV\nconst useSWRV = (swrv.default as (typeof import('swrv'))['default']) || swrv;\nconst store: Record<string, any> = {};\n\nexport function useCompletion({\n  api = '/api/completion',\n  id,\n  initialCompletion = '',\n  initialInput = '',\n  credentials,\n  headers,\n  body,\n  streamProtocol,\n  onFinish,\n  onError,\n  fetch,\n}: UseCompletionOptions = {}): UseCompletionHelpers {\n  // Generate an unique id for the completion if not provided.\n  const completionId = id || `completion-${uniqueId++}`;\n\n  const key = `${api}|${completionId}`;\n  const { data, mutate: originalMutate } = useSWRV<string>(\n    key,\n    () => store[key] || initialCompletion,\n  );\n\n  const { data: isLoading, mutate: mutateLoading } = useSWRV<boolean>(\n    `${completionId}-loading`,\n    null,\n  );\n\n  isLoading.value ??= false;\n\n  // Force the `data` to be `initialCompletion` if it's `undefined`.\n  data.value ||= initialCompletion;\n\n  const mutate = (data: string) => {\n    store[key] = data;\n    return originalMutate();\n  };\n\n  // Because of the `initialData` option, the `data` will never be `undefined`.\n  const completion = data as Ref<string>;\n\n  const error = ref<undefined | Error>(undefined);\n\n  let abortController: AbortController | null = null;\n\n  async function triggerRequest(\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) {\n    return callCompletionApi({\n      api,\n      prompt,\n      credentials,\n      headers: {\n        ...headers,\n        ...options?.headers,\n      },\n      body: {\n        ...unref(body),\n        ...options?.body,\n      },\n      streamProtocol,\n      setCompletion: mutate,\n      setLoading: loading => mutateLoading(() => loading),\n      setError: err => {\n        error.value = err;\n      },\n      setAbortController: controller => {\n        abortController = controller;\n      },\n      onFinish,\n      onError,\n      fetch,\n    });\n  }\n\n  const complete: UseCompletionHelpers['complete'] = async (\n    prompt,\n    options,\n  ) => {\n    return triggerRequest(prompt, options);\n  };\n\n  const stop = () => {\n    if (abortController) {\n      abortController.abort();\n      abortController = null;\n    }\n  };\n\n  const setCompletion = (completion: string) => {\n    mutate(completion);\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = (event?: { preventDefault?: () => void }) => {\n    event?.preventDefault?.();\n    const inputValue = input.value;\n    return inputValue ? complete(inputValue) : undefined;\n  };\n\n  return {\n    completion,\n    complete,\n    error,\n    stop,\n    setCompletion,\n    input,\n    handleSubmit,\n    isLoading,\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACWA,IAAAA,aAIO;AAEP,IAAAC,cAA8B;;;ACjB9B,gBAWO;AACP,iBAAyB;AAEzB,IAAM,UAAN,MAEA;AAAA,EAQE,YAAY,UAAsD;AANlE,SAAQ,gBAAY,gBAAgB,OAAO;AAC3C,SAAQ,eAAW,gBAAuB,MAAS;AAEnD,0BAA+D;AAC/D,uBAAc,IAAI,4BAAkB;AAoBpC,qBAAY,CAAC,WAAuB;AAClC,WAAK,UAAU,QAAQ;AAAA,IACzB;AAEA,oBAAW,CAAC,UAA6B;AACvC,WAAK,SAAS,QAAQ;AAAA,IACxB;AAEA,6BAAoB,CAClB,mBACG;AACH,WAAK,iBAAiB;AAAA,IACxB;AAEA,uBAAc,CAAC,aAAwD;AACrE,WAAK,YAAY,QAAQ;AAAA,IAC3B;AAEA,uBAAc,CAAC,YAAqD;AAClE,WAAK,YAAY,MAAM,KAAK,OAAO;AAAA,IACrC;AAEA,sBAAa,MAAM;AACjB,WAAK,YAAY,MAAM,IAAI;AAAA,IAC7B;AAEA,0BAAiB,CACf,OACA,YACG;AAEH,WAAK,YAAY,MAAM,KAAK,IAAI,EAAE,GAAG,QAAQ;AAAA,IAC/C;AAjDE,SAAK,kBAAc,gBAAI,8BAAY,CAAC,CAAC;AAAA,EAGvC;AAAA,EAEA,IAAI,WAAsD;AACxD,WAAO,KAAK,YAAY;AAAA,EAC1B;AAAA,EAEA,IAAI,SAAqB;AACvB,WAAO,KAAK,UAAU;AAAA,EACxB;AAAA,EAEA,IAAI,QAA2B;AAC7B,WAAO,KAAK,SAAS;AAAA,EACvB;AAmCF;AAEO,SAAS,gBAId,SACgD;AAChD,SAAO,IAAI,oBAA+C;AAAA,IACxD,GAAG;AAAA,IACH,YAAY,CAAAC,aACV,IAAI;AAAA,MACFA,SAAQ;AAAA,IACV;AAAA,EACJ,CAAC;AACH;;;AD2BO,SAAS,QAGd;AAAA,EACA;AAAA,EACA,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA,aAAa,WAAAC;AAAA,EACb;AAAA,EACA,WAAW;AACb,IAAyD,CAAC,GAGxD;AACA,QAAM,eAAe,0BAAU,WAAW;AAE1C,QAAM,YACJ,gBAAgB,OACZ;AAAA,QACE,oCAA6D;AAAA,MAC3D,KAAK;AAAA,MACL;AAAA,IACF,CAAC,EAAE;AAAA,EACL,IACA,OAAO,iBAAiB,aACtB,gBAAgB,aAAa,CAAC,IAC9B;AAER,QAAM,eAAW,sBAAS,MAAM,UAAU,YAAY,YAAY,CAAC;AACnE,QAAM,aAAS,sBAAS,MAAM,UAAU,UAAU,YAAY,CAAC;AAC/D,QAAM,YAAQ,sBAAS,MAAM,UAAU,SAAS,YAAY,CAAC;AAE7D,QAAM,SAAS,OACb,SAIA,EAAE,SAAS,KAAK,IAAwB,CAAC,MAEzC,UAAU,cAAc;AAAA,IACtB,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAEH,QAAM,SAAS,OAAO,EAAE,SAAS,KAAK,IAAwB,CAAC,MAC7D,UAAU,wBAAwB;AAAA,IAChC,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAEH,QAAM,OAAO,MAAM;AACjB,cAAU,WAAW,EAAE,QAAQ,aAAa,CAAC;AAAA,EAC/C;AAEA,QAAM,cAAc,CAClB,kBAWG;AACH,QAAI,OAAO,kBAAkB,YAAY;AACvC,sBAAgB,cAAc,UAAU,YAAY,YAAY,CAAC;AAAA,IACnE;AAEA,cAAU,YAAY;AAAA,MACpB,IAAI;AAAA,MACJ,UAAU;AAAA,IACZ,CAAC;AAAA,EACH;AAEA,QAAM,YAAQ,iBAAI,YAAY;AAE9B,QAAM,eAAe,OACnB,OACA,UAAoE,CAAC,MAClE;AAjNP;AAkNI,yCAAO,mBAAP;AAEA,UAAM,aAAa,MAAM;AAEzB,UAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,UAAM,yCAA6B,mCAAS,KAAK;AAErD,QAAI,CAAC,cAAc,UAAU,WAAW;AAAG;AAE3C,UAAM;AAAA,MACJ;AAAA,QACE,IAAI,WAAW;AAAA,QACf,MAAM;AAAA,QACN,UAAU;AAAA,QACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,WAAW,CAAC;AAAA,MAC1D;AAAA,MACA;AAAA,QACE,SAAS,QAAQ;AAAA,QACjB,MAAM,QAAQ;AAAA,MAChB;AAAA,IACF;AAEA,UAAM,QAAQ;AAAA,EAChB;AAEA,QAAM,gBAAgB,CACpB,YAMG,UAAU,cAAc,EAAE,QAAQ,cAAc,GAAG,QAAQ,CAAC;AAEjE,SAAO;AAAA,IACL,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AEjQA,IAAAC,aAAkC;AAClC,kBAAiB;AAEjB,IAAAC,cAA2B;AAwC3B,IAAI,WAAW;AAGf,IAAM,UAAW,YAAAC,QAAK,WAAkD,YAAAA;AACxE,IAAM,QAA6B,CAAC;AAE7B,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,IAA0B,CAAC,GAAyB;AA9DpD;AAgEE,QAAM,eAAe,MAAM,cAAc,UAAU;AAEnD,QAAM,MAAM,GAAG,GAAG,IAAI,YAAY;AAClC,QAAM,EAAE,MAAM,QAAQ,eAAe,IAAI;AAAA,IACvC;AAAA,IACA,MAAM,MAAM,GAAG,KAAK;AAAA,EACtB;AAEA,QAAM,EAAE,MAAM,WAAW,QAAQ,cAAc,IAAI;AAAA,IACjD,GAAG,YAAY;AAAA,IACf;AAAA,EACF;AAEA,kBAAU,UAAV,sBAAU,QAAU;AAGpB,OAAK,UAAL,KAAK,QAAU;AAEf,QAAM,SAAS,CAACC,UAAiB;AAC/B,UAAM,GAAG,IAAIA;AACb,WAAO,eAAe;AAAA,EACxB;AAGA,QAAM,aAAa;AAEnB,QAAM,YAAQ,iBAAuB,MAAS;AAE9C,MAAI,kBAA0C;AAE9C,iBAAe,eACb,QACA,SACA;AACA,eAAO,8BAAkB;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,GAAG;AAAA,QACH,GAAG,mCAAS;AAAA,MACd;AAAA,MACA,MAAM;AAAA,QACJ,OAAG,mBAAM,IAAI;AAAA,QACb,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf,YAAY,aAAW,cAAc,MAAM,OAAO;AAAA,MAClD,UAAU,SAAO;AACf,cAAM,QAAQ;AAAA,MAChB;AAAA,MACA,oBAAoB,gBAAc;AAChC,0BAAkB;AAAA,MACpB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,WAA6C,OACjD,QACA,YACG;AACH,WAAO,eAAe,QAAQ,OAAO;AAAA,EACvC;AAEA,QAAM,OAAO,MAAM;AACjB,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,wBAAkB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,gBAAgB,CAACC,gBAAuB;AAC5C,WAAOA,WAAU;AAAA,EACnB;AAEA,QAAM,YAAQ,iBAAI,YAAY;AAE9B,QAAM,eAAe,CAAC,UAA4C;AAjJpE,QAAAC;AAkJI,KAAAA,MAAA,+BAAO,mBAAP,gBAAAA,IAAA;AACA,UAAM,aAAa,MAAM;AACzB,WAAO,aAAa,SAAS,UAAU,IAAI;AAAA,EAC7C;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["import_ai","import_vue","options","generateIdFunc","import_ai","import_vue","swrv","data","completion","_a"]}

package/dist/index.mjs

@@ -1,197 +1,137 @@
 // src/use-chat.ts
 import {
-  callChatApi,
   convertFileListToFileUIParts,
-  extractMaxToolInvocationStep,
-  generateId as generateIdFunc,
-  getToolInvocations,
-  isAssistantMessageWithCompletedToolCalls,
-  shouldResubmitMessages,
-  updateToolCallResult
+  defaultChatStoreOptions,
+  generateId as generateIdFunc
 } from "ai";
-import swrv from "swrv";
-import { ref, unref } from "vue";
-var useSWRV = swrv.default || swrv;
-var store = {};
-var statusStore = {};
+import { computed, ref as ref2 } from "vue";
+
+// src/chat-store.ts
+import {
+  ChatStore,
+  SerialJobExecutor
+} from "ai";
+import { ref } from "vue";
+var VueChat = class {
+  constructor(messages) {
+    this.statusRef = ref("ready");
+    this.errorRef = ref(void 0);
+    this.activeResponse = void 0;
+    this.jobExecutor = new SerialJobExecutor();
+    this.setStatus = (status) => {
+      this.statusRef.value = status;
+    };
+    this.setError = (error) => {
+      this.errorRef.value = error;
+    };
+    this.setActiveResponse = (activeResponse) => {
+      this.activeResponse = activeResponse;
+    };
+    this.setMessages = (messages) => {
+      this.messagesRef.value = messages;
+    };
+    this.pushMessage = (message) => {
+      this.messagesRef.value.push(message);
+    };
+    this.popMessage = () => {
+      this.messagesRef.value.pop();
+    };
+    this.replaceMessage = (index, message) => {
+      this.messagesRef.value[index] = { ...message };
+    };
+    this.messagesRef = ref(messages != null ? messages : []);
+  }
+  get messages() {
+    return this.messagesRef.value;
+  }
+  get status() {
+    return this.statusRef.value;
+  }
+  get error() {
+    return this.errorRef.value;
+  }
+};
+function createChatStore(options) {
+  return new ChatStore({
+    ...options,
+    createChat: (options2) => new VueChat(
+      options2.messages
+    )
+  });
+}
+
+// src/use-chat.ts
 function useChat({
-  api = "/api/chat",
   chatId,
-  initialMessages = [],
   initialInput = "",
-  streamProtocol = "ui-message",
   onFinish,
   onError,
-  credentials,
-  headers: metadataHeaders,
-  body: metadataBody,
   generateId = generateIdFunc,
   onToolCall,
-  fetch,
-  maxSteps = 1,
-  experimental_prepareRequestBody,
-  messageMetadataSchema
-} = {
-  maxSteps: 1
-}) {
-  var _a, _b;
+  chatStore: chatStoreArg
+} = {}) {
   const stableChatId = chatId != null ? chatId : generateId();
-  const key = `${api}|${stableChatId}`;
-  const { data: messagesData, mutate: originalMutate } = useSWRV(key, () => {
-    var _a2;
-    return (_a2 = store[key]) != null ? _a2 : initialMessages;
+  const chatStore = chatStoreArg == null ? createChatStore(
+    defaultChatStoreOptions({
+      api: "/api/chat",
+      generateId
+    })()
+  ) : typeof chatStoreArg === "function" ? createChatStore(chatStoreArg()) : chatStoreArg;
+  const messages = computed(() => chatStore.getMessages(stableChatId));
+  const status = computed(() => chatStore.getStatus(stableChatId));
+  const error = computed(() => chatStore.getError(stableChatId));
+  const append = async (message, { headers, body } = {}) => chatStore.submitMessage({
+    chatId: stableChatId,
+    message,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
+  });
+  const reload = async ({ headers, body } = {}) => chatStore.resubmitLastUserMessage({
+    chatId: stableChatId,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
   });
-  const status = (_a = statusStore[stableChatId]) != null ? _a : statusStore[stableChatId] = ref("ready");
-  (_b = messagesData.value) != null ? _b : messagesData.value = initialMessages;
-  const mutate = (data) => {
-    store[key] = data;
-    return originalMutate();
-  };
-  const messages = messagesData;
-  const error = ref(void 0);
-  let abortController = null;
-  async function triggerRequest(messagesSnapshot, { headers, body } = {}) {
-    var _a2;
-    error.value = void 0;
-    status.value = "submitted";
-    const messageCount = messages.value.length;
-    const lastMessage = messages.value.at(-1);
-    const maxStep = lastMessage != null ? extractMaxToolInvocationStep(getToolInvocations(lastMessage)) : 0;
-    try {
-      abortController = new AbortController();
-      mutate(messagesSnapshot);
-      await callChatApi({
-        api,
-        body: (_a2 = experimental_prepareRequestBody == null ? void 0 : experimental_prepareRequestBody({
-          chatId: stableChatId,
-          messages: messagesSnapshot,
-          requestBody: body
-        })) != null ? _a2 : {
-          chatId: stableChatId,
-          messages: messagesSnapshot,
-          ...unref(metadataBody),
-          // Use unref to unwrap the ref value
-          ...body
-        },
-        streamProtocol,
-        headers: {
-          ...metadataHeaders,
-          ...headers
-        },
-        abortController: () => abortController,
-        credentials,
-        onUpdate({ message }) {
-          status.value = "streaming";
-          const replaceLastMessage = message.id === messagesSnapshot[messagesSnapshot.length - 1].id;
-          mutate([
-            ...replaceLastMessage ? messagesSnapshot.slice(0, messagesSnapshot.length - 1) : messagesSnapshot,
-            message
-          ]);
-        },
-        onFinish,
-        generateId,
-        onToolCall,
-        fetch,
-        // enabled use of structured clone in processChatResponse:
-        lastMessage: recursiveToRaw(
-          messagesSnapshot[messagesSnapshot.length - 1]
-        ),
-        messageMetadataSchema
-      });
-      status.value = "ready";
-    } catch (err) {
-      if (err.name === "AbortError") {
-        abortController = null;
-        status.value = "ready";
-        return null;
-      }
-      if (onError && err instanceof Error) {
-        onError(err);
-      }
-      error.value = err;
-      status.value = "error";
-    } finally {
-      abortController = null;
-    }
-    if (shouldResubmitMessages({
-      originalMaxToolInvocationStep: maxStep,
-      originalMessageCount: messageCount,
-      maxSteps,
-      messages: messages.value
-    })) {
-      await triggerRequest(messages.value);
-    }
-  }
-  const append = async (message, options) => {
-    var _a2;
-    return triggerRequest(
-      messages.value.concat({
-        ...message,
-        id: (_a2 = message.id) != null ? _a2 : generateId(),
-        parts: message.parts
-      }),
-      options
-    );
-  };
-  const reload = async (options) => {
-    const messagesSnapshot = messages.value;
-    if (messagesSnapshot.length === 0)
-      return null;
-    const lastMessage = messagesSnapshot[messagesSnapshot.length - 1];
-    if (lastMessage.role === "assistant") {
-      return triggerRequest(messagesSnapshot.slice(0, -1), options);
-    }
-    return triggerRequest(messagesSnapshot, options);
-  };
   const stop = () => {
-    if (abortController) {
-      abortController.abort();
-      abortController = null;
-    }
+    chatStore.stopStream({ chatId: stableChatId });
   };
-  const setMessages = (messagesArg) => {
-    if (typeof messagesArg === "function") {
-      messagesArg = messagesArg(messages.value);
+  const setMessages = (messagesParam) => {
+    if (typeof messagesParam === "function") {
+      messagesParam = messagesParam(chatStore.getMessages(stableChatId));
     }
-    mutate(messagesArg);
+    chatStore.setMessages({
+      id: stableChatId,
+      messages: messagesParam
+    });
   };
-  const input = ref(initialInput);
+  const input = ref2(initialInput);
   const handleSubmit = async (event, options = {}) => {
-    var _a2;
-    (_a2 = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a2.call(event);
+    var _a;
+    (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
     const inputValue = input.value;
     const fileParts = Array.isArray(options == null ? void 0 : options.files) ? options.files : await convertFileListToFileUIParts(options == null ? void 0 : options.files);
     if (!inputValue && fileParts.length === 0)
       return;
-    triggerRequest(
-      messages.value.concat({
+    await append(
+      {
         id: generateId(),
         role: "user",
+        metadata: void 0,
         parts: [...fileParts, { type: "text", text: inputValue }]
-      }),
-      options
+      },
+      {
+        headers: options.headers,
+        body: options.body
+      }
     );
     input.value = "";
   };
-  const addToolResult = ({
-    toolCallId,
-    result
-  }) => {
-    const currentMessages = messages.value;
-    updateToolCallResult({
-      messages: currentMessages,
-      toolCallId,
-      toolResult: result
-    });
-    mutate(currentMessages);
-    if (status.value === "submitted" || status.value === "streaming") {
-      return;
-    }
-    const lastMessage = currentMessages[currentMessages.length - 1];
-    if (isAssistantMessageWithCompletedToolCalls(lastMessage)) {
-      triggerRequest(currentMessages);
-    }
-  };
+  const addToolResult = (options) => chatStore.addToolResult({ chatId: stableChatId, ...options });
   return {
     chatId: stableChatId,
     messages,
@@ -206,27 +146,14 @@ function useChat({
     addToolResult
   };
 }
-function recursiveToRaw(inputValue) {
-  if (Array.isArray(inputValue)) {
-    return [...inputValue.map(recursiveToRaw)];
-  } else if (typeof inputValue === "object" && inputValue !== null) {
-    const clone = {};
-    for (const [key, value] of Object.entries(inputValue)) {
-      clone[key] = recursiveToRaw(value);
-    }
-    return clone;
-  } else {
-    return inputValue;
-  }
-}
 
 // src/use-completion.ts
 import { callCompletionApi } from "ai";
-import swrv2 from "swrv";
-import { ref as ref2, unref as unref2 } from "vue";
+import swrv from "swrv";
+import { ref as ref3, unref } from "vue";
 var uniqueId = 0;
-var useSWRV2 = swrv2.default || swrv2;
-var store2 = {};
+var useSWRV = swrv.default || swrv;
+var store = {};
 function useCompletion({
   api = "/api/completion",
   id,
@@ -243,22 +170,22 @@ function useCompletion({
   var _a;
   const completionId = id || `completion-${uniqueId++}`;
   const key = `${api}|${completionId}`;
-  const { data, mutate: originalMutate } = useSWRV2(
+  const { data, mutate: originalMutate } = useSWRV(
     key,
-    () => store2[key] || initialCompletion
+    () => store[key] || initialCompletion
   );
-  const { data: isLoading, mutate: mutateLoading } = useSWRV2(
+  const { data: isLoading, mutate: mutateLoading } = useSWRV(
     `${completionId}-loading`,
     null
   );
   (_a = isLoading.value) != null ? _a : isLoading.value = false;
   data.value || (data.value = initialCompletion);
   const mutate = (data2) => {
-    store2[key] = data2;
+    store[key] = data2;
     return originalMutate();
   };
   const completion = data;
-  const error = ref2(void 0);
+  const error = ref3(void 0);
   let abortController = null;
   async function triggerRequest(prompt, options) {
     return callCompletionApi({
@@ -270,7 +197,7 @@ function useCompletion({
         ...options == null ? void 0 : options.headers
       },
       body: {
-        ...unref2(body),
+        ...unref(body),
         ...options == null ? void 0 : options.body
       },
       streamProtocol,
@@ -299,7 +226,7 @@ function useCompletion({
   const setCompletion = (completion2) => {
     mutate(completion2);
   };
-  const input = ref2(initialInput);
+  const input = ref3(initialInput);
   const handleSubmit = (event) => {
     var _a2;
     (_a2 = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a2.call(event);

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/use-chat.ts","../src/use-completion.ts"],"sourcesContent":["import type {\n  ChatRequestOptions,\n  CreateUIMessage,\n  FileUIPart,\n  JSONValue,\n  OriginalUseChatOptions,\n  UIMessage,\n  UseChatOptions,\n} from 'ai';\nimport {\n  callChatApi,\n  convertFileListToFileUIParts,\n  extractMaxToolInvocationStep,\n  generateId as generateIdFunc,\n  getToolInvocations,\n  isAssistantMessageWithCompletedToolCalls,\n  shouldResubmitMessages,\n  updateToolCallResult,\n} from 'ai';\nimport swrv from 'swrv';\nimport type { Ref } from 'vue';\nimport { ref, unref } from 'vue';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<MESSAGE_METADATA> = {\n  /** Current messages in the chat */\n  messages: Ref<UIMessage<MESSAGE_METADATA>[]>;\n  /** The error object of the API request */\n  error: Ref<undefined | Error>;\n  /**\n   * Append a user message to the chat list. This triggers the API call to fetch\n   * the assistant's response.\n   */\n  append: (\n    message: UIMessage<MESSAGE_METADATA> | CreateUIMessage<MESSAGE_METADATA>,\n    chatRequestOptions?: ChatRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Reload the last AI chat response for the given chat history. If the last\n   * message isn't from the assistant, it will request the API to generate a\n   * new response.\n   */\n  reload: (\n    chatRequestOptions?: ChatRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Abort the current request immediately, keep the generated tokens if any.\n   */\n  stop: () => void;\n  /**\n   * Update the `messages` state locally. This is useful when you want to\n   * edit the messages on the client, and then trigger the `reload` method\n   * manually to regenerate the AI response.\n   */\n  setMessages: (\n    messages:\n      | UIMessage<MESSAGE_METADATA>[]\n      | ((\n          messages: UIMessage<MESSAGE_METADATA>[],\n        ) => UIMessage<MESSAGE_METADATA>[]),\n  ) => void;\n  /** The current value of the input */\n  input: Ref<string>;\n  /** Form submission handler to automatically reset input and append a user message  */\n  handleSubmit: (\n    event?: { preventDefault?: () => void },\n    chatRequestOptions?: ChatRequestOptions & {\n      files?: FileList | FileUIPart[];\n    },\n  ) => void;\n\n  /**\n   * Hook status:\n   *\n   * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n   * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n   * - `ready`: The full response has been received and processed; a new user message can be submitted.\n   * - `error`: An error occurred during the API request, preventing successful completion.\n   */\n  status: Ref<'submitted' | 'streaming' | 'ready' | 'error'>;\n\n  addToolResult: ({\n    toolCallId,\n    result,\n  }: {\n    toolCallId: string;\n    result: any;\n  }) => void;\n\n  /** The id of the chat */\n  chatId: string;\n};\n\n// @ts-expect-error - some issues with the default export of useSWRV\nconst useSWRV = (swrv.default as (typeof import('swrv'))['default']) || swrv;\nconst store: Record<string, UIMessage<any>[] | undefined> = {};\nconst statusStore: Record<\n  string,\n  Ref<'submitted' | 'streaming' | 'ready' | 'error'>\n> = {};\n\nexport function useChat<MESSAGE_METADATA = unknown>(\n  {\n    api = '/api/chat',\n    chatId,\n    initialMessages = [],\n    initialInput = '',\n    streamProtocol = 'ui-message',\n    onFinish,\n    onError,\n    credentials,\n    headers: metadataHeaders,\n    body: metadataBody,\n    generateId = generateIdFunc,\n    onToolCall,\n    fetch,\n    maxSteps = 1,\n    experimental_prepareRequestBody,\n    messageMetadataSchema,\n  }: OriginalUseChatOptions<MESSAGE_METADATA> & {\n    /**\n     * Experimental (Vue only). When a function is provided, it will be used\n     * to prepare the request body for the chat API. This can be useful for\n     * customizing the request body based on the messages and data in the chat.\n     *\n     * @param id The chat ID\n     * @param messages The current messages in the chat\n     * @param requestData The data object passed in the chat request\n     * @param requestBody The request body object passed in the chat request\n     */\n    experimental_prepareRequestBody?: (options: {\n      chatId: string;\n      messages: UIMessage<MESSAGE_METADATA>[];\n      requestData?: JSONValue;\n      requestBody?: object;\n    }) => unknown;\n  } = {\n    maxSteps: 1,\n  },\n): UseChatHelpers<MESSAGE_METADATA> {\n  // Generate a unique ID for the chat if not provided.\n  const stableChatId = chatId ?? generateId();\n\n  const key = `${api}|${stableChatId}`;\n  const { data: messagesData, mutate: originalMutate } = useSWRV<\n    UIMessage<MESSAGE_METADATA>[]\n  >(key, () => store[key] ?? initialMessages);\n\n  const status =\n    statusStore[stableChatId] ??\n    (statusStore[stableChatId] = ref<\n      'submitted' | 'streaming' | 'ready' | 'error'\n    >('ready'));\n\n  // Force the `data` to be `initialMessages` if it's `undefined`.\n  messagesData.value ??= initialMessages;\n\n  const mutate = (data?: UIMessage<MESSAGE_METADATA>[]) => {\n    store[key] = data;\n    return originalMutate();\n  };\n\n  // Because of the `initialData` option, the `data` will never be `undefined`.\n  const messages = messagesData as Ref<UIMessage<MESSAGE_METADATA>[]>;\n\n  const error = ref<undefined | Error>(undefined);\n\n  let abortController: AbortController | null = null;\n\n  async function triggerRequest(\n    messagesSnapshot: UIMessage<MESSAGE_METADATA>[],\n    { headers, body }: ChatRequestOptions = {},\n  ) {\n    error.value = undefined;\n    status.value = 'submitted';\n\n    const messageCount = messages.value.length;\n    const lastMessage = messages.value.at(-1);\n    const maxStep =\n      lastMessage != null\n        ? extractMaxToolInvocationStep(getToolInvocations(lastMessage))\n        : 0;\n\n    try {\n      abortController = new AbortController();\n\n      // Do an optimistic update to show the updated messages immediately:\n      mutate(messagesSnapshot);\n\n      await callChatApi({\n        api,\n        body: experimental_prepareRequestBody?.({\n          chatId: stableChatId,\n          messages: messagesSnapshot,\n          requestBody: body,\n        }) ?? {\n          chatId: stableChatId,\n          messages: messagesSnapshot,\n          ...unref(metadataBody), // Use unref to unwrap the ref value\n          ...body,\n        },\n        streamProtocol,\n        headers: {\n          ...metadataHeaders,\n          ...headers,\n        },\n        abortController: () => abortController,\n        credentials,\n        onUpdate({ message }) {\n          status.value = 'streaming';\n\n          const replaceLastMessage =\n            message.id === messagesSnapshot[messagesSnapshot.length - 1].id;\n\n          mutate([\n            ...(replaceLastMessage\n              ? messagesSnapshot.slice(0, messagesSnapshot.length - 1)\n              : messagesSnapshot),\n            message,\n          ]);\n        },\n        onFinish,\n        generateId,\n        onToolCall,\n        fetch,\n        // enabled use of structured clone in processChatResponse:\n        lastMessage: recursiveToRaw(\n          messagesSnapshot[messagesSnapshot.length - 1],\n        ),\n        messageMetadataSchema,\n      });\n\n      status.value = 'ready';\n    } catch (err) {\n      // Ignore abort errors as they are expected.\n      if ((err as any).name === 'AbortError') {\n        abortController = null;\n        status.value = 'ready';\n        return null;\n      }\n\n      if (onError && err instanceof Error) {\n        onError(err);\n      }\n\n      error.value = err as Error;\n      status.value = 'error';\n    } finally {\n      abortController = null;\n    }\n\n    // auto-submit when all tool calls in the last assistant message have results:\n    if (\n      shouldResubmitMessages({\n        originalMaxToolInvocationStep: maxStep,\n        originalMessageCount: messageCount,\n        maxSteps,\n        messages: messages.value,\n      })\n    ) {\n      await triggerRequest(messages.value);\n    }\n  }\n\n  const append: UseChatHelpers<MESSAGE_METADATA>['append'] = async (\n    message,\n    options,\n  ) => {\n    return triggerRequest(\n      messages.value.concat({\n        ...message,\n        id: message.id ?? generateId(),\n        parts: message.parts,\n      }),\n      options,\n    );\n  };\n\n  const reload: UseChatHelpers<MESSAGE_METADATA>['reload'] = async options => {\n    const messagesSnapshot = messages.value;\n    if (messagesSnapshot.length === 0) return null;\n\n    const lastMessage = messagesSnapshot[messagesSnapshot.length - 1];\n    if (lastMessage.role === 'assistant') {\n      return triggerRequest(messagesSnapshot.slice(0, -1), options);\n    }\n\n    return triggerRequest(messagesSnapshot, options);\n  };\n\n  const stop = () => {\n    if (abortController) {\n      abortController.abort();\n      abortController = null;\n    }\n  };\n\n  const setMessages = (\n    messagesArg:\n      | UIMessage<MESSAGE_METADATA>[]\n      | ((\n          messages: UIMessage<MESSAGE_METADATA>[],\n        ) => UIMessage<MESSAGE_METADATA>[]),\n  ) => {\n    if (typeof messagesArg === 'function') {\n      messagesArg = messagesArg(messages.value);\n    }\n\n    mutate(messagesArg);\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = async (\n    event?: { preventDefault?: () => void },\n    options: ChatRequestOptions & { files?: FileList | FileUIPart[] } = {},\n  ) => {\n    event?.preventDefault?.();\n\n    const inputValue = input.value;\n\n    const fileParts = Array.isArray(options?.files)\n      ? options.files\n      : await convertFileListToFileUIParts(options?.files);\n\n    if (!inputValue && fileParts.length === 0) return;\n\n    triggerRequest(\n      messages.value.concat({\n        id: generateId(),\n        role: 'user',\n        parts: [...fileParts, { type: 'text', text: inputValue }],\n      }),\n      options,\n    );\n\n    input.value = '';\n  };\n\n  const addToolResult = ({\n    toolCallId,\n    result,\n  }: {\n    toolCallId: string;\n    result: unknown;\n  }) => {\n    const currentMessages = messages.value;\n\n    updateToolCallResult({\n      messages: currentMessages,\n      toolCallId,\n      toolResult: result,\n    });\n\n    mutate(currentMessages);\n\n    // when the request is ongoing, the auto-submit will be triggered after the request is finished\n    if (status.value === 'submitted' || status.value === 'streaming') {\n      return;\n    }\n\n    // auto-submit when all tool calls in the last assistant message have results:\n    const lastMessage = currentMessages[currentMessages.length - 1];\n    if (isAssistantMessageWithCompletedToolCalls(lastMessage)) {\n      triggerRequest(currentMessages);\n    }\n  };\n\n  return {\n    chatId: stableChatId,\n    messages,\n    append,\n    error,\n    reload,\n    stop,\n    setMessages,\n    input,\n    handleSubmit,\n    status: status as Ref<'submitted' | 'streaming' | 'ready' | 'error'>,\n    addToolResult,\n  };\n}\n\n// required for use of structured clone\nfunction recursiveToRaw<T>(inputValue: T): T {\n  if (Array.isArray(inputValue)) {\n    return [...inputValue.map(recursiveToRaw)] as T;\n  } else if (typeof inputValue === 'object' && inputValue !== null) {\n    const clone: any = {};\n    for (const [key, value] of Object.entries(inputValue)) {\n      clone[key] = recursiveToRaw(value);\n    }\n    return clone;\n  } else {\n    return inputValue;\n  }\n}\n","import type { CompletionRequestOptions, UseCompletionOptions } from 'ai';\nimport { callCompletionApi } from 'ai';\nimport swrv from 'swrv';\nimport type { Ref } from 'vue';\nimport { ref, unref } from 'vue';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n  /** The current completion result */\n  completion: Ref<string>;\n  /** The error object of the API request */\n  error: Ref<undefined | Error>;\n  /**\n   * Send a new prompt to the API endpoint and update the completion state.\n   */\n  complete: (\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Abort the current API request but keep the generated tokens.\n   */\n  stop: () => void;\n  /**\n   * Update the `completion` state locally.\n   */\n  setCompletion: (completion: string) => void;\n  /** The current value of the input */\n  input: Ref<string>;\n  /**\n   * Form submission handler to automatically reset input and append a user message\n   * @example\n   * ```jsx\n   * <form @submit=\"handleSubmit\">\n   *  <input @change=\"handleInputChange\" v-model=\"input\" />\n   * </form>\n   * ```\n   */\n  handleSubmit: (event?: { preventDefault?: () => void }) => void;\n  /** Whether the API request is in progress */\n  isLoading: Ref<boolean | undefined>;\n};\n\nlet uniqueId = 0;\n\n// @ts-expect-error - some issues with the default export of useSWRV\nconst useSWRV = (swrv.default as (typeof import('swrv'))['default']) || swrv;\nconst store: Record<string, any> = {};\n\nexport function useCompletion({\n  api = '/api/completion',\n  id,\n  initialCompletion = '',\n  initialInput = '',\n  credentials,\n  headers,\n  body,\n  streamProtocol,\n  onFinish,\n  onError,\n  fetch,\n}: UseCompletionOptions = {}): UseCompletionHelpers {\n  // Generate an unique id for the completion if not provided.\n  const completionId = id || `completion-${uniqueId++}`;\n\n  const key = `${api}|${completionId}`;\n  const { data, mutate: originalMutate } = useSWRV<string>(\n    key,\n    () => store[key] || initialCompletion,\n  );\n\n  const { data: isLoading, mutate: mutateLoading } = useSWRV<boolean>(\n    `${completionId}-loading`,\n    null,\n  );\n\n  isLoading.value ??= false;\n\n  // Force the `data` to be `initialCompletion` if it's `undefined`.\n  data.value ||= initialCompletion;\n\n  const mutate = (data: string) => {\n    store[key] = data;\n    return originalMutate();\n  };\n\n  // Because of the `initialData` option, the `data` will never be `undefined`.\n  const completion = data as Ref<string>;\n\n  const error = ref<undefined | Error>(undefined);\n\n  let abortController: AbortController | null = null;\n\n  async function triggerRequest(\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) {\n    return callCompletionApi({\n      api,\n      prompt,\n      credentials,\n      headers: {\n        ...headers,\n        ...options?.headers,\n      },\n      body: {\n        ...unref(body),\n        ...options?.body,\n      },\n      streamProtocol,\n      setCompletion: mutate,\n      setLoading: loading => mutateLoading(() => loading),\n      setError: err => {\n        error.value = err;\n      },\n      setAbortController: controller => {\n        abortController = controller;\n      },\n      onFinish,\n      onError,\n      fetch,\n    });\n  }\n\n  const complete: UseCompletionHelpers['complete'] = async (\n    prompt,\n    options,\n  ) => {\n    return triggerRequest(prompt, options);\n  };\n\n  const stop = () => {\n    if (abortController) {\n      abortController.abort();\n      abortController = null;\n    }\n  };\n\n  const setCompletion = (completion: string) => {\n    mutate(completion);\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = (event?: { preventDefault?: () => void }) => {\n    event?.preventDefault?.();\n    const inputValue = input.value;\n    return inputValue ? complete(inputValue) : undefined;\n  };\n\n  return {\n    completion,\n    complete,\n    error,\n    stop,\n    setCompletion,\n    input,\n    handleSubmit,\n    isLoading,\n  };\n}\n"],"mappings":";AASA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA,cAAc;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,OAAO,UAAU;AAEjB,SAAS,KAAK,aAAa;AA0E3B,IAAM,UAAW,KAAK,WAAkD;AACxE,IAAM,QAAsD,CAAC;AAC7D,IAAM,cAGF,CAAC;AAEE,SAAS,QACd;AAAA,EACE,MAAM;AAAA,EACN;AAAA,EACA,kBAAkB,CAAC;AAAA,EACnB,eAAe;AAAA,EACf,iBAAiB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAAS;AAAA,EACT,MAAM;AAAA,EACN,aAAa;AAAA,EACb;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA;AACF,IAiBI;AAAA,EACF,UAAU;AACZ,GACkC;AA5IpC;AA8IE,QAAM,eAAe,0BAAU,WAAW;AAE1C,QAAM,MAAM,GAAG,GAAG,IAAI,YAAY;AAClC,QAAM,EAAE,MAAM,cAAc,QAAQ,eAAe,IAAI,QAErD,KAAK,MAAG;AAnJZ,QAAAA;AAmJe,YAAAA,MAAA,MAAM,GAAG,MAAT,OAAAA,MAAc;AAAA,GAAe;AAE1C,QAAM,UACJ,iBAAY,YAAY,MAAxB,YACC,YAAY,YAAY,IAAI,IAE3B,OAAO;AAGX,qBAAa,UAAb,yBAAa,QAAU;AAEvB,QAAM,SAAS,CAAC,SAAyC;AACvD,UAAM,GAAG,IAAI;AACb,WAAO,eAAe;AAAA,EACxB;AAGA,QAAM,WAAW;AAEjB,QAAM,QAAQ,IAAuB,MAAS;AAE9C,MAAI,kBAA0C;AAE9C,iBAAe,eACb,kBACA,EAAE,SAAS,KAAK,IAAwB,CAAC,GACzC;AA7KJ,QAAAA;AA8KI,UAAM,QAAQ;AACd,WAAO,QAAQ;AAEf,UAAM,eAAe,SAAS,MAAM;AACpC,UAAM,cAAc,SAAS,MAAM,GAAG,EAAE;AACxC,UAAM,UACJ,eAAe,OACX,6BAA6B,mBAAmB,WAAW,CAAC,IAC5D;AAEN,QAAI;AACF,wBAAkB,IAAI,gBAAgB;AAGtC,aAAO,gBAAgB;AAEvB,YAAM,YAAY;AAAA,QAChB;AAAA,QACA,OAAMA,MAAA,mFAAkC;AAAA,UACtC,QAAQ;AAAA,UACR,UAAU;AAAA,UACV,aAAa;AAAA,QACf,OAJM,OAAAA,MAIA;AAAA,UACJ,QAAQ;AAAA,UACR,UAAU;AAAA,UACV,GAAG,MAAM,YAAY;AAAA;AAAA,UACrB,GAAG;AAAA,QACL;AAAA,QACA;AAAA,QACA,SAAS;AAAA,UACP,GAAG;AAAA,UACH,GAAG;AAAA,QACL;AAAA,QACA,iBAAiB,MAAM;AAAA,QACvB;AAAA,QACA,SAAS,EAAE,QAAQ,GAAG;AACpB,iBAAO,QAAQ;AAEf,gBAAM,qBACJ,QAAQ,OAAO,iBAAiB,iBAAiB,SAAS,CAAC,EAAE;AAE/D,iBAAO;AAAA,YACL,GAAI,qBACA,iBAAiB,MAAM,GAAG,iBAAiB,SAAS,CAAC,IACrD;AAAA,YACJ;AAAA,UACF,CAAC;AAAA,QACH;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA;AAAA,QAEA,aAAa;AAAA,UACX,iBAAiB,iBAAiB,SAAS,CAAC;AAAA,QAC9C;AAAA,QACA;AAAA,MACF,CAAC;AAED,aAAO,QAAQ;AAAA,IACjB,SAAS,KAAK;AAEZ,UAAK,IAAY,SAAS,cAAc;AACtC,0BAAkB;AAClB,eAAO,QAAQ;AACf,eAAO;AAAA,MACT;AAEA,UAAI,WAAW,eAAe,OAAO;AACnC,gBAAQ,GAAG;AAAA,MACb;AAEA,YAAM,QAAQ;AACd,aAAO,QAAQ;AAAA,IACjB,UAAE;AACA,wBAAkB;AAAA,IACpB;AAGA,QACE,uBAAuB;AAAA,MACrB,+BAA+B;AAAA,MAC/B,sBAAsB;AAAA,MACtB;AAAA,MACA,UAAU,SAAS;AAAA,IACrB,CAAC,GACD;AACA,YAAM,eAAe,SAAS,KAAK;AAAA,IACrC;AAAA,EACF;AAEA,QAAM,SAAqD,OACzD,SACA,YACG;AA5QP,QAAAA;AA6QI,WAAO;AAAA,MACL,SAAS,MAAM,OAAO;AAAA,QACpB,GAAG;AAAA,QACH,KAAIA,MAAA,QAAQ,OAAR,OAAAA,MAAc,WAAW;AAAA,QAC7B,OAAO,QAAQ;AAAA,MACjB,CAAC;AAAA,MACD;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAqD,OAAM,YAAW;AAC1E,UAAM,mBAAmB,SAAS;AAClC,QAAI,iBAAiB,WAAW;AAAG,aAAO;AAE1C,UAAM,cAAc,iBAAiB,iBAAiB,SAAS,CAAC;AAChE,QAAI,YAAY,SAAS,aAAa;AACpC,aAAO,eAAe,iBAAiB,MAAM,GAAG,EAAE,GAAG,OAAO;AAAA,IAC9D;AAEA,WAAO,eAAe,kBAAkB,OAAO;AAAA,EACjD;AAEA,QAAM,OAAO,MAAM;AACjB,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,wBAAkB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,cAAc,CAClB,gBAKG;AACH,QAAI,OAAO,gBAAgB,YAAY;AACrC,oBAAc,YAAY,SAAS,KAAK;AAAA,IAC1C;AAEA,WAAO,WAAW;AAAA,EACpB;AAEA,QAAM,QAAQ,IAAI,YAAY;AAE9B,QAAM,eAAe,OACnB,OACA,UAAoE,CAAC,MAClE;AA7TP,QAAAA;AA8TI,KAAAA,MAAA,+BAAO,mBAAP,gBAAAA,IAAA;AAEA,UAAM,aAAa,MAAM;AAEzB,UAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,MAAM,6BAA6B,mCAAS,KAAK;AAErD,QAAI,CAAC,cAAc,UAAU,WAAW;AAAG;AAE3C;AAAA,MACE,SAAS,MAAM,OAAO;AAAA,QACpB,IAAI,WAAW;AAAA,QACf,MAAM;AAAA,QACN,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,WAAW,CAAC;AAAA,MAC1D,CAAC;AAAA,MACD;AAAA,IACF;AAEA,UAAM,QAAQ;AAAA,EAChB;AAEA,QAAM,gBAAgB,CAAC;AAAA,IACrB;AAAA,IACA;AAAA,EACF,MAGM;AACJ,UAAM,kBAAkB,SAAS;AAEjC,yBAAqB;AAAA,MACnB,UAAU;AAAA,MACV;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,WAAO,eAAe;AAGtB,QAAI,OAAO,UAAU,eAAe,OAAO,UAAU,aAAa;AAChE;AAAA,IACF;AAGA,UAAM,cAAc,gBAAgB,gBAAgB,SAAS,CAAC;AAC9D,QAAI,yCAAyC,WAAW,GAAG;AACzD,qBAAe,eAAe;AAAA,IAChC;AAAA,EACF;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAGA,SAAS,eAAkB,YAAkB;AAC3C,MAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,WAAO,CAAC,GAAG,WAAW,IAAI,cAAc,CAAC;AAAA,EAC3C,WAAW,OAAO,eAAe,YAAY,eAAe,MAAM;AAChE,UAAM,QAAa,CAAC;AACpB,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,YAAM,GAAG,IAAI,eAAe,KAAK;AAAA,IACnC;AACA,WAAO;AAAA,EACT,OAAO;AACL,WAAO;AAAA,EACT;AACF;;;AC5YA,SAAS,yBAAyB;AAClC,OAAOC,WAAU;AAEjB,SAAS,OAAAC,MAAK,SAAAC,cAAa;AAwC3B,IAAI,WAAW;AAGf,IAAMC,WAAWH,MAAK,WAAkDA;AACxE,IAAMI,SAA6B,CAAC;AAE7B,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,IAA0B,CAAC,GAAyB;AA9DpD;AAgEE,QAAM,eAAe,MAAM,cAAc,UAAU;AAEnD,QAAM,MAAM,GAAG,GAAG,IAAI,YAAY;AAClC,QAAM,EAAE,MAAM,QAAQ,eAAe,IAAID;AAAA,IACvC;AAAA,IACA,MAAMC,OAAM,GAAG,KAAK;AAAA,EACtB;AAEA,QAAM,EAAE,MAAM,WAAW,QAAQ,cAAc,IAAID;AAAA,IACjD,GAAG,YAAY;AAAA,IACf;AAAA,EACF;AAEA,kBAAU,UAAV,sBAAU,QAAU;AAGpB,OAAK,UAAL,KAAK,QAAU;AAEf,QAAM,SAAS,CAACE,UAAiB;AAC/B,IAAAD,OAAM,GAAG,IAAIC;AACb,WAAO,eAAe;AAAA,EACxB;AAGA,QAAM,aAAa;AAEnB,QAAM,QAAQJ,KAAuB,MAAS;AAE9C,MAAI,kBAA0C;AAE9C,iBAAe,eACb,QACA,SACA;AACA,WAAO,kBAAkB;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,GAAG;AAAA,QACH,GAAG,mCAAS;AAAA,MACd;AAAA,MACA,MAAM;AAAA,QACJ,GAAGC,OAAM,IAAI;AAAA,QACb,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf,YAAY,aAAW,cAAc,MAAM,OAAO;AAAA,MAClD,UAAU,SAAO;AACf,cAAM,QAAQ;AAAA,MAChB;AAAA,MACA,oBAAoB,gBAAc;AAChC,0BAAkB;AAAA,MACpB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,WAA6C,OACjD,QACA,YACG;AACH,WAAO,eAAe,QAAQ,OAAO;AAAA,EACvC;AAEA,QAAM,OAAO,MAAM;AACjB,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,wBAAkB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,gBAAgB,CAACI,gBAAuB;AAC5C,WAAOA,WAAU;AAAA,EACnB;AAEA,QAAM,QAAQL,KAAI,YAAY;AAE9B,QAAM,eAAe,CAAC,UAA4C;AAjJpE,QAAAM;AAkJI,KAAAA,MAAA,+BAAO,mBAAP,gBAAAA,IAAA;AACA,UAAM,aAAa,MAAM;AACzB,WAAO,aAAa,SAAS,UAAU,IAAI;AAAA,EAC7C;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["_a","swrv","ref","unref","useSWRV","store","data","completion","_a"]}
+{"version":3,"sources":["../src/use-chat.ts","../src/chat-store.ts","../src/use-completion.ts"],"sourcesContent":["import type {\n  ChatRequestOptions,\n  ChatStatus,\n  ChatStore,\n  CreateUIMessage,\n  FileUIPart,\n  InferUIDataParts,\n  UIDataPartSchemas,\n  UIMessage,\n  UseChatOptions,\n} from 'ai';\nimport {\n  convertFileListToFileUIParts,\n  defaultChatStoreOptions,\n  generateId as generateIdFunc,\n} from 'ai';\nimport type { Ref } from 'vue';\nimport { computed, ref } from 'vue';\nimport { createChatStore } from './chat-store';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<\n  MESSAGE_METADATA = unknown,\n  DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> = {\n  /**\n   * The id of the chat.\n   */\n  readonly chatId: string;\n\n  /** Current messages in the chat */\n  readonly messages: Ref<\n    UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n  >;\n\n  /** The error object of the API request */\n  readonly error: Ref<Error | undefined>;\n\n  /**\n   * Append a user message to the chat list. This triggers the API call to fetch\n   * the assistant's response.\n   *\n   * @param message The message to append\n   * @param options Additional options to pass to the API call\n   */\n  append: (\n    message: CreateUIMessage<\n      MESSAGE_METADATA,\n      InferUIDataParts<DATA_PART_SCHEMAS>\n    >,\n    options?: ChatRequestOptions,\n  ) => Promise<void>;\n\n  /**\n   * Reload the last AI chat response for the given chat history. If the last\n   * message isn't from the assistant, it will request the API to generate a\n   * new response.\n   */\n  reload: (\n    chatRequestOptions?: ChatRequestOptions,\n  ) => Promise<string | null | undefined>;\n\n  /**\n   * Abort the current request immediately, keep the generated tokens if any.\n   */\n  stop: () => void;\n\n  /**\n   * Update the `messages` state locally. This is useful when you want to\n   * edit the messages on the client, and then trigger the `reload` method\n   * manually to regenerate the AI response.\n   */\n  setMessages: (\n    messages:\n      | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n      | ((\n          messages: UIMessage<\n            MESSAGE_METADATA,\n            InferUIDataParts<DATA_PART_SCHEMAS>\n          >[],\n        ) => UIMessage<\n          MESSAGE_METADATA,\n          InferUIDataParts<DATA_PART_SCHEMAS>\n        >[]),\n  ) => void;\n\n  /** The current value of the input */\n  input: Ref<string>;\n\n  /** Form submission handler to automatically reset input and append a user message  */\n  handleSubmit: (\n    event?: { preventDefault?: () => void },\n    chatRequestOptions?: ChatRequestOptions & {\n      files?: FileList | FileUIPart[];\n    },\n  ) => void;\n\n  /**\n   * Hook status:\n   *\n   * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n   * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n   * - `ready`: The full response has been received and processed; a new user message can be submitted.\n   * - `error`: An error occurred during the API request, preventing successful completion.\n   */\n  status: Ref<ChatStatus>;\n\n  addToolResult: ({\n    toolCallId,\n    result,\n  }: {\n    toolCallId: string;\n    result: any;\n  }) => void;\n};\n\nexport function useChat<\n  MESSAGE_METADATA = unknown,\n  DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>({\n  chatId,\n  initialInput = '',\n  onFinish,\n  onError,\n  generateId = generateIdFunc,\n  onToolCall,\n  chatStore: chatStoreArg,\n}: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> = {}): UseChatHelpers<\n  MESSAGE_METADATA,\n  DATA_PART_SCHEMAS\n> {\n  const stableChatId = chatId ?? generateId();\n\n  const chatStore =\n    chatStoreArg == null\n      ? createChatStore(\n          defaultChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n            api: '/api/chat',\n            generateId,\n          })(),\n        )\n      : typeof chatStoreArg === 'function'\n        ? createChatStore(chatStoreArg())\n        : chatStoreArg;\n\n  const messages = computed(() => chatStore.getMessages(stableChatId));\n  const status = computed(() => chatStore.getStatus(stableChatId));\n  const error = computed(() => chatStore.getError(stableChatId));\n\n  const append = async (\n    message: CreateUIMessage<\n      MESSAGE_METADATA,\n      InferUIDataParts<DATA_PART_SCHEMAS>\n    >,\n    { headers, body }: ChatRequestOptions = {},\n  ) =>\n    chatStore.submitMessage({\n      chatId: stableChatId,\n      message,\n      headers,\n      body,\n      onError,\n      onToolCall,\n      onFinish,\n    });\n\n  const reload = async ({ headers, body }: ChatRequestOptions = {}) =>\n    chatStore.resubmitLastUserMessage({\n      chatId: stableChatId,\n      headers,\n      body,\n      onError,\n      onToolCall,\n      onFinish,\n    });\n\n  const stop = () => {\n    chatStore.stopStream({ chatId: stableChatId });\n  };\n\n  const setMessages = (\n    messagesParam:\n      | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n      | ((\n          messages: UIMessage<\n            MESSAGE_METADATA,\n            InferUIDataParts<DATA_PART_SCHEMAS>\n          >[],\n        ) => UIMessage<\n          MESSAGE_METADATA,\n          InferUIDataParts<DATA_PART_SCHEMAS>\n        >[]),\n  ) => {\n    if (typeof messagesParam === 'function') {\n      messagesParam = messagesParam(chatStore.getMessages(stableChatId));\n    }\n\n    chatStore.setMessages({\n      id: stableChatId,\n      messages: messagesParam,\n    });\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = async (\n    event?: { preventDefault?: () => void },\n    options: ChatRequestOptions & { files?: FileList | FileUIPart[] } = {},\n  ) => {\n    event?.preventDefault?.();\n\n    const inputValue = input.value;\n\n    const fileParts = Array.isArray(options?.files)\n      ? options.files\n      : await convertFileListToFileUIParts(options?.files);\n\n    if (!inputValue && fileParts.length === 0) return;\n\n    await append(\n      {\n        id: generateId(),\n        role: 'user',\n        metadata: undefined,\n        parts: [...fileParts, { type: 'text', text: inputValue }],\n      },\n      {\n        headers: options.headers,\n        body: options.body,\n      },\n    );\n\n    input.value = '';\n  };\n\n  const addToolResult = (\n    options: Omit<\n      Parameters<\n        ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>['addToolResult']\n      >[0],\n      'chatId'\n    >,\n  ) => chatStore.addToolResult({ chatId: stableChatId, ...options });\n\n  return {\n    chatId: stableChatId,\n    messages,\n    append,\n    error,\n    reload,\n    stop,\n    setMessages,\n    input,\n    handleSubmit,\n    status,\n    addToolResult,\n  };\n}\n","import {\n  ChatStore,\n  SerialJobExecutor,\n  type ActiveResponse,\n  type Chat,\n  type ChatStatus,\n  type ChatStoreOptions,\n  type InferUIDataParts,\n  type UIDataPartSchemas,\n  type UIDataTypes,\n  type UIMessage,\n} from 'ai';\nimport { Ref, ref } from 'vue';\n\nclass VueChat<MESSAGE_METADATA, DATA_TYPES extends UIDataTypes>\n  implements Chat<MESSAGE_METADATA, DATA_TYPES>\n{\n  private messagesRef: Ref<UIMessage<MESSAGE_METADATA, DATA_TYPES>[]>;\n  private statusRef = ref<ChatStatus>('ready');\n  private errorRef = ref<Error | undefined>(undefined);\n\n  activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined = undefined;\n  jobExecutor = new SerialJobExecutor();\n\n  constructor(messages?: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) {\n    this.messagesRef = ref(messages ?? []) as Ref<\n      UIMessage<MESSAGE_METADATA, DATA_TYPES>[]\n    >;\n  }\n\n  get messages(): UIMessage<MESSAGE_METADATA, DATA_TYPES>[] {\n    return this.messagesRef.value;\n  }\n\n  get status(): ChatStatus {\n    return this.statusRef.value;\n  }\n\n  get error(): Error | undefined {\n    return this.errorRef.value;\n  }\n\n  setStatus = (status: ChatStatus) => {\n    this.statusRef.value = status;\n  };\n\n  setError = (error: Error | undefined) => {\n    this.errorRef.value = error;\n  };\n\n  setActiveResponse = (\n    activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined,\n  ) => {\n    this.activeResponse = activeResponse;\n  };\n\n  setMessages = (messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) => {\n    this.messagesRef.value = messages;\n  };\n\n  pushMessage = (message: UIMessage<MESSAGE_METADATA, DATA_TYPES>) => {\n    this.messagesRef.value.push(message);\n  };\n\n  popMessage = () => {\n    this.messagesRef.value.pop();\n  };\n\n  replaceMessage = (\n    index: number,\n    message: UIMessage<MESSAGE_METADATA, DATA_TYPES>,\n  ) => {\n    // message is cloned here because vue's deep reactivity shows unexpected behavior, particularly when updating tool invocation parts\n    this.messagesRef.value[index] = { ...message };\n  };\n}\n\nexport function createChatStore<\n  MESSAGE_METADATA = unknown,\n  DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>(\n  options: ChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>,\n): ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS> {\n  return new ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n    ...options,\n    createChat: options =>\n      new VueChat<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>(\n        options.messages,\n      ),\n  });\n}\n","import type { CompletionRequestOptions, UseCompletionOptions } from 'ai';\nimport { callCompletionApi } from 'ai';\nimport swrv from 'swrv';\nimport type { Ref } from 'vue';\nimport { ref, unref } from 'vue';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n  /** The current completion result */\n  completion: Ref<string>;\n  /** The error object of the API request */\n  error: Ref<undefined | Error>;\n  /**\n   * Send a new prompt to the API endpoint and update the completion state.\n   */\n  complete: (\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) => Promise<string | null | undefined>;\n  /**\n   * Abort the current API request but keep the generated tokens.\n   */\n  stop: () => void;\n  /**\n   * Update the `completion` state locally.\n   */\n  setCompletion: (completion: string) => void;\n  /** The current value of the input */\n  input: Ref<string>;\n  /**\n   * Form submission handler to automatically reset input and append a user message\n   * @example\n   * ```jsx\n   * <form @submit=\"handleSubmit\">\n   *  <input @change=\"handleInputChange\" v-model=\"input\" />\n   * </form>\n   * ```\n   */\n  handleSubmit: (event?: { preventDefault?: () => void }) => void;\n  /** Whether the API request is in progress */\n  isLoading: Ref<boolean | undefined>;\n};\n\nlet uniqueId = 0;\n\n// @ts-expect-error - some issues with the default export of useSWRV\nconst useSWRV = (swrv.default as (typeof import('swrv'))['default']) || swrv;\nconst store: Record<string, any> = {};\n\nexport function useCompletion({\n  api = '/api/completion',\n  id,\n  initialCompletion = '',\n  initialInput = '',\n  credentials,\n  headers,\n  body,\n  streamProtocol,\n  onFinish,\n  onError,\n  fetch,\n}: UseCompletionOptions = {}): UseCompletionHelpers {\n  // Generate an unique id for the completion if not provided.\n  const completionId = id || `completion-${uniqueId++}`;\n\n  const key = `${api}|${completionId}`;\n  const { data, mutate: originalMutate } = useSWRV<string>(\n    key,\n    () => store[key] || initialCompletion,\n  );\n\n  const { data: isLoading, mutate: mutateLoading } = useSWRV<boolean>(\n    `${completionId}-loading`,\n    null,\n  );\n\n  isLoading.value ??= false;\n\n  // Force the `data` to be `initialCompletion` if it's `undefined`.\n  data.value ||= initialCompletion;\n\n  const mutate = (data: string) => {\n    store[key] = data;\n    return originalMutate();\n  };\n\n  // Because of the `initialData` option, the `data` will never be `undefined`.\n  const completion = data as Ref<string>;\n\n  const error = ref<undefined | Error>(undefined);\n\n  let abortController: AbortController | null = null;\n\n  async function triggerRequest(\n    prompt: string,\n    options?: CompletionRequestOptions,\n  ) {\n    return callCompletionApi({\n      api,\n      prompt,\n      credentials,\n      headers: {\n        ...headers,\n        ...options?.headers,\n      },\n      body: {\n        ...unref(body),\n        ...options?.body,\n      },\n      streamProtocol,\n      setCompletion: mutate,\n      setLoading: loading => mutateLoading(() => loading),\n      setError: err => {\n        error.value = err;\n      },\n      setAbortController: controller => {\n        abortController = controller;\n      },\n      onFinish,\n      onError,\n      fetch,\n    });\n  }\n\n  const complete: UseCompletionHelpers['complete'] = async (\n    prompt,\n    options,\n  ) => {\n    return triggerRequest(prompt, options);\n  };\n\n  const stop = () => {\n    if (abortController) {\n      abortController.abort();\n      abortController = null;\n    }\n  };\n\n  const setCompletion = (completion: string) => {\n    mutate(completion);\n  };\n\n  const input = ref(initialInput);\n\n  const handleSubmit = (event?: { preventDefault?: () => void }) => {\n    event?.preventDefault?.();\n    const inputValue = input.value;\n    return inputValue ? complete(inputValue) : undefined;\n  };\n\n  return {\n    completion,\n    complete,\n    error,\n    stop,\n    setCompletion,\n    input,\n    handleSubmit,\n    isLoading,\n  };\n}\n"],"mappings":";AAWA;AAAA,EACE;AAAA,EACA;AAAA,EACA,cAAc;AAAA,OACT;AAEP,SAAS,UAAU,OAAAA,YAAW;;;ACjB9B;AAAA,EACE;AAAA,EACA;AAAA,OASK;AACP,SAAc,WAAW;AAEzB,IAAM,UAAN,MAEA;AAAA,EAQE,YAAY,UAAsD;AANlE,SAAQ,YAAY,IAAgB,OAAO;AAC3C,SAAQ,WAAW,IAAuB,MAAS;AAEnD,0BAA+D;AAC/D,uBAAc,IAAI,kBAAkB;AAoBpC,qBAAY,CAAC,WAAuB;AAClC,WAAK,UAAU,QAAQ;AAAA,IACzB;AAEA,oBAAW,CAAC,UAA6B;AACvC,WAAK,SAAS,QAAQ;AAAA,IACxB;AAEA,6BAAoB,CAClB,mBACG;AACH,WAAK,iBAAiB;AAAA,IACxB;AAEA,uBAAc,CAAC,aAAwD;AACrE,WAAK,YAAY,QAAQ;AAAA,IAC3B;AAEA,uBAAc,CAAC,YAAqD;AAClE,WAAK,YAAY,MAAM,KAAK,OAAO;AAAA,IACrC;AAEA,sBAAa,MAAM;AACjB,WAAK,YAAY,MAAM,IAAI;AAAA,IAC7B;AAEA,0BAAiB,CACf,OACA,YACG;AAEH,WAAK,YAAY,MAAM,KAAK,IAAI,EAAE,GAAG,QAAQ;AAAA,IAC/C;AAjDE,SAAK,cAAc,IAAI,8BAAY,CAAC,CAAC;AAAA,EAGvC;AAAA,EAEA,IAAI,WAAsD;AACxD,WAAO,KAAK,YAAY;AAAA,EAC1B;AAAA,EAEA,IAAI,SAAqB;AACvB,WAAO,KAAK,UAAU;AAAA,EACxB;AAAA,EAEA,IAAI,QAA2B;AAC7B,WAAO,KAAK,SAAS;AAAA,EACvB;AAmCF;AAEO,SAAS,gBAId,SACgD;AAChD,SAAO,IAAI,UAA+C;AAAA,IACxD,GAAG;AAAA,IACH,YAAY,CAAAC,aACV,IAAI;AAAA,MACFA,SAAQ;AAAA,IACV;AAAA,EACJ,CAAC;AACH;;;AD2BO,SAAS,QAGd;AAAA,EACA;AAAA,EACA,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA,aAAa;AAAA,EACb;AAAA,EACA,WAAW;AACb,IAAyD,CAAC,GAGxD;AACA,QAAM,eAAe,0BAAU,WAAW;AAE1C,QAAM,YACJ,gBAAgB,OACZ;AAAA,IACE,wBAA6D;AAAA,MAC3D,KAAK;AAAA,MACL;AAAA,IACF,CAAC,EAAE;AAAA,EACL,IACA,OAAO,iBAAiB,aACtB,gBAAgB,aAAa,CAAC,IAC9B;AAER,QAAM,WAAW,SAAS,MAAM,UAAU,YAAY,YAAY,CAAC;AACnE,QAAM,SAAS,SAAS,MAAM,UAAU,UAAU,YAAY,CAAC;AAC/D,QAAM,QAAQ,SAAS,MAAM,UAAU,SAAS,YAAY,CAAC;AAE7D,QAAM,SAAS,OACb,SAIA,EAAE,SAAS,KAAK,IAAwB,CAAC,MAEzC,UAAU,cAAc;AAAA,IACtB,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAEH,QAAM,SAAS,OAAO,EAAE,SAAS,KAAK,IAAwB,CAAC,MAC7D,UAAU,wBAAwB;AAAA,IAChC,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAEH,QAAM,OAAO,MAAM;AACjB,cAAU,WAAW,EAAE,QAAQ,aAAa,CAAC;AAAA,EAC/C;AAEA,QAAM,cAAc,CAClB,kBAWG;AACH,QAAI,OAAO,kBAAkB,YAAY;AACvC,sBAAgB,cAAc,UAAU,YAAY,YAAY,CAAC;AAAA,IACnE;AAEA,cAAU,YAAY;AAAA,MACpB,IAAI;AAAA,MACJ,UAAU;AAAA,IACZ,CAAC;AAAA,EACH;AAEA,QAAM,QAAQC,KAAI,YAAY;AAE9B,QAAM,eAAe,OACnB,OACA,UAAoE,CAAC,MAClE;AAjNP;AAkNI,yCAAO,mBAAP;AAEA,UAAM,aAAa,MAAM;AAEzB,UAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,MAAM,6BAA6B,mCAAS,KAAK;AAErD,QAAI,CAAC,cAAc,UAAU,WAAW;AAAG;AAE3C,UAAM;AAAA,MACJ;AAAA,QACE,IAAI,WAAW;AAAA,QACf,MAAM;AAAA,QACN,UAAU;AAAA,QACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,WAAW,CAAC;AAAA,MAC1D;AAAA,MACA;AAAA,QACE,SAAS,QAAQ;AAAA,QACjB,MAAM,QAAQ;AAAA,MAChB;AAAA,IACF;AAEA,UAAM,QAAQ;AAAA,EAChB;AAEA,QAAM,gBAAgB,CACpB,YAMG,UAAU,cAAc,EAAE,QAAQ,cAAc,GAAG,QAAQ,CAAC;AAEjE,SAAO;AAAA,IACL,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AEjQA,SAAS,yBAAyB;AAClC,OAAO,UAAU;AAEjB,SAAS,OAAAC,MAAK,aAAa;AAwC3B,IAAI,WAAW;AAGf,IAAM,UAAW,KAAK,WAAkD;AACxE,IAAM,QAA6B,CAAC;AAE7B,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,IAA0B,CAAC,GAAyB;AA9DpD;AAgEE,QAAM,eAAe,MAAM,cAAc,UAAU;AAEnD,QAAM,MAAM,GAAG,GAAG,IAAI,YAAY;AAClC,QAAM,EAAE,MAAM,QAAQ,eAAe,IAAI;AAAA,IACvC;AAAA,IACA,MAAM,MAAM,GAAG,KAAK;AAAA,EACtB;AAEA,QAAM,EAAE,MAAM,WAAW,QAAQ,cAAc,IAAI;AAAA,IACjD,GAAG,YAAY;AAAA,IACf;AAAA,EACF;AAEA,kBAAU,UAAV,sBAAU,QAAU;AAGpB,OAAK,UAAL,KAAK,QAAU;AAEf,QAAM,SAAS,CAACC,UAAiB;AAC/B,UAAM,GAAG,IAAIA;AACb,WAAO,eAAe;AAAA,EACxB;AAGA,QAAM,aAAa;AAEnB,QAAM,QAAQD,KAAuB,MAAS;AAE9C,MAAI,kBAA0C;AAE9C,iBAAe,eACb,QACA,SACA;AACA,WAAO,kBAAkB;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,GAAG;AAAA,QACH,GAAG,mCAAS;AAAA,MACd;AAAA,MACA,MAAM;AAAA,QACJ,GAAG,MAAM,IAAI;AAAA,QACb,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf,YAAY,aAAW,cAAc,MAAM,OAAO;AAAA,MAClD,UAAU,SAAO;AACf,cAAM,QAAQ;AAAA,MAChB;AAAA,MACA,oBAAoB,gBAAc;AAChC,0BAAkB;AAAA,MACpB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,WAA6C,OACjD,QACA,YACG;AACH,WAAO,eAAe,QAAQ,OAAO;AAAA,EACvC;AAEA,QAAM,OAAO,MAAM;AACjB,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,wBAAkB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,gBAAgB,CAACE,gBAAuB;AAC5C,WAAOA,WAAU;AAAA,EACnB;AAEA,QAAM,QAAQF,KAAI,YAAY;AAE9B,QAAM,eAAe,CAAC,UAA4C;AAjJpE,QAAAG;AAkJI,KAAAA,MAAA,+BAAO,mBAAP,gBAAAA,IAAA;AACA,UAAM,aAAa,MAAM;AACzB,WAAO,aAAa,SAAS,UAAU,IAAI;AAAA,EAC7C;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["ref","options","ref","ref","data","completion","_a"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/vue",
-  "version": "2.0.0-alpha.5",
+  "version": "2.0.0-alpha.6",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -20,8 +20,8 @@
   ],
   "dependencies": {
     "swrv": "^1.0.4",
-    "ai": "5.0.0-alpha.5",
-    "@ai-sdk/provider-utils": "3.0.0-alpha.4"
+    "ai": "5.0.0-alpha.6",
+    "@ai-sdk/provider-utils": "3.0.0-alpha.6"
   },
   "devDependencies": {
     "@testing-library/jest-dom": "^6.6.3",