discourse-ai/lib/modules/summarization/strategies/truncate_content.rb

31 lines
717 B
Ruby

# frozen_string_literal: true
module DiscourseAi
module Summarization
module Strategies
class TruncateContent < ::Summarization::Base
def initialize(completion_model)
@completion_model = completion_model
end
attr_reader :completion_model
delegate :correctly_configured?,
:display_name,
:configuration_hint,
:model,
to: :completion_model
def summarize(content)
opts = content.except(:contents)
{
summary: completion_model.summarize_with_truncation(content[:contents], opts),
chunks: [],
}
end
end
end
end
end