Query

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"},
)

result = rag.query("How long do I have to request a refund?", retrieval={"top_k": 5})

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

result = rag.query(
    "What about the second step?",
    retrieval={
        "top_k": 20,
        "query_rewrite": {
            "enabled": True,
            "session_context": "The user is asking about refund setup.",
        },
        "rerank": {
            "enabled": True,
            "provider": "voyage",
            "model": "rerank-2-lite",
            "top_k": 5,
        },
    },
)

history = []
result = rag.chat(
    "How long do I have to request a refund?",
    history=history,
    persona="Answer only from the support policy context.",
)

print(result.answer)
history = result.history

for event in rag.chat_stream("How long do I have to request a refund?", history=[]):
    if event["type"] == "token":
        print(event["data"]["text"], end="")
    elif event["type"] == "final":
        result = event["data"]