47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
#!/usr/bin/env python
|
||
"""
|
||
测试修复后的路径
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
sys.path.append('server')
|
||
|
||
def test_from_server_dir():
|
||
"""从server目录模拟API调用"""
|
||
# 模拟API服务器从server目录运行
|
||
original_cwd = os.getcwd()
|
||
|
||
try:
|
||
# 切换到server目录
|
||
server_dir = os.path.join(os.getcwd(), 'server')
|
||
os.chdir(server_dir)
|
||
print(f"切换到server目录: {os.getcwd()}")
|
||
|
||
# 添加当前目录到路径（server目录）
|
||
if server_dir not in sys.path:
|
||
sys.path.insert(0, server_dir)
|
||
|
||
# 导入模型管理器
|
||
from utils.model_manager import model_manager
|
||
|
||
print(f"模型目录: {model_manager.model_dir}")
|
||
print(f"模型目录存在: {os.path.exists(model_manager.model_dir)}")
|
||
|
||
# 测试模型列表
|
||
models = model_manager.list_models()
|
||
print(f"找到模型: {len(models)} 个")
|
||
|
||
if models:
|
||
model = models[0]
|
||
print(f"模型文件名: {model.get('filename')}")
|
||
print(f"产品名称: {model.get('product_name')}")
|
||
|
||
return len(models) > 0
|
||
|
||
finally:
|
||
# 恢复原始工作目录
|
||
os.chdir(original_cwd)
|
||
|
||
if __name__ == "__main__":
|
||
test_from_server_dir() |