Merge pull request modelcontextprotocol#25 from jkoelker/jk/timezones

feat(tools): always report eastern time

Author: jkoelker

src/schwab_mcp/tools/tools.py

@@ -1,8 +1,8 @@
 #
 
-from typing import Annotated
-
 import datetime
+from typing import Annotated
+from zoneinfo import ZoneInfo
 from mcp.server.fastmcp import FastMCP
 
 from schwab_mcp.context import SchwabContext
@@ -12,9 +12,10 @@
 
 async def get_datetime() -> str:
     """
-    Get the current datetime in ISO format (e.g., '2023-10-27T10:30:00.123456').
+    Get the current datetime in ISO format with Eastern Time offset and abbreviation.
     """
-    return datetime.datetime.now().isoformat()
+    eastern_now = datetime.datetime.now(tz=ZoneInfo("America/New_York"))
+    return f"{eastern_now.isoformat()} {eastern_now.tzname()}"
 
 
 async def get_market_hours(

tests/test_tools.py

@@ -162,3 +162,20 @@ async def fake_call(func, *args, **kwargs):
     kwargs = captured["kwargs"]
     assert isinstance(kwargs, dict)
     assert kwargs["projection"] is client.Instrument.Projection.SYMBOL_SEARCH
+
+
+def test_get_datetime_returns_eastern_time(monkeypatch):
+    class DummyDatetime(datetime.datetime):
+        @classmethod
+        def now(cls, tz=None):
+            assert tz is not None
+            return cls(2024, 1, 15, 12, 30, 45, tzinfo=tz)
+
+    monkeypatch.setattr(tools.datetime, "datetime", DummyDatetime)
+
+    result = run(tools.get_datetime())
+
+    assert result.startswith("2024-01-15T12:30:45")
+    assert "-05:00" in result or "-04:00" in result
+    assert result.endswith("EST") or result.endswith("EDT")
+    assert "Eastern Time" not in result