Class: Raif::EmbeddingModels::Google

Inherits:
Raif::EmbeddingModel
  • Object
show all
Defined in:
app/models/raif/embedding_models/google.rb

Instance Method Summary collapse

Instance Method Details

#generate_embedding!(input, dimensions: nil) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
 
# File 'app/models/raif/embedding_models/google.rb', line 4

def generate_embedding!(input, dimensions: nil)
  unless input.is_a?(String)
    raise ArgumentError, "Raif::EmbeddingModels::Google#generate_embedding! input must be a string"
  end

  response = connection.post("models/#{api_name}:embedContent") do |req|
    req.body = build_request_parameters(input, dimensions:)
  end

  response.body.dig("embedding", "values")
end