module RubyLLM::ActiveRecord::MessageMethods

# File lib/ruby_llm/active_record/message_methods.rb, line 56
def cache_until_here
  update!(cache_until_here: true)
  self
end

# File lib/ruby_llm/active_record/message_methods.rb, line 62
def cache_until_here?
  optional_column(:cache_until_here) || false
end

# File lib/ruby_llm/active_record/message_methods.rb, line 75
def citations
  Array(optional_column(:citations)).map { |citation| RubyLLM::Citation.from_h(citation) }
end

# File lib/ruby_llm/active_record/message_methods.rb, line 160
def content_filtered?
  to_llm.content_filtered?
end

# File lib/ruby_llm/active_record/message_methods.rb, line 90
def cost
  records = ruby_llm_usages.to_a
  RubyLLM::Cost.aggregate(records.map(&:cost), complete: records.all?(&:cost_available?))
end

# File lib/ruby_llm/active_record/message_methods.rb, line 140
def finish_reason
  optional_column(:finish_reason)&.to_sym
end

# File lib/ruby_llm/active_record/message_methods.rb, line 150
def max_tokens?
  to_llm.max_tokens?
end

# File lib/ruby_llm/active_record/message_methods.rb, line 46
def model
  to_llm.model
end

# File lib/ruby_llm/active_record/message_methods.rb, line 51
def model_info
  to_llm.model_info
end

# File lib/ruby_llm/active_record/message_methods.rb, line 101
def parent_tool_call
  ruby_llm_parent_tool_call&.to_llm
end

# File lib/ruby_llm/active_record/message_methods.rb, line 165
def parsed
  text = extract_content
  return if text.nil? || text.empty?

  JSON.parse(text)
end

# File lib/ruby_llm/active_record/message_methods.rb, line 80
def server_tool_calls
  Array(optional_column(:server_tool_calls)).map { |call| RubyLLM::ServerToolCall.from_h(call) }
end

# File lib/ruby_llm/active_record/message_methods.rb, line 145
def stopped?
  to_llm.stopped?
end

# File lib/ruby_llm/active_record/message_methods.rb, line 67
def thinking
  RubyLLM::Thinking.build(
    text: optional_column(:thinking_text),
    signature: optional_column(:thinking_signature)
  )
end

# File lib/ruby_llm/active_record/message_methods.rb, line 21
def to_llm
  entries = ruby_llm_usage_entries
  RubyLLM::Message.new(
    role: role.to_sym,
    content: extract_content,
    attachments: extract_attachments,
    thinking: thinking,
    citations: citations,
    server_tool_calls: server_tool_calls,
    raw_content: optional_column(:raw_content),
    raw_reasoning: optional_column(:raw_reasoning),
    usage_entries: entries,
    tool_calls: tool_calls,
    tool_call_id: parent_tool_call&.id,
    finish_reason: optional_column(:finish_reason),
    model: entries.reverse.find(&:succeeded?)&.model,
    cache_until_here: cache_until_here?
  )
end

# File lib/ruby_llm/active_record/message_methods.rb, line 122
def to_partial_path
  partial_prefix = self.class.name.underscore.pluralize
  role_partial = if tool_call?
                   'tool_calls'
                 elsif role.to_s == 'tool'
                   'tool'
                 else
                   role.to_s.presence || 'assistant'
                 end
  "#{partial_prefix}/#{role_partial}"
end

# File lib/ruby_llm/active_record/message_methods.rb, line 85
def tokens
  RubyLLM::Tokens.aggregate(ruby_llm_usages.map(&:tokens))
end

# File lib/ruby_llm/active_record/message_methods.rb, line 111
def tool_call?
  ruby_llm_tool_calls.any?
end

# File lib/ruby_llm/active_record/message_methods.rb, line 155
def tool_call_stop?
  to_llm.tool_call_stop?
end

# File lib/ruby_llm/active_record/message_methods.rb, line 96
def tool_calls
  ruby_llm_tool_calls.to_h { |record| [record.tool_call_id, record.to_llm] }
end

# File lib/ruby_llm/active_record/message_methods.rb, line 135
def tool_error_message
  payload_error_message(extract_content)
end

# File lib/ruby_llm/active_record/message_methods.rb, line 116
def tool_result?
  ruby_llm_parent_tool_call.present?
end

# File lib/ruby_llm/active_record/message_methods.rb, line 106
def tool_results
  ruby_llm_tool_calls.filter_map(&:result)
end