🎨 更新项目版本

mcp/client/client.go

@@ -0,0 +1,39 @@
+package client
+
+import (
+	"context"
+	"errors"
+	mcpClient "github.com/mark3labs/mcp-go/client"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+func NewClient(baseUrl, name, version, serverName string) (*mcpClient.Client, error) {
+	client, err := mcpClient.NewSSEMCPClient(baseUrl)
+	if err != nil {
+		return nil, err
+	}
+
+	ctx := context.Background()
+
+	// 启动client
+	if err := client.Start(ctx); err != nil {
+		return nil, err
+	}
+
+	// 初始化
+	initRequest := mcp.InitializeRequest{}
+	initRequest.Params.ProtocolVersion = mcp.LATEST_PROTOCOL_VERSION
+	initRequest.Params.ClientInfo = mcp.Implementation{
+		Name:    name,
+		Version: version,
+	}
+
+	result, err := client.Initialize(ctx, initRequest)
+	if err != nil {
+		return nil, err
+	}
+	if result.ServerInfo.Name != serverName {
+		return nil, errors.New("server name mismatch")
+	}
+	return client, nil
+}