Method: GenAI::Language::GooglePalm#embed

Defined in:
lib/gen_ai/language/google_palm.rb

#embed(input, model: nil) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
 
# File 'lib/gen_ai/language/google_palm.rb', line 17

def embed(input, model: nil)
  responses = array_wrap(input).map do |text|
    handle_errors { client.embed(text: text, model: model) }
  end

  build_result(
    model: model || EMBEDDING_MODEL,
    raw: { 'data' => responses, 'usage' => {} },
    parsed: extract_embeddings(responses)
  )
end