This commit is contained in:
45
docs/snippets/ruby/advanced/chunking_rag.md
Normal file
45
docs/snippets/ruby/advanced/chunking_rag.md
Normal file
@@ -0,0 +1,45 @@
|
||||
```ruby title="Ruby"
|
||||
require 'kreuzberg'
|
||||
|
||||
config = Kreuzberg::ExtractionConfig.new(
|
||||
chunking: Kreuzberg::ChunkingConfig.new(
|
||||
max_characters: 500,
|
||||
overlap: 50,
|
||||
embedding: Kreuzberg::EmbeddingConfig.new(
|
||||
model: Kreuzberg::EmbeddingModelType.new(
|
||||
type: 'preset',
|
||||
name: 'all-mpnet-base-v2'
|
||||
),
|
||||
normalize: true,
|
||||
batch_size: 16
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
result = Kreuzberg.extract_file_sync('research_paper.pdf', config: config)
|
||||
|
||||
vector_store = build_vector_store(result.chunks)
|
||||
query = 'machine learning optimization'
|
||||
relevant_chunks = search_vector_store(vector_store, query)
|
||||
|
||||
puts "Found #{relevant_chunks.length} relevant chunks"
|
||||
relevant_chunks.take(3).each do |chunk|
|
||||
puts "Content: #{chunk[:content][0..80]}..."
|
||||
puts "Similarity: #{chunk[:similarity]&.round(3)}\n"
|
||||
end
|
||||
|
||||
def build_vector_store(chunks)
|
||||
chunks.map.with_index do |chunk, idx|
|
||||
{
|
||||
id: idx,
|
||||
content: chunk.content,
|
||||
embedding: chunk.embedding,
|
||||
similarity: 0.0
|
||||
}
|
||||
end
|
||||
end
|
||||
|
||||
def search_vector_store(store, query)
|
||||
store.sort_by { |entry| entry[:similarity] }.reverse
|
||||
end
|
||||
```
|
||||
Reference in New Issue
Block a user