File size: 6,083 Bytes
89879a0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 |
"""
通過 API 測試貓砂查詢功能
"""
import requests
import json
def test_api_endpoints():
"""測試 API 端點"""
base_url = "http://localhost:7860"
print("🚀 開始 API 測試\n")
# 1. 測試健康檢查
print("1. 測試健康檢查...")
try:
response = requests.get(f"{base_url}/health", timeout=10)
print(f" 狀態碼: {response.status_code}")
if response.status_code == 200:
print(f" 回應: {response.json()}")
print(" ✅ 健康檢查通過\n")
except Exception as e:
print(f" ❌ 健康檢查失敗: {str(e)}\n")
return False
# 2. 測試智能路由 API
print("2. 測試智能路由 API...")
test_messages = [
"你好, 請問有沒有貓砂相關產品?",
"是否有推薦貓砂？",
"有什麼寵物用品？",
"查詢狗糧庫存",
"你好！今天天氣如何？" # 非商品查詢
]
for i, message in enumerate(test_messages, 1):
print(f"\n 測試 {i}: '{message}'")
try:
payload = {
"message": message,
"user_id": "test_user_api"
}
response = requests.post(
f"{base_url}/route",
json=payload,
timeout=30,
headers={"Content-Type": "application/json"}
)
print(f" 狀態碼: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f" 模式: {result.get('mode', 'unknown')}")
print(f" 成功: {result.get('success', False)}")
# 顯示回應內容
response_text = result.get('text', 'No response')
if len(response_text) > 150:
print(f" 回應: {response_text[:150]}...")
else:
print(f" 回應: {response_text}")
# 如果是商品查詢，顯示額外資訊
if result.get('products_found') is not None:
print(f" 找到商品: {result['products_found']}")
if result.get('has_recommendations'):
print(f" 包含推薦: {result['has_recommendations']}")
else:
print(f" ❌ API 錯誤: {response.text}")
except Exception as e:
print(f" ❌ 請求失敗: {str(e)}")
# 3. 測試專門的商品查詢 API
print(f"\n3. 測試專門的商品查詢 API...")
product_queries = [
"是否有推薦貓砂？",
"有什麼寵物用品？",
"查詢貓相關產品"
]
for i, query in enumerate(product_queries, 1):
print(f"\n 商品查詢 {i}: '{query}'")
try:
payload = {
"message": query,
"user_id": "test_user_product"
}
response = requests.post(
f"{base_url}/product-query",
json=payload,
timeout=30,
headers={"Content-Type": "application/json"}
)
print(f" 狀態碼: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f" 成功: {result.get('success', False)}")
response_text = result.get('text', 'No response')
if len(response_text) > 150:
print(f" 回應: {response_text[:150]}...")
else:
print(f" 回應: {response_text}")
if result.get('products_found') is not None:
print(f" 找到商品: {result['products_found']}")
if result.get('error'):
print(f" 錯誤: {result['error']}")
else:
print(f" ❌ API 錯誤: {response.text}")
except Exception as e:
print(f" ❌ 請求失敗: {str(e)}")
# 4. 測試傳統搜尋 API
print(f"\n4. 測試傳統搜尋 API...")
search_queries = [
"貓砂",
"寵物用品",
"貓"
]
for i, query in enumerate(search_queries, 1):
print(f"\n 搜尋 {i}: '{query}'")
try:
payload = {
"message": query,
"user_id": "test_user_search"
}
response = requests.post(
f"{base_url}/search",
json=payload,
timeout=30,
headers={"Content-Type": "application/json"}
)
print(f" 狀態碼: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f" 成功: {result.get('success', False)}")
response_text = result.get('text', 'No response')
if len(response_text) > 150:
print(f" 回應: {response_text[:150]}...")
else:
print(f" 回應: {response_text}")
else:
print(f" ❌ API 錯誤: {response.text}")
except Exception as e:
print(f" ❌ 請求失敗: {str(e)}")
def main():
"""主函數"""
print("🧪 貓砂查詢 API 測試")
print("=" * 50)
test_api_endpoints()
print("\n" + "=" * 50)
print("✅ API 測試完成！")
print("\n💡 分析建議：")
print("1. 檢查智能路由是否正確識別商品查詢意圖")
print("2. 比較不同 API 端點的回應差異")
print("3. 確認是否真的沒有貓砂相關商品，還是查詢邏輯問題")
if __name__ == "__main__":
main()
|