LlamaIndex

pip install "decimalai[llamaindex]" "llama-index-core>=0.12.0"

import decimalai
decimalai.init(api_key="dai_sk_...", llamaindex=True)

# Query engines, retrievers, and synthesizers are traced
response = query_engine.query("What is DecimalAI?")

import decimalai
decimalai.init(api_key="dai_sk_...")

from decimalai.llamaindex import instrument
instrument(agent_name="my-rag-agent")

from decimalai.providers import agent_run

with agent_run("support-rag"):
    support_engine.query("Where's my order?")     # filed under support-rag

with agent_run("billing-rag"):
    billing_engine.query("Why was I charged twice?")   # filed under billing-rag

def handle(tenant, question):          # one request, one thread
    with agent_run(f"rag-{tenant}"):
        return engine.query(question)