Files
ailine/backend/rag_core/__init__.py
2026-04-21 19:06:34 +08:00

34 lines
749 B
Python

"""
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",
]