Spaces:
Sleeping
Sleeping
File size: 839 Bytes
939a9f4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
"""
RAG Package
===========
Modular Retrieval-Augmented Generation system with Ollama + Groq
"""
from .chunker import chunk_text, Chunk, chunk_documents
from .embeddings import OllamaEmbeddingClient, cosine_similarity
from .vector_store import ChromaVectorStore, RetrievalResult
from .llm import GroqLLMClient, build_context_string
from .pdf_processor import PDFProcessor
from .pipeline import RAGPipeline, RAGConfig
__all__ = [
# Chunking
"chunk_text",
"Chunk",
"chunk_documents",
# Embeddings
"OllamaEmbeddingClient",
"cosine_similarity",
# Vector Store
"ChromaVectorStore",
"SimpleVectorStore",
"RetrievalResult",
# LLM
"GroqLLMClient",
"build_context_string",
# PDF Processing
"PDFProcessor",
# Pipeline
"RAGPipeline",
"RAGConfig",
]
__version__ = "0.1.0" |