@athenaintel/react 0.10.41-rc.1 → 0.10.41

package/dist/index.d.ts

@@ -82,7 +82,7 @@ export { DEFAULT_STATEWIRE_MODEL, useAthenaStatewireRuntime, } from './runtime/u
 export type { AthenaStatewireRuntimeConfig } from './runtime/useAthenaStatewireRuntime';
 export { buildStatewireRunConfig } from './runtime/statewire-run-config';
 export type { StatewireRunConfigOptions } from './runtime/statewire-run-config';
-export { ATHENA_TRANSPORTS, DEFAULT_ATHENA_TRANSPORT, resolveAthenaTransport, } from './runtime/transport';
+export { allowsMidRunSend, ATHENA_TRANSPORTS, DEFAULT_ATHENA_TRANSPORT, resolveAthenaTransport, } from './runtime/transport';
 export type { AthenaTransport, ResolvedAthenaTransport } from './runtime/transport';
 export { useParentAuth, useParentBridge } from './runtime/auth';
 export type { ParentBridgeState } from './runtime/auth';

package/dist/index.js

@@ -23,6 +23,7 @@ import { fromThreadMessageLike, bindExternalStoreMessage, getExternalStoreMessag
 import * as React from "react";
 import React__default, { useMemo, useState, useRef, useEffect, useContext, createContext, useCallback, useLayoutEffect, useReducer, useDebugValue, forwardRef, createRef, memo, createElement, version as version$1, useImperativeHandle } from "react";
 import { A as AthenaAuthContext } from "./AthenaAuthContext-DQsdayH2.js";
+import { LocalStorageSession } from "@assistant-ui/react-statewire/local-storage";
 import { resource, withKey } from "@assistant-ui/tap";
 import { StatewireSendError, StatewireSSE, StatewireWS } from "statewire";
 import "@assistant-ui/core";
@@ -10837,6 +10838,7 @@ function useDeepAgentThread({
   sse = false,
   headers,
   sessionStore,
+  storage,
   isNewChat = false,
   attachKey,
   suspended = false,
@@ -11004,6 +11006,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
@@ -11241,6 +11246,7 @@ function useAthenaStatewireRuntime(config2) {
   const clientTools = useStatewireClientTools(frontendToolkit);
   const clientToolsRef = useRef(clientTools);
   clientToolsRef.current = clientTools;
+  const [canPersistSession] = useState(() => typeof localStorage !== "undefined");
   const thread = useDeepAgentThread({
     threadId,
     baseUrl: syncUrl,
@@ -11258,6 +11264,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: LocalStorageSession({
+          threadId,
+          key: (id) => `${syncUrl}:${id}`
+        })
+      }
+    },
     capabilities: { edit: true, reload: true, continue: true },
     onConnectionChange,
     onRawConnectionChange,
@@ -11337,7 +11354,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 = createContext(null);
@@ -11365,6 +11384,20 @@ function StatewireClientToolBridge({
   const toolsRef = useRef(tools);
   toolsRef.current = tools;
   const pendingRequest = inputRequests == null ? void 0 : inputRequests.find(isPendingClientToolRequest);
+  const ownedClaimsRef = useRef(/* @__PURE__ */ new Set());
+  const activeRef = useRef(true);
+  const releaseClaim = useCallback((claim) => {
+    ownedClaimsRef.current.delete(claim);
+    clientToolRequests.release(claim);
+  }, []);
+  useEffect(() => {
+    activeRef.current = true;
+    return () => {
+      activeRef.current = false;
+      for (const claim of ownedClaimsRef.current) clientToolRequests.release(claim);
+      ownedClaimsRef.current.clear();
+    };
+  }, []);
   const liveRequestIdsRef = useRef(/* @__PURE__ */ new Set());
   liveRequestIdsRef.current = new Set((inputRequests ?? []).map((request) => request.id));
   const executeRequest = useCallback(
@@ -11375,8 +11408,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);
@@ -11398,8 +11431,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 {
@@ -11409,16 +11442,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]
   );
   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;
@@ -11433,6 +11469,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) {
@@ -18768,16 +18807,19 @@ function StatewireThreadListState({
   initialThreadId,
   children
 }) {
+  var _a3;
   const [active, setActive] = useState(
     () => initialThreadId ? { id: initialThreadId, isNew: false } : { id: mintThreadId(), isNew: true }
   );
   const tokenRef = useRef(token);
   tokenRef.current = token;
+  const authContext = useContext(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,
@@ -18795,6 +18837,7 @@ function StatewireThreadListState({
     void refetch();
   }, [refetch]);
   const selectThread = useCallback((threadId) => {
+    if (!threadId) return;
     setActive(
       (current) => current.id === threadId ? current : { id: threadId, isNew: false }
     );
@@ -18802,6 +18845,12 @@ function StatewireThreadListState({
   const newThread = useCallback(() => {
     setActive({ id: mintThreadId(), isNew: true });
   }, []);
+  const isPersisted = !!(data == null ? void 0 : data.some((thread) => thread.id === active.id));
+  useEffect(() => {
+    if (isPersisted) {
+      setActive((current) => current.isNew ? { ...current, isNew: false } : current);
+    }
+  }, [isPersisted]);
   const value = useMemo(
     () => ({
       threads: data ?? [],
@@ -18971,8 +19020,6 @@ function useAthenaThreadManager() {
   const statewireManager = 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,
@@ -19802,9 +19849,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":
@@ -20170,6 +20224,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({
@@ -20204,7 +20260,7 @@ function AthenaProvider({
       appUrl: effectiveAppUrl,
       apiKey: configuredApiKey,
       token: effectiveToken,
-      getToken,
+      getToken: configuredGetToken,
       model,
       tools,
       frontendToolIds: frontendToolNames,
@@ -20214,7 +20270,7 @@ function AthenaProvider({
       systemPrompt,
       customToolConfigs,
       appId,
-      extraRunConfig,
+      extraRunConfig: configuredExtraRunConfig,
       linkClicks,
       citationLinks
     };
@@ -55267,7 +55323,7 @@ const TiptapComposer = ({ tools = [], rootCategories }) => {
   const isUploadingRef = useRef(isUploading);
   isUploadingRef.current = isUploading;
   const isRunningThread = useAuiState((s) => s.thread.isRunning);
-  const isThreadRunning = transport === "statewire" ? false : isRunningThread;
+  const isThreadRunning = allowsMidRunSend(transport) ? false : isRunningThread;
   const isThreadRunningRef = useRef(isThreadRunning);
   isThreadRunningRef.current = isThreadRunning;
   const handleSubmit = useCallback(() => {
@@ -55289,10 +55345,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();
       }
@@ -55541,14 +55595,20 @@ const StatewireApprovalCardInner = () => {
   }
   const isRunning = useAuiState((s) => s.thread.isRunning);
   const sawResumeRunRef = 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;
-  }
+  useEffect(() => {
+    if (!pending) {
+      sawResumeRunRef.current = false;
+      return;
+    }
+    if (isRunning) {
+      sawResumeRunRef.current = true;
+      return;
+    }
+    if (sawResumeRunRef.current) {
+      sawResumeRunRef.current = false;
+      setPending(false);
+    }
+  }, [pending, isRunning]);
   useEffect(() => {
     if (!pending) return;
     const timer = setTimeout(() => {
@@ -55839,7 +55899,7 @@ const StatewireLegacyReadOnlyBanner = () => {
     /* @__PURE__ */ jsx(Archive, { className: "mt-0.5 size-4 shrink-0" }),
     /* @__PURE__ */ jsxs("div", { className: "min-w-0 flex-1", children: [
       /* @__PURE__ */ jsx("p", { className: "font-semibold text-xs", children: "Read-only conversation" }),
-      /* @__PURE__ */ 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__ */ 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." })
     ] })
   ] });
 };
@@ -61078,7 +61138,7 @@ const ThreadScrollToBottom = () => /* @__PURE__ */ jsx(ThreadPrimitive.ScrollToB
 ) });
 const ComposerAction = () => {
   const { transport } = useAthenaConfig();
-  const queuesWhileRunning = transport === "statewire";
+  const queuesWhileRunning = allowsMidRunSend(transport);
   return /* @__PURE__ */ jsxs("div", { className: "aui-composer-action-wrapper relative mx-2 mb-2 flex items-center justify-between", children: [
     /* @__PURE__ */ jsx("div", { className: "flex items-center gap-1", children: /* @__PURE__ */ jsx(FileUploadButton, {}) }),
     /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-1", children: [
@@ -61105,7 +61165,7 @@ const ComposerSendWithQuote = () => {
   const editorRef = useComposerEditorRef();
   const editorEmpty = useComposerEditorEmpty();
   const isRunningThread = 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 = useCallback(() => {
     var _a3;
@@ -61121,10 +61181,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 {
@@ -62431,6 +62489,7 @@ export {
   TooltipTrigger,
   UpdateSheetRangeToolUI,
   WebSearchToolUI,
+  allowsMidRunSend,
   archiveThread,
   asHitlApproval,
   athenaStatewireErrorNotice,