@athenaintel/react 0.35.0 → 0.36.0

package/dist/catalog/use-athena-chat-mode.cjs

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const react$1 = require("@assistant-ui/react");
+const reactQuery = require("@tanstack/react-query");
+const react = require("react");
+const apiClient = require("../chat-ui/mentions/suggestions/data/api-client.cjs");
+const agentSelectionContext = require("../provider/agent-selection-context.cjs");
+const chatModeContext = require("../provider/chat-mode-context.cjs");
+const agentModeMenu = require("./agent-mode-menu.cjs");
+const runtimeCatalog = require("./runtime-catalog.cjs");
+const OFF = { status: "off" };
+const LOADING = { status: "loading" };
+const NO_MODES = [];
+const ATHENA_SESSION_MODE_QUERY = `
+  query AthenaSessionMode($assetId: String!) {
+    getAssetById(assetId: $assetId, withMetadata: true) { athenaMetadata }
+  }
+`;
+function useAthenaChatMode() {
+  var _a, _b, _c;
+  const selection = react.useContext(chatModeContext.AthenaChatModeSelectionContext);
+  const agentSelection = react.useContext(agentSelectionContext.AthenaAgentSelectionContext);
+  const agentAssetId = (agentSelection == null ? void 0 : agentSelection.agentAssetId) ?? null;
+  const channel = (agentSelection == null ? void 0 : agentSelection.channel) ?? null;
+  const hostModes = (selection == null ? void 0 : selection.hostModes) ?? NO_MODES;
+  const enabled = selection !== null && hostModes.length > 0 && agentAssetId !== null;
+  const auth = runtimeCatalog.useCatalogAuth();
+  const principal = runtimeCatalog.useCatalogPrincipal();
+  const queryClient = runtimeCatalog.useCatalogQueryClient();
+  const agentRead = reactQuery.useQuery(
+    {
+      queryKey: ["athena-sdk", "catalog", "agent-modes", auth.backendUrl, principal, agentAssetId],
+      queryFn: async () => {
+        try {
+          const data = await apiClient.fetchGraphQL(
+            auth,
+            runtimeCatalog.ATHENA_AGENT_MODELS_QUERY,
+            { assetId: agentAssetId }
+          );
+          if (data.getAssetById === null) return { readable: false };
+          return {
+            readable: true,
+            source: agentModeMenu.publishedAgentModeSource(data.getAssetById.athenaMetadata)
+          };
+        } catch (error) {
+          if (runtimeCatalog.isAgentReadDenial(error)) return { readable: false };
+          throw error;
+        }
+      },
+      enabled,
+      staleTime: runtimeCatalog.AGENT_MODELS_REFRESH_MS,
+      refetchInterval: (query) => query.state.data !== void 0 && !query.state.data.readable ? false : runtimeCatalog.AGENT_MODELS_REFRESH_MS
+    },
+    queryClient
+  );
+  const threadEmpty = react$1.useAuiState((state) => {
+    var _a2;
+    return ((_a2 = state.optional.thread) == null ? void 0 : _a2.isEmpty) ?? true;
+  });
+  const threadRunning = react$1.useAuiState((state) => {
+    var _a2;
+    return ((_a2 = state.optional.thread) == null ? void 0 : _a2.isRunning) ?? false;
+  });
+  const threadId = (selection == null ? void 0 : selection.threadId) ?? null;
+  const sessionStarted = (selection == null ? void 0 : selection.isNewChat) === false || !threadEmpty;
+  const readSession = enabled && threadId !== null && sessionStarted;
+  const sessionRead = reactQuery.useQuery(
+    {
+      queryKey: ["athena-sdk", "chat-mode", "session", auth.backendUrl, principal, threadId],
+      queryFn: async () => {
+        const data = await apiClient.fetchGraphQL(
+          auth,
+          ATHENA_SESSION_MODE_QUERY,
+          { assetId: `asset_${threadId}` }
+        );
+        const asset = data.getAssetById;
+        return asset ? { exists: true, channel: agentModeMenu.sessionChannelFromMetadata(asset.athenaMetadata) } : { exists: false };
+      },
+      enabled: readSession,
+      staleTime: 0,
+      refetchOnWindowFocus: true,
+      retry: false
+    },
+    queryClient
+  );
+  const refetchSession = sessionRead.refetch;
+  const wasRunning = react.useRef(threadRunning);
+  react.useEffect(() => {
+    const ended = wasRunning.current && !threadRunning;
+    wasRunning.current = threadRunning;
+    if (ended && readSession) void refetchSession();
+  }, [threadRunning, readSession, refetchSession]);
+  const source = ((_a = agentRead.data) == null ? void 0 : _a.readable) ? agentRead.data.source : null;
+  const menu = react.useMemo(
+    () => source ? agentModeMenu.agentModeMenu({ source, channel, hostModes }) : null,
+    [source, channel, hostModes]
+  );
+  const pick = (selection == null ? void 0 : selection.pick) ?? null;
+  const session = readSession ? sessionRead.data : void 0;
+  const sessionChannel = (session == null ? void 0 : session.exists) ? session.channel : null;
+  const setPick = selection == null ? void 0 : selection.setPick;
+  const setSessionChannel = selection == null ? void 0 : selection.setSessionChannel;
+  react.useEffect(() => {
+    if (!threadId || !setSessionChannel || session === void 0) return;
+    setSessionChannel({ threadId, channel: sessionChannel });
+  }, [threadId, setSessionChannel, session, sessionChannel]);
+  react.useEffect(() => {
+    if (!threadId || !setPick || pick === null) return;
+    const applied = sessionChannel !== null && (sessionChannel === pick || source !== null && agentModeMenu.agentModeOfChannel({ source, channel: sessionChannel }) === pick);
+    const withdrawn = menu !== null && !menu.options.some((option) => option.id === pick);
+    if (applied || withdrawn) setPick({ threadId, modeId: null });
+  }, [threadId, setPick, pick, sessionChannel, source, menu]);
+  const canSwitch = !((menu == null ? void 0 : menu.locked) && sessionStarted);
+  const options = menu == null ? void 0 : menu.options;
+  const select = react.useCallback(
+    (modeId) => {
+      if (!threadId || !setPick || !canSwitch) return;
+      if (!(options == null ? void 0 : options.some((option) => option.id === modeId))) return;
+      setPick({ threadId, modeId });
+    },
+    [threadId, setPick, canSwitch, options]
+  );
+  if (!enabled) return OFF;
+  if (agentRead.isPending) return LOADING;
+  if (!menu || menu.options.length < 2) return OFF;
+  const current = pick ?? (source && sessionChannel !== null ? agentModeMenu.agentModeOfChannel({ source, channel: sessionChannel }) : menu.firstSendModeId);
+  const currentLabel = ((_b = menu.options.find((option) => option.id === current)) == null ? void 0 : _b.label) ?? (current === "base" ? "Default" : ((_c = source == null ? void 0 : source.channels[current]) == null ? void 0 : _c.label) ?? current);
+  return {
+    status: "ready",
+    options: menu.options,
+    current,
+    currentLabel,
+    canSwitch,
+    locked: menu.locked,
+    select
+  };
+}
+exports.useAthenaChatMode = useAthenaChatMode;
+//# sourceMappingURL=use-athena-chat-mode.cjs.map

