83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
|
"""
|
|||
|
|
通过 API 测试知识库检索性能
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
import json
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
API_BASE = "http://localhost:8000"
|
|||
|
|
API_KEY = "oQfkSAbL8iafzyHxqb--G7zRWSOYJHvlzQxia2KpYms"
|
|||
|
|
TENANT_ID = "szmp@ash@2026"
|
|||
|
|
|
|||
|
|
def test_kb_search():
|
|||
|
|
"""测试知识库搜索 API"""
|
|||
|
|
print("=" * 80)
|
|||
|
|
print("测试知识库检索 API")
|
|||
|
|
print("=" * 80)
|
|||
|
|
|
|||
|
|
headers = {
|
|||
|
|
"Content-Type": "application/json",
|
|||
|
|
"X-API-Key": API_KEY,
|
|||
|
|
"X-Tenant-Id": TENANT_ID,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 测试数据
|
|||
|
|
test_cases = [
|
|||
|
|
{
|
|||
|
|
"name": "完整参数（含context过滤）",
|
|||
|
|
"data": {
|
|||
|
|
"query": "三年级语文学习",
|
|||
|
|
"scene": "学习方案",
|
|||
|
|
"top_k": 5,
|
|||
|
|
"context": {"grade": "三年级", "subject": "语文"},
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "简化参数（无context）",
|
|||
|
|
"data": {
|
|||
|
|
"query": "三年级语文学习",
|
|||
|
|
"scene": "学习方案",
|
|||
|
|
"top_k": 5,
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for test_case in test_cases:
|
|||
|
|
print(f"\n{'='*80}")
|
|||
|
|
print(f"测试: {test_case['name']}")
|
|||
|
|
print(f"{'='*80}")
|
|||
|
|
print(f"请求数据: {json.dumps(test_case['data'], ensure_ascii=False)}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
start = time.time()
|
|||
|
|
response = requests.post(
|
|||
|
|
f"{API_BASE}/api/v1/mid/kb-search-dynamic",
|
|||
|
|
headers=headers,
|
|||
|
|
json=test_case['data'],
|
|||
|
|
timeout=30,
|
|||
|
|
)
|
|||
|
|
elapsed = (time.time() - start) * 1000
|
|||
|
|
|
|||
|
|
print(f"\n响应状态: {response.status_code}")
|
|||
|
|
print(f"总耗时: {elapsed:.2f} ms")
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
result = response.json()
|
|||
|
|
print(f"API 结果:")
|
|||
|
|
print(f" success: {result.get('success')}")
|
|||
|
|
print(f" hits count: {len(result.get('hits', []))}")
|
|||
|
|
print(f" duration_ms: {result.get('duration_ms')}")
|
|||
|
|
print(f" applied_filter: {result.get('applied_filter')}")
|
|||
|
|
else:
|
|||
|
|
print(f"错误: {response.text}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"请求失败: {e}")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 80)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_kb_search()
|