fix error

kytpbs · Oct 18, 2023 · a9f0488 · a9f0488
1 parent 25e634c
commit a9f0488
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/GPT.py b/src/GPT.py
@@ -54,11 +54,17 @@ def chat(main_message: str, message_history: dict[discord.User, str]):
         },
     ]
     for user, message in message_history.items():
-        messages.append({
-        "role": "assistant" if user.bot else "user",
-        "name": user.name,
-        "content": message,
-        })
+        if user.bot:
+            messages.append({
+            "role": "assistant",
+            "content": message,
+            })
+        else:
+            messages.append({
+            "role": "user",
+            "name": user.name,
+            "content": message,
+            })
 
     messages.append({
         "role": "user",