ailine/app/rag/__init__.py

"""
RAG 检索与生成模块

提供在线检索与生成功能，包括：
- 基础向量检索
- 重排序
- RAG-Fusion
- Agentic RAG

示例用法：
    >>> from app.rag import RAGPipeline, search_knowledge_base
    >>> from rag_core import LlamaCppEmbedder
    >>> 
    >>> embeddings = LlamaCppEmbedder()
    >>> pipeline = RAGPipeline(embeddings=embeddings)
    >>> 
    >>> documents = pipeline.retrieve("戏耍貂蝉美女")
    >>> context = pipeline.format_context(documents)
"""

from .retriever import (
    create_base_retriever,
    create_hybrid_retriever,
    # create_ensemble_retriever,
    create_qdrant_client,
)
from .reranker import CrossEncoderReranker
from .query_transform import MultiQueryTransformer
from .pipeline import RAGPipeline, RAGLevel
from .tools import search_knowledge_base, search_knowledge_base_sync


__all__ = [
    # 检索器
    "create_base_retriever",
    "create_hybrid_retriever",
    # "create_ensemble_retriever",
    "create_qdrant_client",
    
    # 重排序器
    "CrossEncoderReranker",
    
    # 查询转换器
    "MultiQueryTransformer",
    
    # 流水线
    "RAGPipeline",
    "RAGLevel",
    
    # 工具
    "search_knowledge_base",
    "search_knowledge_base_sync",
]