refactor: Refactor loop for improved efficiency and readabilit

This commit is contained in:
kafka
2023-06-21 14:24:43 +05:45
parent 6ccd05ab65
commit 5d33593517

View File

@@ -34,14 +34,9 @@ def pretty_print_conversation(messages):
formatted_messages.append(role_to_message[message["role"]]) formatted_messages.append(role_to_message[message["role"]])
for formatted_message in formatted_messages: for formatted_message in formatted_messages:
print( role = messages[formatted_messages.index(formatted_message)]["role"]
colored( color = role_to_color[role]
formatted_message, print(colored(formatted_message, color))
role_to_color[
messages[formatted_messages.index(formatted_message)]["role"]
],
)
)
@app.command() @app.command()