mcp-proxy-adapter 6.3.4__py3-none-any.whl → 6.3.5__py3-none-any.whl

mcp_proxy_adapter/examples/scripts/config_generator.py

@@ -20,15 +20,15 @@ logger = logging.getLogger(__name__)
 class ConfigGenerator:
     """
     Configuration generator for unified mcp_proxy_adapter and mcp_security_framework configs.
-    
+
     Generates comprehensive configuration files with detailed comments and examples
     for both the proxy adapter and security framework components.
     """
-    
+
     def __init__(self):
         """Initialize configuration generator."""
         self.template_config = self._get_template_config()
-    
+
     def _get_template_config(self) -> Dict[str, Any]:
         """Get template configuration with all available options."""
         return {
@@ -38,7 +38,7 @@ class ConfigGenerator:
                 "debug": False,
                 "log_level": "INFO",
                 "workers": 1,
-                "reload": False
+                "reload": False,
             },
             "ssl": {
                 "enabled": False,
@@ -47,9 +47,12 @@ class ConfigGenerator:
                 "ca_cert": None,
                 "verify_client": False,
                 "client_cert_required": False,
-                "cipher_suites": ["TLS_AES_256_GCM_SHA384", "TLS_CHACHA20_POLY1305_SHA256"],
+                "cipher_suites": [
+                    "TLS_AES_256_GCM_SHA384",
+                    "TLS_CHACHA20_POLY1305_SHA256",
+                ],
                 "min_tls_version": "TLSv1.2",
-                "max_tls_version": "1.3"
+                "max_tls_version": "1.3",
             },
             "security": {
                 "framework": "mcp_security_framework",
@@ -57,7 +60,6 @@ class ConfigGenerator:
                 "debug": False,
                 "environment": "dev",
                 "version": "1.0.0",
-                
                 "ssl": {
                     "enabled": False,
                     "cert_file": None,
@@ -71,9 +73,8 @@ class ConfigGenerator:
                     "cipher_suite": None,
                     "check_hostname": True,
                     "check_expiry": True,
-                    "expiry_warning_days": 30
+                    "expiry_warning_days": 30,
                 },
-                
                 "auth": {
                     "enabled": False,
                     "methods": [],
@@ -92,10 +93,9 @@ class ConfigGenerator:
                         "X-Content-Type-Options": "nosniff",
                         "X-Frame-Options": "DENY",
                         "X-XSS-Protection": "1; mode=block",
-                        "Strict-Transport-Security": "max-age=31536000; includeSubDomains"
-                    }
+                        "Strict-Transport-Security": "max-age=31536000; includeSubDomains",
+                    },
                 },
-                
                 "certificates": {
                     "enabled": False,
                     "ca_cert_path": None,
@@ -110,9 +110,8 @@ class ConfigGenerator:
                     "crl_url": None,
                     "crl_validity_days": 30,
                     "auto_renewal": False,
-                    "renewal_threshold_days": 30
+                    "renewal_threshold_days": 30,
                 },
-                
                 "permissions": {
                     "enabled": False,
                     "roles_file": None,
@@ -123,9 +122,8 @@ class ConfigGenerator:
                     "permission_cache_ttl": 300,
                     "wildcard_permissions": False,
                     "strict_mode": False,
-                    "roles": {}
+                    "roles": {},
                 },
-                
                 "rate_limit": {
                     "enabled": False,
                     "default_requests_per_minute": 60,
@@ -136,9 +134,8 @@ class ConfigGenerator:
                     "redis_config": None,
                     "cleanup_interval": 300,
                     "exempt_paths": ["/health", "/docs", "/openapi.json"],
-                    "exempt_roles": ["admin"]
+                    "exempt_roles": ["admin"],
                 },
-                
                 "logging": {
                     "enabled": True,
                     "level": "INFO",
@@ -151,10 +148,9 @@ class ConfigGenerator:
                     "json_format": False,
                     "include_timestamp": True,
                     "include_level": True,
-                    "include_module": True
-                }
+                    "include_module": True,
+                },
             },
