@ensembleapp/client-sdk 0.0.50 → 0.0.52

package/dist/index.js

@@ -4217,6 +4217,7 @@ function createAuthAwareFetch(tokenRef, onAuthError) {
     return response;
   };
 }
+var EMPTY_HEADERS = {};
 function useChat({
   api,
   threadId,
@@ -4231,7 +4232,13 @@ function useChat({
   onMessage,
   onUsage
 }) {
-  const { baseUrl, token, headers: customHeaders = {} } = api;
+  const { baseUrl, token, headers } = api;
+  const customHeadersKey = JSON.stringify(headers ?? {});
+  const customHeaders = useMemo(
+    () => headers ?? EMPTY_HEADERS,
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    [customHeadersKey]
+  );
   const dataContextRef = useRef(dataContext);
   dataContextRef.current = dataContext;
   const tokenRef = useRef(token);
@@ -4288,8 +4295,11 @@ function useChat({
     },
     onData: (dataPart) => {
       onData?.(dataPart);
-      if (dataPart && typeof dataPart === "object" && "inputTokens" in dataPart) {
-        onUsage?.(dataPart);
+      if (dataPart && typeof dataPart === "object") {
+        const part = dataPart;
+        if (part.type === "data-usage" && part.data) {
+          onUsage?.(part.data);
+        }
       }
     }
   });
@@ -4310,8 +4320,14 @@ function useChat({
     [sendMessage]
   );
   const [isLoadingInitial, setIsLoadingInitial] = useState(true);
+  const hasFetchedRef = useRef(false);
   useEffect(() => {
+    hasFetchedRef.current = false;
+  }, [threadId]);
+  useEffect(() => {
+    if (hasFetchedRef.current) return;
     const fetchInitialMessages = async () => {
+      hasFetchedRef.current = true;
       try {
         const response = await authFetch(`${baseUrl}/chat/messages`, {
           method: "GET",
@@ -4330,13 +4346,14 @@ function useChat({
         }
       } catch (error) {
         console.error("Error fetching initial messages:", error);
+        hasFetchedRef.current = false;
         onError?.(error);
       } finally {
         setIsLoadingInitial(false);
       }
     };
     fetchInitialMessages();
-  }, [agentExecutionId, agentId, authFetch, baseUrl, customHeaders, onError, setMessages, threadId]);
+  }, [agentExecutionId, agentId, authFetch, baseUrl, customHeadersKey, threadId]);
   useEffect(() => {
     if (!onMessage || messages.length === 0) return;
     const lastMessage = messages[messages.length - 1];