@athenaintel/react 0.10.13 → 0.10.15

package/dist/index.js

@@ -24558,12 +24558,23 @@ const useAthenaRuntime = (config2) => {
     if (isExistingThread && !hasResumedRef.current) {
       hasResumedRef.current = true;
       (async () => {
+        var _a2;
         try {
           const auth = { apiKey: apiKeyRef.current, token: tokenRef.current };
           const state = await getThreadState(backendUrl, auth, threadId);
           runtime.thread.importExternalState({
             messages: state.messages
           });
+          if (state.status === "running") {
+            try {
+              const lastMessageId = ((_a2 = runtime.thread.getState().messages.at(-1)) == null ? void 0 : _a2.id) ?? null;
+              runtime.thread.unstable_resumeRun({ parentId: lastMessageId });
+            } catch (resumeErr) {
+              if (IS_DEV) {
+                console.error("[AthenaSDK] Failed to resume running thread:", resumeErr);
+              }
+            }
+          }
         } catch (err) {
           if (IS_DEV) {
             console.error("[AthenaSDK] Failed to load thread state:", err);
@@ -24722,6 +24733,68 @@ function useAthenaThreadManager() {
     };
   }, [runtime, activeThreadId, isListLoading, isThreadLoading, switchToThread, switchToNewThread]);
 }
+const POLL_DELAY_MS = 5e3;
+const POLL_INTERVAL_MS = 1e3;
+const POLL_MAX_DURATION_MS = 6e4;
+function useThreadTitlePolling(refresh) {
+  const threadKey = useThread({
+    optional: true,
+    selector: (s) => {
+      var _a2, _b;
+      return ((_a2 = s.metadata) == null ? void 0 : _a2.remoteId) ?? ((_b = s.metadata) == null ? void 0 : _b.id) ?? s.threadId;
+    }
+  }) ?? null;
+  const hasMessages = useThread({
+    optional: true,
+    selector: (s) => s.messages.length > 0
+  }) ?? false;
+  const currentTitle = useThreadList({
+    optional: true,
+    selector: (s) => {
+      const main = s.threadItems[s.mainThreadId];
+      return (main == null ? void 0 : main.title) ?? "";
+    }
+  }) ?? "";
+  const hasTitle = currentTitle.trim().length > 0;
+  const polledThreadsRef = useRef(/* @__PURE__ */ new Set());
+  const refreshRef = useRef(refresh);
+  refreshRef.current = refresh;
+  useEffect(() => {
+    if (!threadKey || hasTitle || !hasMessages) {
+      return;
+    }
+    if (polledThreadsRef.current.has(threadKey)) {
+      return;
+    }
+    polledThreadsRef.current.add(threadKey);
+    let stopped = false;
+    let intervalId = null;
+    let maxTimeoutId = null;
+    const stop = () => {
+      stopped = true;
+      if (intervalId !== null) {
+        clearInterval(intervalId);
+        intervalId = null;
+      }
+      if (maxTimeoutId !== null) {
+        clearTimeout(maxTimeoutId);
+        maxTimeoutId = null;
+      }
+    };
+    const startTimeoutId = setTimeout(() => {
+      if (stopped) return;
+      refreshRef.current();
+      intervalId = setInterval(() => {
+        refreshRef.current();
+      }, POLL_INTERVAL_MS);
+      maxTimeoutId = setTimeout(stop, POLL_MAX_DURATION_MS - POLL_DELAY_MS);
+    }, POLL_DELAY_MS);
+    return () => {
+      clearTimeout(startTimeoutId);
+      stop();
+    };
+  }, [threadKey, hasTitle, hasMessages]);
+}
 function createJSONStorage(getStorage, options) {
   let storage;
   try {
@@ -25264,12 +25337,16 @@ const useAthenaConfigValue = ({
   linkClicksRef.current = linkClicks;
   const citationLinksRef = useRef(citationLinks);
   citationLinksRef.current = citationLinks;
+  const tokenRef = useRef(token);
+  tokenRef.current = token;
   return useMemo(
     () => ({
       backendUrl,
       appUrl,
       apiKey,
-      token,
+      get token() {
+        return tokenRef.current;
+      },
       get linkClicks() {
         return linkClicksRef.current;
       },
@@ -25277,7 +25354,7 @@ const useAthenaConfigValue = ({
         return citationLinksRef.current;
       }
     }),
-    [backendUrl, appUrl, apiKey, token]
+    [backendUrl, appUrl, apiKey]
   );
 };
 function AthenaStandalone({
@@ -25438,6 +25515,7 @@ function AthenaWithThreadList({
   });
   return /* @__PURE__ */ jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsx(ThreadListRefreshContext.Provider, { value: handleRefresh, children: /* @__PURE__ */ jsxs(TooltipProvider, { children: [
     /* @__PURE__ */ jsx(AssetPanelThreadSync, {}),
+    /* @__PURE__ */ jsx(ThreadTitlePoller, { refresh: handleRefresh }),
     children
   ] }) }) }) });
 }
@@ -25450,6 +25528,10 @@ function AssetPanelThreadSync() {
   }, [activeThreadId, setCurrentThread]);
   return null;
 }
+function ThreadTitlePoller({ refresh }) {
+  useThreadTitlePolling(refresh);
+  return null;
+}
 function AthenaProvider({
   children,
   config: config2,
@@ -65481,7 +65563,10 @@ const TooltipIconButton = forwardRef(
 TooltipIconButton.displayName = "TooltipIconButton";
 const MAX_FILE_SIZE = 5 * 1024 * 1024 * 1024;
 function useFileUpload() {
-  const { backendUrl, apiKey, token } = useAthenaConfig();
+  const config2 = useAthenaConfig();
+  const configRef = useRef(config2);
+  configRef.current = config2;
+  const { backendUrl, apiKey } = config2;
   const [isUploading, setIsUploading] = useState(false);
   const [progress, setProgress] = useState(null);
   const [error2, setError] = useState(null);
@@ -65514,9 +65599,10 @@ function useFileUpload() {
         }
         const baseUrl = backendUrl.replace(/\/api\/assistant-ui\/?$/, "");
         const uploadUrl = `${baseUrl}/api/upload/`;
+        const currentToken = configRef.current.token;
         const headers = {};
-        if (token) {
-          headers["Authorization"] = `Bearer ${token}`;
+        if (currentToken) {
+          headers["Authorization"] = `Bearer ${currentToken}`;
         } else if (apiKey) {
           headers["X-API-KEY"] = apiKey;
         }
@@ -65566,7 +65652,7 @@ function useFileUpload() {
           for (const [key, value] of Object.entries(headers)) {
             xhr.setRequestHeader(key, value);
           }
-          xhr.withCredentials = !token && !apiKey;
+          xhr.withCredentials = !currentToken && !apiKey;
           xhr.send(formData);
         });
         if (result.errors.length > 0) {
@@ -65588,7 +65674,7 @@ function useFileUpload() {
         setProgress(null);
       }
     },
-    [backendUrl, apiKey, token]
+    [backendUrl, apiKey]
   );
   return { upload, isUploading, progress, error: error2, clearError };
 }