package/dist/catalog/use-athena-chat-mode.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-athena-chat-mode.cjs","sources":["../../src/catalog/use-athena-chat-mode.ts"],"sourcesContent":["import { useAuiState } from '@assistant-ui/react';\nimport { useQuery } from '@tanstack/react-query';\nimport { useCallback, useContext, useEffect, useMemo, useRef } from 'react';\nimport { fetchGraphQL } from '../chat-ui/mentions/suggestions/data/api-client';\nimport { AthenaAgentSelectionContext } from '../provider/agent-selection-context';\nimport { AthenaChatModeSelectionContext } from '../provider/chat-mode-context';\nimport {\n  type AgentModeSource,\n  type AthenaChatModeOption,\n  agentModeMenu,\n  agentModeOfChannel,\n  publishedAgentModeSource,\n  sessionChannelFromMetadata,\n} from './agent-mode-menu';\nimport {\n  AGENT_MODELS_REFRESH_MS,\n  ATHENA_AGENT_MODELS_QUERY,\n  isAgentReadDenial,\n  useCatalogAuth,\n  useCatalogPrincipal,\n  useCatalogQueryClient,\n} from './runtime-catalog';\n\n/**\n * The open chat's mode menu (`AthenaProvider` `modes`):\n *\n * - `off`: no menu: the host listed no modes, no collab agent is selected, the\n *   agent cannot be read or was never published, or it offers fewer than two\n *   of the listed modes on the provider's `channel`;\n * - `loading`: the agent is being read;\n * - `ready`: the modes to offer, the one the open session runs in (or a new\n *   one starts in), and `select` to switch. `canSwitch` is false on a session\n *   of a channel its author locks: there, people pick the mode before a\n *   session's first message.\n */\nexport type AthenaChatMode =\n  | { status: 'off' }\n  | { status: 'loading' }\n  | {\n      status: 'ready';\n      options: readonly AthenaChatModeOption[];\n      /** The mode the open session runs in, or a new session starts in. */\n      current: string;\n      /** The current mode's name, also when it is none of `options`. */\n      currentLabel: string;\n      /** Whether a pick applies now. */\n      canSwitch: boolean;\n      /** The channel keeps each session in the mode it starts in. */\n      locked: boolean;\n      /** Run the open session's next sends in `modeId` (one of `options`). */\n      select: (modeId: string) => void;\n    };\n\ntype AgentModeRead = { readable: true; source: AgentModeSource | null } | { readable: false };\n\n/** The open thread's session asset: absent until its first send mints it. */\ntype SessionRead = { exists: false } | { exists: true; channel: string | null };\n\nconst OFF: AthenaChatMode = { status: 'off' };\nconst LOADING: AthenaChatMode = { status: 'loading' };\nconst NO_MODES: readonly string[] = [];\n\nconst ATHENA_SESSION_MODE_QUERY = `\n  query AthenaSessionMode($assetId: String!) {\n    getAssetById(assetId: $assetId, withMetadata: true) { athenaMetadata }\n  }\n`;\n\n/**\n * The mode menu of the chat inside the nearest `AthenaProvider`, from the\n * agent's published definition, the open session's own mode, and the\n * provider's `modes`, `agent` and `channel`. `AthenaChat` renders it in its\n * composer; use it to build a menu of your own.\n *\n * The open session's mode is read again when the chat regains focus and after\n * every run, and its sends name it, so a session switched in another browser\n * continues where it is. A pick holds until the session runs in it (or the\n * menu stops offering it); only a pick not yet sent overrides the session.\n */\nexport function useAthenaChatMode(): AthenaChatMode {\n  const selection = useContext(AthenaChatModeSelectionContext);\n  const agentSelection = useContext(AthenaAgentSelectionContext);\n  const agentAssetId = agentSelection?.agentAssetId ?? null;\n  const channel = agentSelection?.channel ?? null;\n  const hostModes = selection?.hostModes ?? NO_MODES;\n  const enabled = selection !== null && hostModes.length > 0 && agentAssetId !== null;\n  const auth = useCatalogAuth();\n  const principal = useCatalogPrincipal();\n  const queryClient = useCatalogQueryClient();\n\n  const agentRead = useQuery(\n    {\n      queryKey: ['athena-sdk', 'catalog', 'agent-modes', auth.backendUrl, principal, agentAssetId],\n      queryFn: async (): Promise<AgentModeRead> => {\n        try {\n          const data = await fetchGraphQL<{ getAssetById: { athenaMetadata: unknown } | null }>(\n            auth,\n            ATHENA_AGENT_MODELS_QUERY,\n            { assetId: agentAssetId },\n          );\n          if (data.getAssetById === null) return { readable: false };\n          return {\n            readable: true,\n            source: publishedAgentModeSource(data.getAssetById.athenaMetadata),\n          };\n        } catch (error) {\n          if (isAgentReadDenial(error)) return { readable: false };\n          throw error;\n        }\n      },\n      enabled,\n      staleTime: AGENT_MODELS_REFRESH_MS,\n      refetchInterval: (query) =>\n        query.state.data !== undefined && !query.state.data.readable\n          ? false\n          : AGENT_MODELS_REFRESH_MS,\n    },\n    queryClient,\n  );\n\n  // The open thread as this chat holds it: its first send ends a new chat's\n  // pick-before-start window even where the provider cannot tell (a\n  // standalone chat on a thread id it minted), and the end of each run, which\n  // may have switched the session (here or in another browser), reads it again.\n  const threadEmpty = useAuiState((state) => state.optional.thread?.isEmpty ?? true);\n  const threadRunning = useAuiState((state) => state.optional.thread?.isRunning ?? false);\n\n  const threadId = selection?.threadId ?? null;\n  const sessionStarted = selection?.isNewChat === false || !threadEmpty;\n  const readSession = enabled && threadId !== null && sessionStarted;\n  const sessionRead = useQuery(\n    {\n      queryKey: ['athena-sdk', 'chat-mode', 'session', auth.backendUrl, principal, threadId],\n      queryFn: async (): Promise<SessionRead> => {\n        const data = await fetchGraphQL<{ getAssetById: { athenaMetadata: unknown } | null }>(\n          auth,\n          ATHENA_SESSION_MODE_QUERY,\n          { assetId: `asset_${threadId}` },\n        );\n        const asset = data.getAssetById;\n        return asset\n          ? { exists: true, channel: sessionChannelFromMetadata(asset.athenaMetadata) }\n          : { exists: false };\n      },\n      enabled: readSession,\n      staleTime: 0,\n      refetchOnWindowFocus: true,\n      retry: false,\n    },\n    queryClient,\n  );\n  const refetchSession = sessionRead.refetch;\n  const wasRunning = useRef(threadRunning);\n  useEffect(() => {\n    const ended = wasRunning.current && !threadRunning;\n    wasRunning.current = threadRunning;\n    if (ended && readSession) void refetchSession();\n  }, [threadRunning, readSession, refetchSession]);\n\n  const source = agentRead.data?.readable ? agentRead.data.source : null;\n  const menu = useMemo(\n    () => (source ? agentModeMenu({ source, channel, hostModes }) : null),\n    [source, channel, hostModes],\n  );\n  const pick = selection?.pick ?? null;\n  const session = readSession ? sessionRead.data : undefined;\n  const sessionChannel = session?.exists ? session.channel : null;\n  const setPick = selection?.setPick;\n  const setSessionChannel = selection?.setSessionChannel;\n\n  useEffect(() => {\n    if (!threadId || !setSessionChannel || session === undefined) return;\n    setSessionChannel({ threadId, channel: sessionChannel });\n  }, [threadId, setSessionChannel, session, sessionChannel]);\n\n  useEffect(() => {\n    if (!threadId || !setPick || pick === null) return;\n    const applied =\n      sessionChannel !== null &&\n      (sessionChannel === pick ||\n        (source !== null && agentModeOfChannel({ source, channel: sessionChannel }) === pick));\n    const withdrawn = menu !== null && !menu.options.some((option) => option.id === pick);\n    if (applied || withdrawn) setPick({ threadId, modeId: null });\n  }, [threadId, setPick, pick, sessionChannel, source, menu]);\n\n  const canSwitch = !(menu?.locked && sessionStarted);\n  const options = menu?.options;\n  const select = useCallback(\n    (modeId: string) => {\n      if (!threadId || !setPick || !canSwitch) return;\n      if (!options?.some((option) => option.id === modeId)) return;\n      setPick({ threadId, modeId });\n    },\n    [threadId, setPick, canSwitch, options],\n  );\n\n  if (!enabled) return OFF;\n  if (agentRead.isPending) return LOADING;\n  if (!menu || menu.options.length < 2) return OFF;\n  const current =\n    pick ??\n    (source && sessionChannel !== null\n      ? agentModeOfChannel({ source, channel: sessionChannel })\n      : menu.firstSendModeId);\n  const currentLabel =\n    menu.options.find((option) => option.id === current)?.label ??\n    (current === 'base' ? 'Default' : (source?.channels[current]?.label ?? current));\n  return {\n    status: 'ready',\n    options: menu.options,\n    current,\n    currentLabel,\n    canSwitch,\n    locked: menu.locked,\n    select,\n  };\n}\n"],"names":["useContext","AthenaChatModeSelectionContext","AthenaAgentSelectionContext","useCatalogAuth","useCatalogPrincipal","useCatalogQueryClient","useQuery","fetchGraphQL","ATHENA_AGENT_MODELS_QUERY","publishedAgentModeSource","isAgentReadDenial","AGENT_MODELS_REFRESH_MS","useAuiState","_a","sessionChannelFromMetadata","useRef","useEffect","useMemo","agentModeMenu","agentModeOfChannel","useCallback"],"mappings":";;;;;;;;;;AA0DA,MAAM,MAAsB,EAAE,QAAQ,MAAA;AACtC,MAAM,UAA0B,EAAE,QAAQ,UAAA;AAC1C,MAAM,WAA8B,CAAA;AAEpC,MAAM,4BAA4B;AAAA;AAAA;AAAA;AAAA;AAiB3B,SAAS,oBAAoC;;AAClD,QAAM,YAAYA,MAAAA,WAAWC,8CAA8B;AAC3D,QAAM,iBAAiBD,MAAAA,WAAWE,iDAA2B;AAC7D,QAAM,gBAAe,iDAAgB,iBAAgB;AACrD,QAAM,WAAU,iDAAgB,YAAW;AAC3C,QAAM,aAAY,uCAAW,cAAa;AAC1C,QAAM,UAAU,cAAc,QAAQ,UAAU,SAAS,KAAK,iBAAiB;AAC/E,QAAM,OAAOC,eAAAA,eAAA;AACb,QAAM,YAAYC,eAAAA,oBAAA;AAClB,QAAM,cAAcC,eAAAA,sBAAA;AAEpB,QAAM,YAAYC,WAAAA;AAAAA,IAChB;AAAA,MACE,UAAU,CAAC,cAAc,WAAW,eAAe,KAAK,YAAY,WAAW,YAAY;AAAA,MAC3F,SAAS,YAAoC;AAC3C,YAAI;AACF,gBAAM,OAAO,MAAMC,UAAAA;AAAAA,YACjB;AAAA,YACAC,eAAAA;AAAAA,YACA,EAAE,SAAS,aAAA;AAAA,UAAa;AAE1B,cAAI,KAAK,iBAAiB,KAAM,QAAO,EAAE,UAAU,MAAA;AACnD,iBAAO;AAAA,YACL,UAAU;AAAA,YACV,QAAQC,cAAAA,yBAAyB,KAAK,aAAa,cAAc;AAAA,UAAA;AAAA,QAErE,SAAS,OAAO;AACd,cAAIC,eAAAA,kBAAkB,KAAK,EAAG,QAAO,EAAE,UAAU,MAAA;AACjD,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA;AAAA,MACA,WAAWC,eAAAA;AAAAA,MACX,iBAAiB,CAAC,UAChB,MAAM,MAAM,SAAS,UAAa,CAAC,MAAM,MAAM,KAAK,WAChD,QACAA,eAAAA;AAAAA,IAAA;AAAA,IAER;AAAA,EAAA;AAOF,QAAM,cAAcC,QAAAA,YAAY,CAAC,UAAA;;AAAU,aAAAC,MAAA,MAAM,SAAS,WAAf,gBAAAA,IAAuB,YAAW;AAAA,GAAI;AACjF,QAAM,gBAAgBD,QAAAA,YAAY,CAAC,UAAA;;AAAU,aAAAC,MAAA,MAAM,SAAS,WAAf,gBAAAA,IAAuB,cAAa;AAAA,GAAK;AAEtF,QAAM,YAAW,uCAAW,aAAY;AACxC,QAAM,kBAAiB,uCAAW,eAAc,SAAS,CAAC;AAC1D,QAAM,cAAc,WAAW,aAAa,QAAQ;AACpD,QAAM,cAAcP,WAAAA;AAAAA,IAClB;AAAA,MACE,UAAU,CAAC,cAAc,aAAa,WAAW,KAAK,YAAY,WAAW,QAAQ;AAAA,MACrF,SAAS,YAAkC;AACzC,cAAM,OAAO,MAAMC,UAAAA;AAAAA,UACjB;AAAA,UACA;AAAA,UACA,EAAE,SAAS,SAAS,QAAQ,GAAA;AAAA,QAAG;AAEjC,cAAM,QAAQ,KAAK;AACnB,eAAO,QACH,EAAE,QAAQ,MAAM,SAASO,cAAAA,2BAA2B,MAAM,cAAc,EAAA,IACxE,EAAE,QAAQ,MAAA;AAAA,MAChB;AAAA,MACA,SAAS;AAAA,MACT,WAAW;AAAA,MACX,sBAAsB;AAAA,MACtB,OAAO;AAAA,IAAA;AAAA,IAET;AAAA,EAAA;AAEF,QAAM,iBAAiB,YAAY;AACnC,QAAM,aAAaC,MAAAA,OAAO,aAAa;AACvCC,QAAAA,UAAU,MAAM;AACd,UAAM,QAAQ,WAAW,WAAW,CAAC;AACrC,eAAW,UAAU;AACrB,QAAI,SAAS,YAAa,MAAK,eAAA;AAAA,EACjC,GAAG,CAAC,eAAe,aAAa,cAAc,CAAC;AAE/C,QAAM,WAAS,eAAU,SAAV,mBAAgB,YAAW,UAAU,KAAK,SAAS;AAClE,QAAM,OAAOC,MAAAA;AAAAA,IACX,MAAO,SAASC,cAAAA,cAAc,EAAE,QAAQ,SAAS,UAAA,CAAW,IAAI;AAAA,IAChE,CAAC,QAAQ,SAAS,SAAS;AAAA,EAAA;AAE7B,QAAM,QAAO,uCAAW,SAAQ;AAChC,QAAM,UAAU,cAAc,YAAY,OAAO;AACjD,QAAM,kBAAiB,mCAAS,UAAS,QAAQ,UAAU;AAC3D,QAAM,UAAU,uCAAW;AAC3B,QAAM,oBAAoB,uCAAW;AAErCF,QAAAA,UAAU,MAAM;AACd,QAAI,CAAC,YAAY,CAAC,qBAAqB,YAAY,OAAW;AAC9D,sBAAkB,EAAE,UAAU,SAAS,eAAA,CAAgB;AAAA,EACzD,GAAG,CAAC,UAAU,mBAAmB,SAAS,cAAc,CAAC;AAEzDA,QAAAA,UAAU,MAAM;AACd,QAAI,CAAC,YAAY,CAAC,WAAW,SAAS,KAAM;AAC5C,UAAM,UACJ,mBAAmB,SAClB,mBAAmB,QACjB,WAAW,QAAQG,cAAAA,mBAAmB,EAAE,QAAQ,SAAS,eAAA,CAAgB,MAAM;AACpF,UAAM,YAAY,SAAS,QAAQ,CAAC,KAAK,QAAQ,KAAK,CAAC,WAAW,OAAO,OAAO,IAAI;AACpF,QAAI,WAAW,UAAW,SAAQ,EAAE,UAAU,QAAQ,MAAM;AAAA,EAC9D,GAAG,CAAC,UAAU,SAAS,MAAM,gBAAgB,QAAQ,IAAI,CAAC;AAE1D,QAAM,YAAY,GAAE,6BAAM,WAAU;AACpC,QAAM,UAAU,6BAAM;AACtB,QAAM,SAASC,MAAAA;AAAAA,IACb,CAAC,WAAmB;AAClB,UAAI,CAAC,YAAY,CAAC,WAAW,CAAC,UAAW;AACzC,UAAI,EAAC,mCAAS,KAAK,CAAC,WAAW,OAAO,OAAO,SAAS;AACtD,cAAQ,EAAE,UAAU,QAAQ;AAAA,IAC9B;AAAA,IACA,CAAC,UAAU,SAAS,WAAW,OAAO;AAAA,EAAA;AAGxC,MAAI,CAAC,QAAS,QAAO;AACrB,MAAI,UAAU,UAAW,QAAO;AAChC,MAAI,CAAC,QAAQ,KAAK,QAAQ,SAAS,EAAG,QAAO;AAC7C,QAAM,UACJ,SACC,UAAU,mBAAmB,OAC1BD,cAAAA,mBAAmB,EAAE,QAAQ,SAAS,eAAA,CAAgB,IACtD,KAAK;AACX,QAAM,iBACJ,UAAK,QAAQ,KAAK,CAAC,WAAW,OAAO,OAAO,OAAO,MAAnD,mBAAsD,WACrD,YAAY,SAAS,cAAa,sCAAQ,SAAS,aAAjB,mBAA2B,UAAS;AACzE,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,SAAS,KAAK;AAAA,IACd;AAAA,IACA;AAAA,IACA;AAAA,IACA,QAAQ,KAAK;AAAA,IACb;AAAA,EAAA;AAEJ;;"}

package/dist/catalog/use-athena-chat-mode.d.ts

@@ -0,0 +1,43 @@
+import { type AthenaChatModeOption } from './agent-mode-menu';
+/**
+ * The open chat's mode menu (`AthenaProvider` `modes`):
+ *
+ * - `off`: no menu: the host listed no modes, no collab agent is selected, the
+ *   agent cannot be read or was never published, or it offers fewer than two
+ *   of the listed modes on the provider's `channel`;
+ * - `loading`: the agent is being read;
+ * - `ready`: the modes to offer, the one the open session runs in (or a new
+ *   one starts in), and `select` to switch. `canSwitch` is false on a session
+ *   of a channel its author locks: there, people pick the mode before a
+ *   session's first message.
+ */
+export type AthenaChatMode = {
+    status: 'off';
+} | {
+    status: 'loading';
+} | {
+    status: 'ready';
+    options: readonly AthenaChatModeOption[];
+    /** The mode the open session runs in, or a new session starts in. */
+    current: string;
+    /** The current mode's name, also when it is none of `options`. */
+    currentLabel: string;
+    /** Whether a pick applies now. */
+    canSwitch: boolean;
+    /** The channel keeps each session in the mode it starts in. */
+    locked: boolean;
+    /** Run the open session's next sends in `modeId` (one of `options`). */
+    select: (modeId: string) => void;
+};
+/**
+ * The mode menu of the chat inside the nearest `AthenaProvider`, from the
+ * agent's published definition, the open session's own mode, and the
+ * provider's `modes`, `agent` and `channel`. `AthenaChat` renders it in its
+ * composer; use it to build a menu of your own.
+ *
+ * The open session's mode is read again when the chat regains focus and after
+ * every run, and its sends name it, so a session switched in another browser
+ * continues where it is. A pick holds until the session runs in it (or the
+ * menu stops offering it); only a pick not yet sent overrides the session.
+ */
+export declare function useAthenaChatMode(): AthenaChatMode;

package/dist/catalog/use-athena-chat-mode.js

@@ -0,0 +1,139 @@
+import { useAuiState } from "@assistant-ui/react";
+import { useQuery } from "@tanstack/react-query";
+import { useContext, useRef, useEffect, useMemo, useCallback } from "react";
+import { fetchGraphQL } from "../chat-ui/mentions/suggestions/data/api-client.js";
+import { AthenaAgentSelectionContext } from "../provider/agent-selection-context.js";
+import { AthenaChatModeSelectionContext } from "../provider/chat-mode-context.js";
+import { publishedAgentModeSource, sessionChannelFromMetadata, agentModeMenu, agentModeOfChannel } from "./agent-mode-menu.js";
+import { useCatalogAuth, useCatalogPrincipal, useCatalogQueryClient, AGENT_MODELS_REFRESH_MS, ATHENA_AGENT_MODELS_QUERY, isAgentReadDenial } from "./runtime-catalog.js";
+const OFF = { status: "off" };
+const LOADING = { status: "loading" };
+const NO_MODES = [];
+const ATHENA_SESSION_MODE_QUERY = `
+  query AthenaSessionMode($assetId: String!) {
+    getAssetById(assetId: $assetId, withMetadata: true) { athenaMetadata }
+  }
+`;
+function useAthenaChatMode() {
+  var _a, _b, _c;
+  const selection = useContext(AthenaChatModeSelectionContext);
+  const agentSelection = useContext(AthenaAgentSelectionContext);
+  const agentAssetId = (agentSelection == null ? void 0 : agentSelection.agentAssetId) ?? null;
+  const channel = (agentSelection == null ? void 0 : agentSelection.channel) ?? null;
+  const hostModes = (selection == null ? void 0 : selection.hostModes) ?? NO_MODES;
+  const enabled = selection !== null && hostModes.length > 0 && agentAssetId !== null;
+  const auth = useCatalogAuth();
+  const principal = useCatalogPrincipal();
+  const queryClient = useCatalogQueryClient();
+  const agentRead = useQuery(
+    {
+      queryKey: ["athena-sdk", "catalog", "agent-modes", auth.backendUrl, principal, agentAssetId],
+      queryFn: async () => {
+        try {
+          const data = await fetchGraphQL(
+            auth,
+            ATHENA_AGENT_MODELS_QUERY,
+            { assetId: agentAssetId }
+          );
+          if (data.getAssetById === null) return { readable: false };
+          return {
+            readable: true,
+            source: publishedAgentModeSource(data.getAssetById.athenaMetadata)
+          };
+        } catch (error) {
+          if (isAgentReadDenial(error)) return { readable: false };
+          throw error;
+        }
+      },
+      enabled,
+      staleTime: AGENT_MODELS_REFRESH_MS,
+      refetchInterval: (query) => query.state.data !== void 0 && !query.state.data.readable ? false : AGENT_MODELS_REFRESH_MS
+    },
+    queryClient
+  );
+  const threadEmpty = useAuiState((state) => {
+    var _a2;
+    return ((_a2 = state.optional.thread) == null ? void 0 : _a2.isEmpty) ?? true;
+  });
+  const threadRunning = useAuiState((state) => {
+    var _a2;
+    return ((_a2 = state.optional.thread) == null ? void 0 : _a2.isRunning) ?? false;
+  });
+  const threadId = (selection == null ? void 0 : selection.threadId) ?? null;
+  const sessionStarted = (selection == null ? void 0 : selection.isNewChat) === false || !threadEmpty;
+  const readSession = enabled && threadId !== null && sessionStarted;
+  const sessionRead = useQuery(
+    {
+      queryKey: ["athena-sdk", "chat-mode", "session", auth.backendUrl, principal, threadId],
+      queryFn: async () => {
+        const data = await fetchGraphQL(
+          auth,
+          ATHENA_SESSION_MODE_QUERY,
+          { assetId: `asset_${threadId}` }
+        );
+        const asset = data.getAssetById;
+        return asset ? { exists: true, channel: sessionChannelFromMetadata(asset.athenaMetadata) } : { exists: false };
+      },
+      enabled: readSession,
+      staleTime: 0,
+      refetchOnWindowFocus: true,
+      retry: false
+    },
+    queryClient
+  );
+  const refetchSession = sessionRead.refetch;
+  const wasRunning = useRef(threadRunning);
+  useEffect(() => {
+    const ended = wasRunning.current && !threadRunning;
+    wasRunning.current = threadRunning;
+    if (ended && readSession) void refetchSession();
+  }, [threadRunning, readSession, refetchSession]);
+  const source = ((_a = agentRead.data) == null ? void 0 : _a.readable) ? agentRead.data.source : null;
+  const menu = useMemo(
+    () => source ? agentModeMenu({ source, channel, hostModes }) : null,
+    [source, channel, hostModes]
+  );
+  const pick = (selection == null ? void 0 : selection.pick) ?? null;
+  const session = readSession ? sessionRead.data : void 0;
+  const sessionChannel = (session == null ? void 0 : session.exists) ? session.channel : null;
+  const setPick = selection == null ? void 0 : selection.setPick;
+  const setSessionChannel = selection == null ? void 0 : selection.setSessionChannel;
+  useEffect(() => {
+    if (!threadId || !setSessionChannel || session === void 0) return;
+    setSessionChannel({ threadId, channel: sessionChannel });
+  }, [threadId, setSessionChannel, session, sessionChannel]);
+  useEffect(() => {
+    if (!threadId || !setPick || pick === null) return;
+    const applied = sessionChannel !== null && (sessionChannel === pick || source !== null && agentModeOfChannel({ source, channel: sessionChannel }) === pick);
+    const withdrawn = menu !== null && !menu.options.some((option) => option.id === pick);
+    if (applied || withdrawn) setPick({ threadId, modeId: null });
+  }, [threadId, setPick, pick, sessionChannel, source, menu]);
+  const canSwitch = !((menu == null ? void 0 : menu.locked) && sessionStarted);
+  const options = menu == null ? void 0 : menu.options;
+  const select = useCallback(
+    (modeId) => {
+      if (!threadId || !setPick || !canSwitch) return;
+      if (!(options == null ? void 0 : options.some((option) => option.id === modeId))) return;
+      setPick({ threadId, modeId });
+    },
+    [threadId, setPick, canSwitch, options]
+  );
+  if (!enabled) return OFF;
+  if (agentRead.isPending) return LOADING;
+  if (!menu || menu.options.length < 2) return OFF;
+  const current = pick ?? (source && sessionChannel !== null ? agentModeOfChannel({ source, channel: sessionChannel }) : menu.firstSendModeId);
+  const currentLabel = ((_b = menu.options.find((option) => option.id === current)) == null ? void 0 : _b.label) ?? (current === "base" ? "Default" : ((_c = source == null ? void 0 : source.channels[current]) == null ? void 0 : _c.label) ?? current);
+  return {
+    status: "ready",
+    options: menu.options,
+    current,
+    currentLabel,
+    canSwitch,
+    locked: menu.locked,
+    select
+  };
+}
+export {
+  useAthenaChatMode
+};
+//# sourceMappingURL=use-athena-chat-mode.js.map

package/dist/catalog/use-athena-chat-mode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-athena-chat-mode.js","sources":["../../src/catalog/use-athena-chat-mode.ts"],"sourcesContent":["import { useAuiState } from '@assistant-ui/react';\nimport { useQuery } from '@tanstack/react-query';\nimport { useCallback, useContext, useEffect, useMemo, useRef } from 'react';\nimport { fetchGraphQL } from '../chat-ui/mentions/suggestions/data/api-client';\nimport { AthenaAgentSelectionContext } from '../provider/agent-selection-context';\nimport { AthenaChatModeSelectionContext } from '../provider/chat-mode-context';\nimport {\n  type AgentModeSource,\n  type AthenaChatModeOption,\n  agentModeMenu,\n  agentModeOfChannel,\n  publishedAgentModeSource,\n  sessionChannelFromMetadata,\n} from './agent-mode-menu';\nimport {\n  AGENT_MODELS_REFRESH_MS,\n  ATHENA_AGENT_MODELS_QUERY,\n  isAgentReadDenial,\n  useCatalogAuth,\n  useCatalogPrincipal,\n  useCatalogQueryClient,\n} from './runtime-catalog';\n\n/**\n * The open chat's mode menu (`AthenaProvider` `modes`):\n *\n * - `off`: no menu: the host listed no modes, no collab agent is selected, the\n *   agent cannot be read or was never published, or it offers fewer than two\n *   of the listed modes on the provider's `channel`;\n * - `loading`: the agent is being read;\n * - `ready`: the modes to offer, the one the open session runs in (or a new\n *   one starts in), and `select` to switch. `canSwitch` is false on a session\n *   of a channel its author locks: there, people pick the mode before a\n *   session's first message.\n */\nexport type AthenaChatMode =\n  | { status: 'off' }\n  | { status: 'loading' }\n  | {\n      status: 'ready';\n      options: readonly AthenaChatModeOption[];\n      /** The mode the open session runs in, or a new session starts in. */\n      current: string;\n      /** The current mode's name, also when it is none of `options`. */\n      currentLabel: string;\n      /** Whether a pick applies now. */\n      canSwitch: boolean;\n      /** The channel keeps each session in the mode it starts in. */\n      locked: boolean;\n      /** Run the open session's next sends in `modeId` (one of `options`). */\n      select: (modeId: string) => void;\n    };\n\ntype AgentModeRead = { readable: true; source: AgentModeSource | null } | { readable: false };\n\n/** The open thread's session asset: absent until its first send mints it. */\ntype SessionRead = { exists: false } | { exists: true; channel: string | null };\n\nconst OFF: AthenaChatMode = { status: 'off' };\nconst LOADING: AthenaChatMode = { status: 'loading' };\nconst NO_MODES: readonly string[] = [];\n\nconst ATHENA_SESSION_MODE_QUERY = `\n  query AthenaSessionMode($assetId: String!) {\n    getAssetById(assetId: $assetId, withMetadata: true) { athenaMetadata }\n  }\n`;\n\n/**\n * The mode menu of the chat inside the nearest `AthenaProvider`, from the\n * agent's published definition, the open session's own mode, and the\n * provider's `modes`, `agent` and `channel`. `AthenaChat` renders it in its\n * composer; use it to build a menu of your own.\n *\n * The open session's mode is read again when the chat regains focus and after\n * every run, and its sends name it, so a session switched in another browser\n * continues where it is. A pick holds until the session runs in it (or the\n * menu stops offering it); only a pick not yet sent overrides the session.\n */\nexport function useAthenaChatMode(): AthenaChatMode {\n  const selection = useContext(AthenaChatModeSelectionContext);\n  const agentSelection = useContext(AthenaAgentSelectionContext);\n  const agentAssetId = agentSelection?.agentAssetId ?? null;\n  const channel = agentSelection?.channel ?? null;\n  const hostModes = selection?.hostModes ?? NO_MODES;\n  const enabled = selection !== null && hostModes.length > 0 && agentAssetId !== null;\n  const auth = useCatalogAuth();\n  const principal = useCatalogPrincipal();\n  const queryClient = useCatalogQueryClient();\n\n  const agentRead = useQuery(\n    {\n      queryKey: ['athena-sdk', 'catalog', 'agent-modes', auth.backendUrl, principal, agentAssetId],\n      queryFn: async (): Promise<AgentModeRead> => {\n        try {\n          const data = await fetchGraphQL<{ getAssetById: { athenaMetadata: unknown } | null }>(\n            auth,\n            ATHENA_AGENT_MODELS_QUERY,\n            { assetId: agentAssetId },\n          );\n          if (data.getAssetById === null) return { readable: false };\n          return {\n            readable: true,\n            source: publishedAgentModeSource(data.getAssetById.athenaMetadata),\n          };\n        } catch (error) {\n          if (isAgentReadDenial(error)) return { readable: false };\n          throw error;\n        }\n      },\n      enabled,\n      staleTime: AGENT_MODELS_REFRESH_MS,\n      refetchInterval: (query) =>\n        query.state.data !== undefined && !query.state.data.readable\n          ? false\n          : AGENT_MODELS_REFRESH_MS,\n    },\n    queryClient,\n  );\n\n  // The open thread as this chat holds it: its first send ends a new chat's\n  // pick-before-start window even where the provider cannot tell (a\n  // standalone chat on a thread id it minted), and the end of each run, which\n  // may have switched the session (here or in another browser), reads it again.\n  const threadEmpty = useAuiState((state) => state.optional.thread?.isEmpty ?? true);\n  const threadRunning = useAuiState((state) => state.optional.thread?.isRunning ?? false);\n\n  const threadId = selection?.threadId ?? null;\n  const sessionStarted = selection?.isNewChat === false || !threadEmpty;\n  const readSession = enabled && threadId !== null && sessionStarted;\n  const sessionRead = useQuery(\n    {\n      queryKey: ['athena-sdk', 'chat-mode', 'session', auth.backendUrl, principal, threadId],\n      queryFn: async (): Promise<SessionRead> => {\n        const data = await fetchGraphQL<{ getAssetById: { athenaMetadata: unknown } | null }>(\n          auth,\n          ATHENA_SESSION_MODE_QUERY,\n          { assetId: `asset_${threadId}` },\n        );\n        const asset = data.getAssetById;\n        return asset\n          ? { exists: true, channel: sessionChannelFromMetadata(asset.athenaMetadata) }\n          : { exists: false };\n      },\n      enabled: readSession,\n      staleTime: 0,\n      refetchOnWindowFocus: true,\n      retry: false,\n    },\n    queryClient,\n  );\n  const refetchSession = sessionRead.refetch;\n  const wasRunning = useRef(threadRunning);\n  useEffect(() => {\n    const ended = wasRunning.current && !threadRunning;\n    wasRunning.current = threadRunning;\n    if (ended && readSession) void refetchSession();\n  }, [threadRunning, readSession, refetchSession]);\n\n  const source = agentRead.data?.readable ? agentRead.data.source : null;\n  const menu = useMemo(\n    () => (source ? agentModeMenu({ source, channel, hostModes }) : null),\n    [source, channel, hostModes],\n  );\n  const pick = selection?.pick ?? null;\n  const session = readSession ? sessionRead.data : undefined;\n  const sessionChannel = session?.exists ? session.channel : null;\n  const setPick = selection?.setPick;\n  const setSessionChannel = selection?.setSessionChannel;\n\n  useEffect(() => {\n    if (!threadId || !setSessionChannel || session === undefined) return;\n    setSessionChannel({ threadId, channel: sessionChannel });\n  }, [threadId, setSessionChannel, session, sessionChannel]);\n\n  useEffect(() => {\n    if (!threadId || !setPick || pick === null) return;\n    const applied =\n      sessionChannel !== null &&\n      (sessionChannel === pick ||\n        (source !== null && agentModeOfChannel({ source, channel: sessionChannel }) === pick));\n    const withdrawn = menu !== null && !menu.options.some((option) => option.id === pick);\n    if (applied || withdrawn) setPick({ threadId, modeId: null });\n  }, [threadId, setPick, pick, sessionChannel, source, menu]);\n\n  const canSwitch = !(menu?.locked && sessionStarted);\n  const options = menu?.options;\n  const select = useCallback(\n    (modeId: string) => {\n      if (!threadId || !setPick || !canSwitch) return;\n      if (!options?.some((option) => option.id === modeId)) return;\n      setPick({ threadId, modeId });\n    },\n    [threadId, setPick, canSwitch, options],\n  );\n\n  if (!enabled) return OFF;\n  if (agentRead.isPending) return LOADING;\n  if (!menu || menu.options.length < 2) return OFF;\n  const current =\n    pick ??\n    (source && sessionChannel !== null\n      ? agentModeOfChannel({ source, channel: sessionChannel })\n      : menu.firstSendModeId);\n  const currentLabel =\n    menu.options.find((option) => option.id === current)?.label ??\n    (current === 'base' ? 'Default' : (source?.channels[current]?.label ?? current));\n  return {\n    status: 'ready',\n    options: menu.options,\n    current,\n    currentLabel,\n    canSwitch,\n    locked: menu.locked,\n    select,\n  };\n}\n"],"names":["_a"],"mappings":";;;;;;;;AA0DA,MAAM,MAAsB,EAAE,QAAQ,MAAA;AACtC,MAAM,UAA0B,EAAE,QAAQ,UAAA;AAC1C,MAAM,WAA8B,CAAA;AAEpC,MAAM,4BAA4B;AAAA;AAAA;AAAA;AAAA;AAiB3B,SAAS,oBAAoC;;AAClD,QAAM,YAAY,WAAW,8BAA8B;AAC3D,QAAM,iBAAiB,WAAW,2BAA2B;AAC7D,QAAM,gBAAe,iDAAgB,iBAAgB;AACrD,QAAM,WAAU,iDAAgB,YAAW;AAC3C,QAAM,aAAY,uCAAW,cAAa;AAC1C,QAAM,UAAU,cAAc,QAAQ,UAAU,SAAS,KAAK,iBAAiB;AAC/E,QAAM,OAAO,eAAA;AACb,QAAM,YAAY,oBAAA;AAClB,QAAM,cAAc,sBAAA;AAEpB,QAAM,YAAY;AAAA,IAChB;AAAA,MACE,UAAU,CAAC,cAAc,WAAW,eAAe,KAAK,YAAY,WAAW,YAAY;AAAA,MAC3F,SAAS,YAAoC;AAC3C,YAAI;AACF,gBAAM,OAAO,MAAM;AAAA,YACjB;AAAA,YACA;AAAA,YACA,EAAE,SAAS,aAAA;AAAA,UAAa;AAE1B,cAAI,KAAK,iBAAiB,KAAM,QAAO,EAAE,UAAU,MAAA;AACnD,iBAAO;AAAA,YACL,UAAU;AAAA,YACV,QAAQ,yBAAyB,KAAK,aAAa,cAAc;AAAA,UAAA;AAAA,QAErE,SAAS,OAAO;AACd,cAAI,kBAAkB,KAAK,EAAG,QAAO,EAAE,UAAU,MAAA;AACjD,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA;AAAA,MACA,WAAW;AAAA,MACX,iBAAiB,CAAC,UAChB,MAAM,MAAM,SAAS,UAAa,CAAC,MAAM,MAAM,KAAK,WAChD,QACA;AAAA,IAAA;AAAA,IAER;AAAA,EAAA;AAOF,QAAM,cAAc,YAAY,CAAC,UAAA;;AAAU,aAAAA,MAAA,MAAM,SAAS,WAAf,gBAAAA,IAAuB,YAAW;AAAA,GAAI;AACjF,QAAM,gBAAgB,YAAY,CAAC,UAAA;;AAAU,aAAAA,MAAA,MAAM,SAAS,WAAf,gBAAAA,IAAuB,cAAa;AAAA,GAAK;AAEtF,QAAM,YAAW,uCAAW,aAAY;AACxC,QAAM,kBAAiB,uCAAW,eAAc,SAAS,CAAC;AAC1D,QAAM,cAAc,WAAW,aAAa,QAAQ;AACpD,QAAM,cAAc;AAAA,IAClB;AAAA,MACE,UAAU,CAAC,cAAc,aAAa,WAAW,KAAK,YAAY,WAAW,QAAQ;AAAA,MACrF,SAAS,YAAkC;AACzC,cAAM,OAAO,MAAM;AAAA,UACjB;AAAA,UACA;AAAA,UACA,EAAE,SAAS,SAAS,QAAQ,GAAA;AAAA,QAAG;AAEjC,cAAM,QAAQ,KAAK;AACnB,eAAO,QACH,EAAE,QAAQ,MAAM,SAAS,2BAA2B,MAAM,cAAc,EAAA,IACxE,EAAE,QAAQ,MAAA;AAAA,MAChB;AAAA,MACA,SAAS;AAAA,MACT,WAAW;AAAA,MACX,sBAAsB;AAAA,MACtB,OAAO;AAAA,IAAA;AAAA,IAET;AAAA,EAAA;AAEF,QAAM,iBAAiB,YAAY;AACnC,QAAM,aAAa,OAAO,aAAa;AACvC,YAAU,MAAM;AACd,UAAM,QAAQ,WAAW,WAAW,CAAC;AACrC,eAAW,UAAU;AACrB,QAAI,SAAS,YAAa,MAAK,eAAA;AAAA,EACjC,GAAG,CAAC,eAAe,aAAa,cAAc,CAAC;AAE/C,QAAM,WAAS,eAAU,SAAV,mBAAgB,YAAW,UAAU,KAAK,SAAS;AAClE,QAAM,OAAO;AAAA,IACX,MAAO,SAAS,cAAc,EAAE,QAAQ,SAAS,UAAA,CAAW,IAAI;AAAA,IAChE,CAAC,QAAQ,SAAS,SAAS;AAAA,EAAA;AAE7B,QAAM,QAAO,uCAAW,SAAQ;AAChC,QAAM,UAAU,cAAc,YAAY,OAAO;AACjD,QAAM,kBAAiB,mCAAS,UAAS,QAAQ,UAAU;AAC3D,QAAM,UAAU,uCAAW;AAC3B,QAAM,oBAAoB,uCAAW;AAErC,YAAU,MAAM;AACd,QAAI,CAAC,YAAY,CAAC,qBAAqB,YAAY,OAAW;AAC9D,sBAAkB,EAAE,UAAU,SAAS,eAAA,CAAgB;AAAA,EACzD,GAAG,CAAC,UAAU,mBAAmB,SAAS,cAAc,CAAC;AAEzD,YAAU,MAAM;AACd,QAAI,CAAC,YAAY,CAAC,WAAW,SAAS,KAAM;AAC5C,UAAM,UACJ,mBAAmB,SAClB,mBAAmB,QACjB,WAAW,QAAQ,mBAAmB,EAAE,QAAQ,SAAS,eAAA,CAAgB,MAAM;AACpF,UAAM,YAAY,SAAS,QAAQ,CAAC,KAAK,QAAQ,KAAK,CAAC,WAAW,OAAO,OAAO,IAAI;AACpF,QAAI,WAAW,UAAW,SAAQ,EAAE,UAAU,QAAQ,MAAM;AAAA,EAC9D,GAAG,CAAC,UAAU,SAAS,MAAM,gBAAgB,QAAQ,IAAI,CAAC;AAE1D,QAAM,YAAY,GAAE,6BAAM,WAAU;AACpC,QAAM,UAAU,6BAAM;AACtB,QAAM,SAAS;AAAA,IACb,CAAC,WAAmB;AAClB,UAAI,CAAC,YAAY,CAAC,WAAW,CAAC,UAAW;AACzC,UAAI,EAAC,mCAAS,KAAK,CAAC,WAAW,OAAO,OAAO,SAAS;AACtD,cAAQ,EAAE,UAAU,QAAQ;AAAA,IAC9B;AAAA,IACA,CAAC,UAAU,SAAS,WAAW,OAAO;AAAA,EAAA;AAGxC,MAAI,CAAC,QAAS,QAAO;AACrB,MAAI,UAAU,UAAW,QAAO;AAChC,MAAI,CAAC,QAAQ,KAAK,QAAQ,SAAS,EAAG,QAAO;AAC7C,QAAM,UACJ,SACC,UAAU,mBAAmB,OAC1B,mBAAmB,EAAE,QAAQ,SAAS,eAAA,CAAgB,IACtD,KAAK;AACX,QAAM,iBACJ,UAAK,QAAQ,KAAK,CAAC,WAAW,OAAO,OAAO,OAAO,MAAnD,mBAAsD,WACrD,YAAY,SAAS,cAAa,sCAAQ,SAAS,aAAjB,mBAA2B,UAAS;AACzE,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,SAAS,KAAK;AAAA,IACd;AAAA,IACA;AAAA,IACA;AAAA,IACA,QAAQ,KAAK;AAAA,IACb;AAAA,EAAA;AAEJ;"}

package/dist/chat/AthenaChat.cjs

@@ -6,6 +6,7 @@ const AthenaChatErrorBoundary = require("./AthenaChatErrorBoundary.cjs");
 const StatewireApprovalCard = require("./StatewireApprovalCard.cjs");
 const StatewireQueuedMessages = require("./StatewireQueuedMessages.cjs");
 const statewireBanners = require("./statewire-banners.cjs");
+const chatPickers = require("../catalog/chat-pickers.cjs");
 const StatewireRunRefusalNotice = require("./StatewireRunRefusalNotice.cjs");
 const WaitingOnYouBanner = require("./WaitingOnYouBanner.cjs");
 const wakeMarker = require("../chat-ui/messages/wake-marker.cjs");
@@ -544,6 +545,7 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
                 }
               }
             ),
+            /* @__PURE__ */ jsxRuntime.jsx(chatPickers.AthenaChatModeMenu, {}),
             slots == null ? void 0 : slots.toolbarStart
           ] }) }),
           toolbarEnd: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex items-center gap-1", children: recording ? /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [