ragnar_retrieve_vss_and_bm25 {ragnar} | R Documentation |
Retrieve VSS and BM25
Description
Runs ragnar_retrieve_vss()
and ragnar_retrieve_bm25()
and get the distinct
documents.
Usage
ragnar_retrieve_vss_and_bm25(store, text, top_k = 3, ...)
Arguments
store |
A |
text |
Character. Query string to match. |
top_k |
Integer, the number of entries to retrieve using per method. |
... |
Forwarded to |
Value
A tibble
of retrieved chunks. Each row
represents a chunk and always contains a text
column.
Note
The results are not re-ranked after identifying the unique values.
See Also
Other ragnar_retrieve:
ragnar_retrieve()
,
ragnar_retrieve_bm25()
,
ragnar_retrieve_vss()
Examples
## Build a small store with categories
store <- ragnar_store_create(
embed = \(x) ragnar::embed_openai(x, model = "text-embedding-3-small"),
extra_cols = data.frame(category = character()),
version = 1 # store text chunks directly
)
ragnar_store_insert(
store,
data.frame(
category = c(rep("pets", 3), rep("dessert", 3)),
text = c("playful puppy", "sleepy kitten", "curious hamster",
"chocolate cake", "strawberry tart", "vanilla ice cream")
)
)
ragnar_store_build_index(store)
# Top 3 chunks without filtering
ragnar_retrieve(store, "sweet")
# Combine filter with similarity search
ragnar_retrieve(store, "sweet", filter = category == "dessert")
[Package ragnar version 0.2.0 Index]