@athenaintel/react 0.41.3 → 0.42.1

package/dist/provider/AthenaProvider.cjs

@@ -4,6 +4,7 @@ const jsxRuntime = require("react/jsx-runtime");
 const react$1 = require("@assistant-ui/react");
 const react = require("react");
 const offlineHistory = require("../offline/offline-history.cjs");
+const chatOutbox = require("../offline/chat-outbox.cjs");
 const athenaUrls = require("../lib/athena-urls.cjs");
 const mint = require("../collab/mint.cjs");
 const auth = require("../runtime/auth.cjs");
@@ -12,7 +13,10 @@ const AthenaCredentialSourceContext = require("../auth/AthenaCredentialSourceCon
 const credentialSource = require("../auth/credential-source.cjs");
 const authRefusalRestart = require("../runtime/auth-refusal-restart.cjs");
 const useAthenaStatewireRuntime = require("../runtime/useAthenaStatewireRuntime.cjs");
+const statewireSessionKey = require("../runtime/statewire-session-key.cjs");
+const statewireSessionLane = require("../runtime/statewire-session-lane.cjs");
 const statewireLifecycle = require("../runtime/statewire-lifecycle.cjs");
+const credentialRenewal = require("../runtime/credential-renewal.cjs");
 const statewireRunConfig = require("../runtime/statewire-run-config.cjs");
 const statewireThreadConfigLane = require("../runtime/statewire-thread-config-lane.cjs");
 const mintThreadId = require("../threads/mint-thread-id.cjs");
@@ -40,9 +44,12 @@ const PostHogProvider = require("./PostHogProvider.cjs");
 const threadSnapshotMirror = require("../agent-runtime/src/thread-snapshot-mirror.cjs");
 const durableThreadSnapshotCache = require("../agent-runtime/src/durable-thread-snapshot-cache.cjs");
 const threadSnapshotCache = require("../agent-runtime/src/thread-snapshot-cache.cjs");
+const chatOutboxAttachments = require("../agent-runtime/src/chat-outbox-attachments.cjs");
 const toolOutputs = require("../agent-runtime/src/tool-outputs.cjs");
 const threadStatus = require("../agent-runtime/src/thread-status.cjs");
+const useDeepAgentChatOutbox = require("../agent-runtime/src/use-deep-agent-chat-outbox.cjs");
 const useDeepAgentConfig = require("../agent-runtime/src/use-deep-agent-config.cjs");
+const chatOutboxSignals = require("../agent-runtime/src/chat-outbox-signals.cjs");
 const resolveTokenOverride = ({
   config,
   token
@@ -104,6 +111,7 @@ function ComposerHandoffDrain() {
   return null;
 }
 const EMPTY_MODES = [];
+const CHAT_OUTBOX_ATTACHMENT_READ_TIMEOUT_MS = 1e4;
 function withEntry(record, key, value) {
   if (value === null) {
     if (!(key in record)) return record;
@@ -138,8 +146,12 @@ function AthenaStatewireRuntimeMount({
   appId,
   extraRunConfig,
   snapshotCache,
-  toolOutputs: toolOutputs2
+  toolOutputs: toolOutputs2,
+  chatOutbox: chatOutbox2 = null,
+  sessionLane,
+  subagentLaneHistory
 }) {
+  const outboxRef = react.useRef(null);
   const [lifecycle, dispatchLifecycle] = react.useReducer(
     statewireLifecycle.statewireLifecycleReducer,
     statewireLifecycle.initialStatewireLifecycle
@@ -168,6 +180,7 @@ function AthenaStatewireRuntimeMount({
     extraRunConfig,
     snapshotCache,
     toolOutputs: toolOutputs2,
+    subagentLaneHistory,
     // The runtime already records every transport error on the diagnostics
     // bus (with the channel-teardown split), so this layer only maps it to
     // the banner notice.
@@ -181,6 +194,10 @@ function AthenaStatewireRuntimeMount({
     onRawConnectionChange: (connection) => {
       dispatchLifecycle({ type: "raw-connection", reconnect: connection.reconnect });
     },
+    onCommandChange: (update) => {
+      var _a;
+      return (_a = outboxRef.current) == null ? void 0 : _a.observeCommand(update);
+    },
     // The lane as the host holds it: anything that decides from the lane waits
     // for this, never the placeholder or remembered snapshot rendered before.
     onServerState: (state) => {
@@ -188,6 +205,64 @@ function AthenaStatewireRuntimeMount({
       setServerLane((previous) => statewireServerLaneContext.nextServerLane({ previous, config }));
     }
   });
+  const lanePending = statewireSessionKey.statewireSessionPrincipal(react.useContext(AthenaAuthContext.AthenaAuthContext)) === null;
+  react.useEffect(() => {
+    sessionLane.setPending(lanePending);
+  }, [sessionLane, lanePending]);
+  const connectionRef = react.useRef(lifecycle.connection);
+  connectionRef.current = lifecycle.connection;
+  const [connectionListeners] = react.useState(() => /* @__PURE__ */ new Set());
+  react.useEffect(() => {
+    for (const listener of connectionListeners) listener();
+  }, [lifecycle.connection, connectionListeners]);
+  const [channelReachability] = react.useState(() => ({
+    get: () => {
+      var _a;
+      return ((_a = connectionRef.current) == null ? void 0 : _a.status) !== "retrying";
+    },
+    subscribe: (listener) => {
+      connectionListeners.add(listener);
+      return () => {
+        connectionListeners.delete(listener);
+      };
+    }
+  }));
+  const [browserOnline] = react.useState(chatOutboxSignals.browserOnlineSignal);
+  const hostReachability = chatOutbox2 == null ? void 0 : chatOutbox2.reachability;
+  const reachability = react.useMemo(
+    () => chatOutboxSignals.allChatOutboxSignals(hostReachability ?? browserOnline, channelReachability),
+    [hostReachability, browserOnline, channelReachability]
+  );
+  const [defaultForeground] = react.useState(chatOutboxSignals.documentVisibleSignal);
+  const composerHandoff$1 = composerHandoff.useComposerHandoffStore();
+  const outbox = useDeepAgentChatOutbox.useDeepAgentChatOutbox({
+    client,
+    threadId,
+    isNewChat,
+    store: (chatOutbox2 == null ? void 0 : chatOutbox2.store) ?? null,
+    reachability,
+    foreground: (chatOutbox2 == null ? void 0 : chatOutbox2.foreground) ?? defaultForeground,
+    // Cancel and Edit give the draft back, after anything typed since. Its attachments come back
+    // as mentions in it, remembered so a message held again from it is asked about again.
+    restore: (entry) => {
+      const draft = entry.composerMarkdown ?? entry.text;
+      const current = client.composer.getState().text ?? "";
+      client.composer.setText(current.trim() ? `${current}
+
+${draft}` : draft);
+      if (composerHandoff$1 && entry.attachments) {
+        composerHandoff.rememberRestoredAttachments({
+          store: composerHandoff$1,
+          threadId,
+          attachments: entry.attachments
+        });
+      }
+    },
+    ...(chatOutbox2 == null ? void 0 : chatOutbox2.openThread) && { openThread: chatOutbox2.openThread },
+    ...(chatOutbox2 == null ? void 0 : chatOutbox2.checkAttachments) && { checkAttachments: chatOutbox2.checkAttachments },
+    enabled: chatOutbox2 !== null
+  });
+  outboxRef.current = outbox;
   authRefusalRestart.useRestartAfterAuthRefusal({
     connection: lifecycle.connection,
     reconnect: lifecycle.reconnect
@@ -221,9 +296,10 @@ function AthenaStatewireRuntimeMount({
       client,
       lifecycle: lifecycleValue,
       serverLane,
-      threadConfig
+      threadConfig,
+      outbox
     }),
-    [threadId, isNewChat, client, lifecycleValue, serverLane, threadConfig]
+    [threadId, isNewChat, client, lifecycleValue, serverLane, threadConfig, outbox]
   );
   react.useLayoutEffect(() => {
     onPublish(snapshot);
@@ -323,7 +399,7 @@ function AthenaStatewireStandalone({
       published.threadId
     ),
     /* @__PURE__ */ jsxRuntime.jsx(ComposerHandoffDrain, {}),
-    /* @__PURE__ */ jsxRuntime.jsx(chatModeContext.AthenaChatModeSelectionContext.Provider, { value: chatModeSelection, children })
+    /* @__PURE__ */ jsxRuntime.jsx(useDeepAgentChatOutbox.DeepAgentChatOutboxContext.Provider, { value: published.outbox, children: /* @__PURE__ */ jsxRuntime.jsx(chatModeContext.AthenaChatModeSelectionContext.Provider, { value: chatModeSelection, children }) })
   ] }) }) }) }) }) }) }) }) });
   return /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
     /* @__PURE__ */ jsxRuntime.jsx(
@@ -392,15 +468,25 @@ function AthenaStatewireWithThreadList({
 }
 function AthenaStatewireActiveThreadHost({
   children,
+  chatOutbox: chatOutbox2,
   ...standaloneProps
 }) {
   const list = statewireThreadList.useAthenaStatewireThreadList();
+  const selectThread = list == null ? void 0 : list.selectThread;
+  const outboxHost = react.useMemo(
+    () => chatOutbox2 && !chatOutbox2.openThread && selectThread ? {
+      ...chatOutbox2,
+      openThread: ({ threadId, newChat }) => selectThread(threadId, { isNew: newChat })
+    } : chatOutbox2,
+    [chatOutbox2, selectThread]
+  );
   if (!list) return null;
   return /* @__PURE__ */ jsxRuntime.jsxs(
     AthenaStatewireStandalone,
     {
       threadId: list.activeThreadId,
       threadList: { isNewChat: list.isNewChat },
+      chatOutbox: outboxHost,
       ...standaloneProps,
       children: [
         /* @__PURE__ */ jsxRuntime.jsx(statewireThreadList.StatewireThreadTitlePoller, {}),
@@ -454,7 +540,9 @@ function AthenaProvider({
   debug,
   onDiagnostic,
   onError,
-  offlineHistory: offlineHistory$1
+  offlineHistory: offlineHistory$1,
+  chatOutbox: chatOutbox$1,
+  subagentLaneHistory = false
 }) {
   var _a;
   const agentAssetId = statewireRunConfig.resolveAgentProp({ agent });
@@ -551,6 +639,50 @@ function AthenaProvider({
   const [snapshotCache] = react.useState(
     () => durableSnapshotCache ?? threadSnapshotCache.createDeepAgentThreadSnapshotCache()
   );
+  const chatOutboxRef = react.useRef(chatOutbox$1);
+  chatOutboxRef.current = chatOutbox$1;
+  const chatOutboxOwnerId = (chatOutbox$1 == null ? void 0 : chatOutbox$1.ownerId) ?? null;
+  const chatOutboxStore = react.useMemo(() => {
+    const host = chatOutboxRef.current;
+    return host && host.ownerId === chatOutboxOwnerId ? chatOutbox.chatOutboxStoreFor(host) : null;
+  }, [chatOutboxOwnerId]);
+  react.useEffect(() => {
+    if (chatOutbox$1) chatOutbox.chatOutboxStoreFor(chatOutbox$1);
+  }, [chatOutbox$1]);
+  const chatOutboxReachability = chatOutbox$1 == null ? void 0 : chatOutbox$1.reachability;
+  const chatOutboxForeground = chatOutbox$1 == null ? void 0 : chatOutbox$1.foreground;
+  const chatOutboxOpenThread = chatOutbox$1 == null ? void 0 : chatOutbox$1.openThread;
+  const outboxCheckAuthRef = react.useRef({});
+  const outboxCheckBackendRef = react.useRef("");
+  const [checkChatOutboxAttachments] = react.useState(
+    () => chatOutboxAttachments.createChatOutboxAttachmentCheck({
+      readAsset: (assetId) => credentialRenewal.sendWithCredentialRenewal(
+        outboxCheckAuthRef.current,
+        (headers) => fetch(`${outboxCheckBackendRef.current}${chatOutboxAttachments.chatOutboxAttachmentPath(assetId)}`, {
+          method: "GET",
+          headers,
+          signal: AbortSignal.timeout(CHAT_OUTBOX_ATTACHMENT_READ_TIMEOUT_MS)
+        })
+      )
+    })
+  );
+  const chatOutboxHost = react.useMemo(
+    () => chatOutboxOwnerId === null ? null : {
+      store: chatOutboxStore,
+      ...chatOutboxReachability && { reachability: chatOutboxReachability },
+      ...chatOutboxForeground && { foreground: chatOutboxForeground },
+      ...chatOutboxOpenThread && { openThread: chatOutboxOpenThread },
+      checkAttachments: checkChatOutboxAttachments
+    },
+    [
+      chatOutboxOwnerId,
+      chatOutboxStore,
+      chatOutboxReachability,
+      chatOutboxForeground,
+      chatOutboxOpenThread,
+      checkChatOutboxAttachments
+    ]
+  );
   const offlineOwnerId = (offlineHistory$1 == null ? void 0 : offlineHistory$1.ownerId) ?? null;
   const awaitingKeptSnapshots = offlineHistory.useKeptSnapshotsGate({
     cache: durableSnapshotCache,
@@ -575,8 +707,21 @@ function AthenaProvider({
   react.useEffect(() => {
     snapshotCache.clear();
     toolOutputs$1.clear();
-    composerHandoff.clearComposerHandoff({ store: composerHandoff$1, reason: "the signed-in identity changed" });
-  }, [snapshotCache, toolOutputs$1, composerHandoff$1, principal, configuredApiKey]);
+  }, [snapshotCache, toolOutputs$1, principal, configuredApiKey]);
+  const sessionPrincipal = statewireSessionKey.statewireSessionPrincipal(authContext);
+  const [sessionLane] = react.useState(
+    () => statewireSessionLane.createStatewireSessionLaneGate({ pending: sessionPrincipal === null })
+  );
+  const handoffIdentityRef = react.useRef(null);
+  react.useEffect(() => {
+    if (sessionPrincipal === null) return;
+    const identity = JSON.stringify([sessionPrincipal, configuredApiKey ?? null]);
+    const previous = handoffIdentityRef.current;
+    handoffIdentityRef.current = identity;
+    if (previous !== null && previous !== identity) {
+      composerHandoff.clearComposerHandoff({ store: composerHandoff$1, reason: "the signed-in identity changed" });
+    }
+  }, [composerHandoff$1, sessionPrincipal, configuredApiKey]);
   react.useEffect(
     () => () => composerHandoff.clearComposerHandoff({ store: composerHandoff$1, reason: "the chat provider unmounted" }),
     [composerHandoff$1]
@@ -600,6 +745,12 @@ function AthenaProvider({
   const effectiveBackendUrl = mint.getAthenaApiBaseUrl(
     configuredBackendUrl ?? bridge.backendBaseUrl ?? environmentUrls.backendUrl
   );
+  outboxCheckBackendRef.current = effectiveBackendUrl;
+  outboxCheckAuthRef.current = {
+    apiKey: configuredApiKey,
+    token: effectiveToken,
+    getToken: credentialGetToken
+  };
   const hostStatewireSyncUrl = configuredStatewireSyncUrl ?? bridge.statewireSyncUrl ?? void 0;
   const effectiveAppUrl = configuredAppUrl ?? bridge.appUrl ?? athenaUrls.deriveAthenaAppUrl({ apiUrl: effectiveApiUrl, backendUrl: effectiveBackendUrl }) ?? environmentUrls.appUrl;
   if (!bridge.ready) {
@@ -640,6 +791,9 @@ function AthenaProvider({
     snapshotCache,
     toolOutputs: toolOutputs$1,
     historyStore,
+    chatOutbox: chatOutboxHost,
+    sessionLane,
+    subagentLaneHistory,
     linkClicks,
     citationLinks
   };
@@ -672,7 +826,7 @@ function AthenaProvider({
     {
       className: themeStyleVars ? "athena-sdk athena-themed" : "athena-sdk",
       style: themeStyleVars,
-      children: /* @__PURE__ */ jsxRuntime.jsx(PostHogProvider.PostHogProvider, { config: posthogConfig, children: /* @__PURE__ */ jsxRuntime.jsx(composerHandoff.ComposerHandoffContext.Provider, { value: composerHandoff$1, children: /* @__PURE__ */ jsxRuntime.jsx(themeStyleContext.AthenaThemeStyleContext.Provider, { value: themeStyleVars, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaCredentialSourceContext.AthenaCredentialSourceContext.Provider, { value: credentials, children: /* @__PURE__ */ jsxRuntime.jsx(agentSelectionContext.AthenaAgentSelectionContext.Provider, { value: agentSelection, children: /* @__PURE__ */ jsxRuntime.jsx(offlineHistory.AthenaOfflineSnapshotsContext.Provider, { value: offlineSnapshots, children: awaitingKeptSnapshots ? null : inner }) }) }) }) }) })
+      children: /* @__PURE__ */ jsxRuntime.jsx(PostHogProvider.PostHogProvider, { config: posthogConfig, children: /* @__PURE__ */ jsxRuntime.jsx(composerHandoff.ComposerHandoffContext.Provider, { value: composerHandoff$1, children: /* @__PURE__ */ jsxRuntime.jsx(statewireSessionLane.StatewireSessionLaneContext.Provider, { value: sessionLane, children: /* @__PURE__ */ jsxRuntime.jsx(themeStyleContext.AthenaThemeStyleContext.Provider, { value: themeStyleVars, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaCredentialSourceContext.AthenaCredentialSourceContext.Provider, { value: credentials, children: /* @__PURE__ */ jsxRuntime.jsx(agentSelectionContext.AthenaAgentSelectionContext.Provider, { value: agentSelection, children: /* @__PURE__ */ jsxRuntime.jsx(offlineHistory.AthenaOfflineSnapshotsContext.Provider, { value: offlineSnapshots, children: awaitingKeptSnapshots ? null : inner }) }) }) }) }) }) })
     }
   );
 }