@linktr.ee/messaging-react 2.0.0 → 2.0.1-rc-1778694826

package/dist/index.d.ts

@@ -241,15 +241,38 @@ export declare interface ChannelViewProps {
     sendButton?: ComponentType<any>;
 }
 
-export declare interface CreatorCardProps extends LockedAttachmentBaseProps {
+export declare interface ComposerCardProps extends LockedAttachmentBaseProps {
+    /** Placeholder shown in the title slot before the composer types one. */
     placeholderTitle?: string;
+    /** Placeholder shown in the amount slot before one is configured. */
     placeholderAmountText?: string;
-    isUnlocking?: boolean;
+    /**
+     * When provided, renders a dismiss X in the thumbnail corner. Called when
+     * the composer clicks it to remove the attachment.
+     */
     onDismiss?: () => void;
+    /** Fired the first time the composer taps the thumbnail to preview. */
     onPreviewClick?: () => void;
+    /**
+     * Lazily loads the underlying source so the composer can preview the
+     * attachment they're about to send. Called the first time the thumbnail is
+     * tapped; the returned source is cached and reused on subsequent toggles.
+     */
     onFetchSource?: () => Promise<LockedAttachmentSource | void>;
+    /**
+     * When provided, renders a pencil button in the body bottom-right that the
+     * composer can use to edit the attachment metadata (e.g. open the price /
+     * gallery editor). Matches the Composer "Button" instance in Figma.
+     */
+    onEditClick?: () => void;
 }
 
+/**
+ * @deprecated Renamed to `SentCardProps`. Drafting usages (with `onDismiss`)
+ * should migrate to `ComposerCardProps`.
+ */
+export declare type CreatorCardProps = SentCardProps;
+
 export declare const CustomMessageProvider: Provider<Partial<CustomMessageRegistry>>;
 
 export declare interface CustomMessageRegistry {
@@ -300,9 +323,142 @@ export declare function getMessageDisplayText({ message, viewerLanguage, }: {
 
 export declare function isLinkAttachment(a: Attachment): boolean;
 
+/**
+ * Link attachments (image / file media + 1P/3P Link Apps) shown in the chat
+ * thread. Mirrors the `LockedAttachment` API — render `LinkAttachment.Composer`
+ * while drafting, `LinkAttachment.Sent` after posting, and
+ * `LinkAttachment.Received` in the recipient's thread. Maps to the
+ * "Attachments" and "LinkApps" sections of the messaging design system.
+ *
+ * Two visual layouts via the `layout` prop:
+ *   - **Featured** (default) — 180px hero thumbnail above the body. Used by
+ *     image / file Attachments and by hero-image LinkApps (Spotify with
+ *     cover art, TikTok with a frame, etc.).
+ *   - **Classic** — compact card with no hero thumbnail; title /
+ *     description / URL / CTA only. Used for LinkApp embeds without
+ *     artwork (FAQ, Form) and any link preview that lacks OG imagery.
+ *
+ * Image / file Attachments use `layout="featured"` and skip the title /
+ * description / URL body entirely (`CardBody` collapses to nothing when no
+ * text content is provided). LinkApps always carry a title + description
+ * and prefix the title with an `appIcon` brand badge.
+ */
+export declare const LinkAttachment: {
+    Composer: (props: LinkAttachmentComposerCardProps) => JSX_2.Element;
+    Sent: (props: LinkAttachmentSentCardProps) => JSX_2.Element;
+    Received: (props: LinkAttachmentReceivedCardProps) => JSX_2.Element;
+};
+
+/**
+ * Shared props for the three `LinkAttachment.*` variants (Composer, Sent,
+ * Received). Maps to the "Attachments" + "LinkApps" sections of the messaging
+ * design system in Figma — a 280px-wide card with an optional 180px hero
+ * thumbnail, a title (optionally prefixed with a brand badge for Link Apps),
+ * a description, and either a URL footer or a CTA button.
+ */
+export declare interface LinkAttachmentBaseProps {
+    title?: string;
+    /** Placeholder shown in the title slot before one is configured (dark variants only). */
+    placeholderTitle?: string;
+    /** Secondary description rendered below the title. */
+    description?: string;
+    /**
+     * Optional URL displayed at the bottom of the card (e.g. `tr.ee/briemix`).
+     * Ignored when `cta` is provided. Also used as the navigation target for
+     * the Received card when no `cta` is set.
+     */
+    url?: string;
+    /** MIME type of the hero thumbnail — drives the type icon for empty states. */
+    mimeType?: string;
+    /** Hero thumbnail (180px tall) shown above the title block. */
+    thumbnailUrl?: string;
+    /**
+     * Source URL for playable media (video, audio). When provided alongside a
+     * video / audio `mimeType`, the hero region renders an inline player with
+     * native controls instead of the static thumbnail / type-icon.
+     */
+    sourceUrl?: string;
+    /**
+     * Visual layout — `'featured'` keeps the 180px hero thumbnail above the
+     * body, `'classic'` drops the hero entirely for a compact text-only
+     * card. Defaults to `'featured'`.
+     */
+    layout?: LinkAttachmentLayout;
+    /**
+     * Optional 16x16 brand badge rendered before the title (used by Link Apps:
+     * Spotify, TikTok, FAQ, Form, etc.). Consumers render whatever they want
+     * — typically a colored 4px-rounded square containing a glyph or `<img>`.
+     */
+    appIcon?: default_2.ReactNode;
+    /**
+     * Optional call-to-action rendered below the description. When set,
+     * replaces the URL footer (e.g. FAQ "View FAQs", Form "Complete form").
+     */
+    cta?: LinkAttachmentCta;
+}
+
+export declare interface LinkAttachmentComposerCardProps extends LinkAttachmentBaseProps {
+    /**
+     * When provided, renders a dismiss X in the thumbnail corner. Called when
+     * the composer clicks it to remove the attachment.
+     */
+    onDismiss?: () => void;
+    /**
+     * When provided, renders a pencil button to the right of the description
+     * that the composer can use to edit the attachment metadata.
+     */
+    onEditClick?: () => void;
+}
+
+export declare interface LinkAttachmentCta {
+    label: string;
+    /** When set, the CTA renders as an `<a>` opening in a new tab. */
+    href?: string;
+    /** When set, called on click (in addition to `href` navigation if provided). */
+    onClick?: () => void;
+}
+
+/**
+ * Visual layout for a `LinkAttachment.*` card.
+ *
+ * - `featured` — full card with a 180px hero thumbnail above the body.
+ *   The default; matches the "Attachments" frames and the hero-image
+ *   "LinkApps" frames in Figma.
+ * - `classic` — compact card without a hero thumbnail. Title /
+ *   description / URL / CTA only. Used for Link App embeds that don't
+ *   carry artwork (and for plain link previews without OG imagery).
+ */
+export declare type LinkAttachmentLayout = 'featured' | 'classic';
+
+export declare interface LinkAttachmentReceivedCardProps extends LinkAttachmentBaseProps {
+    /**
+     * Fired when the recipient activates the card. Behavior depends on how
+     * the card is configured:
+     *   - **Link app with a CTA** (FAQ / Form): the CTA owns navigation;
+     *     `onClick` fires when the recipient taps the CTA itself, alongside
+     *     `cta.onClick` (use for analytics).
+     *   - **Link app with a URL** (Spotify / TikTok / generic link): the card
+     *     chrome is an `<a target="_blank">` opening `url` — `onClick` fires
+     *     alongside the navigation (use for analytics).
+     *   - **Image / file / placeholder attachment**: the card has no URL, so
+     *     it renders as a button. `onClick` is the consumer's hook for
+     *     opening an image / file preview (lightbox).
+     *   - **Video / audio attachment**: the shell stays non-interactive so
+     *     the native media controls remain operable — `onClick` is ignored
+     *     in this configuration.
+     */
+    onClick?: () => void;
+}
+
+export declare interface LinkAttachmentSentCardProps extends LinkAttachmentBaseProps {
+}
+
 export declare const LockedAttachment: {
-    Creator: (props: CreatorCardProps) => JSX_2.Element;
-    Visitor: (props: VisitorCardProps) => JSX_2.Element;
+    Composer: (props: ComposerCardProps) => JSX_2.Element;
+    Sent: (props: SentCardProps) => JSX_2.Element;
+    Received: (props: ReceivedCardProps) => JSX_2.Element;
+    Creator: (props: SentCardProps) => JSX_2.Element;
+    Visitor: (props: ReceivedCardProps) => JSX_2.Element;
 };
 
 declare interface LockedAttachmentBaseProps {
@@ -312,6 +468,14 @@ declare interface LockedAttachmentBaseProps {
     detail?: string;
     amountText?: string;
     paymentStatus?: PaymentStatus;
+    /**
+     * When provided with 2+ items, the card renders as a mixed-media carousel
+     * (e.g. a couple of photos + a video) instead of a single thumbnail. Each
+     * item brings its own thumbnail and optional source so that
+     * `LockedAttachment.Composer` / `.Sent` / `.Received` can all share the
+     * same carousel chrome.
+     */
+    gallery?: LockedAttachmentGalleryItem[];
 }
 
 export declare interface LockedAttachmentContextValue {
@@ -321,6 +485,15 @@ export declare interface LockedAttachmentContextValue {
     onFetchSource?: (message: LocalMessage, channel: Channel) => Promise<LockedAttachmentSource | void>;
 }
 
+export declare interface LockedAttachmentGalleryItem {
+    /** MIME type of this carousel item — drives the per-item play / lock affordance. */
+    mimeType: string;
+    /** Poster image used for both the locked (blurred) and unlocked preview state. */
+    thumbnailUrl?: string;
+    /** Underlying source (image or video URL) — shown only when unlocked. */
+    sourceUrl?: string;
+}
+
 export declare interface LockedAttachmentSource {
     /** Proxied URL used by the media player for in-app playback. */
     sourceUrl: string;
@@ -501,10 +674,48 @@ export declare interface Participant {
  */
 declare type PaymentStatus = 'pending' | 'paid' | 'failed' | 'refunded';
 
+export declare interface ReceivedCardProps extends LockedAttachmentBaseProps {
+    /**
+     * Called when the recipient clicks Unlock on an unpaid attachment.
+     * Use this to open a checkout flow. Omit to hide the Unlock button.
+     */
+    onUnlockClick?: () => void;
+    /**
+     * Called to fetch the attachment source — fired automatically when
+     * `paymentStatus` transitions to `'paid'`, or immediately on click when
+     * `paymentStatus` is already `'paid'`. Return a `LockedAttachmentSource`
+     * to unlock the card.
+     */
+    onFetchSource?: () => Promise<LockedAttachmentSource | void>;
+    /**
+     * Called when the recipient clicks Download on an unlocked card.
+     * Omit to hide the Download button.
+     */
+    onDownloadClick?: () => void;
+    /**
+     * When true, shows a loading spinner on the Unlock button.
+     * Driven by the LockedAttachmentContext (e.g. checkout in progress).
+     */
+    isUnlocking?: boolean;
+}
+
 export declare function resolveLinkAttachment(message: LocalMessage): Attachment | undefined;
 
 export declare function resolveMediaFromMessage(message: LocalMessage): MediaMessageResolved | null;
 
+export declare interface SentCardProps extends LockedAttachmentBaseProps {
+    /** Placeholder shown in the title slot when no title is set. */
+    placeholderTitle?: string;
+    /** Fired the first time the sender taps the thumbnail to preview their own attachment. */
+    onPreviewClick?: () => void;
+    /**
+     * Lazily loads the underlying source so the sender can preview the attachment.
+     * Called the first time the thumbnail is tapped; the returned source is cached
+     * and reused on subsequent toggles.
+     */
+    onFetchSource?: () => Promise<LockedAttachmentSource | void>;
+}
+
 export declare function useCustomMessage<K extends keyof CustomMessageRegistry>(key: K): CustomMessageRegistry[K];
 
 /**
@@ -528,30 +739,8 @@ declare interface UseMessageVoteResult {
  */
 export declare const useMessaging: () => MessagingContextValue;
 
-export declare interface VisitorCardProps extends LockedAttachmentBaseProps {
-    /**
-     * Called when the visitor clicks Unlock on an unpaid attachment.
-     * Use this to open a checkout flow. Omit to hide the Unlock button.
-     */
-    onUnlockClick?: () => void;
-    /**
-     * Called to fetch the attachment source — fired automatically when
-     * paymentStatus transitions to 'paid', or immediately on click when
-     * paymentStatus is already 'paid'. Return a LockedAttachmentSource to
-     * unlock the card.
-     */
-    onFetchSource?: () => Promise<LockedAttachmentSource | void>;
-    /**
-     * Called when the visitor clicks Download on an unlocked card.
-     * Omit to hide the Download button.
-     */
-    onDownloadClick?: () => void;
-    /**
-     * When true, shows loading dots on the Unlock button.
-     * Driven by the LockedAttachmentContext (e.g. checkout in progress, payment processing).
-     */
-    isUnlocking?: boolean;
-}
+/** @deprecated Renamed to `ReceivedCardProps`. */
+export declare type VisitorCardProps = ReceivedCardProps;
 
 export declare type VoteSelection = 'up' | 'down' | null;
 

package/dist/index.js

@@ -1,26 +1,27 @@
-import { b as e, c as t, C as i, d as n, e as o, f as g, F as m, g as r, L as M, M as l, h as u, i as h, j as L, k as d, l as C, m as c, n as v, r as A, o as k, u as p, p as F, q as f } from "./index-Brz9orsI.js";
+import { a as e, b as t, C as n, c as i, d as o, e as g, F as m, f as M, L as r, h as l, M as h, i as u, j as L, k as c, l as d, m as C, n as v, o as A, p as k, q as p, u as F, s as f, t as q } from "./index-DfcRe-Hj.js";
 export {
   e as ActionButton,
   t as Avatar,
-  i as ChannelEmptyState,
-  n as ChannelList,
+  n as ChannelEmptyState,
+  i as ChannelList,
   o as ChannelView,
   g as CustomMessageProvider,
   m as FaqList,
-  r as FaqListItem,
-  M as LockedAttachment,
-  l as MediaMessage,
+  M as FaqListItem,
+  r as LinkAttachment,
+  l as LockedAttachment,
+  h as MediaMessage,
   u as MessageVoteButtons,
-  h as MessagingProvider,
-  L as MessagingShell,
+  L as MessagingProvider,
+  c as MessagingShell,
   d as formatRelativeTime,
   C as getMessageDisplayText,
-  c as isLinkAttachment,
-  v as normalizeLanguageCode,
-  A as resolveLinkAttachment,
-  k as resolveMediaFromMessage,
-  p as useCustomMessage,
-  F as useMessageVote,
-  f as useMessaging
+  v as isLinkAttachment,
+  A as normalizeLanguageCode,
+  k as resolveLinkAttachment,
+  p as resolveMediaFromMessage,
+  F as useCustomMessage,
+  f as useMessageVote,
+  q as useMessaging
 };
 //# sourceMappingURL=index.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@linktr.ee/messaging-react",
-  "version": "2.0.0",
+  "version": "2.0.1-rc-1778694826",
   "description": "React messaging components built on messaging-core for web applications",
   "type": "module",
   "main": "dist/index.js",

package/src/components/ChannelView.test.tsx

@@ -28,6 +28,17 @@ vi.mock('stream-chat-react', () => ({
   ),
   useMessageContext: () => ({ message: { id: 'message-1', text: 'hello' } }),
   useChannelStateContext: () => ({ channel: activeChannel }),
+  useChatContext: () => ({ client: { user: { id: 'visitor-1' } } }),
+  useTypingContext: () => ({ typing: {} }),
+  useAIState: () => ({ aiState: 'AI_STATE_IDLE' }),
+  AIStates: {
+    Error: 'AI_STATE_ERROR',
+    ExternalSources: 'AI_STATE_EXTERNAL_SOURCES',
+    Generating: 'AI_STATE_GENERATING',
+    Idle: 'AI_STATE_IDLE',
+    Stop: 'AI_STATE_STOP',
+    Thinking: 'AI_STATE_THINKING',
+  },
 }))
 
 vi.mock('../providers/MessagingProvider', () => ({

package/src/components/ChannelView.tsx

@@ -28,6 +28,7 @@ import { CustomMessage } from './CustomMessage'
 import { CustomMessageInput } from './CustomMessageInput'
 import { CustomSystemMessage } from './CustomSystemMessage'
 import CustomTypingIndicator from './CustomTypingIndicator'
+import { DmAgentEnabledContext } from './CustomTypingIndicator/DmAgentContext'
 import { ChannelEmptyState } from './MessagingShell/ChannelEmptyState'
 import { LoadingState } from './MessagingShell/LoadingState'
 
@@ -495,38 +496,40 @@ export const ChannelView = React.memo<ChannelViewProps>(
           className
         )}
       >
-        <Channel
-          channel={channel}
-          MessageSystem={CustomSystemMessage}
-          EmptyStateIndicator={CustomChannelEmptyState}
-          LoadingIndicator={LoadingState}
-          DateSeparator={CustomDateSeparator}
-          TypingIndicator={CustomTypingIndicator}
-          doSendMessageRequest={doSendMessageRequest}
-          {...(sendButton ? { SendButton: sendButton } : {})}
-        >
-          <ChannelViewInner
-            onBack={onBack}
-            showBackButton={showBackButton}
-            renderMessageInputActions={renderMessageInputActions}
-            renderConversationFooter={renderConversationFooter}
-            onLeaveConversation={onLeaveConversation}
-            onBlockParticipant={onBlockParticipant}
-            CustomChannelEmptyState={CustomChannelEmptyState}
-            showDeleteConversation={showDeleteConversation}
-            onDeleteConversationClick={onDeleteConversationClick}
-            onBlockParticipantClick={onBlockParticipantClick}
-            onReportParticipantClick={onReportParticipantClick}
-            showStarButton={showStarButton}
-            dmAgentEnabled={dmAgentEnabled}
-            chatbotVotingEnabled={chatbotVotingEnabled}
-            renderChannelBanner={renderChannelBanner}
-            customProfileContent={customProfileContent}
-            customChannelActions={customChannelActions}
-            renderMessage={renderMessage}
-            viewerLanguage={viewerLanguage}
-          />
-        </Channel>
+        <DmAgentEnabledContext.Provider value={dmAgentEnabled ?? false}>
+          <Channel
+            channel={channel}
+            MessageSystem={CustomSystemMessage}
+            EmptyStateIndicator={CustomChannelEmptyState}
+            LoadingIndicator={LoadingState}
+            DateSeparator={CustomDateSeparator}
+            TypingIndicator={CustomTypingIndicator}
+            doSendMessageRequest={doSendMessageRequest}
+            {...(sendButton ? { SendButton: sendButton } : {})}
+          >
+            <ChannelViewInner
+              onBack={onBack}
+              showBackButton={showBackButton}
+              renderMessageInputActions={renderMessageInputActions}
+              renderConversationFooter={renderConversationFooter}
+              onLeaveConversation={onLeaveConversation}
+              onBlockParticipant={onBlockParticipant}
+              CustomChannelEmptyState={CustomChannelEmptyState}
+              showDeleteConversation={showDeleteConversation}
+              onDeleteConversationClick={onDeleteConversationClick}
+              onBlockParticipantClick={onBlockParticipantClick}
+              onReportParticipantClick={onReportParticipantClick}
+              showStarButton={showStarButton}
+              dmAgentEnabled={dmAgentEnabled}
+              chatbotVotingEnabled={chatbotVotingEnabled}
+              renderChannelBanner={renderChannelBanner}
+              customProfileContent={customProfileContent}
+              customChannelActions={customChannelActions}
+              renderMessage={renderMessage}
+              viewerLanguage={viewerLanguage}
+            />
+          </Channel>
+        </DmAgentEnabledContext.Provider>
       </div>
     )
   }

package/src/components/CustomMessage/index.tsx

@@ -227,21 +227,20 @@ const CustomMessageWithContext = (props: CustomMessageWithContextProps) => {
           {isAttachment ? (
             <div className="str-chat__message-bubble-wrapper">
               {isMine ? (
-                <LockedAttachment.Creator
+                <LockedAttachment.Sent
                   title={message.metadata?.attachment_title}
                   mimeType={message.metadata?.attachment_mime_type}
                   thumbnailUrl={message.metadata?.attachment_thumbnail}
                   amountText={message.metadata?.amount_text}
                   detail={message.metadata?.attachment_detail}
                   paymentStatus={message.metadata?.payment_status}
-                  isUnlocking={isUnlocking(message.id)}
                   onPreviewClick={() => onUnlockClick?.(message, channel)}
                   onFetchSource={async () =>
                     await onFetchSource?.(message, channel)
                   }
                 />
               ) : (
-                <LockedAttachment.Visitor
+                <LockedAttachment.Received
                   title={message.metadata?.attachment_title}
                   mimeType={message.metadata?.attachment_mime_type}
                   thumbnailUrl={message.metadata?.attachment_thumbnail}

package/src/components/CustomTypingIndicator/CustomTypingIndicator.stories.tsx

@@ -2,16 +2,21 @@ import type { Meta, StoryFn } from '@storybook/react'
 import React from 'react'
 import type { Event } from 'stream-chat'
 import {
+  AIStates,
   ChannelStateProvider,
   ChatProvider,
   TypingProvider,
 } from 'stream-chat-react'
 
+import { DmAgentEnabledContext } from './DmAgentContext'
+
 import CustomTypingIndicator from '.'
 
 type StoryProps = {
   typingEventsEnabled?: boolean
   typing?: Record<string, Event>
+  aiState?: string
+  dmAgentEnabled?: boolean
 }
 
 const currentUser = {
@@ -33,10 +38,38 @@ const defaultTyping: Record<string, Event> = {
   } as Event,
 }
 
+type ListenerMap = Record<string, (event: { ai_state: string; cid: string }) => void>
+
+const createMockChannel = ({ aiState }: { aiState?: string }) => {
+  const listeners: ListenerMap = {}
+  const channel = {
+    cid: 'messaging:test',
+    state: {
+      members: {
+        [currentUser.id]: { user: currentUser },
+        [typingUser.id]: { user: typingUser },
+      },
+    },
+    on(eventType: string, handler: ListenerMap[string]) {
+      listeners[eventType] = handler
+      // Fire the requested AI state synchronously so the hook picks it up.
+      if (eventType === 'ai_indicator.update' && aiState) {
+        handler({ ai_state: aiState, cid: 'messaging:test' })
+      }
+      return { unsubscribe: () => {} }
+    },
+  }
+  return channel
+}
+
 const StoryWrapper: React.FC<StoryProps> = ({
   typingEventsEnabled = true,
-  typing = defaultTyping,
+  typing = {},
+  aiState,
+  dmAgentEnabled = true,
 }) => {
+  const channel = createMockChannel({ aiState })
+
   const chatContextValue = {
     client: {
       user: currentUser,
@@ -57,18 +90,7 @@ const StoryWrapper: React.FC<StoryProps> = ({
   }
 
   const channelStateValue = {
-    channel: {
-      state: {
-        members: {
-          [currentUser.id]: {
-            user: currentUser,
-          },
-          [typingUser.id]: {
-            user: typingUser,
-          },
-        },
-      },
-    },
+    channel,
     channelCapabilities: {},
     channelConfig: {
       typing_events: typingEventsEnabled,
@@ -85,9 +107,11 @@ const StoryWrapper: React.FC<StoryProps> = ({
     <ChatProvider value={chatContextValue as never}>
       <ChannelStateProvider value={channelStateValue as never}>
         <TypingProvider value={{ typing }}>
-          <div className="relative h-20 w-[200px] bg-[#f4f4f4] p-3">
-            <CustomTypingIndicator />
-          </div>
+          <DmAgentEnabledContext.Provider value={dmAgentEnabled}>
+            <div className="relative h-20 w-[200px] bg-[#f4f4f4] p-3">
+              <CustomTypingIndicator />
+            </div>
+          </DmAgentEnabledContext.Provider>
         </TypingProvider>
       </ChannelStateProvider>
     </ChatProvider>
@@ -104,7 +128,9 @@ const meta: Meta<StoryProps> = {
 export default meta
 
 export const Default: StoryFn<StoryProps> = (args) => <StoryWrapper {...args} />
-Default.args = {}
+Default.args = {
+  typing: defaultTyping,
+}
 
 export const HiddenWhenNoTyping: StoryFn<StoryProps> = (args) => (
   <StoryWrapper {...args} />
@@ -112,3 +138,17 @@ export const HiddenWhenNoTyping: StoryFn<StoryProps> = (args) => (
 HiddenWhenNoTyping.args = {
   typing: {},
 }
+
+export const AiAgentThinking: StoryFn<StoryProps> = (args) => (
+  <StoryWrapper {...args} />
+)
+AiAgentThinking.args = {
+  aiState: AIStates.Thinking,
+}
+
+export const AiAgentGenerating: StoryFn<StoryProps> = (args) => (
+  <StoryWrapper {...args} />
+)
+AiAgentGenerating.args = {
+  aiState: AIStates.Generating,
+}