discourse-ai/lib/modules/embeddings/hyde_generators/llama2_ftos.rb

# frozen_string_literal: true

module DiscourseAi
  module Embeddings
    module HydeGenerators
      class Llama2Ftos < DiscourseAi::Embeddings::HydeGenerators::Llama2
        def prompt(search_term)
          <<~TEXT
              ### System:
              You are a helpful bot
              You create forum posts about a given topic
              
              ### User:
              Topic: #{search_term}
    
              ### Assistant:
              Here is a forum post about the above topic:
            TEXT
        end

        def models
          %w[StableBeluga2 Upstage-Llama-2-*-instruct-v2]
        end
      end
    end
  end
end