@athenaintel/react 0.28.1 → 0.30.0

package/dist/catalog/runtime-catalog.d.ts

@@ -0,0 +1,100 @@
+/**
+ * The workspace's runtime catalog for a chat's model and agent pickers: the models the signed-in
+ * person may run (the workspace's allow or deny list, admin-only models for admins), and the
+ * published collab agents they can see. The rules are the Chrome extension's, which every host now
+ * reads from here.
+ */
+export type AthenaModelStatus = 'ACTIVE' | 'LEGACY' | 'RECOMMENDED';
+export interface AthenaModelOption {
+    id: string;
+    label: string;
+    description: string | null;
+    /** Upper-case provider key, e.g. `ANTHROPIC`, `OPENAI`. */
+    provider: string;
+    status: AthenaModelStatus;
+    sortOrder: number;
+    familyId: string;
+    familyName: string;
+    isFamilyDefault: boolean;
+    isFeatured: boolean;
+    variantOrder: number;
+}
+export interface AthenaAgentOption {
+    /** The collab agent's asset id; pass `collab_agent:<id>` as the provider's `agent`. */
+    id: string;
+    name: string;
+}
+/** The provider labels a model picker groups by. */
+export declare const MODEL_PROVIDER_LABELS: Readonly<Record<string, string>>;
+export declare const ATHENA_MODEL_CATALOG_QUERY = "\n  query AthenaModelCatalog {\n    getCurrentUserWithWorkspace { user { isAdmin } }\n    settings { workspace { approvedLargeLlms blacklistedLlms } }\n    availableModels {\n      id name provider status description adminOnly sortOrder\n      familyId familyName isFamilyDefault isFeatured variantOrder\n    }\n  }\n";
+export declare const ATHENA_AGENT_CATALOG_QUERY = "\n  query AthenaAgentCatalog {\n    getAllAssets(filterByAthenaOriginalType: [\"collab_agent\"]) { id title }\n  }\n";
+export interface AthenaModelCatalogResponse {
+    getCurrentUserWithWorkspace: {
+        user: {
+            isAdmin: boolean;
+        };
+    };
+    settings: {
+        workspace: {
+            approvedLargeLlms: string[];
+            blacklistedLlms: string[];
+        };
+    };
+    availableModels: Array<{
+        id: string;
+        name: string;
+        provider: string;
+        status: AthenaModelStatus;
+        description: string | null;
+        adminOnly: boolean;
+        sortOrder: number;
+        familyId: string;
+        familyName: string;
+        isFamilyDefault: boolean;
+        isFeatured: boolean;
+        variantOrder: number;
+    }>;
+}
+/**
+ * The models the person may pick, in catalog order. A workspace with an allow list offers only
+ * those; otherwise everything except its deny list. Admin-only models are offered to admins only.
+ */
+export declare function normalizeModelCatalog({ response, }: {
+    response: AthenaModelCatalogResponse;
+}): AthenaModelOption[];
+/** The published collab agents, named and sorted; assets with a blank id are dropped. */
+export declare function normalizeAgentCatalog({ assets, }: {
+    assets: ReadonlyArray<{
+        id: string;
+        title: string | null;
+    }>;
+}): AthenaAgentOption[];
+/**
+ * The model a picker shows as chosen: the stored choice while the catalog offers it, else the
+ * default (unless retired), else a recommended family default, else any live model. While the
+ * catalog is empty (loading, or unreachable) the stored choice or the default stands.
+ */
+export declare function resolveCatalogModelId({ stored, defaultModelId, models, }: {
+    stored: string | null;
+    defaultModelId: string;
+    models: readonly AthenaModelOption[];
+}): string;
+/** Models grouped under their provider's label, in catalog order. */
+export declare function groupModelsByProvider({ models, }: {
+    models: readonly AthenaModelOption[];
+}): Array<{
+    provider: string;
+    models: AthenaModelOption[];
+}>;
+/**
+ * A short fingerprint (two FNV-1a rounds, 64 bits) of a credential: it tells accounts apart in an
+ * in-memory cache key without the credential itself, or anything it could be recovered from,
+ * entering the key.
+ */
+export declare function credentialFingerprint({ credential }: {
+    credential: string;
+}): string;
+/** The models the signed-in person may pick (`normalizeModelCatalog`). */
+export declare function useAthenaModelCatalog(): import("@tanstack/react-query").UseQueryResult<AthenaModelOption[], Error>;
+/** The published collab agents the signed-in person can see (`normalizeAgentCatalog`). */
+export declare function useAthenaAgentCatalog(): import("@tanstack/react-query").UseQueryResult<AthenaAgentOption[], Error>;

package/dist/catalog/runtime-catalog.js

