diff --git a/client/sse.go b/client/sse.go index e7aaaa49..96db7093 100644 --- a/client/sse.go +++ b/client/sse.go @@ -46,6 +46,18 @@ func WithHeaders(headers map[string]string) ClientOption { } } +func WithSSEReadTimeout(timeout time.Duration) ClientOption { + return func(sc *SSEMCPClient) { + sc.sseReadTimeout = timeout + } +} + +func WithHTTPClient(client *http.Client) ClientOption { + return func(sc *SSEMCPClient) { + sc.httpClient = client + } +} + // NewSSEMCPClient creates a new SSE-based MCP client with the given base URL. // Returns an error if the URL is invalid. func NewSSEMCPClient(baseURL string, options ...ClientOption) (*SSEMCPClient, error) {