ai 2.1.26 → 2.1.27

package/react/dist/index.js

@@ -4,22 +4,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 });
@@ -41,26 +27,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());
-  });
-};
 
 // react/index.ts
 var react_exports = {};
@@ -72,7 +38,6 @@ module.exports = __toCommonJS(react_exports);
 
 // react/use-chat.ts
 var import_react = require("react");
-var import_mutation = __toESM(require("swr/mutation"));
 var import_swr = __toESM(require("swr"));
 
 // shared/utils.ts
@@ -91,44 +56,54 @@ function createChunkDecoder() {
 }
 
 // react/use-chat.ts
-var getStreamedResponse = (api, chatRequest, mutate, extraMetadataRef, messagesRef, abortControllerRef, onFinish, onResponse, sendExtraMessageFields) => __async(void 0, null, function* () {
+var getStreamedResponse = async (api, chatRequest, mutate, extraMetadataRef, messagesRef, abortControllerRef, onFinish, onResponse, sendExtraMessageFields) => {
   var _a, _b;
   const previousMessages = messagesRef.current;
   mutate(chatRequest.messages, false);
-  const res = yield fetch(api, __spreadValues({
+  const res = await fetch(api, {
     method: "POST",
-    body: JSON.stringify(__spreadValues(__spreadValues(__spreadValues(__spreadValues({
+    body: JSON.stringify({
       messages: sendExtraMessageFields ? chatRequest.messages : chatRequest.messages.map(
-        ({ role, content, name, function_call }) => __spreadValues(__spreadValues({
+        ({ role, content, name, function_call }) => ({
           role,
-          content
-        }, name !== void 0 && { name }), function_call !== void 0 && {
-          function_call
+          content,
+          ...name !== void 0 && { name },
+          ...function_call !== void 0 && {
+            function_call
+          }
         })
-      )
-    }, extraMetadataRef.current.body), (_a = chatRequest.options) == null ? void 0 : _a.body), chatRequest.functions !== void 0 && {
-      functions: chatRequest.functions
-    }), chatRequest.function_call !== void 0 && {
-      function_call: chatRequest.function_call
-    })),
+      ),
+      ...extraMetadataRef.current.body,
+      ...(_a = chatRequest.options) == null ? void 0 : _a.body,
+      ...chatRequest.functions !== void 0 && {
+        functions: chatRequest.functions
+      },
+      ...chatRequest.function_call !== void 0 && {
+        function_call: chatRequest.function_call
+      }
+    }),
     credentials: extraMetadataRef.current.credentials,
-    headers: __spreadValues(__spreadValues({}, extraMetadataRef.current.headers), (_b = chatRequest.options) == null ? void 0 : _b.headers)
-  }, abortControllerRef.current !== null && {
-    signal: abortControllerRef.current.signal
-  })).catch((err) => {
+    headers: {
+      ...extraMetadataRef.current.headers,
+      ...(_b = chatRequest.options) == null ? void 0 : _b.headers
+    },
+    ...abortControllerRef.current !== null && {
+      signal: abortControllerRef.current.signal
+    }
+  }).catch((err) => {
     mutate(previousMessages, false);
     throw err;
   });
   if (onResponse) {
     try {
-      yield onResponse(res);
+      await onResponse(res);
     } catch (err) {
       throw err;
     }
   }
   if (!res.ok) {
     mutate(previousMessages, false);
-    throw new Error((yield res.text()) || "Failed to fetch the chat response.");
+    throw new Error(await res.text() || "Failed to fetch the chat response.");
   }
   if (!res.body) {
     throw new Error("The response body is empty.");
@@ -145,7 +120,7 @@ var getStreamedResponse = (api, chatRequest, mutate, extraMetadataRef, messagesR
     role: "assistant"
   };
   while (true) {
-    const { done, value } = yield reader.read();
+    const { done, value } = await reader.read();
     if (done) {
       break;
     }
@@ -155,7 +130,7 @@ var getStreamedResponse = (api, chatRequest, mutate, extraMetadataRef, messagesR
     } else {
       responseMessage["content"] = streamedResponse;
     }
-    mutate([...chatRequest.messages, __spreadValues({}, responseMessage)], false);
+    mutate([...chatRequest.messages, { ...responseMessage }], false);
     if (abortControllerRef.current === null) {
       reader.cancel();
       break;
@@ -164,13 +139,13 @@ var getStreamedResponse = (api, chatRequest, mutate, extraMetadataRef, messagesR
   if (streamedResponse.startsWith('{"function_call":')) {
     const parsedFunctionCall = JSON.parse(streamedResponse).function_call;
     responseMessage["function_call"] = parsedFunctionCall;
-    mutate([...chatRequest.messages, __spreadValues({}, responseMessage)]);
+    mutate([...chatRequest.messages, { ...responseMessage }]);
   }
   if (onFinish) {
     onFinish(responseMessage);
   }
   return responseMessage;
-});
+};
 function useChat({
   api = "/api/chat",
   id,
@@ -190,6 +165,10 @@ function useChat({
   const { data, mutate } = (0, import_swr.default)([api, chatId], null, {
     fallbackData: initialMessages
   });
+  const { data: isLoading = false, mutate: mutateLoading } = (0, import_swr.default)(
+    [api, chatId, "loading"],
+    null
+  );
   const messages = data;
   const messagesRef = (0, import_react.useRef)(messages);
   (0, import_react.useEffect)(() => {
@@ -208,91 +187,88 @@ function useChat({
       body
     };
   }, [credentials, headers, body]);
-  const { error, trigger, isMutating } = (0, import_mutation.default)(
-    [api, chatId],
-    (_0, _1) => __async(this, [_0, _1], function* (_, { arg: initialChatRequest }) {
-      try {
-        const abortController = new AbortController();
-        abortControllerRef.current = abortController;
-        let chatRequest = initialChatRequest;
-        while (true) {
-          const streamedResponseMessage = yield getStreamedResponse(
-            api,
-            chatRequest,
-            mutate,
-            extraMetadataRef,
-            messagesRef,
-            abortControllerRef,
-            onFinish,
-            onResponse,
-            sendExtraMessageFields
-          );
-          if (streamedResponseMessage.function_call === void 0 || typeof streamedResponseMessage.function_call === "string") {
+  const [error, setError] = (0, import_react.useState)();
+  async function triggerRequest(chatRequest) {
+    try {
+      mutateLoading(true);
+      const abortController = new AbortController();
+      abortControllerRef.current = abortController;
+      while (true) {
+        const streamedResponseMessage = await getStreamedResponse(
+          api,
+          chatRequest,
+          mutate,
+          extraMetadataRef,
+          messagesRef,
+          abortControllerRef,
+          onFinish,
+          onResponse,
+          sendExtraMessageFields
+        );
+        if (streamedResponseMessage.function_call === void 0 || typeof streamedResponseMessage.function_call === "string") {
+          break;
+        }
+        if (experimental_onFunctionCall) {
+          const functionCall = streamedResponseMessage.function_call;
+          const functionCallResponse = await experimental_onFunctionCall(messagesRef.current, functionCall);
+          if (functionCallResponse === void 0)
             break;
-          }
-          if (experimental_onFunctionCall) {
-            const functionCall = streamedResponseMessage.function_call;
-            const functionCallResponse = yield experimental_onFunctionCall(
-              messagesRef.current,
-              functionCall
-            );
-            if (functionCallResponse === void 0)
-              break;
-            chatRequest = functionCallResponse;
-          }
+          chatRequest = functionCallResponse;
         }
+      }
+      abortControllerRef.current = null;
+      return null;
+    } catch (err) {
+      if (err.name === "AbortError") {
         abortControllerRef.current = null;
         return null;
-      } catch (err) {
-        if (err.name === "AbortError") {
-          abortControllerRef.current = null;
-          return null;
-        }
-        if (onError && err instanceof Error) {
-          onError(err);
-        }
-        throw err;
       }
-    }),
-    {
-      populateCache: false,
-      revalidate: false,
-      // @ts-expect-error - SWR tries to be clever with the throwOnError type
-      throwOnError: Boolean(!onError)
+      if (onError && err instanceof Error) {
+        onError(err);
+      }
+      setError(err);
+    } finally {
+      mutateLoading(false);
     }
-  );
+  }
   const append = (0, import_react.useCallback)(
-    (_0, ..._1) => __async(this, [_0, ..._1], function* (message, { options, functions, function_call } = {}) {
+    async (message, { options, functions, function_call } = {}) => {
       if (!message.id) {
         message.id = nanoid();
       }
-      const chatRequest = __spreadValues(__spreadValues({
+      const chatRequest = {
         messages: messagesRef.current.concat(message),
-        options
-      }, functions !== void 0 && { functions }), function_call !== void 0 && { function_call });
-      return trigger(chatRequest);
-    }),
-    [trigger]
+        options,
+        ...functions !== void 0 && { functions },
+        ...function_call !== void 0 && { function_call }
+      };
+      return triggerRequest(chatRequest);
+    },
+    [triggerRequest]
   );
   const reload = (0, import_react.useCallback)(
-    (..._0) => __async(this, [..._0], function* ({ options, functions, function_call } = {}) {
+    async ({ options, functions, function_call } = {}) => {
       if (messagesRef.current.length === 0)
         return null;
       const lastMessage = messagesRef.current[messagesRef.current.length - 1];
       if (lastMessage.role === "assistant") {
-        const chatRequest2 = __spreadValues(__spreadValues({
+        const chatRequest2 = {
           messages: messagesRef.current.slice(0, -1),
-          options
-        }, functions !== void 0 && { functions }), function_call !== void 0 && { function_call });
-        return trigger(chatRequest2);
+          options,
+          ...functions !== void 0 && { functions },
+          ...function_call !== void 0 && { function_call }
+        };
+        return triggerRequest(chatRequest2);
       }
-      const chatRequest = __spreadValues(__spreadValues({
+      const chatRequest = {
         messages: messagesRef.current,
-        options
-      }, functions !== void 0 && { functions }), function_call !== void 0 && { function_call });
-      return trigger(chatRequest);
-    }),
-    [trigger]
+        options,
+        ...functions !== void 0 && { functions },
+        ...function_call !== void 0 && { function_call }
+      };
+      return triggerRequest(chatRequest);
+    },
+    [triggerRequest]
   );
   const stop = (0, import_react.useCallback)(() => {
     if (abortControllerRef.current) {
@@ -311,7 +287,10 @@ function useChat({
   const handleSubmit = (0, import_react.useCallback)(
     (e, { options, functions, function_call } = {}, metadata) => {
       if (metadata) {
-        extraMetadataRef.current = __spreadValues(__spreadValues({}, extraMetadataRef.current), metadata);
+        extraMetadataRef.current = {
+          ...extraMetadataRef.current,
+          ...metadata
+        };
       }
       e.preventDefault();
       if (!input)
@@ -342,13 +321,12 @@ function useChat({
     setInput,
     handleInputChange,
     handleSubmit,
-    isLoading: isMutating
+    isLoading
   };
 }
 
 // react/use-completion.ts
 var import_react2 = require("react");
-var import_mutation2 = __toESM(require("swr/mutation"));
 var import_swr2 = __toESM(require("swr"));
 function useCompletion({
   api = "/api/completion",
@@ -367,6 +345,11 @@ function useCompletion({
   const { data, mutate } = (0, import_swr2.default)([api, completionId], null, {
     fallbackData: initialCompletion
   });
+  const { data: isLoading = false, mutate: mutateLoading } = (0, import_swr2.default)(
+    [api, completionId, "loading"],
+    null
+  );
+  const [error, setError] = (0, import_react2.useState)(void 0);
   const completion = data;
   const [abortController, setAbortController] = (0, import_react2.useState)(null);
   const extraMetadataRef = (0, import_react2.useRef)({
@@ -381,78 +364,78 @@ function useCompletion({
       body
     };
   }, [credentials, headers, body]);
-  const { error, trigger, isMutating } = (0, import_mutation2.default)(
-    [api, completionId],
-    (_0, _1) => __async(this, [_0, _1], function* (_, { arg }) {
-      try {
-        const { prompt, options } = arg;
-        const abortController2 = new AbortController();
-        setAbortController(abortController2);
-        mutate("", false);
-        const res = yield fetch(api, {
-          method: "POST",
-          body: JSON.stringify(__spreadValues(__spreadValues({
-            prompt
-          }, extraMetadataRef.current.body), options == null ? void 0 : options.body)),
-          credentials: extraMetadataRef.current.credentials,
-          headers: __spreadValues(__spreadValues({}, extraMetadataRef.current.headers), options == null ? void 0 : options.headers),
-          signal: abortController2.signal
-        }).catch((err) => {
+  async function triggerRequest(prompt, options) {
+    try {
+      mutateLoading(true);
+      const abortController2 = new AbortController();
+      setAbortController(abortController2);
+      mutate("", false);
+      const res = await fetch(api, {
+        method: "POST",
+        body: JSON.stringify({
+          prompt,
+          ...extraMetadataRef.current.body,
+          ...options == null ? void 0 : options.body
+        }),
+        credentials: extraMetadataRef.current.credentials,
+        headers: {
+          ...extraMetadataRef.current.headers,
+          ...options == null ? void 0 : options.headers
+        },
+        signal: abortController2.signal
+      }).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, false);
-          if (abortController2 === 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, false);
+        if (abortController2 === null) {
+          reader.cancel();
+          break;
         }
+      }
+      if (onFinish) {
+        onFinish(prompt, result);
+      }
+      setAbortController(null);
+      return result;
+    } catch (err) {
+      if (err.name === "AbortError") {
         setAbortController(null);
-        return result;
-      } catch (err) {
-        if (err.name === "AbortError") {
-          setAbortController(null);
-          return null;
-        }
-        if (onError && err instanceof Error) {
+        return null;
+      }
+      if (err instanceof Error) {
+        if (onError) {
           onError(err);
         }
-        throw err;
       }
-    }),
-    {
-      populateCache: false,
-      revalidate: false,
-      // @ts-expect-error - SWR tries to be clever with the throwOnError type
-      throwOnError: Boolean(onError)
+      setError(err);
+    } finally {
+      mutateLoading(false);
     }
-  );
+  }
   const stop = (0, import_react2.useCallback)(() => {
     if (abortController) {
       abortController.abort();
@@ -466,13 +449,10 @@ function useCompletion({
     [mutate]
   );
   const complete = (0, import_react2.useCallback)(
-    (prompt, options) => __async(this, null, function* () {
-      return trigger({
-        prompt,
-        options
-      });
-    }),
-    [trigger]
+    async (prompt, options) => {
+      return triggerRequest(prompt, options);
+    },
+    [triggerRequest]
   );
   const [input, setInput] = (0, import_react2.useState)(initialInput);
   const handleSubmit = (0, import_react2.useCallback)(
@@ -497,7 +477,7 @@ function useCompletion({
     setInput,
     handleInputChange,
     handleSubmit,
-    isLoading: isMutating
+    isLoading
   };
 }
 // Annotate the CommonJS export names for ESM import in node: