ailine/rag_core/__init__.py

"""
RAG Core - 公共 RAG 组件包

提供嵌入模型、向量存储和文档存储的公共功能，被 rag_indexer 和 app/rag 共用。
"""

from rag_core.embedders import LlamaCppEmbedder
from rag_core.vector_store import QdrantVectorStore, QDRANT_URL, QDRANT_API_KEY
from rag_core.store import PostgresDocStore, create_docstore
from rag_core.retriever_factory import create_parent_retriever


__all__ = [
    "LlamaCppEmbedder",
    "QdrantVectorStore",
    "QDRANT_URL",
    "QDRANT_API_KEY",
    "PostgresDocStore",
    "create_docstore",
    "create_parent_retriever",
]