Hotfix for chat_memory (#36)

* Hotfix * Lint
steamship-core · Apr 14, 2023 · c659ae2 · c659ae2
1 parent 644d91f
commit c659ae2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/steamship_langchain/memory/chat_memory.py b/src/steamship_langchain/memory/chat_memory.py
@@ -51,7 +51,7 @@ def saved_messages(self) -> List[BaseMessage]:
             if b.text.startswith(self.HUMAN_PREFIX):
                 messages.append(HumanMessage(content=b.text[len(self.HUMAN_PREFIX) :]))
             elif b.text.startswith(self.AI_PREFIX):
-                messages.append(HumanMessage(content=b.text[len(self.HUMAN_PREFIX) :]))
+                messages.append(HumanMessage(content=b.text[len(self.AI_PREFIX) :]))
             else:
                 raise ValueError(f"Found unsupported message type: {b.text}")
         return messages