@athenaintel/react 0.10.41-rc.1 → 0.10.41

package/dist/index.cjs

@@ -45,6 +45,7 @@ const jsxRuntime = require("react/jsx-runtime");
 const react$1 = require("@assistant-ui/react");
 const React = require("react");
 const AthenaAuthContext = require("./AthenaAuthContext-B3AwLA5Z.cjs");
+const localStorage$1 = require("@assistant-ui/react-statewire/local-storage");
 const tap = require("@assistant-ui/tap");
 const statewire = require("statewire");
 require("@assistant-ui/core");
@@ -10876,6 +10877,7 @@ function useDeepAgentThread({
   sse = false,
   headers,
   sessionStore,
+  storage,
   isNewChat = false,
   attachKey,
   suspended = false,
@@ -11043,6 +11045,9 @@ function useDeepAgentThread({
         );
       })()
     ),
+    // Survives the transport remounts above: a bumped attachKey re-attaches the
+    // same lane, and a persistent session storage carries it across reloads.
+    ...storage && { storage },
     // The run subsystem (status, lanes, input requests, runId) is mounted on
     // the state root and always replicated by react-statewire >= 0.14 (the
     // former `runs: true` opt-in is gone) — the converter only maps app-owned
@@ -11280,6 +11285,7 @@ function useAthenaStatewireRuntime(config2) {
   const clientTools = useStatewireClientTools(frontendToolkit);
   const clientToolsRef = React.useRef(clientTools);
   clientToolsRef.current = clientTools;
+  const [canPersistSession] = React.useState(() => typeof localStorage !== "undefined");
   const thread = useDeepAgentThread({
     threadId,
     baseUrl: syncUrl,
@@ -11297,6 +11303,17 @@ function useAthenaStatewireRuntime(config2) {
     // Queued entries render in the SDK's queue panel, so only the steer lane
     // projects into the message tail.
     laneProjection: "steer-only",
+    // Durable session: a reload resumes the lane and resends whatever the
+    // server has not marked durable. Scoped by host — a thread id is only
+    // meaningful against the backend that minted it.
+    ...canPersistSession && {
+      storage: {
+        session: localStorage$1.LocalStorageSession({
+          threadId,
+          key: (id) => `${syncUrl}:${id}`
+        })
+      }
+    },
     capabilities: { edit: true, reload: true, continue: true },
     onConnectionChange,
     onRawConnectionChange,
@@ -11376,7 +11393,9 @@ function athenaStatewireErrorNotice(error2) {
   return {
     kind: "command",
     title: "Message was not sent",
-    message: error2 instanceof Error && error2.message.length > 0 ? error2.message : "The chat runtime rejected the command. Please try again."
+    // The server's own detail names the actionable reason; the Error message is
+    // generic client-side text like `command rejected (500)`.
+    message: readDetailMessage(rejection == null ? void 0 : rejection.detail) ?? (error2 instanceof Error && error2.message.length > 0 ? error2.message : "The chat runtime rejected the command. Please try again.")
   };
 }
 const AthenaStatewireLifecycleContext = React.createContext(null);
@@ -11404,6 +11423,20 @@ function StatewireClientToolBridge({
   const toolsRef = React.useRef(tools);
   toolsRef.current = tools;
   const pendingRequest = inputRequests == null ? void 0 : inputRequests.find(isPendingClientToolRequest);
+  const ownedClaimsRef = React.useRef(/* @__PURE__ */ new Set());
+  const activeRef = React.useRef(true);
+  const releaseClaim = React.useCallback((claim) => {
+    ownedClaimsRef.current.delete(claim);
+    clientToolRequests.release(claim);
+  }, []);
+  React.useEffect(() => {
+    activeRef.current = true;
+    return () => {
+      activeRef.current = false;
+      for (const claim of ownedClaimsRef.current) clientToolRequests.release(claim);
+      ownedClaimsRef.current.clear();
+    };
+  }, []);
   const liveRequestIdsRef = React.useRef(/* @__PURE__ */ new Set());
   liveRequestIdsRef.current = new Set((inputRequests ?? []).map((request) => request.id));
   const executeRequest = React.useCallback(
@@ -11414,8 +11447,8 @@ function StatewireClientToolBridge({
       const registry2 = new Map(toolsRef.current.map((tool) => [tool.name, tool]));
       const results = {};
       for (const call of calls) {
-        if (!liveRequestIdsRef.current.has(request.id)) {
-          clientToolRequests.release(requestKey(threadId, request.id));
+        if (!activeRef.current || !liveRequestIdsRef.current.has(request.id)) {
+          releaseClaim(requestKey(threadId, request.id));
           return;
         }
         const tool = registry2.get(call.tool_name);
@@ -11437,8 +11470,8 @@ function StatewireClientToolBridge({
           results[call.interrupt_id] = clientToolErrorEnvelope(error2);
         }
       }
-      if (!liveRequestIdsRef.current.has(request.id)) {
-        clientToolRequests.release(requestKey(threadId, request.id));
+      if (!activeRef.current || !liveRequestIdsRef.current.has(request.id)) {
+        releaseClaim(requestKey(threadId, request.id));
         return;
       }
       try {
@@ -11448,16 +11481,19 @@ function StatewireClientToolBridge({
           requestId: request.id,
           response: { type: "resume", value: wireValue }
         });
+        ownedClaimsRef.current.delete(requestKey(threadId, request.id));
       } catch (error2) {
-        clientToolRequests.release(requestKey(threadId, request.id));
+        releaseClaim(requestKey(threadId, request.id));
         console.error("[AthenaSDK] failed to resume client tool results:", error2);
       }
     },
-    [sendCommand, threadId]
+    [releaseClaim, sendCommand, threadId]
   );
   React.useEffect(() => {
     if (!pendingRequest) return;
-    if (!clientToolRequests.claim(requestKey(threadId, pendingRequest.id))) return;
+    const claim = requestKey(threadId, pendingRequest.id);
+    if (!clientToolRequests.claim(claim)) return;
+    ownedClaimsRef.current.add(claim);
     void executeRequest(pendingRequest);
   }, [executeRequest, pendingRequest, threadId]);
   return null;
@@ -11472,6 +11508,9 @@ function resolveAthenaTransport({
 }) {
   return { transport: transport ?? DEFAULT_ATHENA_TRANSPORT, fallbackReason: null };
 }
+function allowsMidRunSend(transport) {
+  return transport === ATHENA_TRANSPORTS.statewire;
+}
 var __defProp$i = Object.defineProperty;
 var __name$h = (target, value) => __defProp$i(target, "name", { value, configurable: true });
 function setRef$1(ref, value) {
@@ -18807,16 +18846,19 @@ function StatewireThreadListState({
   initialThreadId,
   children
 }) {
+  var _a3;
   const [active, setActive] = React.useState(
     () => initialThreadId ? { id: initialThreadId, isNew: false } : { id: mintThreadId(), isNew: true }
   );
   const tokenRef = React.useRef(token);
   tokenRef.current = token;
+  const authContext = React.useContext(AthenaAuthContext.AthenaAuthContext);
+  const principal = ((_a3 = authContext == null ? void 0 : authContext.user) == null ? void 0 : _a3.userId) ?? "";
   const hasAuth = !!(apiKey || token);
   const { data, isLoading, refetch } = useQuery({
-    // Keyed on the user identity surface, not the raw token — token rotation
-    // for the same session must not refetch the whole list.
-    queryKey: [THREADS_QUERY_KEY, backendUrl, appId ?? "", apiKey ?? "", hasAuth],
+    queryKey: [THREADS_QUERY_KEY, backendUrl, appId ?? "", principal, hasAuth],
+    // Never serve another principal's list while the new one loads.
+    placeholderData: void 0,
     queryFn: async () => {
       const { threads } = await listThreads(
         backendUrl,
@@ -18834,6 +18876,7 @@ function StatewireThreadListState({
     void refetch();
   }, [refetch]);
   const selectThread = React.useCallback((threadId) => {
+    if (!threadId) return;
     setActive(
       (current) => current.id === threadId ? current : { id: threadId, isNew: false }
     );
@@ -18841,6 +18884,12 @@ function StatewireThreadListState({
   const newThread = React.useCallback(() => {
     setActive({ id: mintThreadId(), isNew: true });
   }, []);
+  const isPersisted = !!(data == null ? void 0 : data.some((thread) => thread.id === active.id));
+  React.useEffect(() => {
+    if (isPersisted) {
+      setActive((current) => current.isNew ? { ...current, isNew: false } : current);
+    }
+  }, [isPersisted]);
   const value = React.useMemo(
     () => ({
       threads: data ?? [],
@@ -19010,8 +19059,6 @@ function useAthenaThreadManager() {
   const statewireManager = React.useMemo(() => {
     if (!statewireList) return null;
     return {
-      // A never-started local chat stays out of the URL, mirroring the
-      // legacy manager's local-placeholder handling.
       activeThreadId: statewireList.isNewChat ? null : statewireList.activeThreadId,
       isListLoading: statewireList.isLoading,
       isThreadLoading: statewireIsThreadLoading,
@@ -19841,9 +19888,16 @@ const initialStatewireLifecycle = {
   reconnect: null
 };
 function statewireLifecycleReducer(state, action) {
+  var _a3, _b2;
   switch (action.type) {
-    case "connection":
-      return { ...state, connection: action.connection };
+    case "connection": {
+      const healthy = ((_a3 = action.connection) == null ? void 0 : _a3.status) === "live" || ((_b2 = action.connection) == null ? void 0 : _b2.status) === "idle";
+      return {
+        ...state,
+        connection: action.connection,
+        error: healthy ? null : state.error
+      };
+    }
     case "error":
       return { ...state, error: action.error };
     case "legacy-read-only":
@@ -20209,6 +20263,8 @@ function AthenaProvider({
   const configuredAppUrl = (config2 == null ? void 0 : config2.appUrl) ?? appUrl;
   const configuredTransport = (config2 == null ? void 0 : config2.transport) ?? transport;
   const configuredStatewireSyncUrl = (config2 == null ? void 0 : config2.statewireSyncUrl) ?? statewireSyncUrl;
+  const configuredGetToken = (config2 == null ? void 0 : config2.getToken) ?? getToken;
+  const configuredExtraRunConfig = (config2 == null ? void 0 : config2.extraRunConfig) ?? extraRunConfig;
   const configuredTrustedParentOrigins = config2 == null ? void 0 : config2.trustedParentOrigins;
   const posthogConfig = (config2 == null ? void 0 : config2.posthog) ?? posthogProp;
   const { transport: effectiveTransport, fallbackReason: transportFallbackReason } = resolveAthenaTransport({
@@ -20243,7 +20299,7 @@ function AthenaProvider({
       appUrl: effectiveAppUrl,
       apiKey: configuredApiKey,
       token: effectiveToken,
-      getToken,
+      getToken: configuredGetToken,
       model,
       tools,
       frontendToolIds: frontendToolNames,
@@ -20253,7 +20309,7 @@ function AthenaProvider({
       systemPrompt,
       customToolConfigs,
       appId,
-      extraRunConfig,
+      extraRunConfig: configuredExtraRunConfig,
       linkClicks,
       citationLinks
     };
@@ -55306,7 +55362,7 @@ const TiptapComposer = ({ tools = [], rootCategories }) => {
   const isUploadingRef = React.useRef(isUploading);
   isUploadingRef.current = isUploading;
   const isRunningThread = react$1.useAuiState((s) => s.thread.isRunning);
-  const isThreadRunning = transport === "statewire" ? false : isRunningThread;
+  const isThreadRunning = allowsMidRunSend(transport) ? false : isRunningThread;
   const isThreadRunningRef = React.useRef(isThreadRunning);
   isThreadRunningRef.current = isThreadRunning;
   const handleSubmit = React.useCallback(() => {
@@ -55328,10 +55384,8 @@ const TiptapComposer = ({ tools = [], rootCategories }) => {
         appUrl
       });
       if (fullMessage) {
-        aui.thread.append({
-          role: "user",
-          content: [{ type: "text", text: fullMessage }]
-        });
+        aui.composer.setText(fullMessage);
+        aui.composer.send();
         clearAttachments();
         clearQuote();
       }
@@ -55580,14 +55634,20 @@ const StatewireApprovalCardInner = () => {
   }
   const isRunning = react$1.useAuiState((s) => s.thread.isRunning);
   const sawResumeRunRef = React.useRef(false);
-  if (pending && isRunning) {
-    sawResumeRunRef.current = true;
-  } else if (pending && sawResumeRunRef.current && !isRunning) {
-    sawResumeRunRef.current = false;
-    setPending(false);
-  } else if (!pending) {
-    sawResumeRunRef.current = false;
-  }
+  React.useEffect(() => {
+    if (!pending) {
+      sawResumeRunRef.current = false;
+      return;
+    }
+    if (isRunning) {
+      sawResumeRunRef.current = true;
+      return;
+    }
+    if (sawResumeRunRef.current) {
+      sawResumeRunRef.current = false;
+      setPending(false);
+    }
+  }, [pending, isRunning]);
   React.useEffect(() => {
     if (!pending) return;
     const timer = setTimeout(() => {
@@ -55878,7 +55938,7 @@ const StatewireLegacyReadOnlyBanner = () => {
     /* @__PURE__ */ jsxRuntime.jsx(Archive, { className: "mt-0.5 size-4 shrink-0" }),
     /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "min-w-0 flex-1", children: [
       /* @__PURE__ */ jsxRuntime.jsx("p", { className: "font-semibold text-xs", children: "Read-only conversation" }),
-      /* @__PURE__ */ jsxRuntime.jsx("p", { className: "pt-0.5 text-xs leading-snug text-amber-800", children: "Its history remains available, but new messages cannot be added. Start a new chat to continue." })
+      /* @__PURE__ */ jsxRuntime.jsx("p", { className: "pt-0.5 text-xs leading-snug text-amber-800", children: "This conversation’s history remains available, but new messages cannot be added. Start a new chat to continue." })
     ] })
   ] });
 };
@@ -61117,7 +61177,7 @@ const ThreadScrollToBottom = () => /* @__PURE__ */ jsxRuntime.jsx(react$1.Thread
 ) });
 const ComposerAction = () => {
   const { transport } = useAthenaConfig();
-  const queuesWhileRunning = transport === "statewire";
+  const queuesWhileRunning = allowsMidRunSend(transport);
   return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "aui-composer-action-wrapper relative mx-2 mb-2 flex items-center justify-between", children: [
     /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex items-center gap-1", children: /* @__PURE__ */ jsxRuntime.jsx(FileUploadButton, {}) }),
     /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1", children: [
@@ -61144,7 +61204,7 @@ const ComposerSendWithQuote = () => {
   const editorRef = useComposerEditorRef();
   const editorEmpty = useComposerEditorEmpty();
   const isRunningThread = react$1.useAuiState((s) => s.thread.isRunning);
-  const isThreadRunning = transport === "statewire" ? false : isRunningThread;
+  const isThreadRunning = allowsMidRunSend(transport) ? false : isRunningThread;
   const hasExtras = !!quote || attachments.length > 0;
   const handleSend = React.useCallback(() => {
     var _a3;
@@ -61160,10 +61220,8 @@ const ComposerSendWithQuote = () => {
         appUrl
       });
       if (!fullMessage) return;
-      aui.thread.append({
-        role: "user",
-        content: [{ type: "text", text: fullMessage }]
-      });
+      aui.composer.setText(fullMessage);
+      aui.composer.send();
       clearQuote();
       clearAttachments();
     } else {
@@ -62469,6 +62527,7 @@ exports.TooltipProvider = TooltipProvider;
 exports.TooltipTrigger = TooltipTrigger;
 exports.UpdateSheetRangeToolUI = UpdateSheetRangeToolUI;
 exports.WebSearchToolUI = WebSearchToolUI;
+exports.allowsMidRunSend = allowsMidRunSend;
 exports.archiveThread = archiveThread;
 exports.asHitlApproval = asHitlApproval;
 exports.athenaStatewireErrorNotice = athenaStatewireErrorNotice;