feat: MessagePrimitive.If hasContent (#975)

assistant-ui · Oct 12, 2024 · c85e77a · c85e77a
1 parent aaa4468
commit c85e77a
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/packages/react/src/primitive-hooks/message/useMessageIf.tsx b/packages/react/src/primitive-hooks/message/useMessageIf.tsx
@@ -15,6 +15,7 @@ type MessageIfFilters = {
   lastOrHover: boolean | undefined;
   speaking: boolean | undefined;
   hasAttachments: boolean | undefined;
+  hasContent: boolean | undefined;
   submittedFeedback: "positive" | "negative" | null | undefined;
 };
 export type UseMessageIfProps = RequireAtLeastOne<MessageIfFilters>;
@@ -26,7 +27,15 @@ export const useMessageIf = (props: UseMessageIfProps) => {
   return useCombinedStore(
     [messageStore, messageUtilsStore],
     (
-      { role, attachments, branchCount, isLast, speech, submittedFeedback },
+      {
+        role,
+        attachments,
+        content,
+        branchCount,
+        isLast,
+        speech,
+        submittedFeedback,
+      },
       { isCopied, isHovering },
     ) => {
       if (props.hasBranches === true && branchCount < 2) return false;
@@ -55,9 +64,12 @@ export const useMessageIf = (props: UseMessageIfProps) => {
       )
         return false;
 
+      if (props.hasContent === true && content.length === 0) return false;
+      if (props.hasContent === false && content.length > 0) return false;
+
       if (
         props.submittedFeedback !== undefined &&
-        (   submittedFeedback?.type ?? null) !== props.submittedFeedback
+        (submittedFeedback?.type ?? null) !== props.submittedFeedback
       )
         return false;