ailine/backend/rag_core/__init__.py

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

from .embedders import get_embeddings, get_embedding_dimension
from .vector_store import QdrantHybridStore
from .sparse_embedder import BM25SparseEmbedder, get_sparse_embedder
from .store import PostgresDocStore, create_docstore
from .client import create_qdrant_client, create_async_qdrant_client
from .config import (
    QDRANT_URL,
    QDRANT_API_KEY,
    LLAMACPP_EMBEDDING_URL,
    LLAMACPP_API_KEY,
    DB_URI,
    DOCSTORE_URI,
)


__all__ = [
    "get_embeddings",
    "get_embedding_dimension",
    "QdrantHybridStore",
    "BM25SparseEmbedder",
    "get_sparse_embedder",
    "QDRANT_URL",
    "QDRANT_API_KEY",
    "LLAMACPP_EMBEDDING_URL",
    "LLAMACPP_API_KEY",
    "DB_URI",
    "DOCSTORE_URI",
    "PostgresDocStore",
    "create_docstore",
    "create_qdrant_client",
    "create_async_qdrant_client",
]