ComfyUI MCP Server

Instructions

Implementation Reference

• comfy_mcp/mcp/server.py:349-387 (handler)

  The main handler function for the 'list_templates' MCP tool. Decorated with @mcp.tool for registration. Accepts optional filters for category, difficulty, and tag, then delegates to template_manager.search_templates and returns the filtered list of templates.
  @mcp.tool def list_templates( category: str = None, difficulty: str = None, tag: str = None ) -> list[dict]: """List available workflow templates. Discover workflow templates by category, difficulty, or tags. Templates provide ready-to-use workflows for common use cases. Args: category: Filter by category (Generation, Enhancement, Editing, etc.) difficulty: Filter by difficulty (beginner, intermediate, advanced) tag: Filter by tag (text2img, inpainting, controlnet, etc.) Returns: List of template metadata with names, descriptions, and parameters Examples: list_templates() list_templates(category="Generation") list_templates(difficulty="beginner") list_templates(tag="text2img") """ try: # Convert filters to search parameters tags = [tag] if tag else None results = template_manager.search_templates( category=category, tags=tags, difficulty=difficulty ) return results except Exception as e: raise ToolError(f"Error listing templates: {e}")
• comfy_mcp/templates/manager.py:65-116 (helper)

  Core filtering logic in TemplateManager.search_templates, directly invoked by the list_templates tool handler to apply category, tags, and difficulty filters to the full list of templates (custom and official).
  def search_templates( self, query: Optional[str] = None, category: Optional[str] = None, tags: Optional[List[str]] = None, difficulty: Optional[str] = None, source: Optional[str] = None, include_official: bool = True ) -> List[Dict[str, Any]]: """Search templates by various criteria.""" # Get all templates first all_templates = self.list_templates(include_official=include_official) results = [] for template_data in all_templates: # Check source filter if source and template_data.get("source") != source: continue # Check query match (name, description, category) if query: query_lower = query.lower() if not any([ query_lower in template_data["name"].lower(), query_lower in template_data["description"].lower(), query_lower in template_data["category"].lower(), ]): # For custom templates, also check tags if template_data.get("source") == "custom" and "tags" in template_data: if not any(query_lower in tag.lower() for tag in template_data["tags"]): continue else: continue # Check category if category and template_data["category"].lower() != category.lower(): continue # Check tags (only for custom templates) if tags and template_data.get("source") == "custom": template_tags = template_data.get("tags", []) if not any(tag.lower() in [t.lower() for t in template_tags] for tag in tags): continue # Check difficulty (only for custom templates) if difficulty and template_data.get("difficulty"): if template_data["difficulty"].lower() != difficulty.lower(): continue results.append(template_data) return results
• comfy_mcp/templates/manager.py:17-39 (helper)

  TemplateManager.list_templates method that aggregates custom templates and official templates into a unified list with metadata, used as the base for searching and filtering.
  def list_templates(self, include_official: bool = True) -> List[Dict[str, Any]]: """List all available templates with metadata.""" results = [] # Add custom templates for name, template in self.custom_templates.items(): results.append({ "name": name, "description": template.description, "category": template.category, "tags": template.tags, "difficulty": template.difficulty, "required_models": template.required_models or [], "parameters": template.parameters or {}, "source": "custom" }) # Add official templates if requested if include_official and official_manager.templates: official_templates = official_manager.list_templates() results.extend(official_templates) return results
• comfy_mcp/templates/official.py:310-325 (helper)

  OfficialTemplateManager.list_templates provides metadata for official templates from Comfy-Org repo, integrated into the main template list by TemplateManager.
  def list_templates(self) -> List[Dict[str, Any]]: """List all official templates with metadata.""" return [ { "name": name, "display_name": template.name, "description": template.description, "category": template.category, "source": "official", "preview_images": template.preview_images or [], "source_url": template.source_url, "has_dsl": template.dsl_content is not None } for name, template in self.templates.items() ]