mcp_tools/get_sampler_scheduler_list.py

"""
MCP Tool: get_sampler_scheduler_list
Query all supported Sampler algorithms and Noise Schedulers available for image generation tasks.
"""

try:
    from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
except Exception:
    SAMPLER_CHOICES = ['euler', 'dpmpp_2m_sde_gpu']
    SCHEDULER_CHOICES = ['normal', 'karras']


def handle_get_sampler_scheduler_list() -> dict:
    """Query all supported Sampler algorithms and Noise Schedulers available for image generation tasks."""
    return {
        "samplers": list(SAMPLER_CHOICES),
        "schedulers": list(SCHEDULER_CHOICES),
    }