Skip to main content

Package Map

Built and consumed in system-building order.

Phase 0 — Foundation

PackageResponsibility
rag_axis.coreDocument, Chunk, RetrievalResult, CostReport, ContextBudget, PipelineResult, named error types
rag_axis.adaptersLLMAdapter, EmbedderAdapter, VectorStoreAdapter Protocols + reference implementations

Phase 1 — Pre-Retrieval

PackageResponsibility
rag_axis.ingestionLoaders, parsers, metadata extraction, document registry
rag_axis.chunkingFixed, semantic, structural, parent-child strategies
rag_axis.embeddingBatch embedding, cost tracking, model version lock

Phase 2 — Retrieval Core

PackageResponsibility
rag_axis.retrievalDense, sparse, hybrid RRF, metadata filters
rag_axis.rerankingCross-encoder, score normalisation, ScoreCollapseWarning
rag_axis.contextBudget enforcement, truncation, deduplication, ordering, provenance
rag_axis.generationPrompt assembly, LLM call, output parsing, citation injection

Phase 3 — System Layer

PackageResponsibility
rag_axis.guardsInput guards, output guards, fallback strategies
rag_axis.cacheSemantic cache, exact cache, embedding cache
rag_axis.telemetryOTel spans, cost aggregation, structlog
rag_axis.benchEval hooks, RAGAS, DeepEval, TruLens bridges, drift detection

Phase 4 — Advanced

PackageResponsibility
rag_axis.corpusVersioning, staleness, incremental updates, embedding model migration
rag_axis.knowledgeGraphRAG, entity extraction, multi-hop traversal

Dependency Rules

rag_axis.core aiprims.rag + aiprims.core + pydantic only rag_axis.adapters rag_axis.core rag_axis.retrieval rag_axis.core + rag_axis.adapters rag_axis.reranking rag_axis.core + rag_axis.retrieval rag_axis.context rag_axis.core + rag_axis.retrieval rag_axis.generation rag_axis.core + rag_axis.adapters rag_axis.bench rag_axis.core only rag_axis.telemetry rag_axis.core
One-way. No circular imports. No exceptions.