feat: enhance MCP server connection management with heartbeat and retry logic

Author: michiosw

internal/mcp/config.go

@@ -2,6 +2,8 @@
 package mcp
 
 import (
+	"time"
+
 	"github.com/co-browser/agent-browser/internal/log"
 	"go.uber.org/fx"
 )
@@ -15,8 +17,12 @@ type RemoteMCPServer struct {
 
 // MCPServerConfig holds the configuration for the MCP server
 type MCPServerConfig struct {
-	RemoteServers []RemoteMCPServer `json:"remote_servers"`
-	Port          int               `json:"port" default:"8087"`
+	RemoteServers        []RemoteMCPServer `json:"remote_servers"`
+	Port                 int               `json:"port" default:"8087"`
+	HeartbeatInterval    time.Duration     `json:"heartbeat_interval" default:"15s"`
+	HealthCheckInterval  time.Duration     `json:"health_check_interval" default:"30s"`
+	ConnectionTimeout    time.Duration     `json:"connection_timeout" default:"5s"`
+	MaxReconnectAttempts int               `json:"max_reconnect_attempts" default:"10"`
 }
 
 // ConfigParams contains the parameters needed for configuration
@@ -35,9 +41,12 @@ type ConfigResult struct {
 
 // NewMCPConfig creates a new MCP configuration
 func NewMCPConfig(p ConfigParams) (ConfigResult, error) {
-	// TODO: Load from environment or config file
 	config := MCPServerConfig{
-		Port: 8087,
+		Port:                 8087,
+		HeartbeatInterval:    15 * time.Second,
+		HealthCheckInterval:  30 * time.Second,
+		ConnectionTimeout:    5 * time.Second,
+		MaxReconnectAttempts: 10,
 		RemoteServers: []RemoteMCPServer{
 			{
 				URL:         "http://0.0.0.0:8001/sse",
@@ -54,6 +63,9 @@ func NewMCPConfig(p ConfigParams) (ConfigResult, error) {
 
 	p.Logger.Info().
 		Int("port", config.Port).
+		Dur("heartbeat", config.HeartbeatInterval).
+		Dur("health_check", config.HealthCheckInterval).
+		Int("max_reconnect", config.MaxReconnectAttempts).
 		Int("remote_servers", len(config.RemoteServers)).
 		Msg("MCP configuration loaded")