@athenaintel/react 0.10.9 → 0.10.11

package/dist/index.cjs

@@ -24729,6 +24729,335 @@ const ThreadListRefreshContext = React.createContext(null);
 function useRefreshThreadList() {
   return React.useContext(ThreadListRefreshContext);
 }
+function useAthenaThreadManager() {
+  const runtime = useAssistantRuntime({ optional: true });
+  const remoteId = useThread({ optional: true, selector: (s) => {
+    var _a2;
+    return (_a2 = s.metadata) == null ? void 0 : _a2.remoteId;
+  } });
+  const isThreadLoading = useThread({ optional: true, selector: (s) => s.isLoading }) ?? false;
+  const isListLoading = useThreadList({ optional: true, selector: (s) => s.isLoading });
+  const runtimeRef = React.useRef(runtime);
+  runtimeRef.current = runtime;
+  const switchToThread = React.useCallback(
+    (id) => runtimeRef.current.threads.switchToThread(id),
+    []
+  );
+  const switchToNewThread = React.useCallback(
+    () => runtimeRef.current.threads.switchToNewThread(),
+    []
+  );
+  const activeThreadId = remoteId ?? null;
+  return React.useMemo(() => {
+    if (!runtime || isListLoading == null) {
+      return null;
+    }
+    return {
+      activeThreadId,
+      isListLoading,
+      isThreadLoading,
+      switchToThread,
+      switchToNewThread
+    };
+  }, [runtime, activeThreadId, isListLoading, isThreadLoading, switchToThread, switchToNewThread]);
+}
+function createJSONStorage(getStorage, options) {
+  let storage;
+  try {
+    storage = getStorage();
+  } catch (e) {
+    return;
+  }
+  const persistStorage = {
+    getItem: (name) => {
+      var _a2;
+      const parse2 = (str2) => {
+        if (str2 === null) {
+          return null;
+        }
+        return JSON.parse(str2, void 0);
+      };
+      const str = (_a2 = storage.getItem(name)) != null ? _a2 : null;
+      if (str instanceof Promise) {
+        return str.then(parse2);
+      }
+      return parse2(str);
+    },
+    setItem: (name, newValue) => storage.setItem(name, JSON.stringify(newValue, void 0)),
+    removeItem: (name) => storage.removeItem(name)
+  };
+  return persistStorage;
+}
+const toThenable = (fn) => (input) => {
+  try {
+    const result = fn(input);
+    if (result instanceof Promise) {
+      return result;
+    }
+    return {
+      then(onFulfilled) {
+        return toThenable(onFulfilled)(result);
+      },
+      catch(_onRejected) {
+        return this;
+      }
+    };
+  } catch (e) {
+    return {
+      then(_onFulfilled) {
+        return this;
+      },
+      catch(onRejected) {
+        return toThenable(onRejected)(e);
+      }
+    };
+  }
+};
+const persistImpl = (config2, baseOptions) => (set2, get2, api) => {
+  let options = {
+    storage: createJSONStorage(() => window.localStorage),
+    partialize: (state) => state,
+    version: 0,
+    merge: (persistedState, currentState) => ({
+      ...currentState,
+      ...persistedState
+    }),
+    ...baseOptions
+  };
+  let hasHydrated = false;
+  let hydrationVersion = 0;
+  const hydrationListeners = /* @__PURE__ */ new Set();
+  const finishHydrationListeners = /* @__PURE__ */ new Set();
+  let storage = options.storage;
+  if (!storage) {
+    return config2(
+      (...args) => {
+        console.warn(
+          `[zustand persist middleware] Unable to update item '${options.name}', the given storage is currently unavailable.`
+        );
+        set2(...args);
+      },
+      get2,
+      api
+    );
+  }
+  const setItem = () => {
+    const state = options.partialize({ ...get2() });
+    return storage.setItem(options.name, {
+      state,
+      version: options.version
+    });
+  };
+  const savedSetState = api.setState;
+  api.setState = (state, replace2) => {
+    savedSetState(state, replace2);
+    return setItem();
+  };
+  const configResult = config2(
+    (...args) => {
+      set2(...args);
+      return setItem();
+    },
+    get2,
+    api
+  );
+  api.getInitialState = () => configResult;
+  let stateFromStorage;
+  const hydrate = () => {
+    var _a2, _b;
+    if (!storage) return;
+    const currentVersion = ++hydrationVersion;
+    hasHydrated = false;
+    hydrationListeners.forEach((cb) => {
+      var _a22;
+      return cb((_a22 = get2()) != null ? _a22 : configResult);
+    });
+    const postRehydrationCallback = ((_b = options.onRehydrateStorage) == null ? void 0 : _b.call(options, (_a2 = get2()) != null ? _a2 : configResult)) || void 0;
+    return toThenable(storage.getItem.bind(storage))(options.name).then((deserializedStorageValue) => {
+      if (deserializedStorageValue) {
+        if (typeof deserializedStorageValue.version === "number" && deserializedStorageValue.version !== options.version) {
+          if (options.migrate) {
+            const migration = options.migrate(
+              deserializedStorageValue.state,
+              deserializedStorageValue.version
+            );
+            if (migration instanceof Promise) {
+              return migration.then((result) => [true, result]);
+            }
+            return [true, migration];
+          }
+          console.error(
+            `State loaded from storage couldn't be migrated since no migrate function was provided`
+          );
+        } else {
+          return [false, deserializedStorageValue.state];
+        }
+      }
+      return [false, void 0];
+    }).then((migrationResult) => {
+      var _a22;
+      if (currentVersion !== hydrationVersion) {
+        return;
+      }
+      const [migrated, migratedState] = migrationResult;
+      stateFromStorage = options.merge(
+        migratedState,
+        (_a22 = get2()) != null ? _a22 : configResult
+      );
+      set2(stateFromStorage, true);
+      if (migrated) {
+        return setItem();
+      }
+    }).then(() => {
+      if (currentVersion !== hydrationVersion) {
+        return;
+      }
+      postRehydrationCallback == null ? void 0 : postRehydrationCallback(stateFromStorage, void 0);
+      stateFromStorage = get2();
+      hasHydrated = true;
+      finishHydrationListeners.forEach((cb) => cb(stateFromStorage));
+    }).catch((e) => {
+      if (currentVersion !== hydrationVersion) {
+        return;
+      }
+      postRehydrationCallback == null ? void 0 : postRehydrationCallback(void 0, e);
+    });
+  };
+  api.persist = {
+    setOptions: (newOptions) => {
+      options = {
+        ...options,
+        ...newOptions
+      };
+      if (newOptions.storage) {
+        storage = newOptions.storage;
+      }
+    },
+    clearStorage: () => {
+      storage == null ? void 0 : storage.removeItem(options.name);
+    },
+    getOptions: () => options,
+    rehydrate: () => hydrate(),
+    hasHydrated: () => hasHydrated,
+    onHydrate: (cb) => {
+      hydrationListeners.add(cb);
+      return () => {
+        hydrationListeners.delete(cb);
+      };
+    },
+    onFinishHydration: (cb) => {
+      finishHydrationListeners.add(cb);
+      return () => {
+        finishHydrationListeners.delete(cb);
+      };
+    }
+  };
+  if (!options.skipHydration) {
+    hydrate();
+  }
+  return stateFromStorage || configResult;
+};
+const persist = persistImpl;
+const useAssetPanelStore = create()(
+  persist(
+    (set2, get2) => ({
+      isOpen: false,
+      tabs: [],
+      activeTabId: null,
+      viewMode: "tabs",
+      isFullscreen: false,
+      currentThreadId: null,
+      threadStates: {},
+      assetPanelHostCount: 0,
+      autoOpenGeneration: 0,
+      autoOpenedAssets: /* @__PURE__ */ new Map(),
+      registerAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: state.assetPanelHostCount + 1 })),
+      unregisterAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: Math.max(0, state.assetPanelHostCount - 1) })),
+      openAsset: (assetId, meta) => set2((s) => {
+        const existing = s.tabs.find((t) => t.id === assetId);
+        if (existing) {
+          const tabs = meta ? s.tabs.map(
+            (t) => t.id === assetId ? { ...t, name: meta.name ?? t.name, type: meta.type ?? t.type } : t
+          ) : s.tabs;
+          return { isOpen: true, tabs, activeTabId: assetId };
+        }
+        const newTab = {
+          id: assetId,
+          name: (meta == null ? void 0 : meta.name) ?? null,
+          type: (meta == null ? void 0 : meta.type) ?? "unknown"
+        };
+        return { isOpen: true, tabs: [...s.tabs, newTab], activeTabId: assetId };
+      }),
+      closeTab: (assetId) => set2((s) => {
+        var _a2;
+        const tabs = s.tabs.filter((t) => t.id !== assetId);
+        if (tabs.length === 0) {
+          return { isOpen: false, tabs: [], activeTabId: null, isFullscreen: false };
+        }
+        const activeTabId = s.activeTabId === assetId ? ((_a2 = tabs[Math.min(s.tabs.findIndex((t) => t.id === assetId), tabs.length - 1)]) == null ? void 0 : _a2.id) ?? null : s.activeTabId;
+        return { tabs, activeTabId };
+      }),
+      setActiveTab: (assetId) => set2({ activeTabId: assetId, viewMode: "tabs" }),
+      setViewMode: (mode) => set2({ viewMode: mode }),
+      closePanel: () => set2({ isOpen: false, tabs: [], activeTabId: null, viewMode: "tabs", isFullscreen: false }),
+      toggleFullscreen: () => set2((s) => ({ isFullscreen: !s.isFullscreen })),
+      setCurrentThread: (threadId) => set2((s) => {
+        if (s.currentThreadId === threadId) {
+          return s;
+        }
+        const threadStates = { ...s.threadStates };
+        if (s.currentThreadId !== null) {
+          threadStates[s.currentThreadId] = {
+            tabs: s.tabs,
+            activeTabId: s.activeTabId,
+            isOpen: s.isOpen
+          };
+        }
+        const restored = threadId !== null ? threadStates[threadId] : void 0;
+        return {
+          currentThreadId: threadId,
+          threadStates,
+          tabs: (restored == null ? void 0 : restored.tabs) ?? [],
+          activeTabId: (restored == null ? void 0 : restored.activeTabId) ?? null,
+          isOpen: (restored == null ? void 0 : restored.isOpen) ?? false,
+          // Reset transient UI state on thread switch
+          isFullscreen: false,
+          viewMode: "tabs"
+        };
+      }),
+      resetAutoOpen: () => set2((s) => ({ autoOpenGeneration: s.autoOpenGeneration + 1 })),
+      markAutoOpened: (assetId) => {
+        const state = get2();
+        if (state.autoOpenedAssets.get(assetId) === state.autoOpenGeneration) {
+          return false;
+        }
+        state.autoOpenedAssets.set(assetId, state.autoOpenGeneration);
+        return true;
+      }
+    }),
+    {
+      name: "athena-react-asset-panel",
+      partialize: ({
+        isOpen,
+        tabs,
+        activeTabId,
+        viewMode,
+        isFullscreen,
+        currentThreadId,
+        threadStates
+      }) => ({
+        isOpen,
+        tabs,
+        activeTabId,
+        viewMode,
+        isFullscreen,
+        currentThreadId,
+        threadStates
+      }),
+      storage: createJSONStorage(() => sessionStorage)
+    }
+  )
+);
 const THEME_TO_CSS = {
   primary: "--primary",
   primaryForeground: "--primary-foreground",
@@ -25146,7 +25475,19 @@ function AthenaWithThreadList({
     linkClicks,
     citationLinks
   });
-  return /* @__PURE__ */ jsxRuntime.jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsx(ThreadListRefreshContext.Provider, { value: handleRefresh, children: /* @__PURE__ */ jsxRuntime.jsx(TooltipProvider, { children }) }) }) });
+  return /* @__PURE__ */ jsxRuntime.jsx(AssistantRuntimeProvider, { aui, runtime, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsx(ThreadListRefreshContext.Provider, { value: handleRefresh, children: /* @__PURE__ */ jsxRuntime.jsxs(TooltipProvider, { children: [
+    /* @__PURE__ */ jsxRuntime.jsx(AssetPanelThreadSync, {}),
+    children
+  ] }) }) }) });
+}
+function AssetPanelThreadSync() {
+  const threads = useAthenaThreadManager();
+  const setCurrentThread = useAssetPanelStore((s) => s.setCurrentThread);
+  const activeThreadId = (threads == null ? void 0 : threads.activeThreadId) ?? null;
+  React.useEffect(() => {
+    setCurrentThread(activeThreadId);
+  }, [activeThreadId, setCurrentThread]);
+  return null;
 }
 function AthenaProvider({
   children,
@@ -58903,273 +59244,6 @@ const MentionExtension = Node3.create({
     };
   }
 });
