src/rag/__init__.py · first-aid-rag-assistant

# RAG class hierarchy:
# RAGBase (rag_helper) → RAGWithUsage (rag_usage) → RAGWithMetrics (metrics)
#   → RAGTraced → RAGProduction (starter)
from rag.starter import RAGProduction, RAGTraced
from rag.metrics import RAGWithMetrics, LLMCallRecord
from rag.rag_usage import RAGWithUsage
from rag.rag_helper import RAGBase, INSTRUCTIONS, PROMPT_TEMPLATE, rewrite_query

__all__ = [
    "RAGProduction", "RAGTraced", "RAGWithMetrics", "LLMCallRecord",
    "RAGWithUsage", "RAGBase", "INSTRUCTIONS", "PROMPT_TEMPLATE", "rewrite_query",
]