diff --git a/letta/orm/agent.py b/letta/orm/agent.py index 4a5dae96..219070f2 100644 --- a/letta/orm/agent.py +++ b/letta/orm/agent.py @@ -190,6 +190,7 @@ class Agent(SqlalchemyBase, OrganizationMixin, AsyncAttrs): "response_format": self.response_format, "last_run_completion": self.last_run_completion, "last_run_duration_ms": self.last_run_duration_ms, + "timezone": self.timezone, # optional field defaults "tags": [], "tools": [], @@ -268,6 +269,7 @@ class Agent(SqlalchemyBase, OrganizationMixin, AsyncAttrs): "response_format": self.response_format, "last_run_completion": self.last_run_completion, "last_run_duration_ms": self.last_run_duration_ms, + "timezone": self.timezone, } optional_fields = { "tags": [], diff --git a/tests/test_client.py b/tests/test_client.py index c957a5b5..62e50495 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -654,3 +654,36 @@ def test_initial_sequence(client: Letta): assert messages[0].message_type == "system_message" assert messages[1].message_type == "assistant_message" assert messages[2].message_type == "user_message" + + +# def test_timezone(client: Letta): +# # create an agent +# agent = client.agents.create( +# memory_blocks=[{"label": "human", "value": ""}, {"label": "persona", "value": ""}], +# model="letta/letta-free", +# embedding="letta/letta-free", +# timezone="America/Los_Angeles", +# ) +# +# # get the timzone +# 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" +# # 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 +# ) +#