Class: RubyLLM::Protocols::InvokeModel::CohereEmbeddings

Inherits:
InvokeModel
  • Object
show all
Defined in:
lib/ruby_llm/protocols/invoke_model/cohere_embeddings.rb

Overview

Cohere embedding models over Bedrock InvokeModel.

Instance Method Summary collapse

Instance Method Details

#embed(text, model:, dimensions:, task_type: nil, title: nil, with: nil, provider_options: {}) ⇒ Object

rubocop:disable-next Lint/UnusedMethodArgument



9
10
11
12
13
14
15
16
17
 
# File 'lib/ruby_llm/protocols/invoke_model/cohere_embeddings.rb', line 9

def embed(text, model:, dimensions:, task_type: nil, title: nil, with: nil, provider_options: {})
  ensure_no_embedding_media!(with)
  track_usage(:embedding) do
    payload = render_embedding_payload(text, model:, dimensions:, task_type:, provider_options:)
    response = signed_post(embedding_url(model:), payload)

    parse_embedding_response(response, model:, text:)
  end
end