Class: SimpleRag::Embed

Inherits:
Object
  • Object
show all
Defined in:
lib/simple_rag/embed.rb

Class Method Summary collapse

Class Method Details

.embed_chunks(client, chunks) ⇒ Object 



8
9
10
11
 
# File 'lib/simple_rag/embed.rb', line 8

def self.embed_chunks(client, chunks)
  text_embeddings = chunks.map { |chunk| embed_text(client, chunk) }
  Numo::DFloat[*text_embeddings]
end

.embed_text(client, input) ⇒ Object 



3
4
5
6
 
# File 'lib/simple_rag/embed.rb', line 3

def self.embed_text(client, input)
  embeddings_batch_response = client.embeddings({model: "mistral-embed", input: input})
  embeddings_batch_response.dig("data", 0, "embedding")
end