UPDATE

utils.py

@@ -31,7 +31,7 @@ def cleanMessages(messages: List[ChatMessage], removeThinkingContent: bool = Fal
         content = message.content.strip()
         content = re.sub(r"\n+", "\n", content)
         promptStrList.append(
-            f"{message.role.strip().lower().capitalize()}: {content if message.role!='Assistant' or not removeThinkingContent else remove_nested_think_tags_stack(content)}"
+            f"{message.role.strip().lower().capitalize()}: {content if message.role.strip().lower().capitalize()!='Assistant' or not removeThinkingContent else remove_nested_think_tags_stack(content)}"
         )
 
     return "\n\n".join(promptStrList)