configureConnection method

void configureConnection({
  1. required McpConnectionParams connectionParams,
  2. McpSamplingCallback? samplingCallback,
  3. Map<String, Object?>? samplingCapabilities,
})

Configures per-connection MCP client-side sampling support.

Implementation

void configureConnection({
  required McpConnectionParams connectionParams,
  McpSamplingCallback? samplingCallback,
  Map<String, Object?>? samplingCapabilities,
}) {
  final String key = _connectionKey(connectionParams);
  if (samplingCallback == null) {
    _samplingCallbacksByUrl.remove(key);
  } else {
    _samplingCallbacksByUrl[key] = samplingCallback;
  }
  if (samplingCapabilities == null || samplingCapabilities.isEmpty) {
    _samplingCapabilitiesByUrl.remove(key);
  } else {
    _samplingCapabilitiesByUrl[key] = Map<String, Object?>.from(
      samplingCapabilities,
    );
  }
}