Class ServerConfig

optillm/plugins/mcp_plugin.py:119–162

Configuration for a single MCP server

@dataclass
class ServerConfig:
    """Configuration for a single MCP server"""
    # Transport type: "stdio" (default), "sse", or "websocket"
    transport: str = "stdio"

    # For stdio transport
    command: Optional[str] = None
    args: List[str] = None

    # For remote transports (SSE/WebSocket)
    url: Optional[str] = None
    headers: Dict[str, str] = None

    # Common fields
    env: Dict[str, str] = None
    description: Optional[str] = None

    # Timeout settings
    timeout: float = 5.0
    sse_read_timeout: float = 300.0

    def __post_init__(self):
        """Initialize default values for mutable fields"""
        if self.args is None:
            self.args = []
        if self.headers is None:
            self.headers = {}
        if self.env is None:
            self.env = {}

    @classmethod
    def from_dict(cls, config: Dict[str, Any]) -> 'ServerConfig':
        """Create ServerConfig from a dictionary"""
        return cls(
            transport=config.get("transport", "stdio"),
            command=config.get("command"),
            args=config.get("args", []),
            url=config.get("url"),
            headers=config.get("headers", {}),
            env=config.get("env", {}),
            description=config.get("description"),
            timeout=config.get("timeout", 5.0),
            sse_read_timeout=config.get("sse_read_timeout", 300.0)
        )

class MCPConfigManager:
    """Manages MCP configuration loading and validation"""