ai 2.1.8 → 2.1.10

package/react/dist/index.js

@@ -65,6 +65,7 @@ var __async = (__this, __arguments, generator) => {
 // react/index.ts
 var react_exports = {};
 __export(react_exports, {
+  Tokens: () => Tokens,
   useChat: () => useChat,
   useCompletion: () => useCompletion
 });
@@ -91,15 +92,97 @@ function createChunkDecoder() {
 }
 
 // react/use-chat.ts
+var getStreamedResponse = (api, chatRequest, mutate, extraMetadataRef, messagesRef, abortControllerRef, onFinish, onResponse, sendExtraMessageFields) => __async(void 0, null, function* () {
+  var _a, _b;
+  const previousMessages = messagesRef.current;
+  mutate(chatRequest.messages, false);
+  const res = yield fetch(api, __spreadValues({
+    method: "POST",
+    body: JSON.stringify(__spreadValues(__spreadValues(__spreadValues(__spreadValues({
+      messages: sendExtraMessageFields ? chatRequest.messages : chatRequest.messages.map(
+        ({ role, content, name, function_call }) => __spreadValues(__spreadValues({
+          role,
+          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
+    })),
+    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) => {
+    mutate(previousMessages, false);
+    throw err;
+  });
+  if (onResponse) {
+    try {
+      yield onResponse(res);
+    } catch (err) {
+      throw err;
+    }
+  }
+  if (!res.ok) {
+    mutate(previousMessages, false);
+    throw new Error((yield res.text()) || "Failed to fetch the chat response.");
+  }
+  if (!res.body) {
+    throw new Error("The response body is empty.");
+  }
+  let streamedResponse = "";
+  const createdAt = /* @__PURE__ */ new Date();
+  const replyId = nanoid();
+  const reader = res.body.getReader();
+  const decode = createChunkDecoder();
+  let responseMessage = {
+    id: replyId,
+    createdAt,
+    content: "",
+    role: "assistant"
+  };
+  while (true) {
+    const { done, value } = yield reader.read();
+    if (done) {
+      break;
+    }
+    streamedResponse += decode(value);
+    if (streamedResponse.startsWith('{"function_call":')) {
+      responseMessage["function_call"] = streamedResponse;
+    } else {
+      responseMessage["content"] = streamedResponse;
+    }
+    mutate([...chatRequest.messages, __spreadValues({}, responseMessage)], false);
+    if (abortControllerRef.current === null) {
+      reader.cancel();
+      break;
+    }
+  }
+  if (streamedResponse.startsWith('{"function_call":')) {
+    const parsedFunctionCall = JSON.parse(streamedResponse).function_call;
+    responseMessage["function_call"] = parsedFunctionCall;
+    mutate([...chatRequest.messages, __spreadValues({}, responseMessage)]);
+  }
+  if (onFinish) {
+    onFinish(responseMessage);
+  }
+  return responseMessage;
+});
 function useChat({
   api = "/api/chat",
   id,
   initialMessages = [],
   initialInput = "",
   sendExtraMessageFields,
+  experimental_onFunctionCall,
   onResponse,
   onFinish,
   onError,
+  credentials,
   headers,
   body
 } = {}) {
@@ -115,92 +198,52 @@ function useChat({
   }, [messages]);
   const abortControllerRef = (0, import_react.useRef)(null);
   const extraMetadataRef = (0, import_react.useRef)({
+    credentials,
     headers,
     body
   });
   (0, import_react.useEffect)(() => {
     extraMetadataRef.current = {
+      credentials,
       headers,
       body
     };
-  }, [headers, body]);
+  }, [credentials, headers, body]);
   const { error, trigger, isMutating } = (0, import_mutation.default)(
     [api, chatId],
-    (_0, _1) => __async(this, [_0, _1], function* (_, { arg }) {
+    (_0, _1) => __async(this, [_0, _1], function* (_, { arg: initialChatRequest }) {
       try {
-        const { messages: messagesSnapshot, options } = arg;
         const abortController = new AbortController();
         abortControllerRef.current = abortController;
-        const previousMessages = messagesRef.current;
-        mutate(messagesSnapshot, false);
-        const res = yield fetch(api, {
-          method: "POST",
-          body: JSON.stringify(__spreadValues(__spreadValues({
-            messages: sendExtraMessageFields ? messagesSnapshot : messagesSnapshot.map(({ role, content }) => ({
-              role,
-              content
-            }))
-          }, extraMetadataRef.current.body), options == null ? void 0 : options.body)),
-          headers: __spreadValues(__spreadValues({}, extraMetadataRef.current.headers), options == null ? void 0 : options.headers),
-          signal: abortController.signal
-        }).catch((err) => {
-          mutate(previousMessages, false);
-          throw err;
-        });
-        if (onResponse) {
-          try {
-            yield onResponse(res);
-          } catch (err) {
-            throw err;
-          }
-        }
-        if (!res.ok) {
-          mutate(previousMessages, false);
-          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 decode = createChunkDecoder();
+        let chatRequest = initialChatRequest;
         while (true) {
-          const { done, value } = yield reader.read();
-          if (done) {
-            break;
-          }
-          result += decode(value);
-          mutate(
-            [
-              ...messagesSnapshot,
-              {
-                id: replyId,
-                createdAt,
-                content: result,
-                role: "assistant"
-              }
-            ],
-            false
+          const streamedResponseMessage = yield getStreamedResponse(
+            api,
+            chatRequest,
+            mutate,
+            extraMetadataRef,
+            messagesRef,
+            abortControllerRef,
+            onFinish,
+            onResponse,
+            sendExtraMessageFields
           );
-          if (abortControllerRef.current === null) {
-            reader.cancel();
+          if (streamedResponseMessage.function_call === void 0 || typeof streamedResponseMessage.function_call === "string") {
             break;
           }
-        }
-        if (onFinish) {
-          onFinish({
-            id: replyId,
-            createdAt,
-            content: result,
-            role: "assistant"
-          });
+          if (experimental_onFunctionCall) {
+            const functionCall = streamedResponseMessage.function_call;
+            const functionCallResponse = yield experimental_onFunctionCall(
+              messagesRef.current,
+              functionCall
+            );
+            if (functionCallResponse === void 0)
+              break;
+            chatRequest = functionCallResponse;
+          }
         }
         abortControllerRef.current = null;
-        return result;
+        return null;
       } catch (err) {
         if (err.name === "AbortError") {
           abortControllerRef.current = null;
@@ -218,32 +261,35 @@ function useChat({
     }
   );
   const append = (0, import_react.useCallback)(
-    (message, options) => __async(this, null, function* () {
+    (_0, ..._1) => __async(this, [_0, ..._1], function* (message, { options, functions, function_call } = {}) {
       if (!message.id) {
         message.id = nanoid();
       }
-      return trigger({
+      const chatRequest = __spreadValues(__spreadValues({
         messages: messagesRef.current.concat(message),
         options
-      });
+      }, functions !== void 0 && { functions }), function_call !== void 0 && { function_call });
+      return trigger(chatRequest);
     }),
     [trigger]
   );
   const reload = (0, import_react.useCallback)(
-    (options) => __async(this, null, function* () {
+    (..._0) => __async(this, [..._0], function* ({ options, functions, function_call } = {}) {
       if (messagesRef.current.length === 0)
         return null;
       const lastMessage = messagesRef.current[messagesRef.current.length - 1];
       if (lastMessage.role === "assistant") {
-        return trigger({
+        const chatRequest2 = __spreadValues(__spreadValues({
           messages: messagesRef.current.slice(0, -1),
           options
-        });
+        }, functions !== void 0 && { functions }), function_call !== void 0 && { function_call });
+        return trigger(chatRequest2);
       }
-      return trigger({
+      const chatRequest = __spreadValues(__spreadValues({
         messages: messagesRef.current,
         options
-      });
+      }, functions !== void 0 && { functions }), function_call !== void 0 && { function_call });
+      return trigger(chatRequest);
     }),
     [trigger]
   );
@@ -262,18 +308,21 @@ function useChat({
   );
   const [input, setInput] = (0, import_react.useState)(initialInput);
   const handleSubmit = (0, import_react.useCallback)(
-    (e, metadata) => {
+    (e, { options, functions, function_call } = {}, metadata) => {
       if (metadata) {
         extraMetadataRef.current = __spreadValues(__spreadValues({}, extraMetadataRef.current), metadata);
       }
       e.preventDefault();
       if (!input)
         return;
-      append({
-        content: input,
-        role: "user",
-        createdAt: /* @__PURE__ */ new Date()
-      });
+      append(
+        {
+          content: input,
+          role: "user",
+          createdAt: /* @__PURE__ */ new Date()
+        },
+        { options, functions, function_call }
+      );
       setInput("");
     },
     [input, append]
@@ -305,6 +354,7 @@ function useCompletion({
   id,
   initialCompletion = "",
   initialInput = "",
+  credentials,
   headers,
   body,
   onResponse,
@@ -319,15 +369,17 @@ function useCompletion({
   const completion = data;
   const [abortController, setAbortController] = (0, import_react2.useState)(null);
   const extraMetadataRef = (0, import_react2.useRef)({
+    credentials,
     headers,
     body
   });
   (0, import_react2.useEffect)(() => {
     extraMetadataRef.current = {
+      credentials,
       headers,
       body
     };
-  }, [headers, body]);
+  }, [credentials, headers, body]);
   const { error, trigger, isMutating } = (0, import_mutation2.default)(
     [api, completionId],
     (_0, _1) => __async(this, [_0, _1], function* (_, { arg }) {
@@ -341,6 +393,7 @@ function useCompletion({
           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) => {
@@ -444,8 +497,33 @@ function useCompletion({
     isLoading: isMutating
   };
 }
+
+// react/tokens.tsx
+var import_react3 = require("react");
+var import_jsx_runtime = require("react/jsx-runtime");
+function Tokens(props) {
+  return __async(this, null, function* () {
+    const { stream } = props;
+    const reader = stream.getReader();
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react3.Suspense, { children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(RecursiveTokens, { reader }) });
+  });
+}
+function RecursiveTokens(_0) {
+  return __async(this, arguments, function* ({ reader }) {
+    const { done, value } = yield reader.read();
+    if (done) {
+      return null;
+    }
+    const text = new TextDecoder().decode(value);
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(import_jsx_runtime.Fragment, { children: [
+      text,
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react3.Suspense, { fallback: null, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(RecursiveTokens, { reader }) })
+    ] });
+  });
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  Tokens,
   useChat,
   useCompletion
 });