fix: refactor to fix tests

Author: isekovanic

src/components/Chat/Chat.tsx

@@ -55,7 +55,7 @@ export const Chat = <
     defaultLanguage,
     i18nInstance,
     initialNavOpen = true,
-    isMessageAIGenerated = () => false,
+    isMessageAIGenerated,
     theme = 'messaging light',
     useImageFlagEmojisOnWindows = false,
   } = props;

src/components/Chat/hooks/useCreateChatContext.ts

@@ -60,6 +60,7 @@ export const useCreateChatContext = <
       getAppSettings,
       mutedUsersLength,
       navOpen,
+      isMessageAIGenerated,
     ],
   );
 

src/components/Message/MessageSimple.tsx

@@ -102,7 +102,7 @@ const MessageSimpleWithContext = <
   const showReplyCountButton = !threadList && !!message.reply_count;
   const allowRetry = message.status === 'failed' && message.errorStatusCode !== 403;
   const isBounced = isMessageBounced(message);
-  const isEdited = isMessageEdited(message) && !isMessageAIGenerated(message);
+  const isEdited = isMessageEdited(message) && !isMessageAIGenerated?.(message);
 
   let handleClick: (() => void) | undefined = undefined;
 
@@ -188,7 +188,7 @@ const MessageSimpleWithContext = <
               {message.attachments?.length && !message.quoted_message ? (
                 <Attachment actionHandler={handleAction} attachments={message.attachments} />
               ) : null}
-              {isMessageAIGenerated(message) ? (
+              {isMessageAIGenerated?.(message) ? (
                 <StreamedMessageText message={message} renderText={renderText} />
               ) : (
                 <MessageText message={message} renderText={renderText} />

src/context/ChatContext.tsx

@@ -56,7 +56,8 @@ export type ChatContextValue<
    */
   customClasses?: CustomClasses;
   navOpen?: boolean;
-} & Required<Pick<ChatProps<StreamChatGenerics>, 'theme' | 'client' | 'isMessageAIGenerated'>>;
+} & Partial<Pick<ChatProps<StreamChatGenerics>, 'isMessageAIGenerated'>> &
+  Required<Pick<ChatProps<StreamChatGenerics>, 'theme' | 'client'>>;
 
 export const ChatContext = React.createContext<ChatContextValue | undefined>(undefined);
 

src/context/MessageContext.tsx

@@ -68,10 +68,6 @@ export type MessageContextValue<
   handleReaction: (reactionType: string, event: React.BaseSyntheticEvent) => Promise<void>;
   /** Function to retry sending a Message */
   handleRetry: ChannelActionContextValue<StreamChatGenerics>['retrySendMessage'];
-  /**
-   * A factory function that determines whether a message is AI generated or not.
-   */
-  isMessageAIGenerated: (message: StreamMessage<StreamChatGenerics>) => boolean;
   /** Function that returns whether the Message belongs to the current user */
   isMyMessage: () => boolean;
   /** The message object */
@@ -110,6 +106,10 @@ export type MessageContextValue<
   highlighted?: boolean;
   /** Whether the threaded message is the first in the thread list */
   initialMessage?: boolean;
+  /**
+   * A factory function that determines whether a message is AI generated or not.
+   */
+  isMessageAIGenerated?: (message: StreamMessage<StreamChatGenerics>) => boolean;
   /** Latest message id on current channel */
   lastReceivedId?: string | null;
   /** DOMRect object for parent MessageList component */