Fix: Regenerate AI Chat Messages

Author: ShenKSPZ

src/components/chat/ChatMessageArea.tsx

@@ -11,7 +11,7 @@ interface ChatMessageAreaProps {
   error: string | null;
   onSendMessage: (content: string) => void;
   onStopStreaming?: () => void;
-  onRegenerateResponse?: () => void;
+  onRegenerateResponse?: (messageId: string) => void;
   onEditMessage?: (messageId: string, newContent: string) => void;
   isCurrentlyStreaming?: boolean;
 }
@@ -62,9 +62,9 @@ export const ChatMessageArea: React.FC<ChatMessageAreaProps> = ({
   };
 
   // Handle regenerate response
-  const handleRegenerateResponse = () => {
+  const handleRegenerateResponse = (messageId: string) => {
     if (onRegenerateResponse) {
-      onRegenerateResponse();
+      onRegenerateResponse(messageId);
     } else {
       console.error('Regenerate response function not provided');
     }
@@ -165,7 +165,7 @@ export const ChatMessageArea: React.FC<ChatMessageAreaProps> = ({
                   id: 'regenerate',
                   icon: RotateCcw,
                   label: 'Regenerate',
-                  onClick: () => handleRegenerateResponse(),
+                  onClick: () => handleRegenerateResponse(message.messageId),
                 }
               ];
 

src/pages/ChatPage.tsx

@@ -230,14 +230,19 @@ export const ChatPage: React.FC<ChatPageProps> = ({
   };
 
   // Handle regenerating the last AI response
-  const handleRegenerateResponse = async () => {
-    if (!isServiceInitialized || !chatServiceRef.current) return;
+  const handleRegenerateResponse = async (messageId: string) => {
+    if (!activeConversationId || !isServiceInitialized || !chatServiceRef.current) return;
 
     try {
       // Use the new regenerateLastMessage method
-      await chatServiceRef.current.regenerateLastMessage((updatedConversation) => {
-        setConversations(updatedConversation);
-      });
+      await chatServiceRef.current.regenerateAiMessage(
+        messageId, 
+        activeConversationId,
+        true,
+        (updatedConversation) => {
+          setConversations(updatedConversation);
+        }
+      );
     } catch (error) {
       console.error('Error regenerating response:', error);
     }

src/pages/SettingsPage.tsx

@@ -20,7 +20,7 @@ export const SettingsPage: React.FC<SettingsPageProps> = ({
   onSave,
 }) => {
   const [activeTab, setActiveTab] = useState<SettingsTab>('api');
-  const [selectedProvider, setSelectedProvider] = useState<AIProvider>('OpenAI');
+  const [selectedProvider, setSelectedProvider] = useState<AIProvider>('TensorBlock');
   const [providerSettings, setProviderSettings] = useState<Record<string, ProviderSettings>>({});
   const [selectedModel, setSelectedModel] = useState('');
   const [useStreaming, setUseStreaming] = useState(true);