feat: 工具部分逻辑

src/components/ChatEditor/index.tsx

@@ -287,13 +287,13 @@ const ChatEditorBase: React.FC<ChatEditorProps & WithAuthProps> = ({ checkAuth,
       {toolList && toolList.length > 0 && (
       {toolList && toolList.length > 0 && (
         <div className="absolute left-4 bottom-2 flex items-center gap-3 pointer-events-auto pl-[16px]">
         <div className="absolute left-4 bottom-2 flex items-center gap-3 pointer-events-auto pl-[16px]">
           {toolList.map((tool: any, index: number) => {
           {toolList.map((tool: any, index: number) => {
-            // index === 1 的按钮（通用模式）在默认状态下高亮
-            const isSelected = (selectedToolId === tool.toolId && !isToolBtnActive) || (index === 1 && isToolBtnActive)
+            // tool.toolName === '通用模式' 的按钮（通用模式）在默认状态下高亮
+            const isSelected = (selectedToolId === tool.toolId && !isToolBtnActive) || (tool.toolName === '通用模式' && isToolBtnActive)
             const handleButtonPress = async () => {
             const handleButtonPress = async () => {
               // 高亮状态直接返回，避免重复触发
               // 高亮状态直接返回，避免重复触发
               if (isSelected)
               if (isSelected)
                 return
                 return
-              if (index === 1)
+              if (tool.toolName === '通用模式')
                 await handleGeneralClick()
                 await handleGeneralClick()
               else
               else
                 await handleToolClick(tool)
                 await handleToolClick(tool)

src/layouts/HistoryBar/components/HistoryBarList/index.tsx

@@ -24,7 +24,6 @@ export const HistoryBarList: React.FC<HistoryBarListProps> = ({ searchValue, onS
     if (isMobile()) {
     if (isMobile()) {
       onSetHistoryVisible(false)
       onSetHistoryVisible(false)
     }
     }
-
     // 直接导航到历史记录，不设置shouldSendQuestion
     // 直接导航到历史记录，不设置shouldSendQuestion
     navigate(`/chat/${conversation.conversationId}`)
     navigate(`/chat/${conversation.conversationId}`)
   }
   }

src/pages/Chat/Chat.tsx

@@ -233,6 +233,14 @@ export const Chat: React.FC = () => {
         return item
         return item
       })
       })
       setAllItems(processedMessages)
       setAllItems(processedMessages)
+      const latestToolRecord = [...processedMessages].reverse().find(item => item.role === 'ai' && item.toolId !== undefined)
+      const trimmedToolId = latestToolRecord?.toolId?.trim?.()
+      if (trimmedToolId) {
+        dispatch(setCurrentToolId(trimmedToolId))
+      }
+      else {
+        dispatch(clearCurrentToolId())
+      }
     }
     }
  catch {
  catch {
       // 错误处理
       // 错误处理
@@ -240,7 +248,7 @@ export const Chat: React.FC = () => {
  finally {
  finally {
       setIsLoading(false)
       setIsLoading(false)
     }
     }
-  }, [])
+  }, [dispatch])
 
 
   /** 点击滚动到底部 */
   /** 点击滚动到底部 */
   const scrollToBottom = () => {
   const scrollToBottom = () => {

src/pages/Chat/helper.ts

@@ -18,6 +18,7 @@ export function processApiResponse(data: OriginalRecord[]): ChatRecord[] {
         role: 'ai',
         role: 'ai',
         answerList: record.answerList,
         answerList: record.answerList,
         question: record.question,
         question: record.question,
+        toolId: record.toolId,
       })
       })
     }
     }
   })
   })

src/types/chat.ts

@@ -45,6 +45,7 @@ export interface Answer {
   conversationId?: string
   conversationId?: string
   terminateFlag?: boolean
   terminateFlag?: boolean
   toolName?: string
   toolName?: string
+  toolId?: string
   cardList: Attachment[]
   cardList: Attachment[]
   step?: AnswerStep
   step?: AnswerStep
 }
 }
@@ -56,6 +57,7 @@ export interface OriginalRecord {
   answerList: Answer[]
   answerList: Answer[]
   productCode?: string
   productCode?: string
   qaTime?: string
   qaTime?: string
+  toolId?: string
 }
 }
 
 
 export type ChatRecordRole = 'system' | 'user' | 'ai'
 export type ChatRecordRole = 'system' | 'user' | 'ai'