Contents

# frozen_string_literal: true

module GenAI
  class Language
    def initialize(provider, token, options: {})
      build_llm(provider, token, options)
    end

    def embed(text, model: nil)
      llm.embed(text, model: model)
    end

    def complete(prompt, options = {})
      llm.complete(prompt, options: options)
    end

    def chat(message, context: nil, history: [], examples: [], **options)
      llm.chat(message, context: context, history: history, examples: examples, options: options)
    end

    # def answer(prompt); end

    # def sentiment(text); end

    # def keywords(text); end

    # def summarization(text); end

    # def translation(text, _target:); end

    # def correction(text); end

    private

    attr_reader :llm

    def build_llm(provider, token, options)
      klass = GenAI::Language.constants.find do |const|
        const.to_s.downcase == provider.to_s.downcase.gsub('_', '')
      end

      raise UnsupportedProvider, "Unsupported LLM provider '#{provider}'" unless klass

      @llm = GenAI::Language.const_get(klass).new(token: token, options: options)
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
gen-ai-0.3.2	lib/gen_ai/language.rb
gen-ai-0.3.1	lib/gen_ai/language.rb
gen-ai-0.3.0	lib/gen_ai/language.rb
gen-ai-0.2.3	lib/gen_ai/language.rb
gen-ai-0.2.2	lib/gen_ai/language.rb
gen-ai-0.2.1	lib/gen_ai/language.rb
gen-ai-0.2.0	lib/gen_ai/language.rb
gen-ai-0.1.0	lib/gen_ai/language.rb