40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""
|
|
测试节点加载和API功能
|
|
"""
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# 添加 server 目录到路径
|
|
sys.path.insert(0, str(Path(__file__).parent))
|
|
|
|
from app.core.node_loader import load_builtin_nodes
|
|
from app.core.node_registry import NodeRegistry
|
|
|
|
print("=" * 60)
|
|
print("测试节点加载功能")
|
|
print("=" * 60)
|
|
|
|
# 加载内置节点
|
|
result = load_builtin_nodes()
|
|
|
|
print(f"\n📊 加载统计:")
|
|
print(f" - 模块加载: {result['modules_loaded']} 个")
|
|
print(f" - 节点注册: {result['nodes_registered']} 个")
|
|
print(f" - 分类数量: {len(result['categories'])} 个")
|
|
|
|
print(f"\n📁 节点分类:")
|
|
for category in result['categories']:
|
|
nodes = NodeRegistry.get_by_category(category)
|
|
print(f" - {category}: {len(nodes)} 个节点")
|
|
|
|
print(f"\n📦 所有节点:")
|
|
all_nodes = NodeRegistry.list_all()
|
|
for class_name, node_class in all_nodes.items():
|
|
metadata = node_class.get_metadata()
|
|
print(f" - {metadata['display_name']} [{class_name}]")
|
|
print(f" 分类: {metadata['category']}")
|
|
print(f" 输入: {len(metadata['inputs'])} | 输出: {len(metadata['outputs'])} | 参数: {len(metadata['params'])}")
|
|
|
|
print(f"\n✅ 测试完成!")
|
|
print(f"总计: {len(all_nodes)} 个节点已注册")
|