From c129ca9862336c687668da17d937e2f2ded9c325 Mon Sep 17 00:00:00 2001
From: James Melvin <melvin@gameface.ai>
Date: Sun, 8 Jan 2023 10:20:38 +0530
Subject: [PATCH] fix: propagated message_id to FlaggableElement.tsx

---
 website/src/components/FlaggableElement.tsx | 2 +-
 website/src/components/Messages.tsx         | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/website/src/components/FlaggableElement.tsx b/website/src/components/FlaggableElement.tsx
index 82db4738..80d97a5b 100644
--- a/website/src/components/FlaggableElement.tsx
+++ b/website/src/components/FlaggableElement.tsx
@@ -42,7 +42,7 @@ export const FlaggableElement = (props) => {
         label_map.set(flag.attributeName, sliderValues[i]);
       }
     });
-    trigger({ post_id: props.post_id, label_map: Object.fromEntries(label_map), text: props.text });
+    trigger({ message_id: props.message_id, post_id: props.post_id, label_map: Object.fromEntries(label_map), text: props.text });
   };
   const [checkboxValues, setCheckboxValues] = useState(new Array(TEXT_LABEL_FLAGS.length).fill(false));
   const [sliderValues, setSliderValues] = useState(new Array(TEXT_LABEL_FLAGS.length).fill(1));
diff --git a/website/src/components/Messages.tsx b/website/src/components/Messages.tsx
index 7b69bc50..750d3946 100644
--- a/website/src/components/Messages.tsx
+++ b/website/src/components/Messages.tsx
@@ -7,14 +7,15 @@ import { FlaggableElement } from "./FlaggableElement";
 export interface Message {
   text: string;
   is_assistant: boolean;
+  message_id: string;
 }
 
 export const Messages = ({ messages, post_id }: { messages: Message[]; post_id: string }) => {
   const items = messages.map((messageProps: Message, i: number) => {
     const { text } = messageProps;
-
+    const { message_id } = messageProps;
     return (
-      <FlaggableElement text={text} post_id={post_id} key={i + text}>
+      <FlaggableElement text={text} post_id={post_id} message_id={message_id} key={i + text}>
         <MessageView {...messageProps} />
       </FlaggableElement>
     );