Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: Temp remove timezone test (#4427)

Browse Source 
Temp remove test
This commit is contained in:
Matthew Zhou
2025-09-04 12:05:16 -07:00
committed by GitHub
parent f6ec6dc3a0
commit ef225d3e49
1 changed files with 34 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
tests/test_client.py
View File

@@ -650,39 +650,40 @@ def test_initial_sequence(client: Letta):
assert messages[2].message_type == "user_message"
def test_timezone(client: Letta):
agent = client.agents.create(
memory_blocks=[{"label": "human", "value": ""}, {"label": "persona", "value": ""}],
model="letta/letta-free",
embedding="letta/letta-free",
timezone="America/Los_Angeles",
)
agent = client.agents.retrieve(agent_id=agent.id)
assert agent.timezone == "America/Los_Angeles"
response = client.agents.messages.create(
agent_id=agent.id,
messages=[
MessageCreate(
role="user",
content="What timezone are you in?",
)
],
)
# second message is assistant message
assert response.messages[1].message_type == "assistant_message"
pacific_tz_indicators = {"America/Los_Angeles", "PDT", "PST", "PT", "Pacific Daylight Time", "Pacific Standard Time", "Pacific Time"}
content = response.messages[1].content
assert any(tz in content for tz in pacific_tz_indicators), (
f"Response content: {response.messages[1].content} does not contain expected timezone"
)
# test updating the timezone
client.agents.modify(agent_id=agent.id, timezone="America/New_York")
agent = client.agents.retrieve(agent_id=agent.id)
assert agent.timezone == "America/New_York"
# TODO: Add back when timezone packing is standardized/settled
# def test_timezone(client: Letta):
# agent = client.agents.create(
# memory_blocks=[{"label": "human", "value": ""}, {"label": "persona", "value": ""}],
# model="letta/letta-free",
# embedding="letta/letta-free",
# timezone="America/Los_Angeles",
# )
#
# agent = client.agents.retrieve(agent_id=agent.id)
# assert agent.timezone == "America/Los_Angeles"
#
# response = client.agents.messages.create(
# agent_id=agent.id,
# messages=[
# MessageCreate(
# role="user",
# content="What timezone are you in?",
# )
# ],
# )
# # second message is assistant message
# assert response.messages[1].message_type == "assistant_message"
#
# pacific_tz_indicators = {"America/Los_Angeles", "PDT", "PST", "PT", "Pacific Daylight Time", "Pacific Standard Time", "Pacific Time"}
# content = response.messages[1].content
# assert any(tz in content for tz in pacific_tz_indicators), (
# f"Response content: {response.messages[1].content} does not contain expected timezone"
# )
#
# # test updating the timezone
# client.agents.modify(agent_id=agent.id, timezone="America/New_York")
# agent = client.agents.retrieve(agent_id=agent.id)
# assert agent.timezone == "America/New_York"
def test_attach_sleeptime_block(client: Letta):