37 lines
871 B
Python
37 lines
871 B
Python
"""
|
|
RAG Core - 公共 RAG 组件包
|
|
|
|
提供嵌入模型、向量存储和文档存储的公共功能，被 rag_indexer 和 app/rag 共用。
|
|
"""
|
|
|
|
from .embedders import LlamaCppEmbedder
|
|
from .vector_store import QdrantVectorStore
|
|
from .sparse_embedder import BM25SparseEmbedder, get_sparse_embedder
|
|
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",
|
|
"BM25SparseEmbedder",
|
|
"get_sparse_embedder",
|
|
"QDRANT_URL",
|
|
"QDRANT_API_KEY",
|
|
"LLAMACPP_EMBEDDING_URL",
|
|
"LLAMACPP_API_KEY",
|
|
"DB_URI",
|
|
"DOCSTORE_URI",
|
|
"PostgresDocStore",
|
|
"create_docstore",
|
|
"create_parent_retriever",
|
|
]
|