57 lines
1.6 KiB
Ruby
57 lines
1.6 KiB
Ruby
# 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.create!(
|
|
target: strategy.target,
|
|
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],
|
|
)
|
|
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)
|
|
#
|