ai 2.1.26 → 2.1.27

package/vue/dist/index.d.ts

@@ -192,7 +192,7 @@ type UseChatHelpers = {
     /** Form submission handler to automattically reset input and append a user message  */
     handleSubmit: (e: any) => void;
     /** Whether the API request is in progress */
-    isLoading: Ref<boolean>;
+    isLoading: Ref<boolean | undefined>;
 };
 declare function useChat({ api, id, initialMessages, initialInput, sendExtraMessageFields, onResponse, onFinish, onError, credentials, headers, body }?: UseChatOptions): UseChatHelpers;
 
@@ -226,7 +226,7 @@ type UseCompletionHelpers = {
      */
     handleSubmit: (e: any) => void;
     /** Whether the API request is in progress */
-    isLoading: Ref<boolean>;
+    isLoading: Ref<boolean | undefined>;
 };
 declare function useCompletion({ api, id, initialCompletion, initialInput, credentials, headers, body, onResponse, onFinish, onError }?: UseCompletionOptions): UseCompletionHelpers;
 

package/vue/dist/index.js

@@ -3,22 +3,8 @@ var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getOwnPropSymbols = Object.getOwnPropertySymbols;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __propIsEnum = Object.prototype.propertyIsEnumerable;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues = (a, b) => {
-  for (var prop in b || (b = {}))
-    if (__hasOwnProp.call(b, prop))
-      __defNormalProp(a, prop, b[prop]);
-  if (__getOwnPropSymbols)
-    for (var prop of __getOwnPropSymbols(b)) {
-      if (__propIsEnum.call(b, prop))
-        __defNormalProp(a, prop, b[prop]);
-    }
-  return a;
-};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -40,26 +26,6 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
   mod
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var __async = (__this, __arguments, generator) => {
-  return new Promise((resolve, reject) => {
-    var fulfilled = (value) => {
-      try {
-        step(generator.next(value));
-      } catch (e) {
-        reject(e);
-      }
-    };
-    var rejected = (value) => {
-      try {
-        step(generator.throw(value));
-      } catch (e) {
-        reject(e);
-      }
-    };
-    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
-    step((generator = generator.apply(__this, __arguments)).next());
-  });
-};
 
 // vue/index.ts
 var vue_exports = {};
