83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
强制重新加载API模块并测试
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
import importlib
|
|||
|
|
|
|||
|
|
# 添加server目录到路径
|
|||
|
|
server_path = os.path.join(os.path.dirname(__file__), 'server')
|
|||
|
|
sys.path.insert(0, server_path)
|
|||
|
|
|
|||
|
|
def force_reload_test():
|
|||
|
|
print("=== 强制重新加载测试 ===")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 清理模块缓存
|
|||
|
|
modules_to_clear = []
|
|||
|
|
for module_name in list(sys.modules.keys()):
|
|||
|
|
if 'model_manager' in module_name or 'utils' in module_name:
|
|||
|
|
modules_to_clear.append(module_name)
|
|||
|
|
|
|||
|
|
for module_name in modules_to_clear:
|
|||
|
|
if module_name in sys.modules:
|
|||
|
|
del sys.modules[module_name]
|
|||
|
|
print(f"清理模块缓存: {module_name}")
|
|||
|
|
|
|||
|
|
# 重新导入
|
|||
|
|
from utils.model_manager import ModelManager
|
|||
|
|
|
|||
|
|
# 测试ModelManager
|
|||
|
|
manager = ModelManager()
|
|||
|
|
models = manager.list_models()
|
|||
|
|
|
|||
|
|
print(f"\n重新加载后，ModelManager找到 {len(models)} 个模型")
|
|||
|
|
|
|||
|
|
# 手动执行API端点逻辑
|
|||
|
|
print("\n=== 手动模拟API端点逻辑 ===")
|
|||
|
|
|
|||
|
|
formatted_models = []
|
|||
|
|
for model in models:
|
|||
|
|
# 生成model_id的逻辑（与API相同）
|
|||
|
|
model_id = model.get('filename', '').replace('.pth', '')
|
|||
|
|
if not model_id:
|
|||
|
|
# 备用方案
|
|||
|
|
product_id = model.get('product_id', 'unknown')
|
|||
|
|
model_type = model.get('model_type', 'unknown')
|
|||
|
|
version = model.get('version', 'v1')
|
|||
|
|
model_id = f"{model_type}_product_{product_id}_{version}"
|
|||
|
|
|
|||
|
|
formatted_model = {
|
|||
|
|
'model_id': model_id,
|
|||
|
|
'filename': model.get('filename', ''), # 关键字段
|
|||
|
|
'product_id': model.get('product_id', ''),
|
|||
|
|
'model_type': model.get('model_type', ''),
|
|||
|
|
'version': model.get('version', 'v1'),
|
|||
|
|
'metrics': model.get('metrics', {}),
|
|||
|
|
}
|
|||
|
|
formatted_models.append(formatted_model)
|
|||
|
|
|
|||
|
|
print(f"格式化后模型数量: {len(formatted_models)}")
|
|||
|
|
|
|||
|
|
for i, model in enumerate(formatted_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')}'")
|
|||
|
|
|
|||
|
|
if model.get('filename') and model.get('model_id'):
|
|||
|
|
print(" ✓ 该模型应该能正常显示详情和删除按钮")
|
|||
|
|
else:
|
|||
|
|
print(" ✗ 该模型有问题")
|
|||
|
|
|
|||
|
|
return formatted_models
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"测试失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
force_reload_test()
|