Function main

examples/mcp/main.go:21–146

()

)

func main() {
	fmt.Println("=== Agent SDK - MCP 集成示例 ===")

	// 1. 创建工具注册表
	toolRegistry := tools.NewRegistry()

	// 2. 注册内置工具
	builtin.RegisterAll(toolRegistry)

	// 3. 创建 MCP Manager 并添加 MCP Server
	mcpManager := mcp.NewMCPManager(toolRegistry)

	// 示例: 添加一个 MCP Server (需要替换为实际的 MCP Server 地址)
	mcpEndpoint := os.Getenv("MCP_ENDPOINT")
	if mcpEndpoint == "" {
		mcpEndpoint = "http://localhost:8080/mcp" // 默认地址
	}

	mcpAccessKey := os.Getenv("MCP_ACCESS_KEY")
	mcpSecretKey := os.Getenv("MCP_SECRET_KEY")

	fmt.Printf("配置 MCP Server:\n")
	fmt.Printf("  Endpoint: %s\n", mcpEndpoint)
	fmt.Printf("  Access Key: %s\n\n", maskString(mcpAccessKey))

	// 添加 MCP Server
	server, err := mcpManager.AddServer(&mcp.MCPServerConfig{
		ServerID:        "my-mcp-server",
		Endpoint:        mcpEndpoint,
		AccessKeyID:     mcpAccessKey,
		AccessKeySecret: mcpSecretKey,
	})

	if err != nil {
		log.Printf("⚠️  添加 MCP Server 失败: %v", err)
		log.Println("   提示: 如果没有实际的 MCP Server，这是正常的")
		log.Println("   继续使用内置工具运行示例...")
	} else {
		fmt.Printf("✓ MCP Server 已添加: %s\n", server.GetServerID())

		// 4. 连接到 MCP Server 并注册工具
		ctx := context.Background()
		if err := mcpManager.ConnectServer(ctx, "my-mcp-server"); err != nil {
			log.Printf("⚠️  连接 MCP Server 失败: %v\n", err)
			log.Println("   提示: 确保 MCP Server 正在运行并且可访问")
			log.Println("   继续使用内置工具运行示例...")
		} else {
			fmt.Printf("✓ 已连接到 MCP Server\n")
			fmt.Printf("  发现工具数量: %d\n\n", server.GetToolCount())

			// 列出所有 MCP 工具
			tools := server.ListTools()
			if len(tools) > 0 {
				fmt.Println("可用的 MCP 工具:")
				for i, tool := range tools {
					fmt.Printf("  %d. %s - %s\n", i+1, tool.Name, tool.Description)
				}
				fmt.Println()