-            
             "registration": {
                 "enabled": False,
                 "server_url": "https://proxy-registry.example.com",
@@ -164,18 +160,18 @@ class ConfigGenerator:
                     "cert_file": "mcp_proxy_adapter/examples/certs/proxy_client.crt",
                     "key_file": "mcp_proxy_adapter/examples/keys/proxy_client.key",
                     "ca_cert_file": "mcp_proxy_adapter/examples/certs/ca.crt",
-                    "verify_server": True
+                    "verify_server": True,
                 },
                 "token": {
                     "enabled": False,
                     "token": "proxy_registration_token_123",
                     "token_type": "bearer",
-                    "refresh_interval": 3600
+                    "refresh_interval": 3600,
                 },
                 "api_key": {
                     "enabled": False,
                     "key": "proxy_api_key_456",
-                    "key_header": "X-Proxy-API-Key"
+                    "key_header": "X-Proxy-API-Key",
                 },
                 "proxy_info": {
                     "name": "mcp_proxy_adapter",
@@ -185,24 +181,23 @@ class ConfigGenerator:
                     "endpoints": {
                         "jsonrpc": "/api/jsonrpc",
                         "rest": "/cmd",
-                        "health": "/health"
-                    }
+                        "health": "/health",
+                    },
                 },
                 "heartbeat": {
                     "enabled": True,
                     "interval": 300,
                     "timeout": 30,
                     "retry_attempts": 3,
-                    "retry_delay": 60
+                    "retry_delay": 60,
                 },
                 "auto_discovery": {
                     "enabled": False,
                     "discovery_urls": [],
                     "discovery_interval": 3600,
-                    "register_on_discovery": True
-                }
+                    "register_on_discovery": True,
+                },
             },
-            
             "logging": {
                 "level": "INFO",
                 "console_output": True,
@@ -210,17 +205,15 @@ class ConfigGenerator:
                 "file_path": None,
                 "max_file_size": 10,
                 "backup_count": 5,
-                "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+                "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
             },
-            
             "commands": {
                 "auto_discovery": True,
                 "commands_directory": "./commands",
                 "builtin_commands": ["echo", "health", "config"],
                 "custom_commands": [],
-                "command_timeout": 30
+                "command_timeout": 30,
             },
-            
             "hooks": {
                 "enabled": True,
                 "application_hooks": {
@@ -228,7 +221,7 @@ class ConfigGenerator:
                     "on_shutdown": [],
                     "before_request": [],
                     "after_request": [],
-                    "on_error": []
+                    "on_error": [],
                 },
                 "command_hooks": {
                     "before_echo_command": [],
@@ -236,22 +229,21 @@ class ConfigGenerator:
                     "before_health_command": [],
                     "after_health_command": [],
                     "before_config_command": [],
-                    "after_config_command": []
-                }
+                    "after_config_command": [],
+                },
             },
-            
             "protocols": {
                 "enabled": True,
                 "allowed_protocols": ["http", "https"],
                 "default_protocol": "http",
-                "strict_mode": False
-            }
+                "strict_mode": False,
+            },
         }
-    
+
     def generate_config_with_comments(self, config_type: str = "full") -> str:
         """
         Generate configuration with detailed comments.
-        
+
         Args:
             config_type: Type of configuration to generate
                 - "full": Complete configuration with all options
@@ -263,24 +255,24 @@ class ConfigGenerator:
                 - "https": HTTPS configuration
                 - "https_token": HTTPS with token authentication
                 - "mtls": mTLS configuration
-                
+
         Returns:
             JSON configuration string with comments
         """
         config = self._get_config_by_type(config_type)
-        
+
         # Convert to JSON with comments
         json_str = json.dumps(config, indent=2, ensure_ascii=False)
-        
+
         # Add comments
         commented_config = self._add_comments(json_str, config_type)
-        
+
         return commented_config
-    
+
     def _get_config_by_type(self, config_type: str) -> Dict[str, Any]:
         """Get configuration based on type."""
         base_config = self.template_config.copy()
-        
+
         if config_type == "minimal":
             return self._get_minimal_config(base_config)
         elif config_type == "secure":
@@ -303,30 +295,30 @@ class ConfigGenerator:
             return self._get_mtls_no_protocol_middleware_config(base_config)
         else:  # full
             return base_config
