def get_placeholders()

in google/generativeai/notebook/lib/prompt_utils.py [0:0]

• 6 lines of code
• 5 McCabe index (conditional complexity)

def get_placeholders(prompt: str) -> AbstractSet[str]:
    """Returns the placeholders for `prompt`.

    E.g. Given "A for {word_one} B for {word_two}", returns {"word_one",
    "word_two"}.

    Args:
      prompt: A prompt template with optional placeholders.

    Returns:
      A sequence of placeholders in `prompt`.
    """
    placeholders: list[str] = []
    for _, field_name, _, _ in string.Formatter().parse(prompt):
        if field_name is not None:
            placeholders.append(field_name)
    return frozenset(placeholders)