class RubyLLM::Message

def ensure_valid_role
  raise Error, "Expected role to be one of: #{ROLES.join(', ')}" unless ROLES.include?(role)
end

def initialize(options = {})
  @role = options[:role].to_sym
  @content = options[:content]
  @tool_calls = options[:tool_calls]
  @input_tokens = options[:input_tokens]
  @output_tokens = options[:output_tokens]
  @model_id = options[:model_id]
  @tool_call_id = options[:tool_call_id]
  ensure_valid_role
end

def to_h
  {
    role: role,
    content: content,
    tool_calls: tool_calls,
    tool_call_id: tool_call_id,
    input_tokens: input_tokens,
    output_tokens: output_tokens,
    model_id: model_id
  }.compact
end

def tool_call?
  !tool_calls.nil? && !tool_calls.empty?
end

def tool_result?
  !tool_call_id.nil? && !tool_call_id.empty?
end

def tool_results
  content if tool_result?
end