From 5d33593517dafa99eceace7c737bcc7b9b7c0306 Mon Sep 17 00:00:00 2001 From: kafka Date: Wed, 21 Jun 2023 14:24:43 +0545 Subject: [PATCH] refactor: Refactor loop for improved efficiency and readabilit --- scripts/print_chat.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/scripts/print_chat.py b/scripts/print_chat.py index eef244f..ab1c09a 100644 --- a/scripts/print_chat.py +++ b/scripts/print_chat.py @@ -34,14 +34,9 @@ def pretty_print_conversation(messages): formatted_messages.append(role_to_message[message["role"]]) for formatted_message in formatted_messages: - print( - colored( - formatted_message, - role_to_color[ - messages[formatted_messages.index(formatted_message)]["role"] - ], - ) - ) + role = messages[formatted_messages.index(formatted_message)]["role"] + color = role_to_color[role] + print(colored(formatted_message, color)) @app.command()