@athenaintel/react 0.60.0 → 0.61.0

package/dist/provider/AthenaProvider.js

@@ -35,7 +35,11 @@ import { StatewireThreadConfigLaneSync } from "../runtime/StatewireThreadConfigL
 import { AthenaStatewireServerLaneContext, STATEWIRE_SERVER_LANE_PENDING, nextServerLane } from "../runtime/statewire-server-lane-context.js";
 import { installAthenaDiagnosticsGlobal, athenaDiagnostics, debugPropToDiagnosticsConfig } from "../diagnostics/athena-diagnostics.js";
 import { AthenaSdkError } from "../diagnostics/errors.js";
-import { backendDeclaredFrontendToolIds, resolveClientToolSurface } from "../chat/statewire-client-tools.js";
+import { backendDeclaredFrontendToolIds, requiresWriteConsent, clientToolWriteConsentSummaries, resolveClientToolSurface } from "../chat/statewire-client-tools.js";
+import { useClientToolWriteConsentStore, ClientToolWriteConsentContext } from "../chat/client-tool-write-consent.js";
+import { AgentUiLayer } from "../agent-ui/AgentUiLayer.js";
+import { AGENT_UI_TOOL_NAMES, AGENT_UI_ACT_SUMMARY } from "../agent-ui/tools.js";
+import { useAgentUi } from "../agent-ui/use-agent-ui.js";
 import { TooltipProvider } from "../ui/tooltip.js";
 import { DEFAULT_AUTO_OPEN_TOOLS } from "../tools/auto-open-tools.js";
 import { AthenaContext } from "./AthenaContext.js";
@@ -58,6 +62,8 @@ import { DeepAgentConnectionContext } from "../agent-runtime/src/thread-status.j
 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";
