Documentation Assistant

from ragrails import QueryRewriteConfig, RagRails

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

for event in rag.scrape_stream("https://docs.example.com", mode="full", max_pages=500):
    if event["type"] == "page":
        print("indexed source page", event["data"]["url"])

rag.ingest(
    urls="https://docs.example.com",
    ingestion={"urls": {"mode": "full", "max_pages": 500}},
    storage={"collection": "docs"},
)

result = rag.chat(
    "How do I rotate an API key?",
    history=[],
    query_rewrite=QueryRewriteConfig(enabled=True),
)

print(result.answer)
print(result.sources)

ragrails setup-url --browser chromium

ragrails ingest \
  --source-url https://docs.example.com \
  --vector-db qdrant \
  --collection docs \
  --url http://localhost:6333 \
  --provider voyage \
  --model voyage-3

ragrails chat "How do I rotate an API key?" \
  --rewrite-query \
  --vector-db qdrant \
  --collection docs \
  --url http://localhost:6333 \
  --embedder-provider voyage \
  --embedder-model voyage-3 \
  --llm-provider openai \
  --llm-model gpt-4o-mini

curl -N -X POST http://127.0.0.1:8000/v1/ingest/url/stream \
  -H "Content-Type: application/json" \
  -d '{"url": "https://docs.example.com", "mode": "full", "max_pages": 500}'

curl -X POST http://127.0.0.1:8000/v1/pipelines/ingest \
  -H "Content-Type: application/json" \
  -d '{
    "urls": "https://docs.example.com",
    "ingestion": {"urls": {"mode": "full", "max_pages": 500}},
    "embedding": {"provider": "voyage", "model": "voyage-3"},
    "storage": {"vector_db": "qdrant", "collection": "docs", "url": "http://localhost:6333"}
  }'

curl -N -X POST http://127.0.0.1:8000/v1/chat/stream \
  -H "Content-Type: application/json" \
  -d '{
    "query": "How do I rotate an API key?",
    "query_rewrite": {"enabled": true},
    "collection": "docs",
    "url": "http://localhost:6333",
    "history": []
  }'