-function createJSONStorage(getStorage, options) {
-  let storage;
-  try {
-    storage = getStorage();
-  } catch (e) {
-    return;
-  }
-  const persistStorage = {
-    getItem: (name) => {
-      var _a2;
-      const parse2 = (str2) => {
-        if (str2 === null) {
-          return null;
-        }
-        return JSON.parse(str2, void 0);
-      };
-      const str = (_a2 = storage.getItem(name)) != null ? _a2 : null;
-      if (str instanceof Promise) {
-        return str.then(parse2);
-      }
-      return parse2(str);
-    },
-    setItem: (name, newValue) => storage.setItem(name, JSON.stringify(newValue, void 0)),
-    removeItem: (name) => storage.removeItem(name)
-  };
-  return persistStorage;
-}
-const toThenable = (fn) => (input) => {
-  try {
-    const result = fn(input);
-    if (result instanceof Promise) {
-      return result;
-    }
-    return {
-      then(onFulfilled) {
-        return toThenable(onFulfilled)(result);
-      },
-      catch(_onRejected) {
-        return this;
-      }
-    };
-  } catch (e) {
-    return {
-      then(_onFulfilled) {
-        return this;
-      },
-      catch(onRejected) {
-        return toThenable(onRejected)(e);
-      }
-    };
-  }
-};
-const persistImpl = (config2, baseOptions) => (set2, get2, api) => {
-  let options = {
-    storage: createJSONStorage(() => window.localStorage),
-    partialize: (state) => state,
-    version: 0,
-    merge: (persistedState, currentState) => ({
-      ...currentState,
-      ...persistedState
-    }),
-    ...baseOptions
-  };
-  let hasHydrated = false;
-  let hydrationVersion = 0;
-  const hydrationListeners = /* @__PURE__ */ new Set();
-  const finishHydrationListeners = /* @__PURE__ */ new Set();
-  let storage = options.storage;
-  if (!storage) {
-    return config2(
-      (...args) => {
-        console.warn(
-          `[zustand persist middleware] Unable to update item '${options.name}', the given storage is currently unavailable.`
-        );
-        set2(...args);
-      },
-      get2,
-      api
-    );
-  }
-  const setItem = () => {
-    const state = options.partialize({ ...get2() });
-    return storage.setItem(options.name, {
-      state,
-      version: options.version
-    });
-  };
-  const savedSetState = api.setState;
-  api.setState = (state, replace2) => {
-    savedSetState(state, replace2);
-    return setItem();
-  };
-  const configResult = config2(
-    (...args) => {
-      set2(...args);
-      return setItem();
-    },
-    get2,
-    api
-  );
-  api.getInitialState = () => configResult;
-  let stateFromStorage;
-  const hydrate = () => {
-    var _a2, _b;
-    if (!storage) return;
-    const currentVersion = ++hydrationVersion;
-    hasHydrated = false;
-    hydrationListeners.forEach((cb) => {
-      var _a22;
-      return cb((_a22 = get2()) != null ? _a22 : configResult);
-    });
-    const postRehydrationCallback = ((_b = options.onRehydrateStorage) == null ? void 0 : _b.call(options, (_a2 = get2()) != null ? _a2 : configResult)) || void 0;
-    return toThenable(storage.getItem.bind(storage))(options.name).then((deserializedStorageValue) => {
-      if (deserializedStorageValue) {
-        if (typeof deserializedStorageValue.version === "number" && deserializedStorageValue.version !== options.version) {
-          if (options.migrate) {
-            const migration = options.migrate(
-              deserializedStorageValue.state,
-              deserializedStorageValue.version
-            );
-            if (migration instanceof Promise) {
-              return migration.then((result) => [true, result]);
-            }
-            return [true, migration];
-          }
-          console.error(
-            `State loaded from storage couldn't be migrated since no migrate function was provided`
-          );
-        } else {
-          return [false, deserializedStorageValue.state];
-        }
-      }
-      return [false, void 0];
-    }).then((migrationResult) => {
-      var _a22;
-      if (currentVersion !== hydrationVersion) {
-        return;
-      }
-      const [migrated, migratedState] = migrationResult;
-      stateFromStorage = options.merge(
-        migratedState,
-        (_a22 = get2()) != null ? _a22 : configResult
-      );
-      set2(stateFromStorage, true);
-      if (migrated) {
-        return setItem();
-      }
-    }).then(() => {
-      if (currentVersion !== hydrationVersion) {
-        return;
-      }
-      postRehydrationCallback == null ? void 0 : postRehydrationCallback(stateFromStorage, void 0);
-      stateFromStorage = get2();
-      hasHydrated = true;
-      finishHydrationListeners.forEach((cb) => cb(stateFromStorage));
-    }).catch((e) => {
-      if (currentVersion !== hydrationVersion) {
-        return;
-      }
-      postRehydrationCallback == null ? void 0 : postRehydrationCallback(void 0, e);
-    });
-  };
-  api.persist = {
-    setOptions: (newOptions) => {
-      options = {
-        ...options,
-        ...newOptions
-      };
-      if (newOptions.storage) {
-        storage = newOptions.storage;
-      }
-    },
-    clearStorage: () => {
-      storage == null ? void 0 : storage.removeItem(options.name);
-    },
-    getOptions: () => options,
-    rehydrate: () => hydrate(),
-    hasHydrated: () => hasHydrated,
-    onHydrate: (cb) => {
-      hydrationListeners.add(cb);
-      return () => {
-        hydrationListeners.delete(cb);
-      };
-    },
-    onFinishHydration: (cb) => {
-      finishHydrationListeners.add(cb);
-      return () => {
-        finishHydrationListeners.delete(cb);
-      };
-    }
-  };
-  if (!options.skipHydration) {
-    hydrate();
-  }
-  return stateFromStorage || configResult;
-};
-const persist = persistImpl;
-const useAssetPanelStore = create()(
-  persist(
-    (set2, get2) => ({
-      isOpen: false,
-      tabs: [],
-      activeTabId: null,
-      viewMode: "tabs",
-      isFullscreen: false,
-      assetPanelHostCount: 0,
-      autoOpenGeneration: 0,
-      autoOpenedAssets: /* @__PURE__ */ new Map(),
-      registerAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: state.assetPanelHostCount + 1 })),
-      unregisterAssetPanelHost: () => set2((state) => ({ assetPanelHostCount: Math.max(0, state.assetPanelHostCount - 1) })),
-      openAsset: (assetId, meta) => set2((s) => {
-        const existing = s.tabs.find((t) => t.id === assetId);
-        if (existing) {
-          const tabs = meta ? s.tabs.map(
-            (t) => t.id === assetId ? { ...t, name: meta.name ?? t.name, type: meta.type ?? t.type } : t
-          ) : s.tabs;
-          return { isOpen: true, tabs, activeTabId: assetId };
-        }
-        const newTab = {
-          id: assetId,
-          name: (meta == null ? void 0 : meta.name) ?? null,
-          type: (meta == null ? void 0 : meta.type) ?? "unknown"
-        };
-        return { isOpen: true, tabs: [...s.tabs, newTab], activeTabId: assetId };
-      }),
-      closeTab: (assetId) => set2((s) => {
-        var _a2;
-        const tabs = s.tabs.filter((t) => t.id !== assetId);
-        if (tabs.length === 0) {
-          return { isOpen: false, tabs: [], activeTabId: null, isFullscreen: false };
-        }
-        const activeTabId = s.activeTabId === assetId ? ((_a2 = tabs[Math.min(s.tabs.findIndex((t) => t.id === assetId), tabs.length - 1)]) == null ? void 0 : _a2.id) ?? null : s.activeTabId;
-        return { tabs, activeTabId };
-      }),
-      setActiveTab: (assetId) => set2({ activeTabId: assetId, viewMode: "tabs" }),
-      setViewMode: (mode) => set2({ viewMode: mode }),
-      closePanel: () => set2({ isOpen: false, tabs: [], activeTabId: null, viewMode: "tabs", isFullscreen: false }),
-      toggleFullscreen: () => set2((s) => ({ isFullscreen: !s.isFullscreen })),
-      resetAutoOpen: () => set2((s) => ({ autoOpenGeneration: s.autoOpenGeneration + 1 })),
-      markAutoOpened: (assetId) => {
-        const state = get2();
-        if (state.autoOpenedAssets.get(assetId) === state.autoOpenGeneration) {
-          return false;
-        }
-        state.autoOpenedAssets.set(assetId, state.autoOpenGeneration);
-        return true;
-      }
-    }),
-    {
-      name: "athena-react-asset-panel",
-      partialize: ({
-        isOpen,
-        tabs,
-        activeTabId,
-        viewMode,
-        isFullscreen
-      }) => ({
-        isOpen,
-        tabs,
-        activeTabId,
-        viewMode,
-        isFullscreen
-      }),
-      storage: createJSONStorage(() => sessionStorage)
-    }
-  )
-);
 const ATHENA_APP_HOSTNAMES = /* @__PURE__ */ new Set(["app.athenaintel.com", "staging-app.athenaintel.com"]);
 const ATHENA_PREVIEW_HOSTNAME_SUFFIX = ".previews.athenaintel.com";
 const ATHENA_WORKSPACE_HOSTNAME_SUFFIXES = [".app.athenaintel.com", ".staging-app.athenaintel.com"];
