File size: 970 Bytes
9dbb7e3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
"""
MCP Tool Handlers — Backward-compatible aggregation entry point.
Core logic has been split into individual files (get_*.py and run_imagegen.py).
"""

from .get_task_list import handle_get_task_list
from .get_model_architecture_list import handle_get_model_architecture_list
from .get_model_list import handle_get_model_list
from .get_feature_list import handle_get_feature_list
from .get_model_features import handle_get_model_features
from .get_chain_schema import handle_get_chain_schema
from .run_imagegen import handle_run_imagegen
from .get_task_status import handle_get_task_status
from .common import (
    _TASK_DEFINITIONS,
    _TASKS_DB,
    _load_yaml,
    _execute_imagegen_pipeline,
)

__all__ = [
    "handle_get_task_list",
    "handle_get_model_architecture_list",
    "handle_get_model_list",
    "handle_get_feature_list",
    "handle_get_model_features",
    "handle_get_chain_schema",
    "handle_run_imagegen",
    "handle_get_task_status",
]