54 lines
2.4 KiB
Python
54 lines
2.4 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
测试新的API端点
|
|||
|
|
"""
|
|||
|
|
import urllib.request
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
def test_new_endpoint():
|
|||
|
|
try:
|
|||
|
|
print("=== 测试新端点 /api/models/test ===")
|
|||
|
|
|
|||
|
|
url = 'http://localhost:5000/api/models/test'
|
|||
|
|
with urllib.request.urlopen(url) as response:
|
|||
|
|
data = response.read().decode('utf-8')
|
|||
|
|
result = json.loads(data)
|
|||
|
|
|
|||
|
|
print(f"响应状态: {response.status}")
|
|||
|
|
print(f"测试名称: {result.get('test_name', 'N/A')}")
|
|||
|
|
print(f"模型目录: {result.get('model_dir', 'N/A')}")
|
|||
|
|
print(f"目录存在: {result.get('dir_exists', 'N/A')}")
|
|||
|
|
print(f"找到模型: {result.get('models_found', 0)} 个")
|
|||
|
|
|
|||
|
|
models = result.get('models', [])
|
|||
|
|
for i, model in enumerate(models):
|
|||
|
|
print(f"\n模型 {i+1}:")
|
|||
|
|
print(f" filename: '{model.get('filename', 'MISSING')}'")
|
|||
|
|
print(f" model_id: '{model.get('model_id', 'MISSING')}'")
|
|||
|
|
print(f" product_id: '{model.get('product_id', 'MISSING')}'")
|
|||
|
|
print(f" model_type: '{model.get('model_type', 'MISSING')}'")
|
|||
|
|
|
|||
|
|
if model.get('filename') != 'MISSING' and model.get('model_id') != 'GENERATED_MISSING':
|
|||
|
|
print(" ✓ 该模型数据正常")
|
|||
|
|
else:
|
|||
|
|
print(" ✗ 该模型数据有问题")
|
|||
|
|
|
|||
|
|
if result.get('models_found', 0) > 0:
|
|||
|
|
working_models = sum(1 for m in models
|
|||
|
|
if m.get('filename') != 'MISSING' and m.get('model_id') != 'GENERATED_MISSING')
|
|||
|
|
if working_models == result.get('models_found', 0):
|
|||
|
|
print(f"\n✓ 新端点测试成功！所有 {working_models} 个模型数据正常")
|
|||
|
|
print("✓ ModelManager修复已生效")
|
|||
|
|
print("✓ 应该重启API服务器以应用到主端点 /api/models")
|
|||
|
|
else:
|
|||
|
|
print(f"\n⚠ 部分模型有问题: {working_models}/{result.get('models_found', 0)} 正常")
|
|||
|
|
else:
|
|||
|
|
print("\n✗ 没有找到模型文件")
|
|||
|
|
|
|||
|
|
except urllib.error.HTTPError as e:
|
|||
|
|
print(f"HTTP错误: {e.code} - 可能端点不存在，需要重启API服务器")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"测试失败: {e}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_new_endpoint()
|