Your agent works.
Until a customer touches it.

tools = [{"name": "retrieve", ...}, {"name": "rerank", ...}, ...]

prompt = f"""You are a RAG assistant. CRITICAL: Use ONLY retrieved info.

## QUERY CLASSIFICATION (classify BEFORE acting):
- Simple factual → retrieve(k=3) → extract_answer
- Complex/deep dive → retrieve(k=8) → rerank(k=3) → extract
- Comparison → retrieve(topic_A) + retrieve(topic_B) → compare

## RESPONSE FORMAT (STRICT):
Return JSON: {{"thinking": "...", "tool_calls": [...],
  "final_answer": "...", "sources": [...], "confidence": 0.0-1.0}}

## TOOL PARAMETER RULES:
- retrieve: k must be 3-10, query must be <100 chars
- rerank: only after retrieve, k <= original k
- extract_answer: requires non-empty doc list

## CRITICAL CONSTRAINTS:
❌ NEVER hallucinate or make up information
❌ NEVER call extract_answer without first calling retrieve
❌ NEVER exceed confidence 0.9 without source validation
✓ ALWAYS cite sources with doc_id references
✓ ALWAYS include confidence scores

REMEMBER: You are a RETRIEVAL assistant, not a knowledge base.
Query: {query}"""
# ... (500 more tokens of examples and error handling)
response = llm.complete(prompt, tools=tools)

class RAGAgent(PlanAndExecute):
    @primitive
    def retrieve(self, q: str, k: int = 5) -> list[Document]: ...
    @primitive
    def rerank(self, docs, q: str) -> list[Document]: ...
    @primitive
    def extract(self, docs, q: str) -> str: ...

    @decomposition(intent="What is machine learning?")
    def simple_qa(self):
        docs = self.retrieve("machine learning definition", k=3)
        return self.extract(docs, "What is machine learning?")

    @decomposition(intent="Explain the architecture of transformers")
    def deep_dive(self):
        docs = self.retrieve("transformer architecture innovations", k=8)
        ranked = self.rerank(docs, "transformer architecture")
        return self.extract(ranked, "Explain transformer architecture")

    @decomposition(intent="Compare React vs Vue")
    def compare(self):
        docs = self.retrieve("React") + self.retrieve("Vue")
        return self.extract(docs, "Compare React vs Vue")

# Intent matching happens automatically:
answer = agent.run("What is attention?")
deep_dive = agent.run("Deep dive on transformers")
comparison = agent.run("React vs Vue")