From dcb9c14d30c01e50ce2a6bd249c0381e5e564388 Mon Sep 17 00:00:00 2001 From: Kurt Heiden Date: Mon, 28 Oct 2024 22:05:55 -0600 Subject: [PATCH] Update simplemind/models.py Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> --- simplemind/models.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/simplemind/models.py b/simplemind/models.py index 47c57e8..9019de1 100644 --- a/simplemind/models.py +++ b/simplemind/models.py @@ -79,9 +79,8 @@ class Conversation(SMBaseModel): def __str__(self): return f"" - def prepend_system_message(self, role: str, text: str, meta: Dict[str, Any] = {}): - self.messages = [Message(role=role, text=text, meta=meta)] + self.messages - + def prepend_system_message(self, role: str, text: str, meta: Optional[Dict[str, Any]] = None): + self.messages = [Message(role=role, text=text, meta=meta or {})] + self.messages def add_message(self, role: str, text: str, meta: Dict[str, Any] = {}): """Add a new message to the conversation.""" self.messages.append(Message(role=role, text=text, meta=meta))