@@ -0,0 +1,181 @@
+import { useQuery, QueryClientContext, QueryClient } from "@tanstack/react-query";
+import { useMemo, useContext, useState } from "react";
+import { AthenaAuthContext } from "../auth/AthenaAuthContext.js";
+import { fetchGraphQL } from "../chat-ui/mentions/suggestions/data/api-client.js";
+import { useAthenaConfig } from "../provider/AthenaContext.js";
+const MODEL_PROVIDER_LABELS = {
+  ANTHROPIC: "Anthropic",
+  OPENAI: "OpenAI",
+  GOOGLE: "Google",
+  XAI: "xAI",
+  META: "Meta",
+  MISTRAL: "Mistral",
+  DEEPSEEK: "DeepSeek",
+  QWEN: "Qwen",
+  MOONSHOT: "Moonshot",
+  MINIMAX: "MiniMax",
+  ZAI: "Z.ai"
+};
+const ATHENA_MODEL_CATALOG_QUERY = `
+  query AthenaModelCatalog {
+    getCurrentUserWithWorkspace { user { isAdmin } }
+    settings { workspace { approvedLargeLlms blacklistedLlms } }
+    availableModels {
+      id name provider status description adminOnly sortOrder
+      familyId familyName isFamilyDefault isFeatured variantOrder
+    }
+  }
+`;
+const ATHENA_AGENT_CATALOG_QUERY = `
+  query AthenaAgentCatalog {
+    getAllAssets(filterByAthenaOriginalType: ["collab_agent"]) { id title }
+  }
+`;
+function normalizeModelCatalog({
+  response
+}) {
+  const isAdmin = response.getCurrentUserWithWorkspace.user.isAdmin;
+  const approved = new Set(response.settings.workspace.approvedLargeLlms);
+  const denied = new Set(response.settings.workspace.blacklistedLlms);
+  const hasAllowlist = approved.size > 0;
+  return response.availableModels.filter(
+    (model) => (!model.adminOnly || isAdmin) && (hasAllowlist ? approved.has(model.id) : !denied.has(model.id))
+  ).map(
+    (model) => ({
+      id: model.id,
+      label: model.name,
+      description: model.description,
+      provider: model.provider,
+      status: model.status,
+      sortOrder: model.sortOrder,
+      familyId: model.familyId,
+      familyName: model.familyName,
+      isFamilyDefault: model.isFamilyDefault,
+      isFeatured: model.isFeatured,
+      variantOrder: model.variantOrder
+    })
+  ).sort(
+    (left, right) => left.sortOrder - right.sortOrder || left.variantOrder - right.variantOrder || left.label.localeCompare(right.label)
+  );
+}
+function normalizeAgentCatalog({
+  assets
+}) {
+  return assets.filter((asset) => asset.id.trim().length > 0).map((asset) => {
+    var _a;
+    return { id: asset.id, name: ((_a = asset.title) == null ? void 0 : _a.trim()) || "Untitled agent" };
+  }).sort((left, right) => left.name.localeCompare(right.name));
+}
+function resolveCatalogModelId({
+  stored,
+  defaultModelId,
+  models
+}) {
+  var _a;
+  if (stored && models.some((model) => model.id === stored)) return stored;
+  return ((_a = models.find((model) => model.id === defaultModelId && model.status !== "LEGACY") ?? models.find((model) => model.status === "RECOMMENDED" && model.isFamilyDefault) ?? models.find((model) => model.status !== "LEGACY" && model.isFamilyDefault) ?? models.find((model) => model.status !== "LEGACY") ?? models[0]) == null ? void 0 : _a.id) ?? (stored || defaultModelId);
+}
+function groupModelsByProvider({
+  models
+}) {
+  const groups = /* @__PURE__ */ new Map();
+  for (const model of models) {
+    const label = MODEL_PROVIDER_LABELS[model.provider] ?? model.provider;
+    groups.set(label, [...groups.get(label) ?? [], model]);
+  }
+  return Array.from(groups, ([provider, grouped]) => ({ provider, models: grouped }));
+}
+function useCatalogAuth() {
+  const config = useAthenaConfig();
+  return useMemo(
+    () => ({
+      backendUrl: config.backendUrl,
+      appUrl: config.appUrl,
+      apiKey: config.apiKey,
+      credentials: "include",
+      get accessToken() {
+        return config.token ?? void 0;
+      },
+      getAccessToken: async () => {
+        var _a;
+        return await ((_a = config.getToken) == null ? void 0 : _a.call(config)) ?? void 0;
+      },
+      renewAccessToken: async () => {
+        var _a;
+        return await ((_a = config.getToken) == null ? void 0 : _a.call(config, { force: true })) ?? void 0;
+      }
+    }),
+    [config]
+  );
+}
+function credentialFingerprint({ credential }) {
+  const round = (seed) => {
+    let hash = seed;
+    for (let index = 0; index < credential.length; index += 1) {
+      hash ^= credential.charCodeAt(index);
+      hash = Math.imul(hash, 16777619) >>> 0;
+    }
+    return hash.toString(16).padStart(8, "0");
+  };
+  return `${round(2166136261)}${round(16777619)}`;
+}
+function useCatalogPrincipal() {
+  var _a, _b;
+  const config = useAthenaConfig();
+  const userId = (_b = (_a = useContext(AthenaAuthContext)) == null ? void 0 : _a.user) == null ? void 0 : _b.userId;
+  if (userId) return `user:${userId}`;
+  if (config.apiKey) return `key:${credentialFingerprint({ credential: config.apiKey })}`;
+  if (config.token) return `token:${credentialFingerprint({ credential: config.token })}`;
+  return "session";
+}
+function useCatalogQueryClient() {
+  const host = useContext(QueryClientContext);
+  const [own] = useState(() => new QueryClient());
+  return host ?? own;
+}
+function useAthenaModelCatalog() {
+  const auth = useCatalogAuth();
+  const principal = useCatalogPrincipal();
+  const queryClient = useCatalogQueryClient();
+  return useQuery(
+    {
+      queryKey: ["athena-sdk", "catalog", "models", auth.backendUrl, principal],
+      queryFn: async () => normalizeModelCatalog({
+        response: await fetchGraphQL(auth, ATHENA_MODEL_CATALOG_QUERY)
+      }),
+      staleTime: 5 * 60 * 1e3
+    },
+    queryClient
+  );
+}
+function useAthenaAgentCatalog() {
+  const auth = useCatalogAuth();
+  const principal = useCatalogPrincipal();
+  const queryClient = useCatalogQueryClient();
+  return useQuery(
+    {
+      queryKey: ["athena-sdk", "catalog", "agents", auth.backendUrl, principal],
+      queryFn: async () => normalizeAgentCatalog({
+        assets: (await fetchGraphQL(
+          auth,
+          ATHENA_AGENT_CATALOG_QUERY
+        )).getAllAssets ?? []
+      }),
+      staleTime: 60 * 1e3
+    },
+    queryClient
+  );
+}
+export {
+  ATHENA_AGENT_CATALOG_QUERY,
+  ATHENA_MODEL_CATALOG_QUERY,
+  MODEL_PROVIDER_LABELS,
+  credentialFingerprint,
+  groupModelsByProvider,
+  normalizeAgentCatalog,
+  normalizeModelCatalog,
+  resolveCatalogModelId,
+  useAthenaAgentCatalog,
+  useAthenaModelCatalog
+};
+//# sourceMappingURL=runtime-catalog.js.map

