aria-code 0.80.0__tar.gz → 0.81.0__tar.gz

{aria_code-0.80.0/src/aria_code.egg-info → aria_code-0.81.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.80.0
+Version: 0.81.0
 Summary: Open-source AI agent for coding, financial analysis and logistics operations: numbers you can check, client data kept apart, actions a person approves
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/artheras/aria-code

{aria_code-0.80.0 → aria_code-0.81.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aria-code"
-version = "0.80.0"
+version = "0.81.0"
 description = "Open-source AI agent for coding, financial analysis and logistics operations: numbers you can check, client data kept apart, actions a person approves"
 readme = "README.md"
 license = { text = "Apache-2.0" }

{aria_code-0.80.0 → aria_code-0.81.0}/src/aria_code/_version.py

@@ -17,4 +17,4 @@ from __future__ import annotations
 
 __all__ = ["__version__"]
 
-__version__ = "0.80.0"
+__version__ = "0.81.0"

{aria_code-0.80.0 → aria_code-0.81.0}/src/aria_code/apps/cli/handlers/market_handlers.py

@@ -23,6 +23,7 @@ from aria_code.apps.cli.utils.market_detect import (
     _detect_market_overview,
     _PRIVATE_COMPANY_PROFILES,
 )
+from aria_code.apps.cli.market_hours import market_session
 from aria_code.apps.cli.market_metadata import enrich_market_quote, market_display_label
 from aria_code.packages.aria_core.paths import aria_home
 
@@ -2143,7 +2144,6 @@ def _try_handle_market_snapshot_analysis(message: str, history: list = None, *,
     _chg_abs    = abs(chg) if chg is not None else None
 
     # ── Build output ──────────────────────────────────────────────────────
-    weekday = datetime.now().weekday()
     ti_provider = ti.get("provider", "")
     quote_chain = quote.get("provider_chain") or [provider]
     data_src = " -> ".join(str(p) for p in quote_chain if p)
@@ -2160,8 +2160,11 @@ def _try_handle_market_snapshot_analysis(message: str, history: list = None, *,
     signal_str = _SIGNAL_LABELS.get(_lang, _SIGNAL_LABELS["zh"])[_sig_key]
     _L = {
         "disclaimer":   "Not investment advice" if _en else "不构成投资建议",
-        "after_hours":  "After-hours" if _en else "休市/盘后",
-        "market_open":  "Market open" if _en else "盘中",
+        "session_open":   "Market open" if _en else "盘中",
+        "session_pre":    "Pre-market" if _en else "盘前",
+        "session_post":   "After-hours" if _en else "盘后",
+        "session_closed": "Market closed" if _en else "休市",
+        "session_24/7":   "24/7 market" if _en else "全天交易",
         "price_hdr":    "Metric" if _en else "指标",
         "value_hdr":    "Value" if _en else "数值",
         "latest":       "Last price" if _en else "最新价",
@@ -2218,7 +2221,9 @@ def _try_handle_market_snapshot_analysis(message: str, history: list = None, *,
     }
 
     _sep = ": " if _en else "："
-    session_note = _L["after_hours"] if weekday >= 5 else _L["market_open"]
+    # On the exchange's clock, not this machine's; nothing when it is unknown.
+    _session = market_session(symbol)
+    session_note = _L.get(f"session_{_session}", "") if _session else ""
 
     def _money(v: float | int | None) -> str:
         try:
@@ -2314,7 +2319,7 @@ def _try_handle_market_snapshot_analysis(message: str, history: list = None, *,
         lines.append(f"## {_header_name}  `{symbol}`")
     else:
         lines.append(f"## `{symbol}`")
-    lines.append(f"*{data_src} · {_now_str} · {session_note} · {_L['disclaimer']}*")
+    lines.append("*" + " · ".join(p for p in (data_src, _now_str, session_note, _L["disclaimer"]) if p) + "*")
     lines.append("")
     lines.append(f"**{'Takeaway' if _en else '结论'}**{_sep}{_summary_line}")
     lines.append(f"**{'Watch' if _en else '观察位'}**{_sep}{_watch_line}")

aria_code-0.81.0/src/aria_code/apps/cli/market_hours.py

@@ -0,0 +1,100 @@
+"""Whether a symbol's exchange is in session, judged on the exchange's clock.
+
+Two places judged it on the clock of the machine running Aria. The snapshot
+header said "Market open" on any local weekday, so at 09:00 on Monday in
+Shanghai — Sunday evening in New York — AAPL was "Market open". The context
+sent to the model compared local hours with New York's 09:30–16:00, so a user
+in Beijing had the US market "open" through the New York night.
+
+Exchange holidays are not known here: on a holiday the usual hours still read
+as a session. An unknown exchange, or a time zone this machine lacks, gives
+``None``, and callers then say nothing rather than guess.
+"""
+
+from __future__ import annotations
+
+import re
+from datetime import datetime, time, timezone
+
+try:
+    from zoneinfo import ZoneInfo
+except ImportError:                    # pragma: no cover - Python < 3.9
+    ZoneInfo = None                    # type: ignore[assignment]
+
+_EU = ((time(9, 0), time(17, 30)),)
+
+# Exchange -> (time zone, regular sessions in local time).
+_EXCHANGES: dict[str, tuple[str, tuple[tuple[time, time], ...]]] = {
+    "US": ("America/New_York", ((time(9, 30), time(16, 0)),)),
+    "CN": ("Asia/Shanghai", ((time(9, 30), time(11, 30)), (time(13, 0), time(15, 0)))),
+    "HK": ("Asia/Hong_Kong", ((time(9, 30), time(12, 0)), (time(13, 0), time(16, 0)))),
+    "JP": ("Asia/Tokyo", ((time(9, 0), time(11, 30)), (time(12, 30), time(15, 30)))),
+    "UK": ("Europe/London", ((time(8, 0), time(16, 30)),)),
+    "FR": ("Europe/Paris", _EU),
+    "DE": ("Europe/Berlin", _EU),
+    "NL": ("Europe/Amsterdam", _EU),
+    "IT": ("Europe/Rome", _EU),
+    "ES": ("Europe/Madrid", _EU),
+}
+
+_SUFFIXES = {
+    ".HK": "HK", ".SS": "CN", ".SZ": "CN", ".SH": "CN", ".T": "JP", ".L": "UK",
+    ".PA": "FR", ".DE": "DE", ".AS": "NL", ".MI": "IT", ".MC": "ES",
+}
+
+# US extended hours, New York time.
+_US_PRE = (time(4, 0), time(9, 30))
+_US_POST = (time(16, 0), time(20, 0))
+
+
+def exchange_for(symbol: str) -> str | None:
+    """"US", "CN", "HK", … for a symbol; "CRYPTO" for a coin pair; None if unknown."""
+    s = str(symbol or "").strip().upper()
+    if not s or s.startswith("^") or s.endswith(("=X", "=F")):
+        return None
+    if re.fullmatch(r"[A-Z0-9]{2,10}-(USD|USDT|USDC|EUR)", s):
+        return "CRYPTO"
+    for suffix, exchange in _SUFFIXES.items():
+        if s.endswith(suffix):
+            return exchange
+    if re.fullmatch(r"(SH|SZ)?\d{6}", s):
+        return "CN"
+    if re.fullmatch(r"[A-Z]{1,5}([.-][A-Z])?", s):
+        return "US"
+    return None
+
+
+def _local_now(exchange: str, now: datetime | None) -> datetime | None:
+    if ZoneInfo is None:
+        return None
+    try:
+        zone = ZoneInfo(_EXCHANGES[exchange][0])
+    except Exception:                  # no tz database on this machine
+        return None
+    moment = now or datetime.now(timezone.utc)
+    if moment.tzinfo is None:
+        moment = moment.astimezone()   # a naive time is this machine's local time
+    return moment.astimezone(zone)
+
+
+def market_session(symbol: str, now: datetime | None = None) -> str | None:
+    """"open", "pre", "post", "closed" or "24/7" for the symbol's exchange; None if unknown."""
+    exchange = exchange_for(symbol)
+    if exchange == "CRYPTO":
+        return "24/7"
+    if exchange is None:
+        return None
+    local = _local_now(exchange, now)
+    if local is None:
+        return None
+    if local.weekday() >= 5:
+        return "closed"
+    clock = local.time()
+    if any(start <= clock < end for start, end in _EXCHANGES[exchange][1]):
+        return "open"
+    if exchange == "US":
+        if _US_PRE[0] <= clock < _US_PRE[1]:
+            return "pre"
+        if _US_POST[0] <= clock < _US_POST[1]:
+            return "post"
+    return "closed"

{aria_code-0.80.0 → aria_code-0.81.0}/src/aria_code/aria_cli.py

@@ -165,6 +165,7 @@ from aria_code.apps.cli.plan_mode import PlanModeState
 from aria_code.workspace import VerificationPlanner, WorkspaceFiles, WorkspaceSecurity
 from aria_code.apps.cli.commands.catalog import VISIBLE_SLASH_COMMANDS
 from aria_code.apps.cli.commands.market_context import build_analyze_context, build_analyze_prompt
+from aria_code.apps.cli.market_hours import market_session
 from aria_code.apps.cli.commands.market import (
     parse_analysis_args,
     parse_symbols,
@@ -2587,15 +2588,11 @@ def _build_user_context(config: dict) -> Optional[dict]:
     now = datetime.now()
     ctx["current_datetime"] = now.strftime("%Y-%m-%d %H:%M")
     ctx["day_of_week"] = now.strftime("%A")
-    # US market session heuristic (Mon-Fri, approximate ET hours)
-    weekday = now.weekday()  # 0=Mon, 6=Sun
-    hour_min = now.hour * 60 + now.minute
-    if weekday < 5 and 570 <= hour_min <= 960:  # 9:30am-4:00pm
-        ctx["market_status"] = "open"
-    elif weekday < 5 and (240 <= hour_min < 570 or 960 <= hour_min < 1200):
-        ctx["market_status"] = "pre/after-hours"
-    else:
-        ctx["market_status"] = "closed"
+    # The US session on New York's clock — this compared local hours with
+    # 09:30-16:00, so a user in Beijing had the market "open" overnight.
+    _us_session = market_session("SPY")
+    if _us_session:
+        ctx["market_status"] = {"pre": "pre-market", "post": "after-hours"}.get(_us_session, _us_session)
     # Active model name
     model_id = config.get("model", "qwen2.5:7b")
     mkey = resolve_model_key(model_id)

{aria_code-0.80.0 → aria_code-0.81.0/src/aria_code.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.80.0
+Version: 0.81.0
 Summary: Open-source AI agent for coding, financial analysis and logistics operations: numbers you can check, client data kept apart, actions a person approves
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/artheras/aria-code

{aria_code-0.80.0 → aria_code-0.81.0}/src/aria_code.egg-info/SOURCES.txt

@@ -175,6 +175,7 @@ src/aria_code/apps/cli/intent_signals.py
 src/aria_code/apps/cli/lifecycle_hooks.py
 src/aria_code/apps/cli/local_tool_schemas.py
 src/aria_code/apps/cli/main.py
+src/aria_code/apps/cli/market_hours.py
 src/aria_code/apps/cli/market_metadata.py
 src/aria_code/apps/cli/market_universe.py
 src/aria_code/apps/cli/message_processing.py
@@ -600,6 +601,7 @@ tests/test_market_answers_in_english.py
 tests/test_market_data_client.py
 tests/test_market_history_inheritance.py
 tests/test_market_history_tool.py
+tests/test_market_hours.py
 tests/test_market_questions_on_backend.py
 tests/test_market_symbol_extraction.py
 tests/test_market_universe.py

{aria_code-0.80.0 → aria_code-0.81.0}/tests/test_market_answers_in_english.py

@@ -199,3 +199,12 @@ def test_analyze_tells_the_gate_whether_its_prompt_carries_data(monkeypatch):
     asyncio.run(Cli("## AAPL Market Data\n- Price: unavailable (configure a data service key via /apikey)")
                 .cmd_analyze("AAPL"))
     assert sent == [True, False]
+
+
+def test_the_snapshot_header_states_the_session_on_the_exchange_clock(offline_market, monkeypatch):
+    monkeypatch.setattr(market_handlers, "market_session", lambda _symbol: "closed")
+    header = market_handlers._try_handle_market_snapshot_analysis("AAPL price")["response"].splitlines()[1]
+    assert "· Market closed ·" in header
+    monkeypatch.setattr(market_handlers, "market_session", lambda _symbol: None)
+    header = market_handlers._try_handle_market_snapshot_analysis("AAPL price")["response"].splitlines()[1]
+    assert "Market" not in header and "· ·" not in header and header.endswith("Not investment advice*")

aria_code-0.81.0/tests/test_market_hours.py

@@ -0,0 +1,73 @@
+"""Market sessions are judged on the exchange's clock, not this machine's.
+
+At 09:18 on Monday 2026-10-05 in Shanghai — Sunday 21:18 in New York — the
+AAPL snapshot said "Market open", and the context sent to the model said the
+US market was open from 09:30 to 16:00 Beijing time.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime, timedelta, timezone
+
+import pytest
+
+from aria_code.apps.cli.market_hours import exchange_for, market_session
+
+BEIJING = timezone(timedelta(hours=8))
+
+
+def at_beijing(*args):
+    return datetime(*args, tzinfo=BEIJING)
+
+
+@pytest.mark.parametrize("symbol, exchange", [
+    ("AAPL", "US"), ("BRK-B", "US"), ("BRK.B", "US"), ("0700.HK", "HK"), ("600519", "CN"),
+    ("000001.SZ", "CN"), ("SH600519", "CN"), ("7203.T", "JP"), ("MC.PA", "FR"), ("SAP.DE", "DE"),
+    ("BTC-USD", "CRYPTO"), ("SOL-USDT", "CRYPTO"),
+    ("^GSPC", None), ("EURUSD=X", None), ("GC=F", None), ("", None),
+])
+def test_the_exchange_follows_the_symbol(symbol, exchange):
+    assert exchange_for(symbol) == exchange
+
+
+@pytest.mark.parametrize("moment, session", [
+    (at_beijing(2026, 10, 5, 9, 18), "closed"),     # Sunday 21:18 in New York
+    (at_beijing(2026, 10, 5, 21, 0), "pre"),        # Monday 09:00 in New York
+    (at_beijing(2026, 10, 5, 22, 0), "open"),       # Monday 10:00
+    (at_beijing(2026, 10, 6, 4, 30), "post"),       # Monday 16:30
+    (at_beijing(2026, 10, 6, 9, 0), "closed"),      # Monday 21:00
+    (at_beijing(2026, 10, 10, 23, 0), "closed"),    # Saturday
+])
+def test_the_us_session_is_on_new_york_time(moment, session):
+    assert market_session("AAPL", moment) == session
+
+
+def test_daylight_saving_moves_the_us_open():
+    # New York leaves daylight saving on 2026-11-01: the open moves from 21:30
+    # to 22:30 Beijing time.
+    assert market_session("AAPL", at_beijing(2026, 10, 30, 21, 45)) == "open"
+    assert market_session("AAPL", at_beijing(2026, 11, 2, 21, 45)) == "pre"
+
+
+def test_asian_lunch_breaks_and_hours():
+    assert market_session("600519", at_beijing(2026, 10, 12, 10, 0)) == "open"
+    assert market_session("600519", at_beijing(2026, 10, 12, 12, 0)) == "closed"
+    assert market_session("600519", at_beijing(2026, 10, 12, 15, 30)) == "closed"
+    assert market_session("0700.HK", at_beijing(2026, 10, 12, 15, 30)) == "open"
+    assert market_session("0700.HK", at_beijing(2026, 10, 12, 12, 30)) == "closed"
+
+
+def test_crypto_never_closes_and_unknown_says_nothing():
+    assert market_session("BTC-USD", at_beijing(2026, 10, 10, 3, 0)) == "24/7"
+    assert market_session("^GSPC", at_beijing(2026, 10, 12, 22, 0)) is None
+
+
+def test_a_machine_without_the_time_zone_gets_no_answer(monkeypatch):
+    import aria_code.apps.cli.market_hours as market_hours
+
+    def missing(_name):
+        raise LookupError("no tz database")
+
+    monkeypatch.setattr(market_hours, "ZoneInfo", missing)
+    assert market_session("AAPL", at_beijing(2026, 10, 12, 22, 0)) is None
+