mcp-server-tree-sitter

Instructions

    Args:
        language: Language name
        template_name: Template name (e.g., "functions", "classes")

    Returns:
        Query template information
    

Implementation Reference

• src/mcp_server_tree_sitter/tools/registration.py:367-389 (handler)

  The handler function for 'get_query_template_tool', decorated with @mcp_server.tool() for registration. It fetches the query template using the helper function and returns it in a structured format, including error handling if the template does not exist.

  @mcp_server.tool()
  def get_query_template_tool(language: str, template_name: str) -> Dict[str, Any]:
      """Get a predefined tree-sitter query template.
  
      Args:
          language: Language name
          template_name: Template name (e.g., "functions", "classes")
  
      Returns:
          Query template information
      """
      from ..language.query_templates import get_query_template
  
      template = get_query_template(language, template_name)
      if not template:
          raise ValueError(f"No template '{template_name}' for language '{language}'")
  
      return {
          "language": language,
          "name": template_name,
          "query": template,
      }

• src/mcp_server_tree_sitter/language/query_templates.py:8-23 (helper)

  Core helper function that retrieves the specific query template string from the QUERY_TEMPLATES dictionary based on language and template name.

  def get_query_template(language: str, template_name: str) -> Optional[str]:
      """
      Get a query template for a language.
  
      Args:
          language: Language identifier
          template_name: Template name
  
      Returns:
          Query string or None if not found
      """
      language_templates = QUERY_TEMPLATES.get(language)
      if language_templates:
          return language_templates.get(template_name)
      return None

• src/mcp_server_tree_sitter/language/templates/__init__.py:20-34 (helper)

  Central data structure aggregating query templates from all language-specific template modules, used by the get_query_template helper.

  # Combine all language templates
  QUERY_TEMPLATES: Dict[str, Dict[str, str]] = {
      "python": python.TEMPLATES,
      "javascript": javascript.TEMPLATES,
      "typescript": typescript.TEMPLATES,
      "go": go.TEMPLATES,
      "rust": rust.TEMPLATES,
      "c": c.TEMPLATES,
      "cpp": cpp.TEMPLATES,
      "swift": swift.TEMPLATES,
      "java": java.TEMPLATES,
      "kotlin": kotlin.TEMPLATES,
      "julia": julia.TEMPLATES,
      "apl": apl.TEMPLATES,
  }