@ai-sdk/vue 1.2.11 → 2.0.0-alpha.1

package/dist/index.js

@@ -30,107 +30,73 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  useAssistant: () => useAssistant,
   useChat: () => useChat,
   useCompletion: () => useCompletion
 });
 module.exports = __toCommonJS(src_exports);
 
 // src/use-chat.ts
-var import_ui_utils = require("@ai-sdk/ui-utils");
+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 = {};
 function useChat({
   api = "/api/chat",
-  id,
+  chatId,
   initialMessages = [],
   initialInput = "",
-  sendExtraMessageFields,
-  streamProtocol = "data",
-  onResponse,
+  streamProtocol = "ui-message",
   onFinish,
   onError,
   credentials,
   headers: metadataHeaders,
   body: metadataBody,
-  generateId: generateId2 = import_ui_utils.generateId,
+  generateId = import_ai.generateId,
   onToolCall,
-  fetch: fetch2,
-  keepLastMessageOnError = true,
+  fetch,
   maxSteps = 1,
-  experimental_prepareRequestBody
+  experimental_prepareRequestBody,
+  messageMetadataSchema
 } = {
   maxSteps: 1
 }) {
   var _a, _b;
-  const chatId = id != null ? id : generateId2();
-  const key = `${api}|${chatId}`;
-  const { data: messagesData, mutate: originalMutate } = useSWRV(
-    key,
-    () => {
-      var _a2;
-      return (_a2 = store[key]) != null ? _a2 : (0, import_ui_utils.fillMessageParts)(initialMessages);
-    }
-  );
-  const { data: status, mutate: mutateStatus } = useSWRV(`${chatId}-status`, null);
-  (_a = status.value) != null ? _a : status.value = "ready";
-  (_b = messagesData.value) != null ? _b : messagesData.value = (0, import_ui_utils.fillMessageParts)(initialMessages);
+  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 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);
-  const streamData = (0, import_vue.ref)(void 0);
   let abortController = null;
