diff --git a/tests/test_client.py b/tests/test_client.py index f4606f99..d3de1bc1 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -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