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"]]) 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()