Add get_sampler_scheduler_list MCP Tools.

mcp_tools/__init__.py

@@ -8,15 +8,22 @@ 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_sampler_scheduler_list import handle_get_sampler_scheduler_list
 from .run import handle_run
 from .get_task_status import handle_get_task_status
 from .error_schema import make_error, make_validation_error, make_not_found_error
-from .mcp_gradio_integration import (
-    register_high_level_mcp_apis,
-    cleanup_dependencies_api_names,
-    patch_gradio_api_suppression,
-    HIGH_LEVEL_MCP_API_NAMES,
-)
+try:
+    from .mcp_gradio_integration import (
+        register_high_level_mcp_apis,
+        cleanup_dependencies_api_names,
+        patch_gradio_api_suppression,
+        HIGH_LEVEL_MCP_API_NAMES,
+    )
+except ImportError:
+    register_high_level_mcp_apis = None
+    cleanup_dependencies_api_names = None
+    patch_gradio_api_suppression = None
+    HIGH_LEVEL_MCP_API_NAMES = set()
 
 MCP_FUNCTIONS = [
     handle_get_task_list,
@@ -24,6 +31,7 @@ MCP_FUNCTIONS = [
     handle_get_model_list,
     handle_get_feature_list,
     handle_get_model_features,
+    handle_get_sampler_scheduler_list,
     handle_run,
     handle_get_task_status,
 ]
@@ -34,6 +42,7 @@ __all__ = [
     "handle_get_model_list",
     "handle_get_feature_list",
     "handle_get_model_features",
+    "handle_get_sampler_scheduler_list",
     "handle_run",
     "handle_get_task_status",
     "make_error",

mcp_tools/get_sampler_scheduler_list.py

@@ -0,0 +1,18 @@
+"""
+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),
+    }

mcp_tools/mcp_gradio_integration.py

@@ -15,6 +15,7 @@ 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_sampler_scheduler_list import handle_get_sampler_scheduler_list
 from .run import handle_run
 from .get_task_status import handle_get_task_status
 
@@ -24,6 +25,7 @@ HIGH_LEVEL_MCP_API_NAMES = {
     "get_model_list",
     "get_feature_list",
     "get_model_features",
+    "get_sampler_scheduler_list",
     "run",
     "get_task_status",
 }
@@ -48,7 +50,7 @@ def patch_gradio_api_suppression():
 def cleanup_dependencies_api_names(demo):
     """
     Clean up residual auto-generated API names in demo.fns and demo.dependencies.
-    Force only the 7 high-level abstract MCP APIs to be exposed as public endpoints.
+    Force only high-level abstract MCP APIs to be exposed as public endpoints.
     """
     for fn in demo.fns.values():
         api_name = getattr(fn, "api_name", None)
@@ -71,7 +73,7 @@ def cleanup_dependencies_api_names(demo):
 
 def register_high_level_mcp_apis(demo):
     """
-    Explicitly register 7 high-level abstract MCP API endpoints on the Gradio demo using gr.api.
+    Explicitly register high-level abstract MCP API endpoints on the Gradio demo using gr.api.
     Using gr.api() never adds any visual UI components (such as Row, Textbox, Button, etc.), avoiding duplicate interface rendering.
     """
     def get_task_list() -> list:
@@ -102,6 +104,10 @@ def register_high_level_mcp_apis(demo):
         model_str = model.strip() if isinstance(model, str) else ""
         return sanitize_keys(handle_get_model_features(model_str))
 
+    def get_sampler_scheduler_list() -> dict:
+        """Query all supported Sampler algorithms and Noise Schedulers available for image generation tasks."""
+        return sanitize_keys(handle_get_sampler_scheduler_list())
+
     def run(json_params: str = "{}") -> dict:
         """[Recommended Discovery Flow Step 4] Unified image generation task execution interface. Accepts a JSON string or dict of parameters.
 
@@ -142,6 +148,7 @@ Paste-and-Run Example (With chain):
         get_model_list,
         get_feature_list,
         get_model_features,
+        get_sampler_scheduler_list,
         run,
         get_task_status,
     ]
@@ -153,4 +160,4 @@ Paste-and-Run Example (With chain):
         if getattr(fn, "api_name", None) in HIGH_LEVEL_MCP_API_NAMES:
             fn.show_api = True
 
-    print("[MCP Integration] Successfully registered 7 High-Level Abstract MCP APIs via gr.api().")
+    print(f"[MCP Integration] Successfully registered {len(funcs)} High-Level Abstract MCP APIs via gr.api().")

mcp_tools/tool_handlers.py

@@ -8,6 +8,7 @@ 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_sampler_scheduler_list import handle_get_sampler_scheduler_list
 from .run import handle_run
 from .get_task_status import handle_get_task_status
 from .common import (
@@ -23,6 +24,7 @@ __all__ = [
     "handle_get_model_list",
     "handle_get_feature_list",
     "handle_get_model_features",
+    "handle_get_sampler_scheduler_list",
     "handle_run",
     "handle_get_task_status",
 ]

requirements.txt

@@ -1,5 +1,5 @@
 comfyui-frontend-package==1.48.7
-comfyui-workflow-templates==0.11.37
+comfyui-workflow-templates==0.11.39
 comfyui-embedded-docs==0.5.9
 torch
 torchsde