ai 2.1.26 → 2.1.27

package/vue/dist/index.mjs

@@ -1,40 +1,3 @@
-var __defProp = Object.defineProperty;
-var __getOwnPropSymbols = Object.getOwnPropertySymbols;
-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 __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/use-chat.ts
 import swrv from "swrv";
 import { ref } from "vue";
@@ -77,6 +40,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;
@@ -84,102 +50,104 @@ function useChat({
   };
   const messages = data;
   const error = ref(void 0);
-  const isLoading = 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;
@@ -188,7 +156,7 @@ function useChat({
       return triggerRequest(messagesSnapshot.slice(0, -1), options);
     }
     return triggerRequest(messagesSnapshot, options);
-  });
+  };
   const stop = () => {
     if (abortController) {
       abortController.abort();
@@ -247,6 +215,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;
@@ -254,77 +225,79 @@ function useCompletion({
   };
   const completion = data;
   const error = ref2(void 0);
-  const isLoading = ref2(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();