root
4209386c77
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m22s
- 重构所有模块导入，移除 sys.path.insert - 统一使用 from backend.xxx 的绝对导入方式 - rag_core 包内使用相对导入（from .xxx） - 移动 visualize_graph.py 到 tools/ 目录 - 添加必要的 __init__.py 文件 - 清理废弃文档和脚本
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
测试 app/rag/retriever.py 里的混合检索函数
|
|
"""
|
|
|
|
import asyncio
|
|
import os
|
|
import sys
|
|
|
|
from backend.app.rag.retriever import create_hybrid_retriever, create_parent_hybrid_retriever
|
|
|
|
|
|
def test_hybrid_retriever():
|
|
"""测试混合检索器"""
|
|
print("="*70)
|
|
print("测试 HybridRetriever...")
|
|
print("="*70)
|
|
|
|
retriever = create_hybrid_retriever(collection_name="rag_documents", search_k=3)
|
|
results = retriever.invoke("黄双银")
|
|
|
|
print(f"\n找到 {len(results)} 个结果\n")
|
|
for i, doc in enumerate(results):
|
|
print(f"--- 结果 {i+1} ---")
|
|
print(doc.page_content[:200])
|
|
print()
|
|
|
|
|
|
def test_parent_hybrid_retriever():
|
|
"""测试父子混合检索器"""
|
|
print("\n" + "="*70)
|
|
print("测试 ParentHybridRetriever...")
|
|
print("="*70)
|
|
|
|
retriever = create_parent_hybrid_retriever(
|
|
collection_name="rag_documents",
|
|
search_k=3,
|
|
use_docstore=False
|
|
)
|
|
results = retriever.invoke("黄双银")
|
|
|
|
print(f"\n找到 {len(results)} 个结果\n")
|
|
for i, doc in enumerate(results):
|
|
print(f"--- 结果 {i+1} ---")
|
|
print(doc.page_content[:300])
|
|
print()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_hybrid_retriever()
|
|
test_parent_hybrid_retriever()
|