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

Many small improvements, new benchmark results!

Browse Source
This commit is contained in:
Anton Osika
2023-06-21 00:37:18 +02:00
parent 4c68126295
commit 16dd96f50d
6 changed files with 97 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
scripts/print_chat.py
View File

@@ -16,19 +16,22 @@ def pretty_print_conversation(messages):
}
formatted_messages = []
for message in messages:
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"]])
if message["role"] == "function":
formatted_messages.append(
f"function ({message['name']}): {message['content']}\n"
)
else:
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",
}
formatted_messages.append(role_to_message[message["role"]])
for formatted_message in formatted_messages:
print(