Merge pull request #320 from SmythOS/fix/chat-stopped-request-response

Chat Interface – Fixed Response to Previous Stopped Request Instead of New Message

Author: hamza-smythos

packages/app/src/react/features/ai-chat/components/system-message.tsx

@@ -34,6 +34,8 @@ export const SystemMessage: FC<ISystemMessageProps> = (props) => {
 
   const contentRef = useRef<HTMLDivElement>(null);
 
+  if (!message) return null; // if message is empty, return null
+
   return (
     <div className="system-message-bubble relative">
       {isError ? (

packages/app/src/react/features/ai-chat/hooks/use-chat.ts

@@ -308,6 +308,23 @@ export const useChat = (config: IUseChatConfig): IUseChatReturn => {
         setIsProcessing(true);
         clearError();
 
+        // Remove any previous error messages and loading messages before sending new message
+        setMessages((prev) => {
+          // Check if there are any error messages (indicates stopped generation)
+          const hasErrorMessages = prev.some((msg) => msg.type === 'error');
+
+          // Remove error messages
+          const filtered = prev.filter((msg) => msg.type !== 'error');
+
+          // If there were error messages and the last message is loading,
+          if (hasErrorMessages && filtered.length > 0) {
+            const lastMessage = filtered[filtered.length - 1];
+            if (lastMessage.type === 'loading') return filtered.slice(0, -1);
+          }
+
+          return filtered;
+        });
+
         // Upload files if present
         let attachments: IFileAttachment[] = [];
         if (files && files.length > 0) {