Merge pull request #286 from kbshal/opensource

refactor: Refactor loop for improved efficiency and readability
This commit is contained in:
Emil Ahlbäck
2023-06-23 14:21:37 +02:00
committed by GitHub

View File

@@ -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()