@ai-sdk/react 1.2.4 → 2.0.0-canary.0

package/dist/index.mjs

@@ -1,173 +1,3 @@
-// src/use-assistant.ts
-import { isAbortError } from "@ai-sdk/provider-utils";
-import {
-  generateId,
-  processAssistantStream
-} from "@ai-sdk/ui-utils";
-import { useCallback, useRef, useState } from "react";
-var getOriginalFetch = () => fetch;
-function useAssistant({
-  api,
-  threadId: threadIdParam,
-  credentials,
-  headers,
-  body,
-  onError,
-  fetch: fetch2
-}) {
-  const [messages, setMessages] = useState([]);
-  const [input, setInput] = useState("");
-  const [currentThreadId, setCurrentThreadId] = useState(
-    void 0
-  );
-  const [status, setStatus] = useState("awaiting_message");
-  const [error, setError] = useState(void 0);
-  const handleInputChange = (event) => {
-    setInput(event.target.value);
-  };
-  const abortControllerRef = useRef(null);
-  const stop = useCallback(() => {
-    if (abortControllerRef.current) {
-      abortControllerRef.current.abort();
-      abortControllerRef.current = null;
-    }
-  }, []);
-  const append = async (message, requestOptions) => {
-    var _a, _b;
-    setStatus("in_progress");
-    setMessages((messages2) => {
-      var _a2;
-      return [
-        ...messages2,
-        {
-          ...message,
-          id: (_a2 = message.id) != null ? _a2 : generateId()
-        }
-      ];
-    });
-    setInput("");
-    const abortController = new AbortController();
-    try {
-      abortControllerRef.current = abortController;
-      const actualFetch = fetch2 != null ? fetch2 : getOriginalFetch();
-      const response = await actualFetch(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 : currentThreadId) != null ? _a : null,
-          message: message.content,
-          // optional request data:
-          data: requestOptions == null ? void 0 : requestOptions.data
-        })
-      });
-      if (!response.ok) {
-        throw new Error(
-          (_b = await response.text()) != null ? _b : "Failed to fetch the assistant response."
-        );
-      }
-      if (response.body == null) {
-        throw new Error("The response body is empty.");
-      }
-      await processAssistantStream({
-        stream: response.body,
-        onAssistantMessagePart(value) {
-          setMessages((messages2) => [
-            ...messages2,
-            {
-              id: value.id,
-              role: value.role,
-              content: value.content[0].text.value,
-              parts: []
-            }
-          ]);
-        },
-        onTextPart(value) {
-          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,
-                parts: lastMessage.parts
-              }
-            ];
-          });
-        },
-        onAssistantControlDataPart(value) {
-          setCurrentThreadId(value.threadId);
-          setMessages((messages2) => {
-            const lastMessage = messages2[messages2.length - 1];
-            lastMessage.id = value.messageId;
-            return [...messages2.slice(0, messages2.length - 1), lastMessage];
-          });
-        },
-        onDataMessagePart(value) {
-          setMessages((messages2) => {
-            var _a2;
-            return [
-              ...messages2,
-              {
-                id: (_a2 = value.id) != null ? _a2 : generateId(),
-                role: "data",
-                content: "",
-                data: value.data,
-                parts: []
-              }
-            ];
-          });
-        },
-        onErrorPart(value) {
-          setError(new Error(value));
-        }
-      });
-    } catch (error2) {
-      if (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, parts: [] }, requestOptions);
-  };
-  const setThreadId = (threadId) => {
-    setCurrentThreadId(threadId);
-    setMessages([]);
-  };
-  return {
-    append,
-    messages,
-    setMessages,
-    threadId: currentThreadId,
-    setThreadId,
-    input,
-    setInput,
-    handleInputChange,
-    submitMessage,
-    status,
-    error,
-    stop
-  };
-}
-
 // src/use-chat.ts
 import {
   callChatApi,
@@ -180,7 +10,7 @@ import {
   shouldResubmitMessages,
   updateToolCallResult
 } from "@ai-sdk/ui-utils";
-import { useCallback as useCallback2, useEffect as useEffect2, useMemo, useRef as useRef2, useState as useState3 } from "react";
+import { useCallback, useEffect as useEffect2, useMemo, useRef, useState as useState2 } from "react";
 import useSWR from "swr";
 
 // src/throttle.ts
@@ -191,9 +21,9 @@ function throttle(fn, waitMs) {
 
 // src/util/use-stable-value.ts
 import { isDeepEqualData } from "@ai-sdk/ui-utils";
-import { useEffect, useState as useState2 } from "react";
+import { useEffect, useState } from "react";
 function useStableValue(latestValue) {
-  const [value, setValue] = useState2(latestValue);
+  const [value, setValue] = useState(latestValue);
   useEffect(() => {
     if (!isDeepEqualData(latestValue, value)) {
       setValue(latestValue);
@@ -219,12 +49,12 @@ function useChat({
   credentials,
   headers,
   body,
-  generateId: generateId2 = generateIdFunc,
+  generateId = generateIdFunc,
   fetch: fetch2,
   keepLastMessageOnError = true,
   experimental_throttle: throttleWaitMs
 } = {}) {
-  const [hookId] = useState3(generateId2);
+  const [hookId] = useState2(generateId);
   const chatId = id != null ? id : hookId;
   const chatKey = typeof api === "string" ? [api, chatId] : chatId;
   const stableInitialMessages = useStableValue(initialMessages != null ? initialMessages : []);
@@ -237,19 +67,19 @@ function useChat({
     null,
     { fallbackData: processedInitialMessages }
   );
-  const messagesRef = useRef2(messages || []);
+  const messagesRef = useRef(messages || []);
   useEffect2(() => {
     messagesRef.current = messages || [];
   }, [messages]);
   const { data: streamData, mutate: mutateStreamData } = useSWR([chatKey, "streamData"], null);
-  const streamDataRef = useRef2(streamData);
+  const streamDataRef = useRef(streamData);
   useEffect2(() => {
     streamDataRef.current = streamData;
   }, [streamData]);
   const { data: status = "ready", mutate: mutateStatus } = useSWR([chatKey, "status"], null);
   const { data: error = void 0, mutate: setError } = useSWR([chatKey, "error"], null);
-  const abortControllerRef = useRef2(null);
-  const extraMetadataRef = useRef2({
+  const abortControllerRef = useRef(null);
+  const extraMetadataRef = useRef({
     credentials,
     headers,
     body
@@ -261,7 +91,7 @@ function useChat({
       body
     };
   }, [credentials, headers, body]);
-  const triggerRequest = useCallback2(
+  const triggerRequest = useCallback(
     async (chatRequest) => {
       var _a, _b;
       mutateStatus("submitted");
@@ -348,7 +178,7 @@ function useChat({
           },
           onToolCall,
           onFinish,
-          generateId: generateId2,
+          generateId,
           fetch: fetch2,
           lastMessage: chatMessages[chatMessages.length - 1]
         });
@@ -394,14 +224,14 @@ function useChat({
       maxSteps,
       messagesRef,
       abortControllerRef,
-      generateId2,
+      generateId,
       fetch2,
       keepLastMessageOnError,
       throttleWaitMs,
       chatId
     ]
   );
-  const append = useCallback2(
+  const append = useCallback(
     async (message, {
       data,
       headers: headers2,
@@ -414,16 +244,16 @@ function useChat({
       );
       const messages2 = messagesRef.current.concat({
         ...message,
-        id: (_a = message.id) != null ? _a : generateId2(),
+        id: (_a = message.id) != null ? _a : generateId(),
         createdAt: (_b = message.createdAt) != null ? _b : /* @__PURE__ */ new Date(),
         experimental_attachments: attachmentsForRequest.length > 0 ? attachmentsForRequest : void 0,
         parts: getMessageParts(message)
       });
       return triggerRequest({ messages: messages2, headers: headers2, body: body2, data });
     },
-    [triggerRequest, generateId2]
+    [triggerRequest, generateId]
   );
-  const reload = useCallback2(
+  const reload = useCallback(
     async ({ data, headers: headers2, body: body2 } = {}) => {
       const messages2 = messagesRef.current;
       if (messages2.length === 0) {
@@ -439,13 +269,13 @@ function useChat({
     },
     [triggerRequest]
   );
-  const stop = useCallback2(() => {
+  const stop = useCallback(() => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
       abortControllerRef.current = null;
     }
   }, []);
-  const setMessages = useCallback2(
+  const setMessages = useCallback(
     (messages2) => {
       if (typeof messages2 === "function") {
         messages2 = messages2(messagesRef.current);
@@ -456,7 +286,7 @@ function useChat({
     },
     [mutate]
   );
-  const setData = useCallback2(
+  const setData = useCallback(
     (data) => {
       if (typeof data === "function") {
         data = data(streamDataRef.current);
@@ -466,8 +296,8 @@ function useChat({
     },
     [mutateStreamData]
   );
-  const [input, setInput] = useState3(initialInput);
-  const handleSubmit = useCallback2(
+  const [input, setInput] = useState2(initialInput);
+  const handleSubmit = useCallback(
     async (event, options = {}, metadata) => {
       var _a;
       (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
@@ -483,7 +313,7 @@ function useChat({
         options.experimental_attachments
       );
       const messages2 = messagesRef.current.concat({
-        id: generateId2(),
+        id: generateId(),
         createdAt: /* @__PURE__ */ new Date(),
         role: "user",
         content: input,
@@ -499,12 +329,12 @@ function useChat({
       triggerRequest(chatRequest);
       setInput("");
     },
-    [input, generateId2, triggerRequest]
+    [input, generateId, triggerRequest]
   );
   const handleInputChange = (e) => {
     setInput(e.target.value);
   };
-  const addToolResult = useCallback2(
+  const addToolResult = useCallback(
     ({ toolCallId, result }) => {
       const currentMessages = messagesRef.current;
       updateToolCallResult({
@@ -512,13 +342,22 @@ function useChat({
         toolCallId,
         toolResult: result
       });
-      mutate(currentMessages, false);
+      mutate(
+        [
+          ...currentMessages.slice(0, currentMessages.length - 1),
+          { ...currentMessages[currentMessages.length - 1] }
+        ],
+        false
+      );
+      if (status === "submitted" || status === "streaming") {
+        return;
+      }
       const lastMessage = currentMessages[currentMessages.length - 1];
       if (isAssistantMessageWithCompletedToolCalls(lastMessage)) {
         triggerRequest({ messages: currentMessages });
       }
     },
-    [mutate, triggerRequest]
+    [mutate, status, triggerRequest]
   );
   return {
     messages: messages != null ? messages : [],
@@ -544,7 +383,7 @@ function useChat({
 import {
   callCompletionApi
 } from "@ai-sdk/ui-utils";
-import { useCallback as useCallback3, useEffect as useEffect3, useId, useRef as useRef3, useState as useState4 } from "react";
+import { useCallback as useCallback2, useEffect as useEffect3, useId, useRef as useRef2, useState as useState3 } from "react";
 import useSWR2 from "swr";
 function useCompletion({
   api = "/api/completion",
@@ -571,10 +410,10 @@ function useCompletion({
     null
   );
   const { data: streamData, mutate: mutateStreamData } = useSWR2([completionId, "streamData"], null);
-  const [error, setError] = useState4(void 0);
+  const [error, setError] = useState3(void 0);
   const completion = data;
-  const [abortController, setAbortController] = useState4(null);
-  const extraMetadataRef = useRef3({
+  const [abortController, setAbortController] = useState3(null);
+  const extraMetadataRef = useRef2({
     credentials,
     headers,
     body
@@ -586,7 +425,7 @@ function useCompletion({
       body
     };
   }, [credentials, headers, body]);
-  const triggerRequest = useCallback3(
+  const triggerRequest = useCallback2(
     async (prompt, options) => callCompletionApi({
       api,
       prompt,
@@ -631,26 +470,26 @@ function useCompletion({
       throttleWaitMs
     ]
   );
-  const stop = useCallback3(() => {
+  const stop = useCallback2(() => {
     if (abortController) {
       abortController.abort();
       setAbortController(null);
     }
   }, [abortController]);
-  const setCompletion = useCallback3(
+  const setCompletion = useCallback2(
     (completion2) => {
       mutate(completion2, false);
     },
     [mutate]
   );
-  const complete = useCallback3(
+  const complete = useCallback2(
     async (prompt, options) => {
       return triggerRequest(prompt, options);
     },
     [triggerRequest]
   );
-  const [input, setInput] = useState4(initialInput);
-  const handleSubmit = useCallback3(
+  const [input, setInput] = useState3(initialInput);
+  const handleSubmit = useCallback2(
     (event) => {
       var _a;
       (_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
@@ -658,7 +497,7 @@ function useCompletion({
     },
     [input, complete]
   );
-  const handleInputChange = useCallback3(
+  const handleInputChange = useCallback2(
     (e) => {
       setInput(e.target.value);
     },
@@ -681,7 +520,7 @@ function useCompletion({
 
 // src/use-object.ts
 import {
-  isAbortError as isAbortError2,
+  isAbortError,
   safeValidateTypes
 } from "@ai-sdk/provider-utils";
 import {
@@ -689,9 +528,9 @@ import {
   isDeepEqualData as isDeepEqualData2,
   parsePartialJson
 } from "@ai-sdk/ui-utils";
-import { useCallback as useCallback4, useId as useId2, useRef as useRef4, useState as useState5 } from "react";
+import { useCallback as useCallback3, useId as useId2, useRef as useRef3, useState as useState4 } from "react";
 import useSWR3 from "swr";
-var getOriginalFetch2 = () => fetch;
+var getOriginalFetch = () => fetch;
 function useObject({
   api,
   id,
@@ -711,10 +550,10 @@ function useObject({
     null,
     { fallbackData: initialValue }
   );
-  const [error, setError] = useState5(void 0);
-  const [isLoading, setIsLoading] = useState5(false);
-  const abortControllerRef = useRef4(null);
-  const stop = useCallback4(() => {
+  const [error, setError] = useState4(void 0);
+  const [isLoading, setIsLoading] = useState4(false);
+  const abortControllerRef = useRef3(null);
+  const stop = useCallback3(() => {
     var _a;
     try {
       (_a = abortControllerRef.current) == null ? void 0 : _a.abort();
@@ -732,7 +571,7 @@ function useObject({
       setError(void 0);
       const abortController = new AbortController();
       abortControllerRef.current = abortController;
-      const actualFetch = fetch2 != null ? fetch2 : getOriginalFetch2();
+      const actualFetch = fetch2 != null ? fetch2 : getOriginalFetch();
       const response = await actualFetch(api, {
         method: "POST",
         headers: {
@@ -780,7 +619,7 @@ function useObject({
         })
       );
     } catch (error2) {
-      if (isAbortError2(error2)) {
+      if (isAbortError(error2)) {
         return;
       }
       if (onError && error2 instanceof Error) {
@@ -801,7 +640,6 @@ function useObject({
 var experimental_useObject = useObject;
 export {
   experimental_useObject,
-  useAssistant,
   useChat,
   useCompletion
 };