fix bug in formatting (#402)

Make sure placeholder is in the text.
microsoft · Sep 11, 2024 · af4be11 · af4be11
2 parents aa23765 + 16e1f80
commit af4be11
Show file tree

Hide file tree

Showing 2 changed files with 2 additions 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

diff --git a/taskweaver/memory/attachment.py b/taskweaver/memory/attachment.py
@@ -41,6 +41,7 @@ class AttachmentType(Enum):
 
     # Misc
     invalid_response = "invalid_response"
+    text = "text"
 
     # board info
     board = "board"