Fix mongodb for cohere

2024-04-22 11:41:14 +00:00
parent 66dfcc2181
commit bbbc56123f
1 changed files with 2 additions and 2 deletions
--- a/utils/db.py
+++ b/utils/db.py
@@ -71,12 +71,12 @@ class MongoDatabase(Database):
        self._client.close()

    def add_chat_history(self, message):
-        chat_history = db.get("core.cohere", "chat_history", default=[])
+        chat_history = db.get("core.cohere", "chat_history", expected_value=[])
        chat_history.append(message)
        db.set("core.cohere", "chat_history", chat_history)

    def get_chat_history(self):
-        return db.get("core.cohere", "chat_history", default=[])
+        return db.get("core.cohere", "chat_history", expected_value=[])


 class SqliteDatabase(Database):