Skip to main content

Integration

You can assign Groq as your generation engine either by passing kwargs into the connection object or through environment variables.
import os
from arivu import Arivu

os.environ["GROQ_API_KEY"] = "..."

db = Arivu.connect(
    # DB arguments...
    llm_provider="groq",
    model="llama3-70b-8192"
)
Ensure your arivu-ai[llm] optional dependencies are installed via pip install arivu-ai[groq] if necessary.