From 846da4042f7ad90a8413a8bd0eb44c9a4895cf69 Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Sun, 26 Oct 2025 11:17:57 +0800
Subject: [PATCH] Add MCP client loading in engine

- Integrated MCP client loading in both Load and Reload functions, enhancing the engine's capability to manage MCP configurations.
- Added error handling for MCP loading, ensuring warnings are generated for any issues encountered during the process.
- This update improves the overall robustness of the engine's loading mechanism and provides better feedback on MCP-related errors.
---
 engine/load.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/engine/load.go b/engine/load.go
index 7a6a82a8..137588ea 100644
--- a/engine/load.go
+++ b/engine/load.go
@@ -21,6 +21,7 @@ import (
 	"github.com/yaoapp/yao/fs"
 	"github.com/yaoapp/yao/i18n"
 	"github.com/yaoapp/yao/kb"
+	"github.com/yaoapp/yao/mcp"
 	"github.com/yaoapp/yao/messenger"
 	"github.com/yaoapp/yao/moapi"
 	"github.com/yaoapp/yao/model"
@@ -282,6 +283,13 @@ func Load(cfg config.Config, options LoadOption) (warnings []Warning, err error)
 		warnings = append(warnings, Warning{Widget: "Pipe", Error: err})
 	}
 
+	// Load MCP Clients
+	err = mcp.Load(cfg)
+	if err != nil {
+		// printErr(cfg.Mode, "MCP", err)
+		warnings = append(warnings, Warning{Widget: "MCP", Error: err})
+	}
+
 	// Load Knowledge Base
 	_, err = kb.Load(cfg)
 	if err != nil {
@@ -506,6 +514,12 @@ func Reload(cfg config.Config, options LoadOption) (err error) {
 		printErr(cfg.Mode, "AIGC", err)
 	}
 
+	// Load MCP Clients
+	err = mcp.Load(cfg)
+	if err != nil {
+		printErr(cfg.Mode, "MCP", err)
+	}
+
 	// Load Knowledge Base
 	_, err = kb.Load(cfg)
 	if err != nil {