discourse-ai/app/models/ai_summary.rb

# frozen_string_literal: true

class AiSummary < ActiveRecord::Base
  belongs_to :target, polymorphic: true

  enum :summary_type, { complete: 0, gist: 1 }
  enum :origin, { human: 0, system: 1 }

  def self.store!(strategy, llm_model, summary, og_content, human:)
    content_ids = og_content.map { |c| c[:id] }

    AiSummary
      .upsert(
        {
          target_id: strategy.target.id,
          target_type: strategy.target.class.name,
          algorithm: llm_model.name,
          content_range: (content_ids.first..content_ids.last),
          summarized_text: summary,
          original_content_sha: build_sha(content_ids.join),
          summary_type: strategy.type,
          origin: !!human ? origins[:human] : origins[:system],
        },
        unique_by: %i[target_id target_type summary_type],
        update_only: %i[summarized_text original_content_sha algorithm origin content_range],
      )
      .first
      .then { AiSummary.find_by(id: _1["id"]) }
  end

  def self.build_sha(joined_ids)
    Digest::SHA256.hexdigest(joined_ids)
  end

  def mark_as_outdated
    @outdated = true
  end

  def outdated
    @outdated || false
  end
end

# == Schema Information
#
# Table name: ai_summaries
#
#  id                   :bigint           not null, primary key
#  target_id            :integer          not null
#  target_type          :string           not null
#  content_range        :int4range
#  summarized_text      :string           not null
#  original_content_sha :string           not null
#  algorithm            :string           not null
#  created_at           :datetime         not null
#  updated_at           :datetime         not null
#  summary_type         :integer          default("complete"), not null
#  origin               :integer
#
# Indexes
#
#  idx_on_target_id_target_type_summary_type_3355609fbb  (target_id,target_type,summary_type) UNIQUE
#  index_ai_summaries_on_target_type_and_target_id       (target_type,target_id)
#