@athenaintel/react 0.57.2 → 0.59.0

package/dist/provider/AthenaProvider.js

@@ -1,13 +1,15 @@
 import { jsx, jsxs, Fragment } from "react/jsx-runtime";
 import { AuiProvider, AuiConfig, useAui } from "@assistant-ui/react";
 import { useMemo, useRef, useEffect, useContext, useState, useCallback, useLayoutEffect, useReducer } from "react";
+import { giveBackDeepAgentOutboxNamedAgents } from "../agent-runtime/src/named-agents.js";
 import { useKeptSnapshotsGate, AthenaOfflineSnapshotsContext } from "../offline/offline-history.js";
-import { chatOutboxStoreFor } from "../offline/chat-outbox.js";
+import { chatOutboxStorageFor, resolveChatSessionKeeping } from "../offline/chat-outbox.js";
 import { getAthenaEnvironmentUrls, deriveAthenaAppUrl, assertHostMintedStatewireMount, resolveStatewireSyncUrl } from "../lib/athena-urls.js";
 import { getAthenaApiBaseUrl } from "../collab/mint.js";
 import { useParentBridge } from "../runtime/auth.js";
 import { AthenaAuthContext } from "../auth/AthenaAuthContext.js";
 import { AthenaCredentialSourceContext } from "../auth/AthenaCredentialSourceContext.js";
+import { AthenaDataQueryClientProvider } from "../data/athena-data-client.js";
 import { createAthenaCredentialSource } from "../auth/credential-source.js";
 import { useRestartAfterAuthRefusal } from "../runtime/auth-refusal-restart.js";
 import { useStatewireClientTools, useAthenaStatewireRuntime } from "../runtime/useAthenaStatewireRuntime.js";
@@ -20,6 +22,7 @@ import { providerThreadConfig } from "../runtime/statewire-thread-config-lane.js
 import { mintThreadId } from "../threads/mint-thread-id.js";
 import { AthenaStatewireLifecycleContext } from "./statewire-lifecycle-context.js";
 import { createComposerHandoffStore, clearComposerHandoff, ComposerHandoffContext, useComposerHandoffStore, useComposerReady, drainComposerHandoff, applyComposerWrite, rememberRestoredAttachments } from "../composer/composer-handoff.js";
+import { giveBackHeldAttachedReferences } from "../composer/attached-references.js";
 import { takeDraftKeptForSignBackIn, keepDraftForSignBackIn } from "../composer/sign-back-in-draft.js";
 import { restoreSignBackInHash, bridgeOwnsCredential, ATHENA_SIGN_BACK_IN_EVENT } from "../auth/sign-back-in.js";
 import { createStatewireHistoryStore, StatewireHistoryContext } from "./statewire-history-context.js";
@@ -27,6 +30,7 @@ import { EarlierStepsButton } from "../chat/super-grouping/earlier-steps-button.
 import { TurnStepsControlContext } from "../chat/super-grouping/use-turn-steps.js";
 import { StatewireToolOutputsContext } from "./statewire-window-context.js";
 import { StatewireClientToolBridge } from "../chat/StatewireClientToolBridge.js";
+import { StatewirePlanModeLaneSync } from "../runtime/StatewirePlanModeLaneSync.js";
 import { StatewireThreadConfigLaneSync } from "../runtime/StatewireThreadConfigLaneSync.js";
 import { AthenaStatewireServerLaneContext, STATEWIRE_SERVER_LANE_PENDING, nextServerLane } from "../runtime/statewire-server-lane-context.js";
 import { installAthenaDiagnosticsGlobal, athenaDiagnostics, debugPropToDiagnosticsConfig } from "../diagnostics/athena-diagnostics.js";
@@ -52,8 +56,8 @@ import { createDeepAgentToolOutputStore } from "../agent-runtime/src/tool-output
 import { clearDeepAgentHistoryStores } from "../agent-runtime/src/history-store.js";
 import { DeepAgentConnectionContext } from "../agent-runtime/src/thread-status.js";
 import { DeepAgentChatOutboxContext, useDeepAgentChatOutbox } from "../agent-runtime/src/use-deep-agent-chat-outbox.js";
+import { documentVisibleSignal, browserOnlineSignal, allChatOutboxSignals } from "../agent-runtime/src/chat-outbox-signals.js";
 import { readServerConfig } from "../agent-runtime/src/use-deep-agent-config.js";
