@ai-sdk/react 0.0.2 → 0.0.4

package/dist/index.js

@@ -31,15 +31,174 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var src_exports = {};
 __export(src_exports, {
   experimental_useAssistant: () => experimental_useAssistant,
+  experimental_useObject: () => experimental_useObject,
   useAssistant: () => useAssistant,
   useChat: () => useChat,
   useCompletion: () => useCompletion
 });
 module.exports = __toCommonJS(src_exports);
 
-// src/use-chat.ts
+// src/use-assistant.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
 var import_ui_utils = require("@ai-sdk/ui-utils");
 var import_react = require("react");
+function useAssistant({
+  api,
+  threadId: threadIdParam,
+  credentials,
+  headers,
+  body,
+  onError
+}) {
+  const [messages, setMessages] = (0, import_react.useState)([]);
+  const [input, setInput] = (0, import_react.useState)("");
+  const [threadId, setThreadId] = (0, import_react.useState)(void 0);
+  const [status, setStatus] = (0, import_react.useState)("awaiting_message");
+  const [error, setError] = (0, import_react.useState)(void 0);
+  const handleInputChange = (event) => {
+    setInput(event.target.value);
+  };
+  const abortControllerRef = (0, import_react.useRef)(null);
+  const stop = (0, import_react.useCallback)(() => {
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+      abortControllerRef.current = null;
+    }
+  }, []);
+  const append = async (message, requestOptions) => {
+    var _a;
+    setStatus("in_progress");
+    setMessages((messages2) => {
+      var _a2;
+      return [
+        ...messages2,
+        {
+          ...message,
+          id: (_a2 = message.id) != null ? _a2 : (0, import_ui_utils.generateId)()
+        }
+      ];
+    });
+    setInput("");
+    const abortController = new AbortController();
+    try {
+      abortControllerRef.current = abortController;
+      const result = await fetch(api, {
+        method: "POST",
+        credentials,
+        signal: abortController.signal,
+        headers: { "Content-Type": "application/json", ...headers },
+        body: JSON.stringify({
+          ...body,
+          // always use user-provided threadId when available:
+          threadId: (_a = threadIdParam != null ? threadIdParam : threadId) != null ? _a : null,
+          message: message.content,
+          // optional request data:
+          data: requestOptions == null ? void 0 : requestOptions.data
+        })
+      });
+      if (result.body == null) {
+        throw new Error("The response body is empty.");
+      }
+      for await (const { type, value } of (0, import_ui_utils.readDataStream)(
+        result.body.getReader()
+      )) {
+        switch (type) {
+          case "assistant_message": {
+            setMessages((messages2) => [
+              ...messages2,
+              {
+                id: value.id,
+                role: value.role,
+                content: value.content[0].text.value
+              }
+            ]);
+            break;
+          }
+          case "text": {
+            setMessages((messages2) => {
+              const lastMessage = messages2[messages2.length - 1];
+              return [
+                ...messages2.slice(0, messages2.length - 1),
+                {
+                  id: lastMessage.id,
+                  role: lastMessage.role,
+                  content: lastMessage.content + value
+                }
+              ];
+            });
+            break;
+          }
+          case "data_message": {
+            setMessages((messages2) => {
+              var _a2;
+              return [
+                ...messages2,
+                {
+                  id: (_a2 = value.id) != null ? _a2 : (0, import_ui_utils.generateId)(),
+                  role: "data",
+                  content: "",
+                  data: value.data
+                }
+              ];
+            });
+            break;
+          }
+          case "assistant_control_data": {
+            setThreadId(value.threadId);
+            setMessages((messages2) => {
+              const lastMessage = messages2[messages2.length - 1];
+              lastMessage.id = value.messageId;
+              return [...messages2.slice(0, messages2.length - 1), lastMessage];
+            });
+            break;
+          }
+          case "error": {
+            setError(new Error(value));
+            break;
+          }
+        }
+      }
+    } catch (error2) {
+      if ((0, import_provider_utils.isAbortError)(error2) && abortController.signal.aborted) {
+        abortControllerRef.current = null;
+        return;
+      }
+      if (onError && error2 instanceof Error) {
+        onError(error2);
+      }
+      setError(error2);
+    } finally {
+      abortControllerRef.current = null;
+      setStatus("awaiting_message");
+    }
+  };
+  const submitMessage = async (event, requestOptions) => {
+    var _a;
+    (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
+    if (input === "") {
+      return;
+    }
+    append({ role: "user", content: input }, requestOptions);
+  };
+  return {
+    append,
+    messages,
+    setMessages,
+    threadId,
+    input,
+    setInput,
+    handleInputChange,
+    submitMessage,
+    status,
+    error,
+    stop
+  };
+}
+var experimental_useAssistant = useAssistant;
+
+// src/use-chat.ts
+var import_ui_utils2 = require("@ai-sdk/ui-utils");
+var import_react2 = require("react");
 var import_swr = __toESM(require("swr"));
 var getStreamedResponse = async (api, chatRequest, mutate, mutateStreamData, existingData, extraMetadataRef, messagesRef, abortControllerRef, generateId2, streamMode, onFinish, onResponse, onToolCall, sendExtraMessageFields) => {
   var _a, _b;
@@ -69,7 +228,7 @@ var getStreamedResponse = async (api, chatRequest, mutate, mutateStreamData, exi
       ...tool_calls !== void 0 && { tool_calls }
     })
   );
