67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
调试API是否使用最新代码
|
||
"""
|
||
import urllib.request
|
||
import json
|
||
import sys
|
||
import os
|
||
|
||
# 添加server目录到路径，直接测试ModelManager
|
||
server_path = os.path.join(os.path.dirname(__file__), 'server')
|
||
sys.path.insert(0, server_path)
|
||
|
||
def test_modelmanager_vs_api():
|
||
print("=== 对比ModelManager与API返回的数据 ===")
|
||
|
||
# 1. 直接测试ModelManager
|
||
print("\n1. 直接测试ModelManager:")
|
||
try:
|
||
from utils.model_manager import ModelManager
|
||
manager = ModelManager()
|
||
models = manager.list_models()
|
||
|
||
print(f"ModelManager找到 {len(models)} 个模型")
|
||
if models:
|
||
model = models[0]
|
||
print(f"第一个模型filename: '{model.get('filename', 'MISSING')}'")
|
||
print(f"第一个模型product_id: '{model.get('product_id', 'MISSING')}'")
|
||
print(f"第一个模型model_type: '{model.get('model_type', 'MISSING')}'")
|
||
|
||
except Exception as e:
|
||
print(f"ModelManager测试失败: {e}")
|
||
|
||
# 2. 测试API端点
|
||
print("\n2. 测试API端点:")
|
||
try:
|
||
url = 'http://localhost:5000/api/models'
|
||
with urllib.request.urlopen(url) as response:
|
||
data = response.read().decode('utf-8')
|
||
result = json.loads(data)
|
||
|
||
models = result.get('data', [])
|
||
print(f"API返回 {len(models)} 个模型")
|
||
if models:
|
||
model = models[0]
|
||
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')}'")
|
||
|
||
except Exception as e:
|
||
print(f"API测试失败: {e}")
|
||
|
||
# 3. 检查API代码修改是否生效
|
||
print("\n3. 检查API代码修改:")
|
||
try:
|
||
with open('server/api.py', 'r', encoding='utf-8') as f:
|
||
content = f.read()
|
||
if "'filename': model.get('filename', '')" in content:
|
||
print("✓ API代码修改已存在")
|
||
else:
|
||
print("✗ API代码修改未找到")
|
||
|
||
except Exception as e:
|
||
print(f"代码检查失败: {e}")
|
||
|
||
if __name__ == "__main__":
|
||
test_modelmanager_vs_api() |