rag/__init__.py

"""RAG module for retrieval-augmented generation."""

from rag.core import RAGSystem
from rag.generator import TinyLlamaGenerator, ResponseGenerator
from rag.gemma_generator import GemmaGenerator
from rag.retriever import VectorStoreFAISS
from rag.embeddings import EmbeddingModel

__all__ = [
    "RAGSystem",
    "TinyLlamaGenerator",
    "GemmaGenerator",
    "ResponseGenerator",
    "VectorStoreFAISS",
    "EmbeddingModel",
]