52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
#!/usr/bin/env python
|
||
"""
|
||
简单的模型测试
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import json
|
||
|
||
# 设置路径
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
server_dir = os.path.join(current_dir, 'server')
|
||
os.chdir(server_dir)
|
||
sys.path.insert(0, server_dir)
|
||
|
||
print(f"工作目录: {os.getcwd()}")
|
||
|
||
try:
|
||
# 导入模型管理器
|
||
from utils.model_manager import model_manager
|
||
|
||
print(f"模型目录: {model_manager.model_dir}")
|
||
|
||
# 获取模型
|
||
models = model_manager.list_models()
|
||
print(f"原始模型数量: {len(models)}")
|
||
|
||
if models:
|
||
model = models[0]
|
||
print("第一个模型原始数据:")
|
||
for key, value in model.items():
|
||
print(f" {key}: {value} ({type(value)})")
|
||
|
||
# 测试JSON序列化
|
||
try:
|
||
json_str = json.dumps(model, ensure_ascii=False, default=str)
|
||
print("JSON序列化成功")
|
||
except Exception as e:
|
||
print(f"JSON序列化失败: {e}")
|
||
# 找出有问题的字段
|
||
for key, value in model.items():
|
||
try:
|
||
json.dumps({key: value}, default=str)
|
||
except Exception as field_error:
|
||
print(f" 问题字段 {key}: {field_error}")
|
||
|
||
print("\n=== 成功：模型管理器在API环境下工作正常 ===")
|
||
|
||
except Exception as e:
|
||
print(f"错误: {e}")
|
||
import traceback
|
||
traceback.print_exc() |