feat: Add follow-up question trigger functionality in Chat components, allowing users to initiate immediate follow-up questions based on AI responses, enhancing interaction and engagement

Author: experdot

src/renderer/src/components/pages/chat/ChatInput.tsx

@@ -6,7 +6,8 @@ import {
   SettingOutlined,
   QuestionCircleOutlined,
   BulbOutlined,
-  DownOutlined
+  DownOutlined,
+  PlaySquareOutlined
 } from '@ant-design/icons'
 import { LLMConfig } from '../../../types/type'
 import { useSettingsStore } from '../../../stores/settingsStore'
@@ -166,6 +167,7 @@ interface ChatInputProps {
   autoQuestionMode?: 'ai' | 'preset'
   autoQuestionListId?: string
   onAutoQuestionChange?: (enabled: boolean, mode: 'ai' | 'preset', listId?: string) => void
+  onTriggerFollowUpQuestion?: () => Promise<void>
 }
 
 export interface ChatInputRef {
@@ -189,7 +191,8 @@ const ChatInput = forwardRef<ChatInputRef, ChatInputProps>(
       autoQuestionEnabled = false,
       autoQuestionMode = 'ai',
       autoQuestionListId,
-      onAutoQuestionChange
+      onAutoQuestionChange,
+      onTriggerFollowUpQuestion
     },
     ref
   ) => {
@@ -295,10 +298,32 @@ const ChatInput = forwardRef<ChatInputRef, ChatInputProps>(
                   mode={autoQuestionMode}
                   selectedListId={autoQuestionListId}
                   promptLists={settings.promptLists || []}
-                  disabled={disabled}
+                  disabled={false}
                   onChange={onAutoQuestionChange}
                 />
               </div>
+
+              {/* 立即追问按钮 */}
+              {autoQuestionEnabled && (
+                <Tooltip title={`立即触发一次${autoQuestionMode === 'ai' ? 'AI' : '预设列表'}追问`}>
+                  <Button
+                    type="text"
+                    size="small"
+                    icon={<PlaySquareOutlined />}
+                    onClick={async () => {
+                      try {
+                        await onTriggerFollowUpQuestion?.()
+                      } catch (error) {
+                        console.error('立即追问失败:', error)
+                      }
+                    }}
+                    disabled={disabled || loading}
+                    style={{ fontSize: '11px', color: '#1890ff' }}
+                  >
+                    立即追问
+                  </Button>
+                </Tooltip>
+              )}
             </Space>
 
             {/* 发送/停止按钮 */}

src/renderer/src/components/pages/chat/ChatLogic.tsx

@@ -22,6 +22,7 @@ interface ChatLogicProps {
     onToggleFavorite: (messageId: string) => void
     onModelChangeForMessage: (messageId: string, newModelId: string) => Promise<void>
     onDeleteMessage: (messageId: string) => Promise<void>
+    onTriggerFollowUpQuestion: () => Promise<void>
   }) => React.ReactNode
 }
 
@@ -77,8 +78,7 @@ export default function ChatLogic({
       taskType,
       taskContext,
       async (messageId: string) => {
-        // 只在普通聊天任务完成后触发自动提问
-        if (autoQuestionEnabled && taskType === 'chat') {
+        if (autoQuestionEnabled) {
           try {
             setTimeout(async () => {
               await autoQuestion.generateAndSendFollowUpQuestion(messageId)
@@ -112,6 +112,54 @@ export default function ChatLogic({
     sendMessageRef.current = messageOperations.handleSendMessage
   }, [messageOperations.handleSendMessage])
 
+  // 获取最新的AI消息ID
+  const getLatestAIMessageId = useCallback(() => {
+    if (!chat?.messages) return null
+    
+    // 获取当前路径的消息
+    const currentPath = chat.currentPath || []
+    let messages: any[] = []
+    
+    if (currentPath.length > 0) {
+      // 使用当前路径
+      messages = currentPath
+        .map((id: string) => chat.messages.find((msg: any) => msg.id === id))
+        .filter(Boolean)
+    } else {
+      // 使用消息树获取默认路径
+      messages = messageTree.getCurrentPathMessages()
+    }
+    
+    // 从后往前找最新的AI消息
+    for (let i = messages.length - 1; i >= 0; i--) {
+      if (messages[i].role === 'assistant') {
+        return messages[i].id
+      }
+    }
+    
+    return null
+  }, [chat, messageTree])
+
+  // 立即触发追问
+  const handleTriggerFollowUpQuestion = useCallback(async () => {
+    if (!autoQuestionEnabled) {
+      console.warn('自动追问功能未开启')
+      return
+    }
+    
+    const latestAIMessageId = getLatestAIMessageId()
+    if (!latestAIMessageId) {
+      console.warn('未找到AI消息，无法触发追问')
+      return
+    }
+    
+    try {
+      await autoQuestion.generateAndSendFollowUpQuestion(latestAIMessageId)
+    } catch (error) {
+      console.error('立即触发追问失败:', error)
+    }
+  }, [autoQuestionEnabled, getLatestAIMessageId, autoQuestion])
+
   const handleModelChange = (modelId: string) => {
     setSelectedModel(modelId)
   }
@@ -133,6 +181,7 @@ export default function ChatLogic({
     onEditAndResendMessage: messageOperations.handleEditAndResendMessage,
     onToggleFavorite: messageOperations.handleToggleFavorite,
     onModelChangeForMessage: messageOperations.handleModelChangeForMessage,
-    onDeleteMessage: messageOperations.handleDeleteMessage
+    onDeleteMessage: messageOperations.handleDeleteMessage,
+    onTriggerFollowUpQuestion: handleTriggerFollowUpQuestion
   })
 }

src/renderer/src/components/pages/chat/ChatWindow.tsx

@@ -180,7 +180,8 @@ const ChatWindow = forwardRef<ChatWindowRef, ChatWindowProps>(({ chatId }, ref)
             onEditAndResendMessage,
             onToggleFavorite,
             onModelChangeForMessage,
-            onDeleteMessage
+            onDeleteMessage,
+            onTriggerFollowUpQuestion
           }) => (
             <>
               {/* 消息树侧边栏 */}
@@ -196,7 +197,7 @@ const ChatWindow = forwardRef<ChatWindowRef, ChatWindowProps>(({ chatId }, ref)
               />
 
               {/* 聊天主内容区 */}
-              <div className="chat-main-content">
+              <div className="chat-content" style={{ flex: 1, display: 'flex', flexDirection: 'column' }}>
                 <ChatHeader
                   chatId={chatId}
                   chatTitle={chat.title}
@@ -228,7 +229,6 @@ const ChatWindow = forwardRef<ChatWindowRef, ChatWindowProps>(({ chatId }, ref)
                   onToggleMessageCollapse={handleToggleMessageCollapse}
                   // 设置相关props
                   onOpenSettings={handleOpenSettings}
-                  onDeleteMessage={onDeleteMessage}
                 />
                 <ChatInput
                   ref={chatInputRef}
@@ -253,6 +253,7 @@ const ChatWindow = forwardRef<ChatWindowRef, ChatWindowProps>(({ chatId }, ref)
                   autoQuestionMode={autoQuestionMode}
                   autoQuestionListId={autoQuestionListId}
                   onAutoQuestionChange={handleAutoQuestionChange}
+                  onTriggerFollowUpQuestion={onTriggerFollowUpQuestion}
                 />
               </div>
             </>