FinRAG-Agent/FinRAG-Pro

1"""测试RAG API"""2import requests3 4BASE = "http://127.0.0.1:5000"5 6# 测试状态7r = requests.get(f"{BASE}/api/status")8print("=== 系统状态 ===")9status = r.json()10print(f"RAG就绪: {status['rag_ready']}")11print(f"知识库: {status['knowledge_base']['total_documents']}份文档")12print()13 14# 测试查询（无API Key）15test_queries = [16    "比亚迪2023年营收和净利润是多少？",17    "华为2023年研发投入多少？",18    "腾讯2023年游戏业务收入",19    "宁德时代动力电池市场份额",20    "贵州茅台2023年净利润",21]22 23for q in test_queries:24    r = requests.post(f"{BASE}/api/query", json={"query": q})25    data = r.json()26    print(f"=== 查询: {q} ===")27    print(f"Success: {data['success']}")28    sources = data.get("sources", [])29    if sources:30        print("来源:")31        for s in sources:32            print(f"  - {s['doc_title']}")33    else:34        print("  (无匹配来源)")35    print(f"Answer前200字: {data.get('answer','')[:200]}")36    print()37