kennethreitz/simplemind
1
0
Fork 0
mirror of https://github.com/kennethreitz/simplemind.git synced 2026-06-05 22:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update simplemind/models.py

Browse Source 
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Kurt Heiden
2024-10-28 22:05:55 -06:00
committed by GitHub
parent 87d636ca55
commit dcb9c14d30
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simplemind/models.py
+2 -3
View File
@@ -79,9 +79,8 @@ class Conversation(SMBaseModel):
def __str__(self):
return f"<Conversation id={self.id!r}>"
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))