diff --git a/chatmastermind/utils.py b/chatmastermind/utils.py index 2dceb69..387db7b 100644 --- a/chatmastermind/utils.py +++ b/chatmastermind/utils.py @@ -74,9 +74,10 @@ def display_chat(chat, dump=False, source_code=False) -> None: else: print(f"{message['role'].upper()}: {message['content']}") + def display_tags_frequency(tags: List[str], dump=False) -> None: if dump: pp(tags) return for tag in set(tags): - print(f"-{tag} : {tags.count(tag)}") + print(f"- {tag}: {tags.count(tag)}")