检索器重构

app/rag/__init__.py

@@ -1,22 +1,53 @@
 """
-在线 RAG 检索与生成系统
+RAG 检索与生成模块
 
-提供高级RAG检索功能，支持混合检索、重排序、RAG-Fusion和多路查询改写。
+提供在线检索与生成功能，包括：
+- 基础向量检索
+- 重排序
+- 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 .pipeline import RAGPipeline
-from .retriever import create_hybrid_retriever, create_base_retriever
+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 .tools import search_knowledge_base_tool
+from .pipeline import RAGPipeline, RAGLevel
+from .tools import search_knowledge_base, search_knowledge_base_sync
+
 
 __all__ = [
-    "RAGPipeline",
-    "create_hybrid_retriever",
+    # 检索器
     "create_base_retriever",
+    "create_hybrid_retriever",
+    # "create_ensemble_retriever",
+    "create_qdrant_client",
+    
+    # 重排序器
     "CrossEncoderReranker",
+    
+    # 查询转换器
     "MultiQueryTransformer",
-    "search_knowledge_base_tool",
+    
+    # 流水线
+    "RAGPipeline",
+    "RAGLevel",
+    
+    # 工具
+    "search_knowledge_base",
+    "search_knowledge_base_sync",
 ]
-
-__version__ = "0.1.0"