class RubyLLM::Moderation

# File lib/ruby_llm/moderation.rb, line 15
def self.moderate(input,
                  model: nil,
                  provider: nil,
                  assume_model_exists: false,
                  context: nil)
  config = context&.config || RubyLLM.config
  model ||= config.default_moderation_model || 'omni-moderation-latest'
  model, provider_instance = Models.resolve(model, provider: provider, assume_exists: assume_model_exists,
                                                   config: config)
  model_id = model.id

  provider_instance.moderate(input, model: model_id)
end

# File lib/ruby_llm/moderation.rb, line 9
def initialize(id:, model:, results:)
  @id = id
  @model = model
  @results = results
end

# File lib/ruby_llm/moderation.rb, line 52
def categories
  results.first&.dig('categories') || {}
end

# File lib/ruby_llm/moderation.rb, line 47
def category_scores
  results.first&.dig('category_scores') || {}
end

# File lib/ruby_llm/moderation.rb, line 30
def content
  results
end

# File lib/ruby_llm/moderation.rb, line 35
def flagged?
  results.any? { |result| result['flagged'] }
end

# File lib/ruby_llm/moderation.rb, line 40
def flagged_categories
  results.flat_map do |result|
    result['categories']&.select { |_category, flagged| flagged }&.keys || []
  end.uniq
end