Update main.py

main.py

@@ -151,13 +151,19 @@ def reply(message: str, history: list[str]) -> str:
     message, bypass = preprocess(message)
 
     # This is some handling that is applied to the history variable to put it in a good format
-    if isinstance(history, list):
-        if len(history) > 0:
-            history = history[-1]
-    print(history)
+    # if isinstance(history, list):
+    #     if len(history) > 0:
+    #         history = history[-1]
+    # print(history)
+    # history_transformer_format = [
+    #     {"role": "assistant" if idx & 1 else "user", "content": msg}
+    #     for idx, msg in enumerate(history)
+    # ] + [{"role": "user", "content": message}]
     history_transformer_format = [
-        {"role": "assistant" if idx & 1 else "user", "content": msg}
-        for idx, msg in enumerate(history)
+        {"role": role, "content": message_pair[idx]}
+        for message_pair in history
+        for idx, role in enumerate(["user", "assistant"])
+        if message_pair[idx] is not None
     ] + [{"role": "user", "content": message}]
 
     # Stream a response from pipe