-    
+
     def _get_minimal_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get minimal working configuration."""
         config = base_config.copy()
-        
+
         # Disable security for minimal config
         config["security"]["enabled"] = False
         config["security"]["auth"]["enabled"] = False
         config["security"]["permissions"]["enabled"] = False
         config["security"]["rate_limit"]["enabled"] = False
-        
+
         # Disable registration for minimal config
         config["registration"]["enabled"] = False
-        
+
         # Keep only essential settings
         config["server"]["port"] = 8000
         config["server"]["debug"] = False
-        
+
         return config
-    
+
     def _get_basic_http_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get basic HTTP configuration."""
         config = base_config.copy()
-        
+
         # Basic HTTP settings
         config["server"]["port"] = 8000
         config["ssl"]["enabled"] = False
@@ -346,17 +338,20 @@ class ConfigGenerator:
         config["registration"]["server_url"] = "http://127.0.0.1:3004/proxy"
         config["registration"]["proxy_info"]["name"] = "mcp_example_server"
         config["registration"]["proxy_info"]["capabilities"] = [
-            "jsonrpc", "rest", "security", "proxy_registration"
+            "jsonrpc",
+            "rest",
+            "security",
+            "proxy_registration",
         ]
         config["registration"]["heartbeat"]["enabled"] = True
         config["registration"]["heartbeat"]["interval"] = 30
-        
+
         return config
-    
+
     def _get_http_token_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get HTTP with token authentication configuration."""
         config = base_config.copy()
-        
+
         # HTTP with token auth
         config["server"]["port"] = 8001
         config["ssl"]["enabled"] = False
@@ -367,89 +362,105 @@ class ConfigGenerator:
             "test-token-123": {
                 "roles": ["admin"],
                 "permissions": ["*"],
-                "expires": None
+                "expires": None,
             },
             "user-token-456": {
                 "roles": ["user"],
                 "permissions": ["read", "execute"],
-                "expires": None
-            }
+                "expires": None,
+            },
         }
         config["security"]["permissions"]["enabled"] = True
-        config["security"]["permissions"]["roles_file"] = "mcp_proxy_adapter/examples/server_configs/roles.json"
+        config["security"]["permissions"][
+            "roles_file"
+        ] = "mcp_proxy_adapter/examples/server_configs/roles.json"
         config["protocols"]["enabled"] = True
         config["protocols"]["allowed_protocols"] = ["http"]
         config["protocols"]["default_protocol"] = "http"
-        
+
         return config
-    
+
     def _get_https_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get HTTPS configuration."""
         config = base_config.copy()
