fix bug in formatting

microsoft · Sep 4, 2024 · 9d8acd9 · 9d8acd9
1 parent aa23765
commit 9d8acd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/taskweaver/code_interpreter/code_interpreter/code_generator.py b/taskweaver/code_interpreter/code_interpreter/code_generator.py
@@ -212,7 +212,7 @@ def compose_prompt(
         return chat_history
 
     def format_attachment(self, attachment: Attachment):
-        if attachment.type == AttachmentType.thought:
+        if attachment.type == AttachmentType.thought and "{ROLE_NAME}" in attachment.content:
             return attachment.content.format(ROLE_NAME=self.role_name)
         else:
             return attachment.content