From aa33f0242ab29164f5c00342f8be0bbeafb2adf5 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Sun, 9 Mar 2025 10:49:28 +0800
Subject: [PATCH] feat: Add support for enabled MCPs in message sending

---
 src/renderer/src/pages/home/Inputbar/Inputbar.tsx | 2 +-
 src/renderer/src/services/ApiService.ts           | 1 +
 src/renderer/src/store/messages.ts                | 4 ++++
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/pages/home/Inputbar/Inputbar.tsx b/src/renderer/src/pages/home/Inputbar/Inputbar.tsx
index e7d9a600..589ac1e6 100644
--- a/src/renderer/src/pages/home/Inputbar/Inputbar.tsx
+++ b/src/renderer/src/pages/home/Inputbar/Inputbar.tsx
@@ -158,7 +158,7 @@ const Inputbar: FC<Props> = ({ assistant: _assistant, setActiveTopic }) => {
     } catch (error) {
       console.error('Failed to send message:', error)
     }
-  }, [inputEmpty, text, assistant, files, selectedKnowledgeBases, mentionModels, dispatch])
+  }, [inputEmpty, files, dispatch, text, assistant, selectedKnowledgeBases, mentionModels, enabledMCPs])
 
   const translate = async () => {
     if (isTranslating) {
diff --git a/src/renderer/src/services/ApiService.ts b/src/renderer/src/services/ApiService.ts
index 537627df..00421d19 100644
--- a/src/renderer/src/services/ApiService.ts
+++ b/src/renderer/src/services/ApiService.ts
@@ -80,6 +80,7 @@ export async function fetchChatCompletion({
     }
 
     const allMCPTools = await window.api.mcp.listTools()
+
     await AI.completions({
       messages: filterUsefulMessages(messages),
       assistant,
diff --git a/src/renderer/src/store/messages.ts b/src/renderer/src/store/messages.ts
index bda9e35a..08223921 100644
--- a/src/renderer/src/store/messages.ts
+++ b/src/renderer/src/store/messages.ts
@@ -245,6 +245,10 @@ export const sendMessage =
         if (options?.mentionModels) {
           userMessage.mentions = options.mentionModels
         }
+
+        if (options?.enabledMCPs) {
+          userMessage.enabledMCPs = options.enabledMCPs
+        }
       }
 
       // 如果不是重发，才添加新的用户消息