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 = [] formatted_messages = []
for message in messages: for message in messages:
if message["role"] == "system": assistant_content = (
formatted_messages.append(f"system: {message['content']}\n") message["function_call"]
elif message["role"] == "user": if message.get("function_call")
formatted_messages.append(f"user: {message['content']}\n") else message["content"]
elif message["role"] == "assistant" and message.get("function_call"): )
formatted_messages.append(f"assistant: {message['function_call']}\n") role_to_message = {
elif message["role"] == "assistant" and not message.get("function_call"): "system": f"system: {message['content']}\n",
formatted_messages.append(f"assistant: {message['content']}\n") "user": f"user: {message['content']}\n",
elif message["role"] == "function": "assistant": f"assistant: {assistant_content}\n",
formatted_messages.append( "function": f"function ({message['name']}): {message['content']}\n",
f"function ({message['name']}): {message['content']}\n" }
)
formatted_messages.append(role_to_message[message["role"]])
for formatted_message in formatted_messages: for formatted_message in formatted_messages:
print( print(
colored( colored(