@@ -66626,38 +66700,6 @@ function ThreadListItem() {
     /* @__PURE__ */ jsxRuntime.jsx("span", { className: "flex-1 truncate text-left", children: /* @__PURE__ */ jsxRuntime.jsx(ThreadListItemPrimitiveTitle, { fallback: "Untitled" }) })
   ] }) });
 }
-function useAthenaThreadManager() {
-  const runtime = useAssistantRuntime({ optional: true });
-  const remoteId = useThread({ optional: true, selector: (s) => {
-    var _a2;
-    return (_a2 = s.metadata) == null ? void 0 : _a2.remoteId;
-  } });
-  const isThreadLoading = useThread({ optional: true, selector: (s) => s.isLoading }) ?? false;
-  const isListLoading = useThreadList({ optional: true, selector: (s) => s.isLoading });
-  const runtimeRef = React.useRef(runtime);
-  runtimeRef.current = runtime;
-  const switchToThread = React.useCallback(
-    (id) => runtimeRef.current.threads.switchToThread(id),
-    []
-  );
-  const switchToNewThread = React.useCallback(
-    () => runtimeRef.current.threads.switchToNewThread(),
-    []
-  );
-  const activeThreadId = remoteId ?? null;
-  return React.useMemo(() => {
-    if (!runtime || isListLoading == null) {
-      return null;
-    }
-    return {
-      activeThreadId,
-      isListLoading,
-      isThreadLoading,
-      switchToThread,
-      switchToNewThread
-    };
-  }, [runtime, activeThreadId, isListLoading, isThreadLoading, switchToThread, switchToNewThread]);
-}
 function useAppendToComposer() {
   const aui = useAui();
   return React.useCallback(