+const EMPTY_AGENT_UI_OPTIONS = {};
+const EMPTY_TOOLKIT = {};
 const resolveTokenOverride = ({
   config,
   token
@@ -389,6 +395,8 @@ function AthenaStatewireStandalone({
   linkClicks,
   citationLinks,
   modes,
+  confirmClientToolWrites,
+  clientToolConsentAgentName,
   ...runtimeProps
 }) {
   const generatedIdRef = useRef(null);
@@ -422,6 +430,13 @@ function AthenaStatewireStandalone({
   };
   const threadMode = hostModes.length > 0 ? listedPick(threadId) ?? sessionChannels[`${modeScope}\0${threadId}`] : void 0;
   const clientTools = useStatewireClientTools(runtimeProps.frontendTools);
+  const writeConsentStore = useClientToolWriteConsentStore(
+    `${principal ?? "signing-in"}\0${runtimeProps.syncUrl}`
+  );
+  const writeConsentSummaries = useMemo(
+    () => clientToolWriteConsentSummaries(runtimeProps.frontendTools),
+    [runtimeProps.frontendTools]
+  );
   const clientToolNamesKey = clientTools.map((tool) => tool.definition.name).join(",");
   const clientToolAppId = runtimeProps.appId;
   useEffect(() => {
@@ -461,8 +476,17 @@ function AthenaStatewireStandalone({
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [modeKey, modeScope, modePicks, sessionChannels, published, setModePick, setSessionChannel]
   );
-  const scope = published && /* @__PURE__ */ jsx(AuiProvider, { value: published.client, children: /* @__PURE__ */ jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsx(AthenaStatewireThreadIdContext.Provider, { value: published.threadId, children: /* @__PURE__ */ jsx(AthenaStatewireLifecycleContext.Provider, { value: published.lifecycle, children: /* @__PURE__ */ jsx(DeepAgentConnectionContext.Provider, { value: published.lifecycle.connection, children: /* @__PURE__ */ jsx(AthenaStatewireServerLaneContext.Provider, { value: published.serverLane, children: /* @__PURE__ */ jsx(StatewireToolOutputsContext.Provider, { value: runtimeProps.toolOutputs, children: /* @__PURE__ */ jsx(TurnStepsControlContext.Provider, { value: EarlierStepsButton, children: /* @__PURE__ */ jsx(StatewireHistoryContext.Provider, { value: runtimeProps.historyStore, children: /* @__PURE__ */ jsxs(TooltipProvider, { children: [
-    /* @__PURE__ */ jsx(StatewireClientToolBridge, { tools: clientTools, threadId: published.threadId }),
+  const scope = published && /* @__PURE__ */ jsx(AuiProvider, { value: published.client, children: /* @__PURE__ */ jsx(AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsx(AthenaStatewireThreadIdContext.Provider, { value: published.threadId, children: /* @__PURE__ */ jsx(AthenaStatewireLifecycleContext.Provider, { value: published.lifecycle, children: /* @__PURE__ */ jsx(DeepAgentConnectionContext.Provider, { value: published.lifecycle.connection, children: /* @__PURE__ */ jsx(AthenaStatewireServerLaneContext.Provider, { value: published.serverLane, children: /* @__PURE__ */ jsx(StatewireToolOutputsContext.Provider, { value: runtimeProps.toolOutputs, children: /* @__PURE__ */ jsx(TurnStepsControlContext.Provider, { value: EarlierStepsButton, children: /* @__PURE__ */ jsx(StatewireHistoryContext.Provider, { value: runtimeProps.historyStore, children: /* @__PURE__ */ jsx(TooltipProvider, { children: /* @__PURE__ */ jsxs(ClientToolWriteConsentContext.Provider, { value: writeConsentStore, children: [
+    /* @__PURE__ */ jsx(
+      StatewireClientToolBridge,
+      {
+        tools: clientTools,
+        threadId: published.threadId,
+        writeConsentSummaries,
+        writeConsentAgentName: clientToolConsentAgentName,
+        confirmWrites: confirmClientToolWrites
+      }
+    ),
     /* @__PURE__ */ jsx(
       StatewireThreadConfigLaneSync,
       {
@@ -492,7 +516,7 @@ function AthenaStatewireStandalone({
         children
       }
     ) }) })
-  ] }) }) }) }) }) }) }) }) }) });
+  ] }) }) }) }) }) }) }) }) }) }) });
   return /* @__PURE__ */ jsxs(Fragment, { children: [
     /* @__PURE__ */ jsx(
       AthenaStatewireRuntimeMount,
@@ -618,6 +642,8 @@ function AthenaProvider({
   model,
   tools = [],
   frontendTools = {},
+  confirmClientToolWrites,
+  agentUi: agentUiProp = false,
   disableAutoOpen = false,
   apiUrl,
   backendUrl,
@@ -692,9 +718,26 @@ function AthenaProvider({
     () => backendDeclaredFrontendToolIds(frontendTools),
     [frontendTools]
   );
+  const agentUiEnabled = agentUiProp === true || typeof agentUiProp === "object" && agentUiProp !== null;
+  const { classNames: agentUiClassNames, ...agentUiOptions } = typeof agentUiProp === "object" && agentUiProp !== null ? agentUiProp : EMPTY_AGENT_UI_OPTIONS;
+  const agentUiActConsent = agentUiOptions.actConsent ?? "host";
+  const agentUiHost = useAgentUi({ ...agentUiOptions, actConsent: agentUiActConsent });
+  const agentUiTools = useMemo(() => {
+    if (!agentUiEnabled) return EMPTY_TOOLKIT;
+    const act = agentUiHost.tools[AGENT_UI_TOOL_NAMES.actOn];
+    if (!act || agentUiActConsent !== "host") return agentUiHost.tools;
+    return {
+      ...agentUiHost.tools,
+      [AGENT_UI_TOOL_NAMES.actOn]: requiresWriteConsent(act, { summary: AGENT_UI_ACT_SUMMARY })
+    };
+  }, [agentUiActConsent, agentUiEnabled, agentUiHost.tools]);
   const effectiveFrontendTools = useMemo(
-    () => disableAutoOpen ? frontendTools : { ...DEFAULT_AUTO_OPEN_TOOLS, ...frontendTools },
-    [disableAutoOpen, frontendTools]
+    () => ({
+      ...disableAutoOpen ? {} : DEFAULT_AUTO_OPEN_TOOLS,
+      ...agentUiTools,
+      ...frontendTools
+    }),
+    [agentUiTools, disableAutoOpen, frontendTools]
   );
   const themeStyleVars = useMemo(() => theme ? themeToStyleVars(theme) : void 0, [theme]);
   const configuredEnvironment = (config == null ? void 0 : config.environment) ?? environment;
@@ -919,6 +962,8 @@ function AthenaProvider({
     tools,
     frontendToolIds: frontendToolNames,
     frontendTools: effectiveFrontendTools,
+    confirmClientToolWrites,
+    clientToolConsentAgentName: agentUiEnabled ? agentUiOptions.agentName : void 0,
     workbench,
     knowledgeBase,
     systemPrompt,
@@ -964,7 +1009,10 @@ 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: /* @__PURE__ */ jsx(AthenaDataQueryClientProvider, { 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__ */ jsxs(AthenaThemeStyleContext.Provider, { value: themeStyleVars, children: [
+        agentUiEnabled ? /* @__PURE__ */ jsx(AgentUiLayer, { agentUi: agentUiHost.agentUi, classNames: agentUiClassNames }) : null,
+        /* @__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 }) }) }) })
+      ] }) }) }) })
     }
   );
 }