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

make prepend_system_message use system reole by default

Browse Source
This commit is contained in:
Luciano Scarpulla
2024-11-02 22:46:54 +08:00
parent fa3ee731df
commit 3d8e169a08
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/simple_memory.py
+1 -1
View File
@@ -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")
simplemind/models.py
+2 -2
View File
@@ -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