From 28ca7f18513ca76e686bf6f600b5a7961b31b588 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=85=AC=E6=98=8E?= <83812544+Ed1s0nZ@users.noreply.github.com> Date: Fri, 9 Jan 2026 18:52:38 +0800 Subject: [PATCH] Add files via upload --- internal/agent/agent.go | 3 +- internal/handler/config.go | 6 +- internal/mcp/client.go | 128 +++++++++++++++++++++++++++++++ internal/mcp/external_manager.go | 6 +- 4 files changed, 138 insertions(+), 5 deletions(-) diff --git a/internal/agent/agent.go b/internal/agent/agent.go index 29c3db85..a319890c 100644 --- a/internal/agent/agent.go +++ b/internal/agent/agent.go @@ -865,7 +865,8 @@ func (a *Agent) getAvailableTools() []Tool { // 获取外部MCP工具 if a.externalMCPMgr != nil { - ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) + // 增加超时时间到30秒，因为通过代理连接远程服务器可能需要更长时间 + ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second) defer cancel() externalTools, err := a.externalMCPMgr.GetAllTools(ctx) diff --git a/internal/handler/config.go b/internal/handler/config.go index 93233de7..0176c360 100644 --- a/internal/handler/config.go +++ b/internal/handler/config.go @@ -203,7 +203,8 @@ func (h *ConfigHandler) GetConfig(c *gin.Context) { // 获取外部MCP工具 if h.externalMCPMgr != nil { - ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) + // 增加超时时间到30秒，因为通过代理连接远程服务器可能需要更长时间 + ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second) defer cancel() externalTools, err := h.externalMCPMgr.GetAllTools(ctx) @@ -375,7 +376,8 @@ func (h *ConfigHandler) GetTools(c *gin.Context) { // 获取外部MCP工具 if h.externalMCPMgr != nil { - ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) + // 增加超时时间到30秒，因为通过代理连接远程服务器可能需要更长时间 + ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second) defer cancel() externalTools, err := h.externalMCPMgr.GetAllTools(ctx) diff --git a/internal/mcp/client.go b/internal/mcp/client.go index d3571d40..36514ae4 100644 --- a/internal/mcp/client.go +++ b/internal/mcp/client.go @@ -102,6 +102,20 @@ func (c *HTTPMCPClient) Initialize(ctx context.Context) error { return fmt.Errorf("初始化失败: %w", err) } + // 发送 initialized 通知（MCP 协议要求：收到 initialize 响应后必须发送此通知） + notifyReq := Message{ + ID: MessageID{value: nil}, // 通知没有 ID + Method: "notifications/initialized", + Version: "2.0", + } + notifyReq.Params = json.RawMessage("{}") + + // 发送通知（不需要等待响应） + if err := c.sendNotification(¬ifyReq); err != nil { + c.logger.Warn("发送 initialized 通知失败", zap.Error(err)) + // 通知失败不应该导致初始化失败，只记录警告 + } + c.setStatus("connected") return nil } @@ -195,6 +209,34 @@ func (c *HTTPMCPClient) sendRequest(ctx context.Context, msg *Message) (*Message return &mcpResp, nil } +func (c *HTTPMCPClient) sendNotification(msg *Message) error { + // 通知没有 ID，不需要等待响应 + body, err := json.Marshal(msg) + if err != nil { + return fmt.Errorf("序列化通知失败: %w", err) + } + + // 使用较短的超时发送通知 + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + httpReq, err := http.NewRequestWithContext(ctx, http.MethodPost, c.url, bytes.NewReader(body)) + if err != nil { + return fmt.Errorf("创建HTTP请求失败: %w", err) + } + + httpReq.Header.Set("Content-Type", "application/json") + + // 发送通知，不等待响应（通知不需要响应） + resp, err := c.client.Do(httpReq) + if err != nil { + return fmt.Errorf("发送通知失败: %w", err) + } + resp.Body.Close() + + return nil +} + func (c *HTTPMCPClient) Close() error { c.setStatus("disconnected") return nil @@ -291,6 +333,20 @@ func (c *StdioMCPClient) Initialize(ctx context.Context) error { return fmt.Errorf("初始化失败: %w", err) } + // 发送 initialized 通知（MCP 协议要求：收到 initialize 响应后必须发送此通知） + notifyReq := Message{ + ID: MessageID{value: nil}, // 通知没有 ID + Method: "notifications/initialized", + Version: "2.0", + } + notifyReq.Params = json.RawMessage("{}") + + // 发送通知（不需要等待响应） + if err := c.sendNotification(¬ifyReq); err != nil { + c.logger.Warn("发送 initialized 通知失败", zap.Error(err)) + // 通知失败不应该导致初始化失败，只记录警告 + } + c.setStatus("connected") return nil } @@ -426,6 +482,20 @@ func (c *StdioMCPClient) ListTools(ctx context.Context) ([]Tool, error) { return listResp.Tools, nil } +func (c *StdioMCPClient) sendNotification(msg *Message) error { + // 通知没有 ID，不需要等待响应 + if c.encoder == nil { + return fmt.Errorf("进程未启动") + } + + // 直接发送通知，不等待响应 + if err := c.encoder.Encode(msg); err != nil { + return fmt.Errorf("发送通知失败: %w", err) + } + + return nil +} + func (c *StdioMCPClient) CallTool(ctx context.Context, name string, args map[string]interface{}) (*ToolResult, error) { req := Message{ ID: MessageID{value: uuid.New().String()}, @@ -563,6 +633,20 @@ func (c *SSEMCPClient) Initialize(ctx context.Context) error { return fmt.Errorf("初始化失败: %w", err) } + // 发送 initialized 通知（MCP 协议要求：收到 initialize 响应后必须发送此通知） + notifyReq := Message{ + ID: MessageID{value: nil}, // 通知没有 ID + Method: "notifications/initialized", + Version: "2.0", + } + notifyReq.Params = json.RawMessage("{}") + + // 发送通知（不需要等待响应） + if err := c.sendNotification(¬ifyReq); err != nil { + c.logger.Warn("发送 initialized 通知失败", zap.Error(err)) + // 通知失败不应该导致初始化失败，只记录警告 + } + c.setStatus("connected") return nil } @@ -836,6 +920,50 @@ func (c *SSEMCPClient) ListTools(ctx context.Context) ([]Tool, error) { return listResp.Tools, nil } +func (c *SSEMCPClient) sendNotification(msg *Message) error { + // 通知没有 ID，不需要等待响应 + if c.sseConn == nil { + return fmt.Errorf("SSE连接未建立") + } + + body, err := json.Marshal(msg) + if err != nil { + return fmt.Errorf("序列化通知失败: %w", err) + } + + // 使用 POST 发送通知（与 sendRequest 类似的逻辑） + postURL := c.url + if strings.HasSuffix(postURL, "/sse") { + postURL = strings.TrimSuffix(postURL, "/sse") + postURL += "/message" + } else if strings.HasSuffix(postURL, "/events") { + postURL = strings.TrimSuffix(postURL, "/events") + postURL += "/message" + } else if !strings.Contains(postURL, "/message") { + postURL = strings.TrimSuffix(postURL, "/") + postURL += "/message" + } + + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + httpReq, err := http.NewRequestWithContext(ctx, http.MethodPost, postURL, bytes.NewReader(body)) + if err != nil { + return fmt.Errorf("创建POST请求失败: %w", err) + } + + httpReq.Header.Set("Content-Type", "application/json") + + // 发送通知，不等待响应（通知不需要响应） + resp, err := c.client.Do(httpReq) + if err != nil { + return fmt.Errorf("发送通知失败: %w", err) + } + resp.Body.Close() + + return nil +} + func (c *SSEMCPClient) CallTool(ctx context.Context, name string, args map[string]interface{}) (*ToolResult, error) { req := Message{ ID: MessageID{value: uuid.New().String()}, diff --git a/internal/mcp/external_manager.go b/internal/mcp/external_manager.go index 58d094c4..b59745dc 100644 --- a/internal/mcp/external_manager.go +++ b/internal/mcp/external_manager.go @@ -543,7 +543,8 @@ func (m *ExternalMCPManager) GetToolCount(name string) (int, error) { return 0, nil } - ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + // 增加超时时间到30秒，因为通过代理连接远程服务器可能需要更长时间 + ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second) defer cancel() tools, err := client.ListTools(ctx) @@ -570,7 +571,8 @@ func (m *ExternalMCPManager) GetToolCounts() map[string]int { continue } - ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + // 增加超时时间到30秒，因为通过代理连接远程服务器可能需要更长时间 + ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second) tools, err := client.ListTools(ctx) cancel()