ComfyUI MCP Server

Instructions

Implementation Reference

• comfy_mcp/mcp/server.py:493-517 (handler)

  MCP tool handler for 'search_templates'. Delegates to TemplateManager.search_templates with the provided query. Includes input schema via docstring and type hints.

  @mcp.tool
  def search_templates(query: str) -> list[dict]:
      """Search templates by name, description, or tags.
      
      Performs fuzzy search across template metadata to find
      relevant workflow templates.
      
      Args:
          query: Search query (searches name, description, tags)
      
      Returns:
          List of matching templates
      
      Examples:
          search_templates("pose")
          search_templates("upscaling")
          search_templates("text to image")
      """
      try:
          results = template_manager.search_templates(query=query)
          return results
          
      except Exception as e:
          raise ToolError(f"Error searching templates: {e}")

• comfy_mcp/templates/manager.py:65-116 (helper)

  Core implementation of template search logic in TemplateManager class. Performs filtering on all templates based on query, category, tags, difficulty, and source.

  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