@athenaintel/react 0.10.2 → 0.10.4

package/dist/index.cjs

@@ -1399,8 +1399,8 @@ const createStoreImpl = (createState) => {
   return api;
 };
 const createStore = ((createState) => createState ? createStoreImpl(createState) : createStoreImpl);
-const identity = (arg) => arg;
-function useStore$1(api, selector = identity) {
+const identity$1 = (arg) => arg;
+function useStore$1(api, selector = identity$1) {
   const slice2 = React.useSyncExternalStore(
     api.subscribe,
     React.useCallback(() => selector(api.getState()), [api, selector]),
@@ -9610,6 +9610,72 @@ const AssistantRuntimeProviderImpl = ({ children, aui, runtime }) => {
   return jsxRuntime.jsxs(AssistantProviderBase, { runtime, aui: aui ?? null, children: [jsxRuntime.jsx(DevToolsRegistration, {}), jsxRuntime.jsx(ThreadPrimitiveViewportProvider, { children })] });
 };
 const AssistantRuntimeProvider = React.memo(AssistantRuntimeProviderImpl);
+const debugVerifyPrototype = (runtime, prototype) => {
+  const unboundMethods = Object.getOwnPropertyNames(prototype).filter((methodStr) => {
+    const descriptor = Object.getOwnPropertyDescriptor(prototype, methodStr);
+    const isMethod = descriptor && typeof descriptor.value === "function";
+    if (!isMethod)
+      return false;
+    const methodName = methodStr;
+    return isMethod && !methodName.startsWith("_") && methodName !== "constructor" && prototype[methodName] === runtime[methodName];
+  });
+  if (unboundMethods.length > 0) {
+    throw new Error(`The following methods are not bound: ${JSON.stringify(unboundMethods)}`);
+  }
+  const prototypePrototype = Object.getPrototypeOf(prototype);
+  if (prototypePrototype && prototypePrototype !== Object.prototype) {
+    debugVerifyPrototype(runtime, prototypePrototype);
+  }
+};
+const ensureBinding = (r2) => {
+  var _a2;
+  const runtime = r2;
+  if (runtime.__isBound)
+    return;
+  (_a2 = runtime.__internal_bindMethods) == null ? void 0 : _a2.call(runtime);
+  runtime.__isBound = true;
+  if (process.env.NODE_ENV !== "production") {
+    debugVerifyPrototype(runtime, Object.getPrototypeOf(runtime));
+  }
+};
+function useRuntimeStateInternal(runtime, selector = identity) {
+  ensureBinding(runtime);
+  const slice2 = React.useSyncExternalStore(runtime.subscribe, () => selector(runtime.getState()), () => selector(runtime.getState()));
+  React.useDebugValue(slice2);
+  return slice2;
+}
+const identity = (arg) => arg;
+function createStateHookForRuntime(useRuntime) {
+  function useStoreHook(param) {
+    let optional = false;
+    let selector;
+    if (typeof param === "function") {
+      selector = param;
+    } else if (param) {
+      optional = !!param.optional;
+      selector = param.selector;
+    }
+    const store = useRuntime({ optional });
+    if (!store)
+      return null;
+    return useRuntimeStateInternal(store, selector);
+  }
+  return useStoreHook;
+}
+function useAssistantRuntime(options) {
+  var _a2, _b;
+  const aui = useAui();
+  const runtime = ((_b = (_a2 = aui.threads()).__internal_getAssistantRuntime) == null ? void 0 : _b.call(_a2)) ?? null;
+  if (!runtime && !(options == null ? void 0 : options.optional)) {
+    throw new Error("AssistantRuntime is not available");
+  }
+  return runtime;
+}
+const useThreadListRuntime = (opt) => {
+  var _a2;
+  return ((_a2 = useAssistantRuntime(opt)) == null ? void 0 : _a2.threads) ?? null;
+};
+const useThreadList = createStateHookForRuntime(useThreadListRuntime);
 function useComposerRuntime(options) {
   const aui = useAui();
   const runtime = useAuiState(() => {
@@ -9621,6 +9687,18 @@ function useComposerRuntime(options) {
   }
   return runtime;
 }
+function useThreadRuntime(options) {
+  const aui = useAui();
+  const runtime = useAuiState(() => {
+    var _a2, _b;
+    return aui.thread.source ? ((_b = (_a2 = aui.thread()).__internal_getRuntime) == null ? void 0 : _b.call(_a2)) ?? null : null;
+  });
+  if (!runtime && !(options == null ? void 0 : options.optional)) {
+    throw new Error("ThreadRuntime is not available");
+  }
+  return runtime;
+}
+const useThread = createStateHookForRuntime(useThreadRuntime);
 function setRef(ref, value) {
   if (typeof ref === "function") {
     return ref(value);
@@ -24996,6 +25074,7 @@ function AthenaWithThreadList({
   linkClicks,
   citationLinks
 }) {
+  var _a2;
   const adapter = useAthenaThreadListAdapter({
     backendUrl,
     apiKey,
@@ -25038,15 +25117,17 @@ function AthenaWithThreadList({
     adapter
   });
   const handleRefresh = React.useCallback(() => {
-    var _a2;
-    const core = (_a2 = runtime == null ? void 0 : runtime._core) == null ? void 0 : _a2.threads;
+    var _a3;
+    const core = (_a3 = runtime == null ? void 0 : runtime._core) == null ? void 0 : _a3.threads;
     if (core) {
       core._loadThreadsPromise = void 0;
       core.__internal_load();
     }
   }, [runtime]);
   const previousAuthRefreshKeyRef = React.useRef(null);
-  const authRefreshKey = `${backendUrl}::${apiKey ?? ""}::${token ?? ""}`;
+  const authContext = React.useContext(AthenaAuthContext.AthenaAuthContext);
+  const authUserId = ((_a2 = authContext == null ? void 0 : authContext.user) == null ? void 0 : _a2.userId) ?? "";
+  const authRefreshKey = `${backendUrl}::${apiKey ?? ""}::${authUserId}`;
   React.useEffect(() => {
     const previousAuthRefreshKey = previousAuthRefreshKeyRef.current;
     previousAuthRefreshKeyRef.current = authRefreshKey;
@@ -64865,11 +64946,24 @@ function highlightSql(code2) {
       i += str.length;
       continue;
     }
+    if (code2.slice(i, i + 2) === "/*") {
+      const end = code2.indexOf("*/", i + 2);
+      const str = end >= 0 ? code2.slice(i, end + 2) : code2.slice(i);
+      result += `<span style="color:var(--aui-syn-comment, #6a9955)">${escapeHtml(str)}</span>`;
+      i += str.length;
+      continue;
+    }
     if (code2[i] === "'" || code2[i] === '"') {
       const q = code2[i];
       let j = i + 1;
-      while (j < code2.length && code2[j] !== q) {
-        if (code2[j] === "\\") j++;
+      while (j < code2.length) {
+        if (code2[j] === q) {
+          if (j + 1 < code2.length && code2[j + 1] === q) {
+            j += 2;
+            continue;
+          }
+          break;
+        }
         j++;
       }
       const str = code2.slice(i, j + 1);
@@ -65018,9 +65112,11 @@ const RunSqlToolUIImpl = ({
               i
             )) })
           ] }) }),
-          parsed.rows.length > 50 && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/30 px-4 py-1.5 text-[10px] text-muted-foreground", children: [
-            "Showing first 50 of ",
-            parsed.rows.length,
+          (parsed.rows.length > 50 || parsed.rowCount !== null && parsed.rowCount > parsed.rows.length) && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/30 px-4 py-1.5 text-[10px] text-muted-foreground", children: [
+            "Showing first ",
+            Math.min(50, parsed.rows.length),
+            " of ",
+            parsed.rowCount ?? parsed.rows.length,
             " rows"
           ] })
         ] }),
@@ -66529,6 +66625,21 @@ 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 threadId = useThread({ optional: true, selector: (s) => s.threadId });
+  const threadList = useThreadList({ optional: true });
+  if (!runtime || !threadList) {
+    return null;
+  }
+  return {
+    activeThreadId: threadId ?? null,
+    threadIds: threadList.threadIds,
+    isLoading: threadList.isLoading,
+    switchToThread: (id) => runtime.threads.switchToThread(id),
+    switchToNewThread: () => runtime.threads.switchToNewThread()
+  };
+}
 function useAppendToComposer() {
   const aui = useAui();
   return React.useCallback(
@@ -66706,6 +66817,7 @@ exports.useAthenaCitationLinkHandler = useAthenaCitationLinkHandler;
 exports.useAthenaConfig = useAthenaConfig;
 exports.useAthenaLinkClickHandler = useAthenaLinkClickHandler;
 exports.useAthenaRuntime = useAthenaRuntime;
+exports.useAthenaThreadManager = useAthenaThreadManager;
 exports.useComposerAttachment = useComposerAttachment;
 exports.useFileUpload = useFileUpload;
 exports.useMentionSuggestions = useMentionSuggestions;