Fix current message

Signed-off-by: Lin Wang <[email protected]>

Author: wanglam

src/plugins/chat/public/components/chat_messages.tsx

@@ -54,20 +54,21 @@ export const ChatMessages: React.FC<ChatMessagesProps> = ({
 
   // Context is now handled by RFC hooks - no subscriptions needed
 
-  // Only show suggestion on llm outputs after last user input
-  const showSuggestions = useMemo(() => {
+  const lastAssistantMessageIndex = useMemo(
+    () => timeline.findLastIndex((message) => message.role === 'assistant'),
+    [timeline]
+  );
+  // Only enable suggestion on llm outputs after last user input
+  const suggestionsEnabled = useMemo(() => {
     if (isStreaming) {
       return false;
     }
     if (timeline.length === 0) {
       return false;
     }
-    const lastAssistantMessageIndex = timeline.findLastIndex(
-      (message) => message.role === 'assistant'
-    );
     const lastUserMessageIndex = timeline.findLastIndex((message) => message.role === 'user');
     return lastAssistantMessageIndex > lastUserMessageIndex;
-  }, [timeline, isStreaming]);
+  }, [timeline, isStreaming, lastAssistantMessageIndex]);
 
   return (
     <>
@@ -87,7 +88,7 @@ export const ChatMessages: React.FC<ChatMessagesProps> = ({
           </div>
         )}
 
-        {timeline.map((message) => {
+        {timeline.map((message, index) => {
           // Handle different message types
           if (message.role === 'user') {
             return <MessageRow key={message.id} message={message} onResend={onResendMessage} />;
@@ -117,7 +118,9 @@ export const ChatMessages: React.FC<ChatMessagesProps> = ({
                   <MessageRow message={assistantMsg} />
                 )}
 
-                {showSuggestions && <ChatSuggestions messages={timeline} />}
+                {suggestionsEnabled && lastAssistantMessageIndex === index && (
+                  <ChatSuggestions messages={timeline} currentMessage={message} />
+                )}
 
                 {/* Tool calls below the message */}
                 {assistantMsg.toolCalls?.map((toolCall) => {

src/plugins/chat/public/components/chat_suggestions.tsx

@@ -78,7 +78,13 @@ const SuggestionBubble: React.FC<SuggestionBubbleProps> = ({
   );
 };
 
-export const ChatSuggestions = ({ messages }: { messages: Message[] }) => {
+export const ChatSuggestions = ({
+  messages,
+  currentMessage,
+}: {
+  messages: Message[];
+  currentMessage: Message;
+}) => {
   const { suggestedActionsService, chatService } = useChatContext();
 
   const [customSuggestions, setCustomSuggestions] = useState<SuggestedActions[]>([]);
@@ -93,7 +99,7 @@ export const ChatSuggestions = ({ messages }: { messages: Message[] }) => {
         // Create ChatContext object from current chat state
         const context: ChatContext = {
           conversationId: chatService.getThreadId(),
-          currentMessage: messages[messages.length - 1],
+          currentMessage,
           messageHistory: messages,
         };
 
@@ -109,7 +115,7 @@ export const ChatSuggestions = ({ messages }: { messages: Message[] }) => {
     };
 
     loadCustomSuggestions();
-  }, [suggestedActionsService, chatService, messages]);
+  }, [suggestedActionsService, chatService, messages, currentMessage]);
 
   if (isLoadingCustomSuggestions || customSuggestions.length === 0) {
     return null;