-        
+
         # HTTPS settings
         config["server"]["port"] = 8443
         config["ssl"]["enabled"] = True
         config["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
         config["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
         config["ssl"]["ca_cert"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        
+
         config["security"]["ssl"]["enabled"] = True
-        config["security"]["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
-        config["security"]["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
-        config["security"]["ssl"]["ca_cert_file"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        
+        config["security"]["ssl"][
+            "cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
+        config["security"]["ssl"][
+            "key_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_key.pem"
+        config["security"]["ssl"][
+            "ca_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
+
         config["security"]["auth"]["enabled"] = False
         config["security"]["permissions"]["enabled"] = False
         config["security"]["permissions"]["roles_file"] = None
         config["protocols"]["enabled"] = True
         config["protocols"]["allowed_protocols"] = ["http", "https"]
         config["protocols"]["default_protocol"] = "https"
-        
+
         return config
-    
+
     def _get_https_token_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get HTTPS with token authentication configuration."""
         config = base_config.copy()
-        
+
         # HTTPS with token auth
         config["server"]["port"] = 8444
         config["ssl"]["enabled"] = True
         config["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
         config["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
         config["ssl"]["ca_cert"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        
+
         config["security"]["ssl"]["enabled"] = True
-        config["security"]["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
-        config["security"]["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
-        config["security"]["ssl"]["ca_cert_file"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        
+        config["security"]["ssl"][
+            "cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
+        config["security"]["ssl"][
+            "key_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_key.pem"
+        config["security"]["ssl"][
+            "ca_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
+
         config["security"]["auth"]["enabled"] = True
         config["security"]["auth"]["methods"] = ["api_key"]
         config["security"]["auth"]["api_keys"] = {
             "test-token-123": {
                 "roles": ["admin"],
                 "permissions": ["*"],
-                "expires": None
+                "expires": None,
             },
             "user-token-456": {
                 "roles": ["user"],
                 "permissions": ["read", "execute"],
-                "expires": None
-            }
+                "expires": None,
+            },
         }
         config["security"]["permissions"]["enabled"] = True
-        config["security"]["permissions"]["roles_file"] = "mcp_proxy_adapter/examples/server_configs/roles.json"
+        config["security"]["permissions"][
+            "roles_file"
+        ] = "mcp_proxy_adapter/examples/server_configs/roles.json"
         config["protocols"]["enabled"] = True
         config["protocols"]["allowed_protocols"] = ["http", "https"]
         config["protocols"]["default_protocol"] = "https"
-        
+
         return config
-    
+
     def _get_mtls_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get mTLS configuration."""
         config = base_config.copy()
-        
+
         # mTLS settings
         config["server"]["port"] = 8445
         config["ssl"]["enabled"] = True
@@ -458,66 +469,90 @@ class ConfigGenerator:
         config["ssl"]["ca_cert"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
         config["ssl"]["verify_client"] = True
         config["ssl"]["client_cert_required"] = True
-        
+
         config["security"]["ssl"]["enabled"] = True
-        config["security"]["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
-        config["security"]["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
-        config["security"]["ssl"]["ca_cert_file"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        config["security"]["ssl"]["client_cert_file"] = "mcp_proxy_adapter/examples/certs/client_cert.pem"
-        config["security"]["ssl"]["client_key_file"] = "mcp_proxy_adapter/examples/certs/client_key.pem"
+        config["security"]["ssl"][
+            "cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
+        config["security"]["ssl"][
+            "key_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_key.pem"
+        config["security"]["ssl"][
+            "ca_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
+        config["security"]["ssl"][
+            "client_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/client_cert.pem"
+        config["security"]["ssl"][
+            "client_key_file"
+        ] = "mcp_proxy_adapter/examples/certs/client_key.pem"
         config["security"]["ssl"]["verify_mode"] = "CERT_REQUIRED"
-        
+
         config["security"]["auth"]["enabled"] = True
         config["security"]["auth"]["methods"] = ["certificate"]
         config["security"]["auth"]["certificate_auth"] = True
         config["security"]["permissions"]["enabled"] = True
-        config["security"]["permissions"]["roles_file"] = "mcp_proxy_adapter/examples/server_configs/roles.json"
+        config["security"]["permissions"][
+            "roles_file"
+        ] = "mcp_proxy_adapter/examples/server_configs/roles.json"
         config["protocols"]["enabled"] = True
         config["protocols"]["allowed_protocols"] = ["https", "mtls"]
         config["protocols"]["default_protocol"] = "https"
-        
+
         return config
-    
-    def _get_https_no_protocol_middleware_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
+
+    def _get_https_no_protocol_middleware_config(
+        self, base_config: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Get HTTPS configuration without ProtocolMiddleware."""
         config = base_config.copy()
-        
+
         # HTTPS settings
         config["server"]["port"] = 8445
         config["ssl"]["enabled"] = True
         config["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
         config["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
         config["ssl"]["ca_cert"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        
+
         config["security"]["ssl"]["enabled"] = True
-        config["security"]["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
-        config["security"]["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
-        config["security"]["ssl"]["ca_cert_file"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        
+        config["security"]["ssl"][
+            "cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
+        config["security"]["ssl"][
+            "key_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_key.pem"
+        config["security"]["ssl"][
+            "ca_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
+
         config["security"]["auth"]["enabled"] = True
         config["security"]["auth"]["methods"] = ["api_key"]
         config["security"]["auth"]["api_keys"] = {
             "test-token-123": {
                 "roles": ["admin"],
                 "permissions": ["*"],
-                "expires": None
+                "expires": None,
             },
             "user-token-456": {
                 "roles": ["user"],
                 "permissions": ["read", "execute"],
-                "expires": None
-            }
+                "expires": None,
+            },
         }
         config["security"]["permissions"]["enabled"] = True
-        config["security"]["permissions"]["roles_file"] = "mcp_proxy_adapter/examples/server_configs/roles.json"
+        config["security"]["permissions"][
+            "roles_file"
+        ] = "mcp_proxy_adapter/examples/server_configs/roles.json"
         config["protocols"]["enabled"] = False  # Disable ProtocolMiddleware
-        
+
         return config
-    
-    def _get_mtls_no_protocol_middleware_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
+
+    def _get_mtls_no_protocol_middleware_config(
+        self, base_config: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Get mTLS configuration without ProtocolMiddleware."""
         config = base_config.copy()
-        
+
         # mTLS settings
         config["server"]["port"] = 8447
         config["ssl"]["enabled"] = True
@@ -526,72 +561,84 @@ class ConfigGenerator:
         config["ssl"]["ca_cert"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
         config["ssl"]["verify_client"] = True
         config["ssl"]["client_cert_required"] = True
-        
+
         config["security"]["ssl"]["enabled"] = True
-        config["security"]["ssl"]["cert_file"] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
-        config["security"]["ssl"]["key_file"] = "mcp_proxy_adapter/examples/certs/server_key.pem"
-        config["security"]["ssl"]["ca_cert_file"] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
-        config["security"]["ssl"]["client_cert_file"] = "mcp_proxy_adapter/examples/certs/client_cert.pem"
-        config["security"]["ssl"]["client_key_file"] = "mcp_proxy_adapter/examples/certs/client_key.pem"
+        config["security"]["ssl"][
+            "cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_cert.pem"
+        config["security"]["ssl"][
+            "key_file"
+        ] = "mcp_proxy_adapter/examples/certs/server_key.pem"
+        config["security"]["ssl"][
+            "ca_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/ca_cert.pem"
+        config["security"]["ssl"][
+            "client_cert_file"
+        ] = "mcp_proxy_adapter/examples/certs/client_cert.pem"
+        config["security"]["ssl"][
+            "client_key_file"
+        ] = "mcp_proxy_adapter/examples/certs/client_key.pem"
         config["security"]["ssl"]["verify_mode"] = "CERT_REQUIRED"
-        
+
         config["security"]["auth"]["enabled"] = True
         config["security"]["auth"]["methods"] = ["certificate"]
         config["security"]["auth"]["certificate_auth"] = True
         config["security"]["permissions"]["enabled"] = True
-        config["security"]["permissions"]["roles_file"] = "mcp_proxy_adapter/examples/server_configs/roles.json"
+        config["security"]["permissions"][
+            "roles_file"
+        ] = "mcp_proxy_adapter/examples/server_configs/roles.json"
         config["protocols"]["enabled"] = False  # Disable ProtocolMiddleware
-        
+
         return config
-    
+
     def _get_secure_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get secure configuration with all security features enabled."""
         config = base_config.copy()
-        
+
         # Enable all security features
         config["security"]["enabled"] = True
         config["security"]["ssl"]["enabled"] = True
         config["security"]["auth"]["enabled"] = True
         config["security"]["permissions"]["enabled"] = True
         config["security"]["rate_limit"]["enabled"] = True
-        
+
         # Enable registration with certificate auth
         config["registration"]["enabled"] = True
         config["registration"]["auth_method"] = "certificate"
         config["registration"]["certificate"]["enabled"] = True
-        
+
         # Set secure defaults
         config["security"]["ssl"]["min_tls_version"] = "TLSv1.2"
         config["security"]["auth"]["methods"] = ["api_key", "jwt"]
         config["security"]["permissions"]["strict_mode"] = True
         config["security"]["rate_limit"]["burst_limit"] = 1
-        
+
         return config
-    
+
     def _get_development_config(self, base_config: Dict[str, Any]) -> Dict[str, Any]:
         """Get development configuration with debug enabled."""
         config = base_config.copy()
-        
+
         # Enable debug features
         config["server"]["debug"] = True
         config["security"]["debug"] = True
         config["logging"]["level"] = "DEBUG"
-        
+
         # Enable registration with token auth for development
         config["registration"]["enabled"] = True
         config["registration"]["auth_method"] = "token"
         config["registration"]["token"]["enabled"] = True
-        
+
         # Relax security for development
         config["security"]["rate_limit"]["default_requests_per_minute"] = 1000
         config["security"]["permissions"]["strict_mode"] = False
-        
+
         return config
-    
+
     def _add_comments(self, json_str: str, config_type: str) -> str:
         """Add comments to JSON configuration."""
         comments = self._get_comments_for_type(config_type)
-        
+
         # Add header comment
         commented_config = f"""/**
  * MCP Proxy Adapter Configuration
@@ -607,7 +654,7 @@ class ConfigGenerator:
  */
 
 """
-        
+
         # Add section comments
         for section, comment in comments.items():
             if section in json_str:
@@ -615,15 +662,15 @@ class ConfigGenerator:
                 section_start = json_str.find(f'"{section}":')
                 if section_start != -1:
                     # Find the line start
-                    line_start = json_str.rfind('\n', 0, section_start) + 1
+                    line_start = json_str.rfind("\n", 0, section_start) + 1
                     json_str = (
-                        json_str[:line_start] +
-                        f"  // {comment}\n" +
-                        json_str[line_start:]
+                        json_str[:line_start]
+                        + f"  // {comment}\n"
+                        + json_str[line_start:]
                     )
-        
+
         return commented_config + json_str
-    
+
     def _get_comments_for_type(self, config_type: str) -> Dict[str, str]:
         """Get comments for configuration sections."""
         base_comments = {
@@ -634,103 +681,157 @@ class ConfigGenerator:
             "logging": "Logging configuration for the application",
             "commands": "Command management and discovery settings",
             "hooks": "Application and command hooks configuration",
-            "protocols": "Protocol endpoints and settings"
+            "protocols": "Protocol endpoints and settings",
         }
-        
+
         if config_type == "minimal":
-            base_comments["security"] = "Security framework configuration (disabled for minimal setup)"
-            base_comments["registration"] = "Proxy registration configuration (disabled for minimal setup)"
+            base_comments["security"] = (
+                "Security framework configuration (disabled for minimal setup)"
+            )
+            base_comments["registration"] = (
+                "Proxy registration configuration (disabled for minimal setup)"
+            )
         elif config_type == "secure":
-            base_comments["security"] = "Security framework configuration (all features enabled)"
-            base_comments["registration"] = "Proxy registration configuration (certificate authentication enabled)"
+            base_comments["security"] = (
+                "Security framework configuration (all features enabled)"
+            )
+            base_comments["registration"] = (
+                "Proxy registration configuration (certificate authentication enabled)"
+            )
         elif config_type == "development":
-            base_comments["security"] = "Security framework configuration (development mode with relaxed settings)"
-            base_comments["registration"] = "Proxy registration configuration (token authentication for development)"
+            base_comments["security"] = (
+                "Security framework configuration (development mode with relaxed settings)"
+            )
+            base_comments["registration"] = (
+                "Proxy registration configuration (token authentication for development)"
+            )
         elif config_type in ["basic_http", "http_token"]:
             base_comments["ssl"] = "SSL/TLS configuration (disabled for HTTP)"
-            base_comments["security"] = f"Security framework configuration ({config_type} mode)"
+            base_comments["security"] = (
+                f"Security framework configuration ({config_type} mode)"
+            )
         elif config_type in ["https", "https_token"]:
             base_comments["ssl"] = "SSL/TLS configuration (enabled for HTTPS)"
-            base_comments["security"] = f"Security framework configuration ({config_type} mode)"
+            base_comments["security"] = (
+                f"Security framework configuration ({config_type} mode)"
+            )
         elif config_type == "mtls":
-            base_comments["ssl"] = "SSL/TLS configuration (enabled for mTLS with client certificate verification)"
-            base_comments["security"] = "Security framework configuration (mTLS mode with certificate authentication)"
+            base_comments["ssl"] = (
+                "SSL/TLS configuration (enabled for mTLS with client certificate verification)"
+            )
+            base_comments["security"] = (
+                "Security framework configuration (mTLS mode with certificate authentication)"
+            )
         elif config_type == "https_no_protocol_middleware":
-            base_comments["ssl"] = "SSL/TLS configuration (enabled for HTTPS without ProtocolMiddleware)"
-            base_comments["security"] = "Security framework configuration (HTTPS mode without ProtocolMiddleware)"
+            base_comments["ssl"] = (
+                "SSL/TLS configuration (enabled for HTTPS without ProtocolMiddleware)"
+            )
+            base_comments["security"] = (
+                "Security framework configuration (HTTPS mode without ProtocolMiddleware)"
+            )
         elif config_type == "mtls_no_protocol_middleware":
-            base_comments["ssl"] = "SSL/TLS configuration (enabled for mTLS without ProtocolMiddleware)"
-            base_comments["security"] = "Security framework configuration (mTLS mode without ProtocolMiddleware)"
-        
+            base_comments["ssl"] = (
+                "SSL/TLS configuration (enabled for mTLS without ProtocolMiddleware)"
+            )
+            base_comments["security"] = (
+                "Security framework configuration (mTLS mode without ProtocolMiddleware)"
+            )
+
         return base_comments
-    
+
     def generate_config_file(self, output_path: str, config_type: str = "full") -> None:
         """
         Generate configuration file and save to disk.
-        
+
         Args:
             output_path: Path to save the configuration file
             config_type: Type of configuration to generate
         """
         try:
             config_content = self.generate_config_with_comments(config_type)
-            
+
             # Create directory if it doesn't exist
             output_file = Path(output_path)
             output_file.parent.mkdir(parents=True, exist_ok=True)
-            
+
             # Write configuration file
-            with open(output_file, 'w', encoding='utf-8') as f:
+            with open(output_file, "w", encoding="utf-8") as f:
                 f.write(config_content)
-            
+
             logger.info(f"Configuration file generated: {output_path}")
             logger.info(f"Configuration type: {config_type}")
-            
+
         except Exception as e:
             logger.error(f"Failed to generate configuration file: {e}")
             raise
-    
+
     def generate_all_configs(self, output_dir: str) -> None:
         """
         Generate all configuration types.
-        
+
         Args:
             output_dir: Directory to save configuration files
         """
         config_types = [
-            "minimal", "development", "secure", "full",
-            "basic_http", "http_token", "https", "https_token", "mtls",
-            "https_no_protocol_middleware", "mtls_no_protocol_middleware"
+            "minimal",
+            "development",
+            "secure",
+            "full",
+            "basic_http",
+            "http_token",
+            "https",
+            "https_token",
+            "mtls",
+            "https_no_protocol_middleware",
+            "mtls_no_protocol_middleware",
         ]
-        
+
         for config_type in config_types:
             output_path = Path(output_dir) / f"config_{config_type}.json"
             self.generate_config_file(str(output_path), config_type)
-        
-        logger.info(f"Generated {len(config_types)} configuration files in {output_dir}")
+
+        logger.info(
+            f"Generated {len(config_types)} configuration files in {output_dir}"
+        )
 
 
 def main():
     """Main function for command-line usage."""
     import argparse
-    
-    parser = argparse.ArgumentParser(description="Generate MCP Proxy Adapter configuration files")
-    parser.add_argument("--type", 
-                       choices=["minimal", "development", "secure", "full", 
-                               "basic_http", "http_token", "https", "https_token", "mtls",
-                               "https_no_protocol_middleware", "mtls_no_protocol_middleware"], 
-                       default="full", help="Configuration type to generate")
-    parser.add_argument("--output", default="./config.json", 
-                       help="Output file path")
-    parser.add_argument("--all", action="store_true", 
-                       help="Generate all configuration types")
-    parser.add_argument("--output-dir", default="./configs", 
-                       help="Output directory for all configs")
-    
+
+    parser = argparse.ArgumentParser(
+        description="Generate MCP Proxy Adapter configuration files"
+    )
+    parser.add_argument(
+        "--type",
+        choices=[
+            "minimal",
+            "development",
+            "secure",
+            "full",
+            "basic_http",
+            "http_token",
+            "https",
+            "https_token",
+            "mtls",
+            "https_no_protocol_middleware",
+            "mtls_no_protocol_middleware",
+        ],
+        default="full",
+        help="Configuration type to generate",
+    )
+    parser.add_argument("--output", default="./config.json", help="Output file path")
+    parser.add_argument(
+        "--all", action="store_true", help="Generate all configuration types"
+    )
+    parser.add_argument(
+        "--output-dir", default="./configs", help="Output directory for all configs"
+    )
+
     args = parser.parse_args()
-    
+
     generator = ConfigGenerator()
-    
+
     if args.all:
         generator.generate_all_configs(args.output_dir)
     else: