Retrieval

from ragrails import RagRails

rag = RagRails(
    collection="support",
    vector_store={"provider": "qdrant", "url": "http://localhost:6333"},
    embedding={"provider": "voyage", "model": "voyage-3"},
    llm={"provider": "openai", "model": "gpt-4o-mini"},
    reranker={"provider": "voyage", "model": "rerank-2-lite"},
)

result = rag.query("How long do refunds take?", retrieval={"top_k": 5})

result = rag.retrieve("How long do refunds take?", top_k=5)

for item in result.items:
    print(item.score, item.text)

result = rag.retrieve(
    "How long do refunds take?",
    top_k=20,
    use_rerank=True,
    rerank_top_k=5,
)

result = rag.query(
    "How long do refunds take?",
    retrieval={"top_k": 20, "rerank": {"enabled": True, "top_k": 5}},
)

result = rag.retrieve(
    "What about the second step?",
    use_query_rewrite=True,
    session_context="The user is asking about refund setup.",
)

print(result.search_query)

result = rag.query(
    "What about the second step?",
    retrieval={"query_rewrite": {"enabled": True, "session_context": "Refund policy thread"}},
)