@athenaintel/react 0.60.0 → 0.61.0

package/dist/provider/AthenaProvider.cjs

@@ -38,6 +38,10 @@ const statewireServerLaneContext = require("../runtime/statewire-server-lane-con
 const athenaDiagnostics = require("../diagnostics/athena-diagnostics.cjs");
 const errors = require("../diagnostics/errors.cjs");
 const statewireClientTools = require("../chat/statewire-client-tools.cjs");
+const clientToolWriteConsent = require("../chat/client-tool-write-consent.cjs");
+const AgentUiLayer = require("../agent-ui/AgentUiLayer.cjs");
+const tools = require("../agent-ui/tools.cjs");
+const useAgentUi = require("../agent-ui/use-agent-ui.cjs");
 const tooltip = require("../ui/tooltip.cjs");
 const autoOpenTools = require("../tools/auto-open-tools.cjs");
 const AthenaContext = require("./AthenaContext.cjs");
@@ -60,6 +64,8 @@ const threadStatus = require("../agent-runtime/src/thread-status.cjs");
 const useDeepAgentChatOutbox = require("../agent-runtime/src/use-deep-agent-chat-outbox.cjs");
 const chatOutboxSignals = require("../agent-runtime/src/chat-outbox-signals.cjs");
 const useDeepAgentConfig = require("../agent-runtime/src/use-deep-agent-config.cjs");
+const EMPTY_AGENT_UI_OPTIONS = {};
+const EMPTY_TOOLKIT = {};
 const resolveTokenOverride = ({
   config,
   token
@@ -179,7 +185,7 @@ function AthenaStatewireRuntimeMount({
   agent,
   channel,
   mode,
-  tools,
+  tools: tools2,
   frontendToolIds,
   frontendTools,
   workbench,
@@ -237,7 +243,7 @@ function AthenaStatewireRuntimeMount({
     agent,
     channel,
     mode,
-    tools,
+    tools: tools2,
     frontendToolIds,
     frontendToolkit: frontendTools,
     workbench,
@@ -355,7 +361,7 @@ ${draft}` : draft);
   const threadConfig = useProviderThreadConfig({
     model,
     systemPrompt,
-    tools,
+    tools: tools2,
     frontendToolIds,
     workbench,
     knowledgeBase,
@@ -391,6 +397,8 @@ function AthenaStatewireStandalone({
   linkClicks,
   citationLinks,
   modes,
+  confirmClientToolWrites,
+  clientToolConsentAgentName,
   ...runtimeProps
 }) {
   const generatedIdRef = require$$0.useRef(null);
@@ -424,6 +432,13 @@ function AthenaStatewireStandalone({
   };
   const threadMode = hostModes.length > 0 ? listedPick(threadId) ?? sessionChannels[`${modeScope}\0${threadId}`] : void 0;
   const clientTools = useAthenaStatewireRuntime.useStatewireClientTools(runtimeProps.frontendTools);
+  const writeConsentStore = clientToolWriteConsent.useClientToolWriteConsentStore(
+    `${principal ?? "signing-in"}\0${runtimeProps.syncUrl}`
+  );
+  const writeConsentSummaries = require$$0.useMemo(
+    () => statewireClientTools.clientToolWriteConsentSummaries(runtimeProps.frontendTools),
+    [runtimeProps.frontendTools]
+  );
   const clientToolNamesKey = clientTools.map((tool) => tool.definition.name).join(",");
   const clientToolAppId = runtimeProps.appId;
   require$$0.useEffect(() => {
@@ -463,8 +478,17 @@ function AthenaStatewireStandalone({
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [modeKey, modeScope, modePicks, sessionChannels, published, setModePick, setSessionChannel]
   );
-  const scope = published && /* @__PURE__ */ jsxRuntime.jsx(react.AuiProvider, { value: published.client, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsx(threadIdContext.AthenaStatewireThreadIdContext.Provider, { value: published.threadId, children: /* @__PURE__ */ jsxRuntime.jsx(statewireLifecycleContext.AthenaStatewireLifecycleContext.Provider, { value: published.lifecycle, children: /* @__PURE__ */ jsxRuntime.jsx(threadStatus.DeepAgentConnectionContext.Provider, { value: published.lifecycle.connection, children: /* @__PURE__ */ jsxRuntime.jsx(statewireServerLaneContext.AthenaStatewireServerLaneContext.Provider, { value: published.serverLane, children: /* @__PURE__ */ jsxRuntime.jsx(statewireWindowContext.StatewireToolOutputsContext.Provider, { value: runtimeProps.toolOutputs, children: /* @__PURE__ */ jsxRuntime.jsx(useTurnSteps.TurnStepsControlContext.Provider, { value: earlierStepsButton.EarlierStepsButton, children: /* @__PURE__ */ jsxRuntime.jsx(statewireHistoryContext.StatewireHistoryContext.Provider, { value: runtimeProps.historyStore, children: /* @__PURE__ */ jsxRuntime.jsxs(tooltip.TooltipProvider, { children: [
-    /* @__PURE__ */ jsxRuntime.jsx(StatewireClientToolBridge.StatewireClientToolBridge, { tools: clientTools, threadId: published.threadId }),
+  const scope = published && /* @__PURE__ */ jsxRuntime.jsx(react.AuiProvider, { value: published.client, children: /* @__PURE__ */ jsxRuntime.jsx(AthenaContext.AthenaContext.Provider, { value: athenaConfig, children: /* @__PURE__ */ jsxRuntime.jsx(threadIdContext.AthenaStatewireThreadIdContext.Provider, { value: published.threadId, children: /* @__PURE__ */ jsxRuntime.jsx(statewireLifecycleContext.AthenaStatewireLifecycleContext.Provider, { value: published.lifecycle, children: /* @__PURE__ */ jsxRuntime.jsx(threadStatus.DeepAgentConnectionContext.Provider, { value: published.lifecycle.connection, children: /* @__PURE__ */ jsxRuntime.jsx(statewireServerLaneContext.AthenaStatewireServerLaneContext.Provider, { value: published.serverLane, children: /* @__PURE__ */ jsxRuntime.jsx(statewireWindowContext.StatewireToolOutputsContext.Provider, { value: runtimeProps.toolOutputs, children: /* @__PURE__ */ jsxRuntime.jsx(useTurnSteps.TurnStepsControlContext.Provider, { value: earlierStepsButton.EarlierStepsButton, children: /* @__PURE__ */ jsxRuntime.jsx(statewireHistoryContext.StatewireHistoryContext.Provider, { value: runtimeProps.historyStore, children: /* @__PURE__ */ jsxRuntime.jsx(tooltip.TooltipProvider, { children: /* @__PURE__ */ jsxRuntime.jsxs(clientToolWriteConsent.ClientToolWriteConsentContext.Provider, { value: writeConsentStore, children: [
+    /* @__PURE__ */ jsxRuntime.jsx(
+      StatewireClientToolBridge.StatewireClientToolBridge,
+      {
+        tools: clientTools,
+        threadId: published.threadId,
+        writeConsentSummaries,
+        writeConsentAgentName: clientToolConsentAgentName,
+        confirmWrites: confirmClientToolWrites
+      }
+    ),
     /* @__PURE__ */ jsxRuntime.jsx(
       StatewireThreadConfigLaneSync.StatewireThreadConfigLaneSync,
       {
@@ -494,7 +518,7 @@ function AthenaStatewireStandalone({
         children
       }
     ) }) })
-  ] }) }) }) }) }) }) }) }) }) });
+  ] }) }) }) }) }) }) }) }) }) }) });
   return /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
     /* @__PURE__ */ jsxRuntime.jsx(
       AthenaStatewireRuntimeMount,
@@ -618,8 +642,10 @@ function AthenaProvider({
   channel,
   modes,
   model,
-  tools = [],
+  tools: tools$1 = [],
   frontendTools = {},
+  confirmClientToolWrites,
+  agentUi: agentUiProp = false,
   disableAutoOpen = false,
   apiUrl,
   backendUrl,
@@ -694,9 +720,26 @@ function AthenaProvider({
     () => statewireClientTools.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.useAgentUi({ ...agentUiOptions, actConsent: agentUiActConsent });
+  const agentUiTools = require$$0.useMemo(() => {
+    if (!agentUiEnabled) return EMPTY_TOOLKIT;
+    const act = agentUiHost.tools[tools.AGENT_UI_TOOL_NAMES.actOn];
+    if (!act || agentUiActConsent !== "host") return agentUiHost.tools;
+    return {
+      ...agentUiHost.tools,
+      [tools.AGENT_UI_TOOL_NAMES.actOn]: statewireClientTools.requiresWriteConsent(act, { summary: tools.AGENT_UI_ACT_SUMMARY })
+    };
+  }, [agentUiActConsent, agentUiEnabled, agentUiHost.tools]);
   const effectiveFrontendTools = require$$0.useMemo(
-    () => disableAutoOpen ? frontendTools : { ...autoOpenTools.DEFAULT_AUTO_OPEN_TOOLS, ...frontendTools },
-    [disableAutoOpen, frontendTools]
+    () => ({
+      ...disableAutoOpen ? {} : autoOpenTools.DEFAULT_AUTO_OPEN_TOOLS,
+      ...agentUiTools,
+      ...frontendTools
+    }),
+    [agentUiTools, disableAutoOpen, frontendTools]
   );
   const themeStyleVars = require$$0.useMemo(() => theme ? index.themeToStyleVars(theme) : void 0, [theme]);
   const configuredEnvironment = (config == null ? void 0 : config.environment) ?? environment;
@@ -918,9 +961,11 @@ function AthenaProvider({
     agent,
     channel,
     modes,
-    tools,
+    tools: tools$1,
     frontendToolIds: frontendToolNames,
     frontendTools: effectiveFrontendTools,
+    confirmClientToolWrites,
+    clientToolConsentAgentName: agentUiEnabled ? agentUiOptions.agentName : void 0,
     workbench,
     knowledgeBase,
     systemPrompt,
@@ -966,7 +1011,10 @@ 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(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: /* @__PURE__ */ jsxRuntime.jsx(athenaDataClient.AthenaDataQueryClientProvider, { 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.jsxs(themeStyleContext.AthenaThemeStyleContext.Provider, { value: themeStyleVars, children: [
+        agentUiEnabled ? /* @__PURE__ */ jsxRuntime.jsx(AgentUiLayer.AgentUiLayer, { agentUi: agentUiHost.agentUi, classNames: agentUiClassNames }) : null,
+        /* @__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: /* @__PURE__ */ jsxRuntime.jsx(athenaDataClient.AthenaDataQueryClientProvider, { children: awaitingKeptSnapshots ? null : inner }) }) }) })
+      ] }) }) }) })
     }
   );
 }