ailine/backend/rag_core/__init__.py

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

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

from .embedders import LlamaCppEmbedder
from .vector_store import QdrantVectorStore
from .store import PostgresDocStore, create_docstore
from .retriever_factory import create_parent_retriever
from .config import (
    QDRANT_URL,
    QDRANT_API_KEY,
    LLAMACPP_EMBEDDING_URL,
    LLAMACPP_API_KEY,
    DB_URI,
    DOCSTORE_URI,
)


__all__ = [
    "LlamaCppEmbedder",
    "QdrantVectorStore",
    "QDRANT_URL",
    "QDRANT_API_KEY",
    "LLAMACPP_EMBEDDING_URL",
    "LLAMACPP_API_KEY",
    "DB_URI",
    "DOCSTORE_URI",
    "PostgresDocStore",
    "create_docstore",
    "create_parent_retriever",
]