test: reduce flakiness of tz testing

This commit is contained in:
Andy Li
2025-08-12 14:53:36 -07:00
committed by GitHub
parent 1eb692f62a
commit 96efbf317f

View File

@@ -651,7 +651,6 @@ def test_initial_sequence(client: Letta):
def test_timezone(client: Letta):
# create an agent
agent = client.agents.create(
memory_blocks=[{"label": "human", "value": ""}, {"label": "persona", "value": ""}],
model="letta/letta-free",
@@ -659,7 +658,6 @@ def test_timezone(client: Letta):
timezone="America/Los_Angeles",
)
# get the timzone
agent = client.agents.retrieve(agent_id=agent.id)
assert agent.timezone == "America/Los_Angeles"
@@ -674,13 +672,11 @@ def test_timezone(client: Letta):
)
# second message is assistant message
assert response.messages[1].message_type == "assistant_message"
# content is similar to current timezone
assert (
"America/Los_Angeles" in response.messages[1].content
or "PDT" in response.messages[1].content
or "PST" in response.messages[1].content
or "Pacific Daylight Time" in response.messages[1].content
or "Pacific Standard Time" in response.messages[1].content
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