mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 14:50:16 +00:00
Siddhesh Agarwal
29 lines
721 B
Python
29 lines
721 B
Python
from _context import sm
|
|
|
|
|
|
class SimpleMemoryPlugin:
|
|
def __init__(self):
|
|
self.memories = [
|
|
"the earth has fictionally beeen destroyed.",
|
|
"the moon is made of cheese.",
|
|
]
|
|
|
|
def yield_memories(self):
|
|
return (m for m in self.memories)
|
|
|
|
def initialize_hook(self, conversation: sm.Conversation):
|
|
for m in self.yield_memories():
|
|
conversation.prepend_system_message(role="system", text=m)
|
|
|
|
|
|
conversation = sm.create_conversation(llm_model="grok-beta", llm_provider="xai")
|
|
conversation.add_plugin(SimpleMemoryPlugin())
|
|
|
|
conversation.add_message(
|
|
role="user",
|
|
text="Please write a poem about the moon",
|
|
)
|
|
|
|
r = conversation.send()
|
|
print(r.text)
|