mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from letta_client import Letta, SleeptimeManagerUpdate
|
|
|
|
client = Letta(base_url="http://localhost:8283")
|
|
|
|
agent = client.agents.create(
|
|
memory_blocks=[
|
|
{"value": "Name: ?", "label": "human"},
|
|
{"value": "You are a helpful assistant.", "label": "persona"},
|
|
],
|
|
model="openai/gpt-4.1",
|
|
embedding="openai/text-embedding-3-small",
|
|
enable_sleeptime=True,
|
|
)
|
|
print(f"Created agent id {agent.id}")
|
|
|
|
# get the group
|
|
group_id = agent.multi_agent_group.id
|
|
current_frequence = agent.multi_agent_group.sleeptime_agent_frequency
|
|
print(f"Group id: {group_id}, frequency: {current_frequence}")
|
|
|
|
group = client.groups.modify(
|
|
group_id=group_id,
|
|
manager_config=SleeptimeManagerUpdate(
|
|
sleeptime_agent_frequency=1
|
|
),
|
|
)
|
|
print(f"Updated group id {group_id} with frequency {group.sleeptime_agent_frequency}")
|
|
print(f"Group members", group.agent_ids)
|
|
sleeptime_ids = []
|
|
for agent_id in group.agent_ids:
|
|
if client.agents.retrieve(agent_id=agent_id).agent_type == "sleeptime_agent":
|
|
sleeptime_ids.append(agent_id)
|
|
print(f"Sleeptime agent ids: {sleeptime_ids}")
|
|
sleeptime_agent_id = sleeptime_ids[0]
|
|
|
|
# check the frequency
|
|
agent = client.agents.retrieve(agent_id=agent.id)
|
|
print(f"Updated agent id {agent.id} with frequency {agent.multi_agent_group.sleeptime_agent_frequency}")
|
|
|
|
|
|
response = client.agents.messages.create(
|
|
agent_id=agent.id,
|
|
messages=[
|
|
{"role": "user", "content": "Hello can you echo back this input?"},
|
|
],
|
|
)
|
|
response = client.agents.messages.create(
|
|
agent_id=agent.id,
|
|
messages=[
|
|
{"role": "user", "content": "My name is sarah"},
|
|
],
|
|
)
|
|
for message in response.messages:
|
|
print(message)
|
|
|
|
print("---------------- SLEEPTIME AGENT ----------------")
|
|
for message in client.agents.messages.list(agent_id=sleeptime_agent_id):
|
|
print(message)
|
|
|