Merge pull request #32 from lucianosrp/fix-sys-prompt

fix: `prepend_system_message` use system role by default
This commit is contained in:
2024-11-02 10:54:47 -04:00
committed by GitHub
2 changed files with 3 additions and 3 deletions
+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")
+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