From 3d8e169a08f4ff8d33490f423a33bdb5abccf211 Mon Sep 17 00:00:00 2001 From: Luciano Scarpulla Date: Sat, 2 Nov 2024 22:46:54 +0800 Subject: [PATCH] make prepend_system_message use system reole by default --- examples/simple_memory.py | 2 +- simplemind/models.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/simple_memory.py b/examples/simple_memory.py index a0d119e..8d757c9 100644 --- a/examples/simple_memory.py +++ b/examples/simple_memory.py @@ -13,7 +13,7 @@ class SimpleMemoryPlugin: def initialize_hook(self, conversation: sm.Conversation): for m in self.yield_memories(): - conversation.prepend_system_message(role="system", text=m) + conversation.prepend_system_message(text=m) conversation = sm.create_conversation(llm_model="grok-beta", llm_provider="xai") diff --git a/simplemind/models.py b/simplemind/models.py index 54eade1..8a32fb9 100644 --- a/simplemind/models.py +++ b/simplemind/models.py @@ -117,10 +117,10 @@ class Conversation(SMBaseModel): pass def prepend_system_message( - self, role: MESSAGE_ROLE, text: str, meta: Dict[str, Any] | None = None + self, text: str, meta: Dict[str, Any] | None = None ): """Prepend a system message to the conversation.""" - self.messages = [Message(role=role, text=text, meta=meta or {})] + self.messages + self.messages = [Message(role="system", text=text, meta=meta or {})] + self.messages def add_message( self, role: MESSAGE_ROLE, text: str, meta: Optional[Dict[str, Any]] = None