ML / src /__init__.py
vivekbarsagadey's picture
feat: implement RAG and CAG document ingestion and knowledge base services
29118da
Raw History Blame Contribute Delete
1.04 kB
"""Package initialization for src."""
from src.config import (
HardwareSpecs,
ModelConfig,
GenerationConfig,
DocumentConfig,
RAGConfig,
)
from src.domain import ChatTurn, MediaItem, MediaKind
from src.registry import ModelRegistry, BaseModelRegistry, TransformersModelRegistry
from src.ingestion import (
KnowledgeBaseService,
BaseRetrievalStrategy,
DocumentItem,
DocumentChunk,
DocumentParser,
RAGStrategy,
CAGStrategy,
)
from src.services.inference_service import InferenceService
from src.ui import render_hardware_header, create_control_panel, STUDIO_CSS
from src.utils import get_spaces_accelerator
__all__ = [
"HardwareSpecs",
"ModelConfig",
"GenerationConfig",
"DocumentConfig",
"RAGConfig",
"ChatTurn",
"MediaItem",
"MediaKind",
"ModelRegistry",
"BaseModelRegistry",
"TransformersModelRegistry",
"KnowledgeBaseService",
"BaseRetrievalStrategy",
"DocumentItem",
"DocumentChunk",
"DocumentParser",
"InferenceService",
]