FinRAG-Agent/FinRAG-Pro

1"""RAG系统测试脚本"""2import sys3sys.path.insert(0, ".")4 5from rag_pipeline import get_rag_pipeline6 7# 初始化8pipe = get_rag_pipeline()9 10# 测试查询11test_queries = [12    "比亚迪2023年营收和净利润",13    "华为2023年研发投入",14    "营收",15    "净利润",16    "腾讯游戏收入",17]18 19for query in test_queries:20    print(f"\n{'='*50}")21    print(f"查询: {query}")22    results = pipe.retrieve(query, top_k=3)23    print(f"检索到 {len(results)} 个结果:")24    for r in results:25        print(f"  Score={r['score']:.4f}  |  {r['source']}")26        print(f"  → {r['content'][:80]}...")27    print()28