make sure human message is sent first

StampyAI · Jun 5, 2024 · dbdd215 · dbdd215
1 parent fbad8ae
commit dbdd215
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/stampy_chat/chat.py b/api/src/stampy_chat/chat.py
@@ -77,7 +77,7 @@ class PrefixedPrompt(BaseChatPromptTemplate):
     def format_messages(self, **kwargs: Any) -> List[BaseMessage]:
         history = kwargs[self.messages_field]
         if history and self.prompt:
-            return [AIMessage(content=self.prompt)] + [self.transformer(i) for i in history]
+            return [HumanMessage(content=self.prompt)] + [self.transformer(i) for i in history]
         return []