-import { browserOnlineSignal, allChatOutboxSignals, documentVisibleSignal } from "../agent-runtime/src/chat-outbox-signals.js";
 const resolveTokenOverride = ({
   config,
   token
@@ -146,6 +150,7 @@ function ComposerDraftCarry({ isNewChat }) {
 }
 const EMPTY_MODES = [];
 const CHAT_OUTBOX_ATTACHMENT_READ_TIMEOUT_MS = 1e4;
+const CHAT_SESSION_KEEPING_PENDING = { kind: "pending" };
 function withEntry(record, key, value) {
   if (value === null) {
     if (!(key in record)) return record;
@@ -180,6 +185,7 @@ function AthenaStatewireRuntimeMount({
   systemPrompt,
   appId,
   extraRunConfig,
+  planMode,
   snapshotCache,
   toolOutputs,
   chatOutbox = null,
@@ -209,7 +215,7 @@ function AthenaStatewireRuntimeMount({
     },
     [binding, threadId]
   );
-  const outboxRef = useRef(null);
+  const [defaultForeground] = useState(documentVisibleSignal);
   const [lifecycle, dispatchLifecycle] = useReducer(
     statewireLifecycleReducer,
     initialStatewireLifecycle
@@ -237,6 +243,7 @@ function AthenaStatewireRuntimeMount({
     systemPrompt,
     appId,
     extraRunConfig,
+    planMode,
     snapshotCache,
     toolOutputs,
     subagentLaneHistory,
@@ -253,9 +260,9 @@ 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);
+    ...chatOutbox && {
+      chatSession: chatOutbox.keeping,
+      foreground: chatOutbox.foreground ?? defaultForeground
     },
     // The lane as the host holds it: anything that decides from the lane waits
     // for this, never the placeholder or remembered snapshot rendered before.
@@ -292,17 +299,20 @@ function AthenaStatewireRuntimeMount({
     () => allChatOutboxSignals(hostReachability ?? browserOnline, channelReachability),
     [hostReachability, browserOnline, channelReachability]
   );
-  const [defaultForeground] = useState(documentVisibleSignal);
   const composerHandoff = useComposerHandoffStore();
   const outbox = useDeepAgentChatOutbox({
     client,
     threadId,
     isNewChat,
-    store: (chatOutbox == null ? void 0 : chatOutbox.store) ?? null,
+    ...chatOutbox && {
+      directory: chatOutbox.storage.directory,
+      legacy: { reader: chatOutbox.storage.legacy }
+    },
     reachability,
     foreground: (chatOutbox == null ? void 0 : chatOutbox.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.
+    // as mentions in it, remembered so a message held again from it is asked about again, the
+    // agents it named come back to the To line, and what an app attached to it comes back as chips.
     restore: (entry) => {
       const draft = entry.composerMarkdown ?? entry.text;
       const current = client.composer.getState().text ?? "";
@@ -316,12 +326,13 @@ ${draft}` : draft);
           attachments: entry.attachments
         });
       }
+      giveBackDeepAgentOutboxNamedAgents(entry);
+      giveBackHeldAttachedReferences(entry);
     },
     ...(chatOutbox == null ? void 0 : chatOutbox.openThread) && { openThread: chatOutbox.openThread },
     ...(chatOutbox == null ? void 0 : chatOutbox.checkAttachments) && { checkAttachments: chatOutbox.checkAttachments },
-    enabled: chatOutbox !== null
+    enabled: chatOutbox !== null && chatOutbox.keeping.kind !== "pending"
   });
-  outboxRef.current = outbox;
   useRestartAfterAuthRefusal({
     connection: lifecycle.connection,
     reconnect: lifecycle.reconnect
@@ -358,9 +369,10 @@ ${draft}` : draft);
       lifecycle: lifecycleValue,
       serverLane,
       threadConfig,
+      planMode: planMode ?? null,
       outbox
     }),
-    [binding, sessionPersistence, threadId, isNewChat, client, lifecycleValue, serverLane, threadConfig, outbox]
+    [binding, sessionPersistence, threadId, isNewChat, client, lifecycleValue, serverLane, threadConfig, planMode, outbox]
   );
   useLayoutEffect(() => {
     onPublish(snapshot);
@@ -461,6 +473,16 @@ function AthenaStatewireStandalone({
       },
       published.threadId
     ),
+    /* @__PURE__ */ jsx(
+      StatewirePlanModeLaneSync,
+      {
+        threadId: published.threadId,
+        planMode: published.planMode,
+        lane: published.serverLane,
+        channelLive: isStatewireChannelLive(published.lifecycle.connection)
+      },
+      `plan-mode:${published.threadId}`
+    ),
     /* @__PURE__ */ jsx(ComposerHandoffDrain, {}),
     /* @__PURE__ */ jsx(ComposerDraftCarry, { isNewChat: published.isNewChat }),
     /* @__PURE__ */ jsx(DeepAgentChatOutboxContext.Provider, { value: published.outbox, children: /* @__PURE__ */ jsx(AthenaChatModeSelectionContext.Provider, { value: chatModeSelection, children: /* @__PURE__ */ jsx(
@@ -606,6 +628,7 @@ function AthenaProvider({
   workbench,
   knowledgeBase,
   systemPrompt,
+  planMode,
   appId,
   threadId: threadIdProp,
   threadIsNew,
@@ -735,13 +758,25 @@ function AthenaProvider({
   const chatOutboxRef = useRef(chatOutbox);
   chatOutboxRef.current = chatOutbox;
   const chatOutboxOwnerId = (chatOutbox == null ? void 0 : chatOutbox.ownerId) ?? null;
-  const chatOutboxStore = useMemo(() => {
+  const chatOutboxStorage = useMemo(() => {
     const host = chatOutboxRef.current;
-    return host && host.ownerId === chatOutboxOwnerId ? chatOutboxStoreFor(host) : null;
+    return host && host.ownerId === chatOutboxOwnerId ? chatOutboxStorageFor(host) : null;
   }, [chatOutboxOwnerId]);
   useEffect(() => {
-    if (chatOutbox) chatOutboxStoreFor(chatOutbox);
+    if (chatOutbox) chatOutboxStorageFor(chatOutbox);
   }, [chatOutbox]);
+  const [chatSessionKeeping, setChatSessionKeeping] = useState(null);
+  useEffect(() => {
+    if (!chatOutboxStorage) return;
+    let live = true;
+    void resolveChatSessionKeeping(chatOutboxStorage).then((keeping) => {
+      if (live) setChatSessionKeeping({ storage: chatOutboxStorage, keeping });
+    });
+    return () => {
+      live = false;
+    };
+  }, [chatOutboxStorage]);
+  const chatOutboxKeeping = (chatSessionKeeping == null ? void 0 : chatSessionKeeping.storage) === chatOutboxStorage && chatSessionKeeping ? chatSessionKeeping.keeping : CHAT_SESSION_KEEPING_PENDING;
   const chatOutboxReachability = chatOutbox == null ? void 0 : chatOutbox.reachability;
   const chatOutboxForeground = chatOutbox == null ? void 0 : chatOutbox.foreground;
   const chatOutboxOpenThread = chatOutbox == null ? void 0 : chatOutbox.openThread;
@@ -760,16 +795,17 @@ function AthenaProvider({
     })
   );
   const chatOutboxHost = useMemo(
-    () => chatOutboxOwnerId === null ? null : {
-      store: chatOutboxStore,
+    () => chatOutboxStorage === null ? null : {
+      storage: chatOutboxStorage,
+      keeping: chatOutboxKeeping,
       ...chatOutboxReachability && { reachability: chatOutboxReachability },
       ...chatOutboxForeground && { foreground: chatOutboxForeground },
       ...chatOutboxOpenThread && { openThread: chatOutboxOpenThread },
       checkAttachments: checkChatOutboxAttachments
     },
     [
-      chatOutboxOwnerId,
-      chatOutboxStore,
+      chatOutboxStorage,
+      chatOutboxKeeping,
       chatOutboxReachability,
       chatOutboxForeground,
       chatOutboxOpenThread,
@@ -885,6 +921,7 @@ function AthenaProvider({
     workbench,
     knowledgeBase,
     systemPrompt,
+    planMode,
     appId,
     extraRunConfig: configuredExtraRunConfig,
     snapshotCache,
@@ -926,7 +963,7 @@ function AthenaProvider({
     {
       className: themeStyleVars ? "athena-sdk athena-themed" : "athena-sdk",
       style: themeStyleVars,
-      children: /* @__PURE__ */ jsx(PostHogProvider, { config: posthogConfig, children: /* @__PURE__ */ jsx(ComposerHandoffContext.Provider, { value: composerHandoff, children: /* @__PURE__ */ jsx(StatewireSessionLaneContext.Provider, { value: sessionLane, children: /* @__PURE__ */ jsx(AthenaThemeStyleContext.Provider, { value: themeStyleVars, children: /* @__PURE__ */ jsx(AthenaCredentialSourceContext.Provider, { value: credentials, children: /* @__PURE__ */ jsx(AthenaAgentSelectionContext.Provider, { value: agentSelection, children: /* @__PURE__ */ jsx(AthenaOfflineSnapshotsContext.Provider, { value: offlineSnapshots, children: awaitingKeptSnapshots ? null : inner }) }) }) }) }) }) })
+      children: /* @__PURE__ */ jsx(PostHogProvider, { config: posthogConfig, children: /* @__PURE__ */ jsx(ComposerHandoffContext.Provider, { value: composerHandoff, children: /* @__PURE__ */ jsx(StatewireSessionLaneContext.Provider, { value: sessionLane, children: /* @__PURE__ */ jsx(AthenaThemeStyleContext.Provider, { value: themeStyleVars, children: /* @__PURE__ */ jsx(AthenaCredentialSourceContext.Provider, { value: credentials, children: /* @__PURE__ */ jsx(AthenaAgentSelectionContext.Provider, { value: agentSelection, children: /* @__PURE__ */ jsx(AthenaOfflineSnapshotsContext.Provider, { value: offlineSnapshots, children: /* @__PURE__ */ jsx(AthenaDataQueryClientProvider, { children: awaitingKeptSnapshots ? null : inner }) }) }) }) }) }) }) })
     }
   );
 }