From c659ae20a383fa0a2eae2793521e6726bbe9ae91 Mon Sep 17 00:00:00 2001
From: Enias Cailliau <enias@steamship.com>
Date: Fri, 14 Apr 2023 15:57:57 +0100
Subject: [PATCH] Hotfix for chat_memory (#36)

* Hotfix

* Lint
---
 src/steamship_langchain/memory/chat_memory.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/steamship_langchain/memory/chat_memory.py b/src/steamship_langchain/memory/chat_memory.py
index 3adb5f7..394b7c3 100644
--- 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