llama-index

1
"""Prompt display utils."""
2

3
from llama_index.legacy.prompts.mixin import PromptDictType
4

5

6
# define prompt viewing function
7
def display_prompt_dict(prompts_dict: PromptDictType) -> None:
8
    """Display prompt dict.
9

10
    Args:
11
        prompts_dict: prompt dict
12

13
    """
14
    from IPython.display import Markdown, display
15

16
    for k, p in prompts_dict.items():
17
        text_md = f"**Prompt Key**: {k}<br>" f"**Text:** <br>"
18
        display(Markdown(text_md))
19
        print(p.get_template())
20
        display(Markdown("<br><br>"))
21