aljaz/gpt-engineer
1
0
Fork 0
mirror of https://github.com/aljazceru/gpt-engineer.git synced 2025-12-17 12:45:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Simplify if statements

Browse Source
This commit is contained in:
Anton Osika
2023-06-19 00:02:19 +02:00
parent e767edf108
commit 35c35b3ffa
1 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
scripts/print_chat.py
View File

@@ -16,18 +16,20 @@ def pretty_print_conversation(messages):
}
formatted_messages = []
for message in messages:
if message["role"] == "system":
formatted_messages.append(f"system: {message['content']}\n")
elif message["role"] == "user":
formatted_messages.append(f"user: {message['content']}\n")
elif message["role"] == "assistant" and message.get("function_call"):
formatted_messages.append(f"assistant: {message['function_call']}\n")
elif message["role"] == "assistant" and not message.get("function_call"):
formatted_messages.append(f"assistant: {message['content']}\n")
elif message["role"] == "function":
formatted_messages.append(
f"function ({message['name']}): {message['content']}\n"
)
assistant_content = (
message["function_call"]
if message.get("function_call")
else message["content"]
)
role_to_message = {
"system": f"system: {message['content']}\n",
"user": f"user: {message['content']}\n",
"assistant": f"assistant: {assistant_content}\n",
"function": f"function ({message['name']}): {message['content']}\n",
}
formatted_messages.append(role_to_message[message["role"]])
for formatted_message in formatted_messages:
print(
colored(