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