package/dist/catalog/runtime-catalog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-catalog.js","sources":["../../src/catalog/runtime-catalog.ts"],"sourcesContent":["import { QueryClient, QueryClientContext, useQuery } from '@tanstack/react-query';\nimport { useContext, useMemo, useState } from 'react';\nimport { AthenaAuthContext } from '../auth/AthenaAuthContext';\nimport { fetchGraphQL, type MentionAuthContext } from '../chat-ui/mentions/suggestions/data/api-client';\nimport { useAthenaConfig } from '../provider/AthenaContext';\n\n/**\n * The workspace's runtime catalog for a chat's model and agent pickers: the models the signed-in\n * person may run (the workspace's allow or deny list, admin-only models for admins), and the\n * published collab agents they can see. The rules are the Chrome extension's, which every host now\n * reads from here.\n */\n\nexport type AthenaModelStatus = 'ACTIVE' | 'LEGACY' | 'RECOMMENDED';\n\nexport interface AthenaModelOption {\n  id: string;\n  label: string;\n  description: string | null;\n  /** Upper-case provider key, e.g. `ANTHROPIC`, `OPENAI`. */\n  provider: string;\n  status: AthenaModelStatus;\n  sortOrder: number;\n  familyId: string;\n  familyName: string;\n  isFamilyDefault: boolean;\n  isFeatured: boolean;\n  variantOrder: number;\n}\n\nexport interface AthenaAgentOption {\n  /** The collab agent's asset id; pass `collab_agent:<id>` as the provider's `agent`. */\n  id: string;\n  name: string;\n}\n\n/** The provider labels a model picker groups by. */\nexport const MODEL_PROVIDER_LABELS: Readonly<Record<string, string>> = {\n  ANTHROPIC: 'Anthropic',\n  OPENAI: 'OpenAI',\n  GOOGLE: 'Google',\n  XAI: 'xAI',\n  META: 'Meta',\n  MISTRAL: 'Mistral',\n  DEEPSEEK: 'DeepSeek',\n  QWEN: 'Qwen',\n  MOONSHOT: 'Moonshot',\n  MINIMAX: 'MiniMax',\n  ZAI: 'Z.ai',\n};\n\nexport const ATHENA_MODEL_CATALOG_QUERY = `\n  query AthenaModelCatalog {\n    getCurrentUserWithWorkspace { user { isAdmin } }\n    settings { workspace { approvedLargeLlms blacklistedLlms } }\n    availableModels {\n      id name provider status description adminOnly sortOrder\n      familyId familyName isFamilyDefault isFeatured variantOrder\n    }\n  }\n`;\n\nexport const ATHENA_AGENT_CATALOG_QUERY = `\n  query AthenaAgentCatalog {\n    getAllAssets(filterByAthenaOriginalType: [\"collab_agent\"]) { id title }\n  }\n`;\n\nexport interface AthenaModelCatalogResponse {\n  getCurrentUserWithWorkspace: { user: { isAdmin: boolean } };\n  settings: { workspace: { approvedLargeLlms: string[]; blacklistedLlms: string[] } };\n  availableModels: Array<{\n    id: string;\n    name: string;\n    provider: string;\n    status: AthenaModelStatus;\n    description: string | null;\n    adminOnly: boolean;\n    sortOrder: number;\n    familyId: string;\n    familyName: string;\n    isFamilyDefault: boolean;\n    isFeatured: boolean;\n    variantOrder: number;\n  }>;\n}\n\n/**\n * The models the person may pick, in catalog order. A workspace with an allow list offers only\n * those; otherwise everything except its deny list. Admin-only models are offered to admins only.\n */\nexport function normalizeModelCatalog({\n  response,\n}: {\n  response: AthenaModelCatalogResponse;\n}): AthenaModelOption[] {\n  const isAdmin = response.getCurrentUserWithWorkspace.user.isAdmin;\n  const approved = new Set(response.settings.workspace.approvedLargeLlms);\n  const denied = new Set(response.settings.workspace.blacklistedLlms);\n  const hasAllowlist = approved.size > 0;\n  return response.availableModels\n    .filter(\n      (model) =>\n        (!model.adminOnly || isAdmin) && (hasAllowlist ? approved.has(model.id) : !denied.has(model.id)),\n    )\n    .map(\n      (model): AthenaModelOption => ({\n        id: model.id,\n        label: model.name,\n        description: model.description,\n        provider: model.provider,\n        status: model.status,\n        sortOrder: model.sortOrder,\n        familyId: model.familyId,\n        familyName: model.familyName,\n        isFamilyDefault: model.isFamilyDefault,\n        isFeatured: model.isFeatured,\n        variantOrder: model.variantOrder,\n      }),\n    )\n    .sort(\n      (left, right) =>\n        left.sortOrder - right.sortOrder ||\n        left.variantOrder - right.variantOrder ||\n        left.label.localeCompare(right.label),\n    );\n}\n\n/** The published collab agents, named and sorted; assets with a blank id are dropped. */\nexport function normalizeAgentCatalog({\n  assets,\n}: {\n  assets: ReadonlyArray<{ id: string; title: string | null }>;\n}): AthenaAgentOption[] {\n  return assets\n    .filter((asset) => asset.id.trim().length > 0)\n    .map((asset) => ({ id: asset.id, name: asset.title?.trim() || 'Untitled agent' }))\n    .sort((left, right) => left.name.localeCompare(right.name));\n}\n\n/**\n * The model a picker shows as chosen: the stored choice while the catalog offers it, else the\n * default (unless retired), else a recommended family default, else any live model. While the\n * catalog is empty (loading, or unreachable) the stored choice or the default stands.\n */\nexport function resolveCatalogModelId({\n  stored,\n  defaultModelId,\n  models,\n}: {\n  stored: string | null;\n  defaultModelId: string;\n  models: readonly AthenaModelOption[];\n}): string {\n  if (stored && models.some((model) => model.id === stored)) return stored;\n  return (\n    (\n      models.find((model) => model.id === defaultModelId && model.status !== 'LEGACY') ??\n      models.find((model) => model.status === 'RECOMMENDED' && model.isFamilyDefault) ??\n      models.find((model) => model.status !== 'LEGACY' && model.isFamilyDefault) ??\n      models.find((model) => model.status !== 'LEGACY') ??\n      models[0]\n    )?.id ?? (stored || defaultModelId)\n  );\n}\n\n/** Models grouped under their provider's label, in catalog order. */\nexport function groupModelsByProvider({\n  models,\n}: {\n  models: readonly AthenaModelOption[];\n}): Array<{ provider: string; models: AthenaModelOption[] }> {\n  const groups = new Map<string, AthenaModelOption[]>();\n  for (const model of models) {\n    const label = MODEL_PROVIDER_LABELS[model.provider] ?? model.provider;\n    groups.set(label, [...(groups.get(label) ?? []), model]);\n  }\n  return Array.from(groups, ([provider, grouped]) => ({ provider, models: grouped }));\n}\n\n/** The provider's credentials as the SDK's GraphQL client takes them. */\nfunction useCatalogAuth(): MentionAuthContext {\n  const config = useAthenaConfig();\n  return useMemo<MentionAuthContext>(\n    () => ({\n      backendUrl: config.backendUrl,\n      appUrl: config.appUrl,\n      apiKey: config.apiKey,\n      credentials: 'include',\n      get accessToken() {\n        return config.token ?? undefined;\n      },\n      getAccessToken: async () => (await config.getToken?.()) ?? undefined,\n      renewAccessToken: async () => (await config.getToken?.({ force: true })) ?? undefined,\n    }),\n    [config],\n  );\n}\n\n/**\n * A short fingerprint (two FNV-1a rounds, 64 bits) of a credential: it tells accounts apart in an\n * in-memory cache key without the credential itself, or anything it could be recovered from,\n * entering the key.\n */\nexport function credentialFingerprint({ credential }: { credential: string }): string {\n  const round = (seed: number) => {\n    let hash = seed;\n    for (let index = 0; index < credential.length; index += 1) {\n      hash ^= credential.charCodeAt(index);\n      hash = Math.imul(hash, 0x01000193) >>> 0;\n    }\n    return hash.toString(16).padStart(8, '0');\n  };\n  return `${round(0x811c9dc5)}${round(0x01000193)}`;\n}\n\n/**\n * Who is asking, for the cache key: the catalog differs per account (admin-only models, the\n * workspace's lists, the agents shared with them), and a host can switch accounts while keeping its\n * query client. The signed-in user's id when the SDK knows it, else a fingerprint of the credential\n * in use. Never the credential itself.\n */\nfunction useCatalogPrincipal(): string {\n  const config = useAthenaConfig();\n  const userId = useContext(AthenaAuthContext)?.user?.userId;\n  if (userId) return `user:${userId}`;\n  if (config.apiKey) return `key:${credentialFingerprint({ credential: config.apiKey })}`;\n  if (config.token) return `token:${credentialFingerprint({ credential: config.token })}`;\n  return 'session';\n}\n\n/**\n * The host's QueryClient when there is one (the thread list provides one), else one of the\n * pickers' own, so the catalog hooks work with or without the thread list.\n */\nfunction useCatalogQueryClient(): QueryClient {\n  const host = useContext(QueryClientContext);\n  const [own] = useState(() => new QueryClient());\n  return host ?? own;\n}\n\n/** The models the signed-in person may pick (`normalizeModelCatalog`). */\nexport function useAthenaModelCatalog() {\n  const auth = useCatalogAuth();\n  const principal = useCatalogPrincipal();\n  const queryClient = useCatalogQueryClient();\n  return useQuery(\n    {\n      queryKey: ['athena-sdk', 'catalog', 'models', auth.backendUrl, principal],\n      queryFn: async () =>\n        normalizeModelCatalog({\n          response: await fetchGraphQL<AthenaModelCatalogResponse>(auth, ATHENA_MODEL_CATALOG_QUERY),\n        }),\n      staleTime: 5 * 60 * 1000,\n    },\n    queryClient,\n  );\n}\n\n/** The published collab agents the signed-in person can see (`normalizeAgentCatalog`). */\nexport function useAthenaAgentCatalog() {\n  const auth = useCatalogAuth();\n  const principal = useCatalogPrincipal();\n  const queryClient = useCatalogQueryClient();\n  return useQuery(\n    {\n      queryKey: ['athena-sdk', 'catalog', 'agents', auth.backendUrl, principal],\n      queryFn: async () =>\n        normalizeAgentCatalog({\n          assets:\n            (\n              await fetchGraphQL<{ getAllAssets: Array<{ id: string; title: string | null }> }>(\n                auth,\n                ATHENA_AGENT_CATALOG_QUERY,\n              )\n            ).getAllAssets ?? [],\n        }),\n      staleTime: 60 * 1000,\n    },\n    queryClient,\n  );\n}\n"],"names":[],"mappings":";;;;;AAqCO,MAAM,wBAA0D;AAAA,EACrE,WAAW;AAAA,EACX,QAAQ;AAAA,EACR,QAAQ;AAAA,EACR,KAAK;AAAA,EACL,MAAM;AAAA,EACN,SAAS;AAAA,EACT,UAAU;AAAA,EACV,MAAM;AAAA,EACN,UAAU;AAAA,EACV,SAAS;AAAA,EACT,KAAK;AACP;AAEO,MAAM,6BAA6B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAWnC,MAAM,6BAA6B;AAAA;AAAA;AAAA;AAAA;AA6BnC,SAAS,sBAAsB;AAAA,EACpC;AACF,GAEwB;AACtB,QAAM,UAAU,SAAS,4BAA4B,KAAK;AAC1D,QAAM,WAAW,IAAI,IAAI,SAAS,SAAS,UAAU,iBAAiB;AACtE,QAAM,SAAS,IAAI,IAAI,SAAS,SAAS,UAAU,eAAe;AAClE,QAAM,eAAe,SAAS,OAAO;AACrC,SAAO,SAAS,gBACb;AAAA,IACC,CAAC,WACE,CAAC,MAAM,aAAa,aAAa,eAAe,SAAS,IAAI,MAAM,EAAE,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE;AAAA,EAAA,EAEjG;AAAA,IACC,CAAC,WAA8B;AAAA,MAC7B,IAAI,MAAM;AAAA,MACV,OAAO,MAAM;AAAA,MACb,aAAa,MAAM;AAAA,MACnB,UAAU,MAAM;AAAA,MAChB,QAAQ,MAAM;AAAA,MACd,WAAW,MAAM;AAAA,MACjB,UAAU,MAAM;AAAA,MAChB,YAAY,MAAM;AAAA,MAClB,iBAAiB,MAAM;AAAA,MACvB,YAAY,MAAM;AAAA,MAClB,cAAc,MAAM;AAAA,IAAA;AAAA,EACtB,EAED;AAAA,IACC,CAAC,MAAM,UACL,KAAK,YAAY,MAAM,aACvB,KAAK,eAAe,MAAM,gBAC1B,KAAK,MAAM,cAAc,MAAM,KAAK;AAAA,EAAA;AAE5C;AAGO,SAAS,sBAAsB;AAAA,EACpC;AACF,GAEwB;AACtB,SAAO,OACJ,OAAO,CAAC,UAAU,MAAM,GAAG,KAAA,EAAO,SAAS,CAAC,EAC5C,IAAI,CAAC,UAAA;;AAAW,aAAE,IAAI,MAAM,IAAI,QAAM,WAAM,UAAN,mBAAa,WAAU,iBAAA;AAAA,GAAmB,EAChF,KAAK,CAAC,MAAM,UAAU,KAAK,KAAK,cAAc,MAAM,IAAI,CAAC;AAC9D;AAOO,SAAS,sBAAsB;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AACF,GAIW;;AACT,MAAI,UAAU,OAAO,KAAK,CAAC,UAAU,MAAM,OAAO,MAAM,EAAG,QAAO;AAClE,WAEI,YAAO,KAAK,CAAC,UAAU,MAAM,OAAO,kBAAkB,MAAM,WAAW,QAAQ,KAC/E,OAAO,KAAK,CAAC,UAAU,MAAM,WAAW,iBAAiB,MAAM,eAAe,KAC9E,OAAO,KAAK,CAAC,UAAU,MAAM,WAAW,YAAY,MAAM,eAAe,KACzE,OAAO,KAAK,CAAC,UAAU,MAAM,WAAW,QAAQ,KAChD,OAAO,CAAC,MAJR,mBAKC,QAAO,UAAU;AAExB;AAGO,SAAS,sBAAsB;AAAA,EACpC;AACF,GAE6D;AAC3D,QAAM,6BAAa,IAAA;AACnB,aAAW,SAAS,QAAQ;AAC1B,UAAM,QAAQ,sBAAsB,MAAM,QAAQ,KAAK,MAAM;AAC7D,WAAO,IAAI,OAAO,CAAC,GAAI,OAAO,IAAI,KAAK,KAAK,IAAK,KAAK,CAAC;AAAA,EACzD;AACA,SAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,UAAU,OAAO,OAAO,EAAE,UAAU,QAAQ,QAAA,EAAU;AACpF;AAGA,SAAS,iBAAqC;AAC5C,QAAM,SAAS,gBAAA;AACf,SAAO;AAAA,IACL,OAAO;AAAA,MACL,YAAY,OAAO;AAAA,MACnB,QAAQ,OAAO;AAAA,MACf,QAAQ,OAAO;AAAA,MACf,aAAa;AAAA,MACb,IAAI,cAAc;AAChB,eAAO,OAAO,SAAS;AAAA,MACzB;AAAA,MACA,gBAAgB,YAAA;;AAAa,uBAAM,YAAO,aAAP,oCAAwB;AAAA;AAAA,MAC3D,kBAAkB,YAAA;;AAAa,uBAAM,YAAO,aAAP,gCAAkB,EAAE,OAAO,KAAA,OAAY;AAAA;AAAA,IAAA;AAAA,IAE9E,CAAC,MAAM;AAAA,EAAA;AAEX;AAOO,SAAS,sBAAsB,EAAE,cAA8C;AACpF,QAAM,QAAQ,CAAC,SAAiB;AAC9B,QAAI,OAAO;AACX,aAAS,QAAQ,GAAG,QAAQ,WAAW,QAAQ,SAAS,GAAG;AACzD,cAAQ,WAAW,WAAW,KAAK;AACnC,aAAO,KAAK,KAAK,MAAM,QAAU,MAAM;AAAA,IACzC;AACA,WAAO,KAAK,SAAS,EAAE,EAAE,SAAS,GAAG,GAAG;AAAA,EAC1C;AACA,SAAO,GAAG,MAAM,UAAU,CAAC,GAAG,MAAM,QAAU,CAAC;AACjD;AAQA,SAAS,sBAA8B;;AACrC,QAAM,SAAS,gBAAA;AACf,QAAM,UAAS,sBAAW,iBAAiB,MAA5B,mBAA+B,SAA/B,mBAAqC;AACpD,MAAI,OAAQ,QAAO,QAAQ,MAAM;AACjC,MAAI,OAAO,OAAQ,QAAO,OAAO,sBAAsB,EAAE,YAAY,OAAO,OAAA,CAAQ,CAAC;AACrF,MAAI,OAAO,MAAO,QAAO,SAAS,sBAAsB,EAAE,YAAY,OAAO,MAAA,CAAO,CAAC;AACrF,SAAO;AACT;AAMA,SAAS,wBAAqC;AAC5C,QAAM,OAAO,WAAW,kBAAkB;AAC1C,QAAM,CAAC,GAAG,IAAI,SAAS,MAAM,IAAI,aAAa;AAC9C,SAAO,QAAQ;AACjB;AAGO,SAAS,wBAAwB;AACtC,QAAM,OAAO,eAAA;AACb,QAAM,YAAY,oBAAA;AAClB,QAAM,cAAc,sBAAA;AACpB,SAAO;AAAA,IACL;AAAA,MACE,UAAU,CAAC,cAAc,WAAW,UAAU,KAAK,YAAY,SAAS;AAAA,MACxE,SAAS,YACP,sBAAsB;AAAA,QACpB,UAAU,MAAM,aAAyC,MAAM,0BAA0B;AAAA,MAAA,CAC1F;AAAA,MACH,WAAW,IAAI,KAAK;AAAA,IAAA;AAAA,IAEtB;AAAA,EAAA;AAEJ;AAGO,SAAS,wBAAwB;AACtC,QAAM,OAAO,eAAA;AACb,QAAM,YAAY,oBAAA;AAClB,QAAM,cAAc,sBAAA;AACpB,SAAO;AAAA,IACL;AAAA,MACE,UAAU,CAAC,cAAc,WAAW,UAAU,KAAK,YAAY,SAAS;AAAA,MACxE,SAAS,YACP,sBAAsB;AAAA,QACpB,SAEI,MAAM;AAAA,UACJ;AAAA,UACA;AAAA,QAAA,GAEF,gBAAgB,CAAA;AAAA,MAAC,CACtB;AAAA,MACH,WAAW,KAAK;AAAA,IAAA;AAAA,IAElB;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 StatewireRunRefusalNotice = require("./StatewireRunRefusalNotice.cjs");
 const WaitingOnYouBanner = require("./WaitingOnYouBanner.cjs");
 const wakeMarker = require("../chat-ui/messages/wake-marker.cjs");
 const toolFallback = require("../tools/tool-fallback.cjs");