-  async function triggerRequest(messagesSnapshot, { data, headers, body } = {}) {
-    var _a2, _b2, _c;
+  async function triggerRequest(messagesSnapshot, { headers, body } = {}) {
+    var _a2;
     error.value = void 0;
-    mutateStatus(() => "submitted");
+    status.value = "submitted";
     const messageCount = messages.value.length;
-    const maxStep = (0, import_ui_utils.extractMaxToolInvocationStep)(
-      (_a2 = messages.value[messages.value.length - 1]) == null ? void 0 : _a2.toolInvocations
-    );
+    const lastMessage = messages.value.at(-1);
+    const maxStep = lastMessage != null ? (0, import_ai.extractMaxToolInvocationStep)((0, import_ai.getToolInvocations)(lastMessage)) : 0;
     try {
       abortController = new AbortController();
-      const previousMessages = (0, import_ui_utils.fillMessageParts)(messagesSnapshot);
-      const chatMessages = previousMessages;
-      mutate(chatMessages);
-      const existingData = (_b2 = streamData.value) != null ? _b2 : [];
-      const constructedMessagesPayload = sendExtraMessageFields ? chatMessages : chatMessages.map(
-        ({
-          role,
-          content,
-          experimental_attachments,
-          data: data2,
-          annotations,
-          toolInvocations,
-          parts
-        }) => ({
-          role,
-          content,
-          ...experimental_attachments !== void 0 && {
-            experimental_attachments
-          },
-          ...data2 !== void 0 && { data: data2 },
-          ...annotations !== void 0 && { annotations },
-          ...toolInvocations !== void 0 && { toolInvocations },
-          ...parts !== void 0 && { parts }
-        })
-      );
-      await (0, import_ui_utils.callChatApi)({
+      mutate(messagesSnapshot);
+      await (0, import_ai.callChatApi)({
         api,
-        body: (_c = experimental_prepareRequestBody == null ? void 0 : experimental_prepareRequestBody({
-          id: chatId,
-          messages: chatMessages,
-          requestData: data,
+        body: (_a2 = experimental_prepareRequestBody == null ? void 0 : experimental_prepareRequestBody({
+          chatId: stableChatId,
+          messages: messagesSnapshot,
           requestBody: body
-        })) != null ? _c : {
-          id: chatId,
-          messages: constructedMessagesPayload,
-          data,
+        })) != null ? _a2 : {
+          chatId: stableChatId,
+          messages: messagesSnapshot,
           ...(0, import_vue.unref)(metadataBody),
           // Use unref to unwrap the ref value
           ...body
@@ -142,45 +108,40 @@ function useChat({
         },
         abortController: () => abortController,
         credentials,
-        onResponse,
-        onUpdate({ message, data: data2, replaceLastMessage }) {
-          mutateStatus(() => "streaming");
+        onUpdate({ message }) {
+          status.value = "streaming";
+          const replaceLastMessage = message.id === messagesSnapshot[messagesSnapshot.length - 1].id;
           mutate([
-            ...replaceLastMessage ? chatMessages.slice(0, chatMessages.length - 1) : chatMessages,
+            ...replaceLastMessage ? messagesSnapshot.slice(0, messagesSnapshot.length - 1) : messagesSnapshot,
             message
           ]);
-          if (data2 == null ? void 0 : data2.length) {
-            streamData.value = [...existingData, ...data2];
-          }
         },
         onFinish,
-        restoreMessagesOnFailure() {
-          if (!keepLastMessageOnError) {
-            mutate(previousMessages);
-          }
-        },
-        generateId: generateId2,
+        generateId,
         onToolCall,
-        fetch: fetch2,
+        fetch,
         // enabled use of structured clone in processChatResponse:
-        lastMessage: recursiveToRaw(chatMessages[chatMessages.length - 1])
+        lastMessage: recursiveToRaw(
+          messagesSnapshot[messagesSnapshot.length - 1]
+        ),
+        messageMetadataSchema
       });
-      mutateStatus(() => "ready");
+      status.value = "ready";
     } catch (err) {
       if (err.name === "AbortError") {
         abortController = null;
-        mutateStatus(() => "ready");
+        status.value = "ready";
         return null;
       }
       if (onError && err instanceof Error) {
         onError(err);
       }
       error.value = err;
-      mutateStatus(() => "error");
+      status.value = "error";
     } finally {
       abortController = null;
     }
-    if ((0, import_ui_utils.shouldResubmitMessages)({
+    if ((0, import_ai.shouldResubmitMessages)({
       originalMaxToolInvocationStep: maxStep,
       originalMessageCount: messageCount,
       maxSteps,
@@ -190,17 +151,12 @@ function useChat({
     }
   }
   const append = async (message, options) => {
-    var _a2, _b2, _c;
-    const attachmentsForRequest = await (0, import_ui_utils.prepareAttachmentsForRequest)(
-      (_a2 = options == null ? void 0 : options.experimental_attachments) != null ? _a2 : message.experimental_attachments
-    );
+    var _a2;
     return triggerRequest(
       messages.value.concat({
         ...message,
-        id: (_b2 = message.id) != null ? _b2 : generateId2(),
-        createdAt: (_c = message.createdAt) != null ? _c : /* @__PURE__ */ new Date(),
-        experimental_attachments: attachmentsForRequest.length > 0 ? attachmentsForRequest : void 0,
-        parts: (0, import_ui_utils.getMessageParts)(message)
+        id: (_a2 = message.id) != null ? _a2 : generateId(),
+        parts: message.parts
       }),
       options
     );
@@ -225,32 +181,21 @@ function useChat({
     if (typeof messagesArg === "function") {
       messagesArg = messagesArg(messages.value);
     }
-    mutate((0, import_ui_utils.fillMessageParts)(messagesArg));
-  };
-  const setData = (dataArg) => {
-    if (typeof dataArg === "function") {
-      dataArg = dataArg(streamData.value);
-    }
-    streamData.value = dataArg;
+    mutate(messagesArg);
   };
   const input = (0, import_vue.ref)(initialInput);
   const handleSubmit = async (event, options = {}) => {
     var _a2;
     (_a2 = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a2.call(event);
     const inputValue = input.value;
-    if (!inputValue && !options.allowEmptySubmit)
+    const fileParts = Array.isArray(options == null ? void 0 : options.files) ? options.files : await (0, import_ai.convertFileListToFileUIParts)(options == null ? void 0 : options.files);
+    if (!inputValue && fileParts.length === 0)
       return;
-    const attachmentsForRequest = await (0, import_ui_utils.prepareAttachmentsForRequest)(
-      options.experimental_attachments
-    );
     triggerRequest(
       messages.value.concat({
-        id: generateId2(),
-        createdAt: /* @__PURE__ */ new Date(),
-        content: inputValue,
+        id: generateId(),
         role: "user",
-        experimental_attachments: attachmentsForRequest.length > 0 ? attachmentsForRequest : void 0,
-        parts: [{ type: "text", text: inputValue }]
+        parts: [...fileParts, { type: "text", text: inputValue }]
       }),
       options
     );
@@ -261,7 +206,7 @@ function useChat({
     result
   }) => {
     const currentMessages = messages.value;
-    (0, import_ui_utils.updateToolCallResult)({
+    (0, import_ai.updateToolCallResult)({
       messages: currentMessages,
       toolCallId,
       toolResult: result
@@ -271,12 +216,12 @@ function useChat({
       return;
     }
     const lastMessage = currentMessages[currentMessages.length - 1];
-    if ((0, import_ui_utils.isAssistantMessageWithCompletedToolCalls)(lastMessage)) {
+    if ((0, import_ai.isAssistantMessageWithCompletedToolCalls)(lastMessage)) {
       triggerRequest(currentMessages);
     }
   };
   return {
-    id: chatId,
+    chatId: stableChatId,
     messages,
     append,
     error,
@@ -285,12 +230,7 @@ function useChat({
     setMessages,
     input,
     handleSubmit,
-    isLoading: (0, import_vue.computed)(
-      () => status.value === "submitted" || status.value === "streaming"
-    ),
     status,
-    data: streamData,
-    setData,
     addToolResult
   };
 }
@@ -309,7 +249,7 @@ function recursiveToRaw(inputValue) {
 }
 
 // src/use-completion.ts
-var import_ui_utils2 = require("@ai-sdk/ui-utils");
+var import_ai2 = require("ai");
 var import_swrv2 = __toESM(require("swrv"));
 var import_vue2 = require("vue");
 var uniqueId = 0;
@@ -324,10 +264,9 @@ function useCompletion({
   headers,
   body,
   streamProtocol,
-  onResponse,
   onFinish,
   onError,
-  fetch: fetch2
+  fetch
 } = {}) {
   var _a;
   const completionId = id || `completion-${uniqueId++}`;
@@ -341,7 +280,6 @@ function useCompletion({
     null
   );
   (_a = isLoading.value) != null ? _a : isLoading.value = false;
-  const { data: streamData, mutate: mutateStreamData } = useSWRV2(`${completionId}-data`, null);
   data.value || (data.value = initialCompletion);
   const mutate = (data2) => {
     store2[key] = data2;
@@ -351,9 +289,7 @@ function useCompletion({
   const error = (0, import_vue2.ref)(void 0);
   let abortController = null;
   async function triggerRequest(prompt, options) {
-    var _a2;
-    const existingData = (_a2 = streamData.value) != null ? _a2 : [];
-    return (0, import_ui_utils2.callCompletionApi)({
+    return (0, import_ai2.callCompletionApi)({
       api,
       prompt,
       credentials,
@@ -374,13 +310,9 @@ function useCompletion({
       setAbortController: (controller) => {
         abortController = controller;
       },
-      onResponse,
       onFinish,
       onError,
-      onData: (data2) => {
-        mutateStreamData(() => [...existingData, ...data2 != null ? data2 : []]);
-      },
-      fetch: fetch2
+      fetch
     });
   }
   const complete = async (prompt, options) => {
@@ -410,182 +342,11 @@ function useCompletion({
     setCompletion,
     input,
     handleSubmit,
-    isLoading,
-    data: streamData
-  };
-}
-
-// src/use-assistant.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
-var import_ui_utils3 = require("@ai-sdk/ui-utils");
-var import_vue3 = require("vue");
-function useAssistant({
-  api,
-  threadId: threadIdParam,
-  credentials,
-  headers,
-  body,
-  onError
-}) {
-  const messages = (0, import_vue3.ref)([]);
-  const input = (0, import_vue3.ref)("");
-  const currentThreadId = (0, import_vue3.ref)(void 0);
-  const status = (0, import_vue3.ref)("awaiting_message");
-  const error = (0, import_vue3.ref)(void 0);
-  const setMessages = (messageFactory) => {
-    messages.value = messageFactory(messages.value);
-  };
-  const setCurrentThreadId = (newThreadId) => {
-    currentThreadId.value = newThreadId;
-    messages.value = [];
-  };
-  const handleInputChange = (event) => {
-    var _a;
-    input.value = (_a = event == null ? void 0 : event.target) == null ? void 0 : _a.value;
-  };
-  const isSending = (0, import_vue3.computed)(() => status.value === "in_progress");
-  const abortController = (0, import_vue3.ref)(null);
-  const stop = (0, import_vue3.computed)(() => {
-    return () => {
-      if (abortController.value) {
-        abortController.value.abort();
-        abortController.value = null;
-      }
-    };
-  });
-  const append = async (message, requestOptions) => {
-    var _a, _b, _c, _d;
-    status.value = "in_progress";
-    const newMessage = {
-      ...message,
-      id: (_a = message.id) != null ? _a : (0, import_ui_utils3.generateId)()
-    };
-    setMessages((messages2) => [...messages2, newMessage]);
-    input.value = "";
-    const controller = new AbortController();
-    try {
-      abortController.value = controller;
-      const response = await fetch(api, {
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-          ...headers
-        },
-        body: JSON.stringify({
-          ...body,
-          // Message Content
-          message: message.content,
-          // Always Use User Provided Thread ID When Available
-          threadId: (_b = threadIdParam != null ? threadIdParam : currentThreadId.value) != null ? _b : null,
-          // Optional Request Data
-          ...(requestOptions == null ? void 0 : requestOptions.data) && { data: requestOptions == null ? void 0 : requestOptions.data }
-        }),
-        signal: controller.signal,
-        credentials
-      });
-      if (!response.ok) {
-        throw new Error(
-          (_c = response.statusText) != null ? _c : "An error occurred while sending the message"
-        );
-      }
-      if (!response.body) {
-        throw new Error("The response body is empty");
-      }
-      await (0, import_ui_utils3.processAssistantStream)({
-        stream: response.body,
-        onAssistantMessagePart(value) {
-          messages.value = [
-            ...messages.value,
-            {
-              id: value.id,
-              content: value.content[0].text.value,
-              role: value.role,
-              parts: []
-            }
-          ];
-        },
-        onTextPart(value) {
-          setMessages((messages2) => {
-            const lastMessage = messages2[messages2.length - 1];
-            lastMessage.content += value;
-            return [...messages2.slice(0, -1), lastMessage];
-          });
-        },
-        onAssistantControlDataPart(value) {
-          if (value.threadId) {
-            currentThreadId.value = value.threadId;
-          }
-          setMessages((messages2) => {
-            const lastMessage = messages2[messages2.length - 1];
-            lastMessage.id = value.messageId;
-            return [...messages2.slice(0, -1), lastMessage];
-          });
-        },
-        onDataMessagePart(value) {
-          setMessages((messages2) => {
-            var _a2;
-            return [
-              ...messages2,
-              {
-                id: (_a2 = value.id) != null ? _a2 : (0, import_ui_utils3.generateId)(),
-                role: "data",
-                content: "",
-                data: value.data,
-                parts: []
-              }
-            ];
-          });
-        },
-        onErrorPart(value) {
-          error.value = new Error(value);
-        }
-      });
-    } catch (err) {
-      if ((0, import_provider_utils.isAbortError)(err) && ((_d = abortController.value) == null ? void 0 : _d.signal.aborted)) {
-        abortController.value = null;
-        return;
-      }
-      if (onError && err instanceof Error) {
-        onError(err);
-      }
-      error.value = err;
-    } finally {
-      abortController.value = null;
-      status.value = "awaiting_message";
-    }
-  };
-  const submitMessage = async (event, requestOptions) => {
-    var _a;
-    (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
-    if (!input.value)
-      return;
-    append(
-      {
-        role: "user",
-        content: input.value,
-        parts: []
-      },
-      requestOptions
-    );
-  };
-  return {
-    append,
-    messages,
-    setMessages,
-    threadId: (0, import_vue3.readonly)(currentThreadId),
-    setThreadId: setCurrentThreadId,
-    input,
-    handleInputChange,
-    handleSubmit: submitMessage,
-    isSending,
-    status,
-    error,
-    stop
+    isLoading
   };
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  useAssistant,
   useChat,
   useCompletion
 });