@@ -111,6 +77,9 @@ function useChat({
     key,
     () => store[key] || initialMessages
   );
+  const { data: isLoading, mutate: mutateLoading } = useSWRV(
+    `${key}-loading`
+  );
   data.value || (data.value = initialMessages);
   const mutate = (data2) => {
     store[key] = data2;
@@ -118,102 +87,104 @@ function useChat({
   };
   const messages = data;
   const error = (0, import_vue.ref)(void 0);
-  const isLoading = (0, import_vue.ref)(false);
   let abortController = null;
-  function triggerRequest(messagesSnapshot, options) {
-    return __async(this, null, function* () {
-      try {
-        isLoading.value = true;
-        abortController = new AbortController();
-        const previousMessages = messages.value;
-        mutate(messagesSnapshot);
-        const res = yield fetch(api, {
-          method: "POST",
-          body: JSON.stringify(__spreadValues(__spreadValues({
-            messages: sendExtraMessageFields ? messagesSnapshot : messagesSnapshot.map(({ role, content }) => ({
-              role,
-              content
-            }))
-          }, body), options == null ? void 0 : options.body)),
-          headers: __spreadValues(__spreadValues({}, headers), options == null ? void 0 : options.headers),
-          signal: abortController.signal,
-          credentials
-        }).catch((err) => {
-          mutate(previousMessages);
+  async function triggerRequest(messagesSnapshot, options) {
+    try {
+      mutateLoading(() => true);
+      abortController = new AbortController();
+      const previousMessages = messages.value;
+      mutate(messagesSnapshot);
+      const res = await fetch(api, {
+        method: "POST",
+        body: JSON.stringify({
+          messages: sendExtraMessageFields ? messagesSnapshot : messagesSnapshot.map(({ role, content }) => ({
+            role,
+            content
+          })),
+          ...body,
+          ...options == null ? void 0 : options.body
+        }),
+        headers: {
+          ...headers,
+          ...options == null ? void 0 : options.headers
+        },
+        signal: abortController.signal,
+        credentials
+      }).catch((err) => {
+        mutate(previousMessages);
+        throw err;
+      });
+      if (onResponse) {
+        try {
+          await onResponse(res);
+        } catch (err) {
           throw err;
-        });
-        if (onResponse) {
-          try {
-            yield onResponse(res);
-          } catch (err) {
-            throw err;
-          }
-        }
-        if (!res.ok) {
-          mutate(previousMessages);
-          throw new Error(
-            (yield res.text()) || "Failed to fetch the chat response."
-          );
         }
-        if (!res.body) {
-          throw new Error("The response body is empty.");
-        }
-        let result = "";
-        const createdAt = /* @__PURE__ */ new Date();
-        const replyId = nanoid();
-        const reader = res.body.getReader();
-        const decoder = createChunkDecoder();
-        while (true) {
-          const { done, value } = yield reader.read();
-          if (done) {
-            break;
-          }
-          result += decoder(value);
-          mutate([
-            ...messagesSnapshot,
-            {
-              id: replyId,
-              createdAt,
-              content: result,
-              role: "assistant"
-            }
-          ]);
-          if (abortController === null) {
-            reader.cancel();
-            break;
-          }
+      }
+      if (!res.ok) {
+        mutate(previousMessages);
+        throw new Error(
+          await res.text() || "Failed to fetch the chat response."
+        );
+      }
+      if (!res.body) {
+        throw new Error("The response body is empty.");
+      }
+      let result = "";
+      const createdAt = /* @__PURE__ */ new Date();
+      const replyId = nanoid();
+      const reader = res.body.getReader();
+      const decoder = createChunkDecoder();
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) {
+          break;
         }
-        if (onFinish) {
-          onFinish({
+        result += decoder(value);
+        mutate([
+          ...messagesSnapshot,
+          {
             id: replyId,
             createdAt,
             content: result,
             role: "assistant"
-          });
+          }
+        ]);
+        if (abortController === null) {
+          reader.cancel();
+          break;
         }
+      }
+      if (onFinish) {
+        onFinish({
+          id: replyId,
+          createdAt,
+          content: result,
+          role: "assistant"
+        });
+      }
+      abortController = null;
+      return result;
+    } catch (err) {
+      if (err.name === "AbortError") {
         abortController = null;
-        return result;
-      } catch (err) {
-        if (err.name === "AbortError") {
-          abortController = null;
-          return null;
-        }
-        if (onError && err instanceof Error) {
-          onError(err);
-        }
-        error.value = err;
-      } finally {
-        isLoading.value = false;
+        return null;
       }
-    });
+      if (onError && err instanceof Error) {
+        onError(err);
+      }
+      error.value = err;
+    } finally {
+      mutateLoading(() => false);
+    }
   }
-  const append = (message, options) => __async(this, null, function* () {
+  const append = async (message, options) => {
     if (!message.id) {
       message.id = nanoid();
     }
     return triggerRequest(messages.value.concat(message), options);
-  });
-  const reload = (options) => __async(this, null, function* () {
+  };
+  const reload = async (options) => {
     const messagesSnapshot = messages.value;
     if (messagesSnapshot.length === 0)
       return null;
@@ -222,7 +193,7 @@ function useChat({
       return triggerRequest(messagesSnapshot.slice(0, -1), options);
     }
     return triggerRequest(messagesSnapshot, options);
-  });
+  };
   const stop = () => {
     if (abortController) {
       abortController.abort();
@@ -281,6 +252,9 @@ function useCompletion({
     key,
     () => store2[key] || initialCompletion
   );
+  const { data: isLoading, mutate: mutateLoading } = useSWRV2(
+    `${key}-loading`
+  );
   data.value || (data.value = initialCompletion);
   const mutate = (data2) => {
     store2[key] = data2;
@@ -288,77 +262,79 @@ function useCompletion({
   };
   const completion = data;
   const error = (0, import_vue2.ref)(void 0);
-  const isLoading = (0, import_vue2.ref)(false);
   let abortController = null;
-  function triggerRequest(prompt, options) {
-    return __async(this, null, function* () {
-      try {
-        isLoading.value = true;
-        abortController = new AbortController();
-        mutate("");
-        const res = yield fetch(api, {
-          method: "POST",
-          body: JSON.stringify(__spreadValues(__spreadValues({
-            prompt
-          }, body), options == null ? void 0 : options.body)),
-          headers: __spreadValues(__spreadValues({}, headers), options == null ? void 0 : options.headers),
-          signal: abortController.signal,
-          credentials
-        }).catch((err) => {
+  async function triggerRequest(prompt, options) {
+    try {
+      mutateLoading(() => true);
+      abortController = new AbortController();
+      mutate("");
+      const res = await fetch(api, {
+        method: "POST",
+        body: JSON.stringify({
+          prompt,
+          ...body,
+          ...options == null ? void 0 : options.body
+        }),
+        headers: {
+          ...headers,
+          ...options == null ? void 0 : options.headers
+        },
+        signal: abortController.signal,
+        credentials
+      }).catch((err) => {
+        throw err;
+      });
+      if (onResponse) {
+        try {
+          await onResponse(res);
+        } catch (err) {
           throw err;
-        });
-        if (onResponse) {
-          try {
-            yield onResponse(res);
-          } catch (err) {
-            throw err;
-          }
-        }
-        if (!res.ok) {
-          throw new Error(
-            (yield res.text()) || "Failed to fetch the chat response."
-          );
-        }
-        if (!res.body) {
-          throw new Error("The response body is empty.");
         }
-        let result = "";
-        const reader = res.body.getReader();
-        const decoder = createChunkDecoder();
-        while (true) {
-          const { done, value } = yield reader.read();
-          if (done) {
-            break;
-          }
-          result += decoder(value);
-          mutate(result);
-          if (abortController === null) {
-            reader.cancel();
-            break;
-          }
+      }
+      if (!res.ok) {
+        throw new Error(
+          await res.text() || "Failed to fetch the chat response."
+        );
+      }
+      if (!res.body) {
+        throw new Error("The response body is empty.");
+      }
+      let result = "";
+      const reader = res.body.getReader();
+      const decoder = createChunkDecoder();
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) {
+          break;
         }
-        if (onFinish) {
-          onFinish(prompt, result);
+        result += decoder(value);
+        mutate(result);
+        if (abortController === null) {
+          reader.cancel();
+          break;
         }
+      }
+      if (onFinish) {
+        onFinish(prompt, result);
+      }
+      abortController = null;
+      return result;
+    } catch (err) {
+      if (err.name === "AbortError") {
         abortController = null;
-        return result;
-      } catch (err) {
-        if (err.name === "AbortError") {
-          abortController = null;
-          return null;
-        }
-        if (onError && error instanceof Error) {
-          onError(error);
-        }
-        error.value = err;
-      } finally {
-        isLoading.value = false;
+        return null;
       }
-    });
+      if (onError && error instanceof Error) {
+        onError(error);
+      }
+      error.value = err;
+    } finally {
+      mutateLoading(() => false);
+    }
   }
-  const complete = (prompt, options) => __async(this, null, function* () {
+  const complete = async (prompt, options) => {
     return triggerRequest(prompt, options);
-  });
+  };
   const stop = () => {
     if (abortController) {
       abortController.abort();