@@ -13,6 +14,7 @@ const appendDocumentToolUi = require("../tools/append-document-tool-ui.cjs");
 const readAssetToolUi = require("../tools/read-asset-tool-ui.cjs");
 const toolUis = require("../tools/tool-uis.cjs");
 const SuperGroupingCard = require("./super-grouping/SuperGroupingCard.cjs");
+const SuperGroupingTranscript = require("./super-grouping/SuperGroupingTranscript.cjs");
 const NestedPtcMessages = require("../tools/NestedPtcMessages.cjs");
 const tooltipIconButton = require("./tooltip-icon-button.cjs");
 const button = require("../ui/button.cjs");
@@ -23,6 +25,7 @@ const statewireHistoryEdge = require("../chat-ui/thread/statewire-history-edge.c
 const threadLoadingOverlay = require("../chat-ui/thread/thread-loading-overlay.cjs");
 const composerSendOrStop = require("../chat-ui/thread/composer-send-or-stop.cjs");
 const composerAction = require("../chat-ui/composer/composer-action.cjs");
+const composerMessage = require("../chat-ui/composer/composer-message.cjs");
 const athenaDiagnostics = require("../diagnostics/athena-diagnostics.cjs");
 const useDictation = require("../dictation/use-dictation.cjs");
 const DictationStrip = require("../dictation/DictationStrip.cjs");
@@ -42,6 +45,7 @@ const react = require("@assistant-ui/react");
 const lucideReact = require("lucide-react");
 const react$1 = require("react");
 const threadIdContext = require("../threads/thread-id-context.cjs");
+const statewireThreadList = require("../threads/statewire-thread-list.cjs");
 const FileUploadButton = require("../composer/FileUploadButton.cjs");
 const ComposerQuotePreview = require("../composer/ComposerQuotePreview.cjs");
 const ComposerAttachmentPreview = require("../composer/ComposerAttachmentPreview.cjs");
@@ -57,11 +61,15 @@ const index = require("../toolkits/index.cjs");
 const runActivity = require("./run-activity.cjs");
 const messageTime = require("./message-time.cjs");
 const threadScrollToBottom = require("./thread-scroll-to-bottom.cjs");
-const githubWakeEvent = require("../packages/agent-runtime/src/github-wake-event.cjs");
+const githubWakeEvent = require("../agent-runtime/src/github-wake-event.cjs");
+const subscriptionWakeEvent = require("../agent-runtime/src/subscription-wake-event.cjs");
+const threadStatus = require("../agent-runtime/src/thread-status.cjs");
 const cnfast = require("cnfast");
+const DEFAULT_COMPOSER_PLACEHOLDER = "Type @ to tag an asset";
 function QuotePostMessageBridge() {
-  QuoteContext.useQuoteFromPostMessage();
-  return null;
+  const marker = react$1.useRef(null);
+  QuoteContext.useQuoteFromPostMessage({ scope: marker });
+  return /* @__PURE__ */ jsxRuntime.jsx("span", { ref: marker, hidden: true });
 }
 const AthenaChatDefaultComponentsContext = react$1.createContext(null);
 const AthenaReasoningTextComponentContext = react$1.createContext(null);
@@ -96,8 +104,8 @@ const AthenaDefaultAssistantMessage = () => {
   );
 };
 const AthenaDefaultUserMessage = () => {
-  const { TextComponent } = useAthenaChatDefaultComponents();
-  return /* @__PURE__ */ jsxRuntime.jsx(AthenaUserMessage, { TextComponent });
+  const { UserTextComponent } = useAthenaChatDefaultComponents();
+  return /* @__PURE__ */ jsxRuntime.jsx(AthenaUserMessage, { TextComponent: UserTextComponent });
 };
 const getReasoningTokensFromMetadata = (metadata) => {
   if (!utils.isRecord(metadata)) {
@@ -140,9 +148,10 @@ const AthenaChat = ({
   composer: composerSlots,
   dictation = true
 }) => {
-  var _a, _b, _c;
+  var _a, _b, _c, _d, _e;
   const athenaConfig = AthenaContext.useAthenaConfig();
   const statewireThreadId = react$1.useContext(threadIdContext.AthenaStatewireThreadIdContext);
+  const startNewSession = (_a = statewireThreadList.useAthenaStatewireThreadList()) == null ? void 0 : _a.newThread;
   const mentionAuth = react$1.useMemo(
     () => ({
       backendUrl: athenaConfig.backendUrl,
@@ -187,15 +196,17 @@ const AthenaChat = ({
     }),
     [toolUIs]
   );
-  const textComponent = ((_a = components == null ? void 0 : components.messageParts) == null ? void 0 : _a.Text) ?? tiptapText.TiptapText;
-  const reasoningComponent = (_b = components == null ? void 0 : components.messageParts) == null ? void 0 : _b.Reasoning;
-  const emptyComponent = ((_c = components == null ? void 0 : components.messageParts) == null ? void 0 : _c.Empty) ?? AthenaAssistantMessageEmpty;
+  const textComponent = ((_b = components == null ? void 0 : components.messageParts) == null ? void 0 : _b.Text) ?? tiptapText.TiptapText;
+  const userTextComponent = ((_c = components == null ? void 0 : components.messageParts) == null ? void 0 : _c.Text) ?? messageComponents.UserMessageText;
+  const reasoningComponent = (_d = components == null ? void 0 : components.messageParts) == null ? void 0 : _d.Reasoning;
+  const emptyComponent = ((_e = components == null ? void 0 : components.messageParts) == null ? void 0 : _e.Empty) ?? AthenaAssistantMessageEmpty;
   const actionBarComponent = (components == null ? void 0 : components.AssistantActionBar) ?? AthenaAssistantActionBar;
   const avatarComponent = components == null ? void 0 : components.AssistantAvatar;
   const defaultComponentsContextValue = react$1.useMemo(
     () => ({
       toolUIs: mergedToolUIs,
       TextComponent: textComponent,
+      UserTextComponent: userTextComponent,
       ReasoningComponent: reasoningComponent,
       EmptyComponent: emptyComponent,
       ActionBarComponent: actionBarComponent,
@@ -209,12 +220,13 @@ const AthenaChat = ({
       groupToolCalls,
       mergedToolUIs,
       reasoningComponent,
-      textComponent
+      textComponent,
+      userTextComponent
     ]
   );
   const AssistantMessageComponent = (components == null ? void 0 : components.AssistantMessage) ?? AthenaDefaultAssistantMessage;
   const UserMessageComponent = (components == null ? void 0 : components.UserMessage) ?? AthenaDefaultUserMessage;
-  const messageComponents2 = react$1.useMemo(
+  const messageComponents$1 = react$1.useMemo(
     () => ({
       UserMessage: UserMessageComponent,
       AssistantMessage: AssistantMessageComponent
@@ -238,14 +250,15 @@ const AthenaChat = ({
       viewportOverlay: /* @__PURE__ */ jsxRuntime.jsx(ThreadScrollToTop, {}),
       welcome: /* @__PURE__ */ jsxRuntime.jsx(athenaChatWelcome.AthenaChatWelcome, { welcomeMessage, welcomeSubtext }),
       beforeMessages: /* @__PURE__ */ jsxRuntime.jsx(statewireHistoryEdge.StatewireHistoryEdge, { className: "aui-thread-history-edge" }),
-      components: messageComponents2,
+      components: messageComponents$1,
       runningIndicator: null,
       scrollToBottom: null,
       viewportFooterClassName: "aui-thread-viewport-footer sticky bottom-0 mx-auto mt-auto flex w-full max-w-(--thread-max-width) flex-col gap-4 overflow-visible rounded-t-3xl bg-background pt-4 pb-4 md:pb-6",
       viewportFooter: /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
         welcomeSuggestions.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(react.ThreadPrimitive.Empty, { children: /* @__PURE__ */ jsxRuntime.jsx(athenaChatWelcome.AthenaChatWelcomeSuggestions, { suggestions: welcomeSuggestions }) }),
         /* @__PURE__ */ jsxRuntime.jsx(statewireBanners.StatewireConnectionBanner, {}),
-        /* @__PURE__ */ jsxRuntime.jsx(statewireBanners.StatewireErrorBanner, {}),
+        /* @__PURE__ */ jsxRuntime.jsx(statewireBanners.StatewireErrorBanner, { onStartNewSession: startNewSession }),
+        /* @__PURE__ */ jsxRuntime.jsx(StatewireRunRefusalNotice.StatewireRunRefusalNotice, { onStartNewSession: startNewSession }),
         /* @__PURE__ */ jsxRuntime.jsx(StatewireApprovalCard.StatewireApprovalCard, {}),
         /* @__PURE__ */ jsxRuntime.jsx(apiClient.MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsxRuntime.jsx(StatewireQueuedMessages.StatewireQueuedMessages, {}) }),
         /* @__PURE__ */ jsxRuntime.jsx(AttachmentContext.AttachmentProvider, { children: /* @__PURE__ */ jsxRuntime.jsxs(QuoteContext.QuoteProvider, { children: [
@@ -327,6 +340,11 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     var _a;
     return ((_a = state.optional.thread) == null ? void 0 : _a.isRunning) ?? false;
   });
+  const connection = threadStatus.useDeepAgentConnection();
+  const runLive = react.useAuiState((state) => {
+    const thread = state.optional.thread;
+    return thread ? threadStatus.readRunLive(thread, connection) : false;
+  });
   const [dragActive, setDragActive] = react$1.useState(false);
   const keyboardHint = composerKeyboardHint.useComposerKeyboardHint();
   const reducedMotion = usePrefersReducedMotion.usePrefersReducedMotion();
@@ -341,6 +359,11 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
   const appendSpokenText = react$1.useCallback(
     (spoken) => {
       if (!spoken) return;
+      const composer = composerRef.current;
+      if (composer == null ? void 0 : composer.appendText) {
+        composer.appendText(spoken);
+        return;
+      }
       aui.composer.setText(dictationState.composeDictationDraft(aui.composer.getState().text ?? "", spoken));
     },
     [aui]
@@ -386,24 +409,25 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     });
   }, []);
   const sendComposed = react$1.useCallback(
-    (userText) => {
-      const message = buildComposedMessage.buildComposedMessage({
+    (editorSend) => {
+      const send = buildComposedMessage.buildComposedSend({
         attachments: attachmentsRef.current,
         quote: quoteRef.current,
-        userText,
+        send: editorSend,
         appUrl
       });
-      if (!message) return false;
-      aui.composer.setText(message);
+      if (!send) return false;
+      composerMessage.prepareComposerSend({ threadId, ...send });
+      aui.composer.setText(send.text);
       aui.composer.send({ steer: false });
       clearAttachments();
       clearQuote();
       return true;
     },
-    [appUrl, aui, clearAttachments, clearQuote]
+    [appUrl, aui, clearAttachments, clearQuote, threadId]
   );
   const interceptSubmit = react$1.useCallback(
-    (text) => attachmentsRef.current.length > 0 || quoteRef.current !== null ? sendComposed(text) : false,
+    (text, draftMarkdown) => attachmentsRef.current.length > 0 || quoteRef.current !== null ? sendComposed({ text, draftMarkdown }) : false,
     [sendComposed]
   );
   const fileIntake = react$1.useMemo(
@@ -418,20 +442,20 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     appendSpokenText((await dictationCtl.stop()).text);
   }, [appendSpokenText, dictationCtl]);
   const sendDictation = react$1.useCallback(async () => {
-    var _a;
+    var _a, _b, _c;
     const { text: spoken, error } = await dictationCtl.stop();
-    const text = dictationState.composeDictationDraft(aui.composer.getState().text ?? "", spoken);
-    if (error) {
-      aui.composer.setText(text);
-      return;
-    }
+    appendSpokenText(spoken);
+    if (error) return;
+    const send = ((_b = (_a = composerRef.current) == null ? void 0 : _a.captureSend) == null ? void 0 : _b.call(_a)) ?? {
+      text: aui.composer.getState().text ?? "",
+      draftMarkdown: aui.composer.getState().text ?? ""
+    };
     const extras = attachmentsRef.current.length > 0 || quoteRef.current !== null;
-    if (!text && !extras) return;
-    aui.composer.setText(text);
+    if (!send.text && !extras) return;
     const action = composerAction.resolveComposerAction({ hasDraft: true, isRunning });
     if (action === "stop") return;
     try {
-      await ((_a = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _a.call(slots, { action }));
+      await ((_c = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _c.call(slots, { action }));
     } catch (error2) {
       athenaDiagnostics.athenaDiagnostics.emit("composer.dictation", {
         level: "warn",
@@ -443,11 +467,13 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
       return;
     }
     if (extras) {
-      sendComposed(text);
+      sendComposed(send);
       return;
     }
+    composerMessage.prepareComposerSend({ threadId, ...send });
+    aui.composer.setText(send.text);
     aui.composer.send({ steer: false });
-  }, [aui, dictationCtl, isRunning, sendComposed, slots]);
+  }, [appendSpokenText, aui, dictationCtl, isRunning, sendComposed, slots, threadId]);
   return /* @__PURE__ */ jsxRuntime.jsx(ComposerDropZone.ComposerDropZone, { onDragActiveChange: setDragActive, children: /* @__PURE__ */ jsxRuntime.jsx(
     "div",
     {
@@ -471,7 +497,7 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
                 ref: composerRef,
                 threadId,
                 appearance: "paper",
-                placeholder: "Type @ to tag an asset",
+                placeholder: (runLive ? slots == null ? void 0 : slots.runningPlaceholder : void 0) ?? (slots == null ? void 0 : slots.placeholder) ?? DEFAULT_COMPOSER_PLACEHOLDER,
                 tools,
                 rootCategories,
                 onDraftStateChange: setHasText,
@@ -555,7 +581,7 @@ const MessageError = () => /* @__PURE__ */ jsxRuntime.jsx(react.MessagePrimitive
     }
   ) })
 ] }) });
-const RunStatusLine = ({ verb, resetKey, counterOnly = false, className }) => {
+function useSecondsSince(resetKey) {
   const [seconds, setSeconds] = react$1.useState(0);
   react$1.useEffect(() => {
     setSeconds(0);
@@ -565,8 +591,11 @@ const RunStatusLine = ({ verb, resetKey, counterOnly = false, className }) => {
     }, 1e3);
     return () => clearInterval(timer);
   }, [resetKey]);
+  return seconds;
+}
+const RunStatusLine = ({ verb, resetKey, className }) => {
+  const seconds = useSecondsSince(resetKey);
   const stalled = seconds >= runActivity.RUN_ACTIVITY_STALL_SECONDS;
-  if (counterOnly && seconds < runActivity.RUN_ACTIVITY_COUNTER_SECONDS) return null;
   return /* @__PURE__ */ jsxRuntime.jsxs(
     "div",
     {
@@ -627,13 +656,22 @@ const AthenaMessageActivityIndicator = () => {
 };
 const AthenaGroupedRunMeta = () => {
   const activityKey = react.useAuiState((s) => runActivity.runActivityKey(s.message.parts));
-  return /* @__PURE__ */ jsxRuntime.jsx(
-    RunStatusLine,
+  const seconds = useSecondsSince(activityKey ?? "thinking");
+  const stalled = seconds >= runActivity.RUN_ACTIVITY_STALL_SECONDS;
+  if (!stalled) return null;
+  return /* @__PURE__ */ jsxRuntime.jsxs(
+    "span",
     {
-      verb: "Working",
-      resetKey: activityKey ?? "thinking",
-      counterOnly: true,
-      className: "mt-1 px-0"
+      "data-slot": "run-stalled",
+      "data-stalled": "true",
+      className: "tabular-nums text-amber-600 dark:text-amber-400",
+      children: [
+        /* @__PURE__ */ jsxRuntime.jsxs("span", { "aria-hidden": "true", children: [
+          "Nothing new for ",
+          runActivity.runElapsedLabel(seconds)
+        ] }),
+        /* @__PURE__ */ jsxRuntime.jsx("span", { "aria-live": "polite", className: "sr-only", children: "Assistant is still working; nothing new for over thirty seconds." })
+      ]
     }
   );
 };
@@ -818,7 +856,7 @@ const AthenaAssistantMessage = ({
             /* @__PURE__ */ jsxRuntime.jsx(AthenaReasoningTextComponentContext.Provider, { value: TextComponent, children: /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "aui-assistant-message-content wrap-break-word px-2 text-foreground leading-relaxed", children: [
               /* @__PURE__ */ jsxRuntime.jsx(NestedPtcMessages.NestedPtcComponentsProvider, { components: nestedPtcComponents, children: groupToolCalls ? /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex flex-col space-y-2", children: [
                 /* @__PURE__ */ jsxRuntime.jsx(
-                  SuperGroupingCard.SuperGroupingCard,
+                  SuperGroupingTranscript.SuperGroupingTranscript,
                   {
                     toolUIs: toolUIsWithNestedMessages,
                     TextComponent,
@@ -959,12 +997,14 @@ const AthenaAssistantActionBar = ({ className }) => {
   );
 };
 const AthenaUserMessage = ({
-  TextComponent = tiptapText.TiptapText
+  TextComponent = messageComponents.UserMessageText
 }) => {
-  const isWake = react.useAuiState((s) => {
-    var _a;
-    return githubWakeEvent.isGithubWakeMessage((_a = s.message.metadata) == null ? void 0 : _a.custom);
-  });
+  const isWake = react.useAuiState(
+    (s) => {
+      var _a, _b;
+      return githubWakeEvent.isGithubWakeMessage((_a = s.message.metadata) == null ? void 0 : _a.custom) || subscriptionWakeEvent.isSubscriptionWakeMessage((_b = s.message.metadata) == null ? void 0 : _b.custom);
+    }
+  );
   if (isWake) {
     return /* @__PURE__ */ jsxRuntime.jsxs(
       react.MessagePrimitive.Root,