display feedback state when loading a past conversation (#41)

* feat: display feedback state when loading a past conversation Signed-off-by: tygao <[email protected]> * update additional_info type Signed-off-by: tygao <[email protected]> * update interface Signed-off-by: tygao <[email protected]> --------- Signed-off-by: tygao <[email protected]>
opensearch-project · Dec 6, 2023 · 36ac34b · 36ac34b
1 parent 40838c6
commit 36ac34b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 11 deletions.
diff --git a/common/types/chat_saved_object_attributes.ts b/common/types/chat_saved_object_attributes.ts
@@ -12,7 +12,7 @@ export interface Interaction {
   conversation_id: string;
   interaction_id: string;
   create_time: string;
-  additional_info?: Record<string, unknown>;
+  additional_info?: { feedback?: SendFeedbackBody; [key: string]: unknown };
   parent_interaction_id?: string;
 }
 
@@ -64,3 +64,6 @@ export type ISuggestedAction = ISuggestedActionBase &
         metadata: { traceId: string; icon: string };
       }
   );
+export interface SendFeedbackBody {
+  satisfaction: boolean;
+}
diff --git a/public/hooks/use_feed_back.tsx b/public/hooks/use_feed_back.tsx
@@ -5,20 +5,17 @@
 
 import { useState } from 'react';
 import { ASSISTANT_API } from '../../common/constants/llm';
-import { IOutput } from '../../common/types/chat_saved_object_attributes';
-import { useChatContext } from '../contexts/chat_context';
+import { IOutput, Interaction } from '../../common/types/chat_saved_object_attributes';
 import { useCore } from '../contexts/core_context';
 import { useChatState } from './use_chat_state';
+import { SendFeedbackBody } from '../../common/types/chat_saved_object_attributes';
 
-interface SendFeedbackBody {
-  satisfaction: boolean;
-}
-
-export const useFeedback = () => {
-  const chatContext = useChatContext();
+export const useFeedback = (interaction?: Interaction | null) => {
   const core = useCore();
   const { chatState } = useChatState();
-  const [feedbackResult, setFeedbackResult] = useState<undefined | boolean>(undefined);
+  const [feedbackResult, setFeedbackResult] = useState<undefined | boolean>(
+    interaction?.additional_info?.feedback?.satisfaction ?? undefined
+  );
 
   const sendFeedback = async (message: IOutput, correct: boolean) => {
     const outputMessage = message;

diff --git a/public/tabs/chat/messages/message_bubble.tsx b/public/tabs/chat/messages/message_bubble.tsx
@@ -42,7 +42,9 @@ type MessageBubbleProps = {
 );
 
 export const MessageBubble: React.FC<MessageBubbleProps> = React.memo((props) => {
-  const { feedbackResult, sendFeedback } = useFeedback();
+  const { feedbackResult, sendFeedback } = useFeedback(
+    'interaction' in props ? props.interaction : null
+  );
 
   // According to the design of the feedback, only markdown type output is supported.
   const showFeedback =