28 lines
927 B
Ruby
28 lines
927 B
Ruby
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Summarization
|
|
class EntryPoint
|
|
def load_files
|
|
require_relative "strategies/anthropic"
|
|
require_relative "strategies/discourse_ai"
|
|
require_relative "strategies/open_ai"
|
|
end
|
|
|
|
def inject_into(plugin)
|
|
[
|
|
Strategies::OpenAi.new("gpt-4"),
|
|
Strategies::OpenAi.new("gpt-4-32k"),
|
|
Strategies::OpenAi.new("gpt-3.5-turbo"),
|
|
Strategies::OpenAi.new("gpt-3.5-turbo-16k"),
|
|
Strategies::DiscourseAi.new("bart-large-cnn-samsum"),
|
|
Strategies::DiscourseAi.new("flan-t5-base-samsum"),
|
|
Strategies::DiscourseAi.new("long-t5-tglobal-base-16384-book-summary"),
|
|
Strategies::Anthropic.new("claude-v1"),
|
|
Strategies::Anthropic.new("claude-v1-100k"),
|
|
].each { |strategy| plugin.register_summarization_strategy(strategy) }
|
|
end
|
|
end
|
|
end
|
|
end
|