Class: Maglev::Adapters::RubyLLMProvider

Inherits:
Object
  • Object
show all
Defined in:
lib/maglev/adapters/ruby_llm_provider.rb

Instance Method Summary collapse

Constructor Details

#initialize(provider) ⇒ RubyLLMProvider

Returns a new instance of RubyLLMProvider.



8
9
10
 
# File 'lib/maglev/adapters/ruby_llm_provider.rb', line 8

def initialize(provider)
  @provider = provider
end

Instance Method Details

#embed(text) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/maglev/adapters/ruby_llm_provider.rb', line 12

def embed(text)
  result = translate_errors do
    context.embed(
      text,
      model: @provider.model,
      provider: :openai,
      assume_model_exists: true,
      dimensions: @provider.dimensions
    )
  end
  result.vectors.first.is_a?(Array) ? result.vectors.first : result.vectors
end

#generate(prompt) ⇒ Object 



25
26
27
28
29
30
31
32
33
34
 
# File 'lib/maglev/adapters/ruby_llm_provider.rb', line 25

def generate(prompt)
  response = translate_errors do
    context.chat(
      model: @provider.model,
      provider: :openai,
      assume_model_exists: true
    ).ask(prompt)
  end
  response.respond_to?(:content) ? response.content : response.to_s
end