-  return await (0, import_ui_utils.callChatApi)({
+  return await (0, import_ui_utils2.callChatApi)({
     api,
     messages: constructedMessagesPayload,
     body: {
@@ -128,12 +287,12 @@ function useChat({
   credentials,
   headers,
   body,
-  generateId: generateId2 = import_ui_utils.generateId
+  generateId: generateId2 = import_ui_utils2.generateId
 } = {}) {
-  const hookId = (0, import_react.useId)();
+  const hookId = (0, import_react2.useId)();
   const idKey = id != null ? id : hookId;
   const chatKey = typeof api === "string" ? [api, idKey] : idKey;
-  const [initialMessagesFallback] = (0, import_react.useState)([]);
+  const [initialMessagesFallback] = (0, import_react2.useState)([]);
   const { data: messages, mutate } = (0, import_swr.default)(
     [chatKey, "messages"],
     null,
@@ -145,31 +304,31 @@ function useChat({
   );
   const { data: streamData, mutate: mutateStreamData } = (0, import_swr.default)([chatKey, "streamData"], null);
   const { data: error = void 0, mutate: setError } = (0, import_swr.default)([chatKey, "error"], null);
-  const messagesRef = (0, import_react.useRef)(messages || []);
-  (0, import_react.useEffect)(() => {
+  const messagesRef = (0, import_react2.useRef)(messages || []);
+  (0, import_react2.useEffect)(() => {
     messagesRef.current = messages || [];
   }, [messages]);
-  const abortControllerRef = (0, import_react.useRef)(null);
-  const extraMetadataRef = (0, import_react.useRef)({
+  const abortControllerRef = (0, import_react2.useRef)(null);
+  const extraMetadataRef = (0, import_react2.useRef)({
     credentials,
     headers,
     body
   });
-  (0, import_react.useEffect)(() => {
+  (0, import_react2.useEffect)(() => {
     extraMetadataRef.current = {
       credentials,
       headers,
       body
     };
   }, [credentials, headers, body]);
-  const triggerRequest = (0, import_react.useCallback)(
+  const triggerRequest = (0, import_react2.useCallback)(
     async (chatRequest) => {
       try {
         mutateLoading(true);
         setError(void 0);
         const abortController = new AbortController();
         abortControllerRef.current = abortController;
-        await (0, import_ui_utils.processChatStream)({
+        await (0, import_ui_utils2.processChatStream)({
           getStreamedResponse: () => getStreamedResponse(
             api,
             chatRequest,
@@ -240,7 +399,7 @@ function useChat({
       generateId2
     ]
   );
-  const append = (0, import_react.useCallback)(
+  const append = (0, import_react2.useCallback)(
     async (message, {
       options,
       functions,
@@ -265,7 +424,7 @@ function useChat({
     },
     [triggerRequest, generateId2]
   );
-  const reload = (0, import_react.useCallback)(
+  const reload = (0, import_react2.useCallback)(
     async ({
       options,
       functions,
@@ -299,21 +458,21 @@ function useChat({
     },
     [triggerRequest]
   );
-  const stop = (0, import_react.useCallback)(() => {
+  const stop = (0, import_react2.useCallback)(() => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
       abortControllerRef.current = null;
     }
   }, []);
-  const setMessages = (0, import_react.useCallback)(
+  const setMessages = (0, import_react2.useCallback)(
     (messages2) => {
       mutate(messages2, false);
       messagesRef.current = messages2;
     },
     [mutate]
   );
-  const [input, setInput] = (0, import_react.useState)(initialInput);
-  const handleSubmit = (0, import_react.useCallback)(
+  const [input, setInput] = (0, import_react2.useState)(initialInput);
+  const handleSubmit = (0, import_react2.useCallback)(
     (e, options = {}, metadata) => {
       if (metadata) {
         extraMetadataRef.current = {
@@ -393,8 +552,8 @@ function countTrailingAssistantMessages(messages) {
 }
 
 // src/use-completion.ts
-var import_ui_utils2 = require("@ai-sdk/ui-utils");
-var import_react2 = require("react");
+var import_ui_utils3 = require("@ai-sdk/ui-utils");
+var import_react3 = require("react");
 var import_swr2 = __toESM(require("swr"));
 function useCompletion({
   api = "/api/completion",
@@ -409,7 +568,7 @@ function useCompletion({
   onFinish,
   onError
 } = {}) {
-  const hookId = (0, import_react2.useId)();
+  const hookId = (0, import_react3.useId)();
   const completionId = id || hookId;
   const { data, mutate } = (0, import_swr2.default)([api, completionId], null, {
     fallbackData: initialCompletion
@@ -419,23 +578,23 @@ function useCompletion({
     null
   );
   const { data: streamData, mutate: mutateStreamData } = (0, import_swr2.default)([completionId, "streamData"], null);
-  const [error, setError] = (0, import_react2.useState)(void 0);
+  const [error, setError] = (0, import_react3.useState)(void 0);
   const completion = data;
-  const [abortController, setAbortController] = (0, import_react2.useState)(null);
-  const extraMetadataRef = (0, import_react2.useRef)({
+  const [abortController, setAbortController] = (0, import_react3.useState)(null);
+  const extraMetadataRef = (0, import_react3.useRef)({
     credentials,
     headers,
     body
   });
-  (0, import_react2.useEffect)(() => {
+  (0, import_react3.useEffect)(() => {
     extraMetadataRef.current = {
       credentials,
       headers,
       body
     };
   }, [credentials, headers, body]);
-  const triggerRequest = (0, import_react2.useCallback)(
-    async (prompt, options) => (0, import_ui_utils2.callCompletionApi)({
+  const triggerRequest = (0, import_react3.useCallback)(
+    async (prompt, options) => (0, import_ui_utils3.callCompletionApi)({
       api,
       prompt,
       credentials: extraMetadataRef.current.credentials,
@@ -471,26 +630,26 @@ function useCompletion({
       mutateStreamData
     ]
   );
-  const stop = (0, import_react2.useCallback)(() => {
+  const stop = (0, import_react3.useCallback)(() => {
     if (abortController) {
       abortController.abort();
       setAbortController(null);
     }
   }, [abortController]);
-  const setCompletion = (0, import_react2.useCallback)(
+  const setCompletion = (0, import_react3.useCallback)(
     (completion2) => {
       mutate(completion2, false);
     },
     [mutate]
   );
-  const complete = (0, import_react2.useCallback)(
+  const complete = (0, import_react3.useCallback)(
     async (prompt, options) => {
       return triggerRequest(prompt, options);
     },
     [triggerRequest]
   );
-  const [input, setInput] = (0, import_react2.useState)(initialInput);
-  const handleSubmit = (0, import_react2.useCallback)(
+  const [input, setInput] = (0, import_react3.useState)(initialInput);
+  const handleSubmit = (0, import_react3.useCallback)(
     (e) => {
       e.preventDefault();
       if (!input)
@@ -517,166 +676,72 @@ function useCompletion({
   };
 }
 
-// src/use-assistant.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
-var import_ui_utils3 = require("@ai-sdk/ui-utils");
-var import_react3 = require("react");
-function useAssistant({
+// src/use-object.ts
+var import_ui_utils4 = require("@ai-sdk/ui-utils");
+var import_react4 = require("react");
+var import_swr3 = __toESM(require("swr"));
+function useObject({
   api,
-  threadId: threadIdParam,
-  credentials,
-  headers,
-  body,
-  onError
+  id,
+  schema,
+  // required, in the future we will use it for validation
+  initialValue
 }) {
-  const [messages, setMessages] = (0, import_react3.useState)([]);
-  const [input, setInput] = (0, import_react3.useState)("");
-  const [threadId, setThreadId] = (0, import_react3.useState)(void 0);
-  const [status, setStatus] = (0, import_react3.useState)("awaiting_message");
-  const [error, setError] = (0, import_react3.useState)(void 0);
-  const handleInputChange = (event) => {
-    setInput(event.target.value);
-  };
-  const abortControllerRef = (0, import_react3.useRef)(null);
-  const stop = (0, import_react3.useCallback)(() => {
-    if (abortControllerRef.current) {
-      abortControllerRef.current.abort();
-      abortControllerRef.current = null;
-    }
-  }, []);
-  const append = async (message, requestOptions) => {
-    var _a;
-    setStatus("in_progress");
-    setMessages((messages2) => {
-      var _a2;
-      return [
-        ...messages2,
-        {
-          ...message,
-          id: (_a2 = message.id) != null ? _a2 : (0, import_ui_utils3.generateId)()
+  const hookId = (0, import_react4.useId)();
+  const completionId = id != null ? id : hookId;
+  const { data, mutate } = (0, import_swr3.default)(
+    [api, completionId],
+    null,
+    { fallbackData: initialValue }
+  );
+  const [error, setError] = (0, import_react4.useState)(void 0);
+  return {
+    async setInput(input) {
+      var _a;
+      try {
+        const response = await fetch(api, {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify(input)
+        });
+        if (!response.ok) {
+          throw new Error(
+            (_a = await response.text()) != null ? _a : "Failed to fetch the response."
+          );
         }
-      ];
-    });
-    setInput("");
-    const abortController = new AbortController();
-    try {
-      abortControllerRef.current = abortController;
-      const result = await fetch(api, {
-        method: "POST",
-        credentials,
-        signal: abortController.signal,
-        headers: { "Content-Type": "application/json", ...headers },
-        body: JSON.stringify({
-          ...body,
-          // always use user-provided threadId when available:
-          threadId: (_a = threadIdParam != null ? threadIdParam : threadId) != null ? _a : null,
-          message: message.content,
-          // optional request data:
-          data: requestOptions == null ? void 0 : requestOptions.data
-        })
-      });
-      if (result.body == null) {
-        throw new Error("The response body is empty.");
-      }
-      for await (const { type, value } of (0, import_ui_utils3.readDataStream)(
-        result.body.getReader()
-      )) {
-        switch (type) {
-          case "assistant_message": {
-            setMessages((messages2) => [
-              ...messages2,
-              {
-                id: value.id,
-                role: value.role,
-                content: value.content[0].text.value
-              }
-            ]);
-            break;
-          }
-          case "text": {
-            setMessages((messages2) => {
-              const lastMessage = messages2[messages2.length - 1];
-              return [
-                ...messages2.slice(0, messages2.length - 1),
-                {
-                  id: lastMessage.id,
-                  role: lastMessage.role,
-                  content: lastMessage.content + value
-                }
-              ];
-            });
-            break;
-          }
-          case "data_message": {
-            setMessages((messages2) => {
-              var _a2;
-              return [
-                ...messages2,
-                {
-                  id: (_a2 = value.id) != null ? _a2 : (0, import_ui_utils3.generateId)(),
-                  role: "data",
-                  content: "",
-                  data: value.data
-                }
-              ];
-            });
-            break;
-          }
-          case "assistant_control_data": {
-            setThreadId(value.threadId);
-            setMessages((messages2) => {
-              const lastMessage = messages2[messages2.length - 1];
-              lastMessage.id = value.messageId;
-              return [...messages2.slice(0, messages2.length - 1), lastMessage];
-            });
-            break;
-          }
-          case "error": {
-            setError(new Error(value));
-            break;
-          }
+        if (response.body == null) {
+          throw new Error("The response body is empty.");
         }
+        let accumulatedText = "";
+        let latestObject = void 0;
+        response.body.pipeThrough(new TextDecoderStream()).pipeTo(
+          new WritableStream({
+            write(chunk) {
+              accumulatedText += chunk;
+              const currentObject = (0, import_ui_utils4.parsePartialJson)(
+                accumulatedText
+              );
+              if (!(0, import_ui_utils4.isDeepEqualData)(latestObject, currentObject)) {
+                latestObject = currentObject;
+                mutate(currentObject);
+              }
+            }
+          })
+        );
+        setError(void 0);
+      } catch (error2) {
+        setError(error2);
       }
-    } catch (error2) {
-      if ((0, import_provider_utils.isAbortError)(error2) && abortController.signal.aborted) {
-        abortControllerRef.current = null;
-        return;
-      }
-      if (onError && error2 instanceof Error) {
-        onError(error2);
-      }
-      setError(error2);
-    } finally {
-      abortControllerRef.current = null;
-      setStatus("awaiting_message");
-    }
-  };
-  const submitMessage = async (event, requestOptions) => {
-    var _a;
-    (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
-    if (input === "") {
-      return;
-    }
-    append({ role: "user", content: input }, requestOptions);
-  };
-  return {
-    append,
-    messages,
-    setMessages,
-    threadId,
-    input,
-    setInput,
-    handleInputChange,
-    submitMessage,
-    status,
-    error,
-    stop
+    },
+    object: data,
+    error
   };
 }
-var experimental_useAssistant = useAssistant;
+var experimental_useObject = useObject;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   experimental_useAssistant,
+  experimental_useObject,
   useAssistant,
   useChat,
   useCompletion