class RubyLLM::MCP::Configuration::Sampling

def enabled?
  @enabled
end

def guard(&block)
  @guard = block if block_given?
  @guard
end

def initialize
  set_defaults
end

def preferred_model(&block)
  @preferred_model = block if block_given?
  @preferred_model
end

def reset!
  set_defaults
end

def set_defaults
  @enabled = false
  @preferred_model = nil
  @guard = nil
end