aria-code 0.155.0__tar.gz → 0.156.0__tar.gz

{aria_code-0.155.0/src/aria_code.egg-info → aria_code-0.156.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.155.0
+Version: 0.156.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
@@ -172,7 +172,7 @@ Numbers you can check, client data kept apart, actions a person approves.</p>
 <p align="center">
   <a href="https://www.npmjs.com/package/@artheras/aria-code"><img src="https://img.shields.io/npm/v/@artheras/aria-code/latest?style=flat-square&logo=npm&label=npm" alt="npm version"></a>
   <!-- Release automation pins PyPI: the default endpoint still selects legacy 4.x. -->
-  <a href="https://pypi.org/project/aria-code/0.155.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.155.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/aria-code/0.156.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.156.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
   <a href="https://github.com/artheras/aria-code/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/artheras/aria-code/ci.yml?branch=main&style=flat-square&logo=githubactions&label=CI" alt="CI status"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-Apache%202.0-64748b?style=flat-square" alt="Apache License 2.0"></a>
 </p>

{aria_code-0.155.0 → aria_code-0.156.0}/README.md

@@ -15,7 +15,7 @@ Numbers you can check, client data kept apart, actions a person approves.</p>
 <p align="center">
   <a href="https://www.npmjs.com/package/@artheras/aria-code"><img src="https://img.shields.io/npm/v/@artheras/aria-code/latest?style=flat-square&logo=npm&label=npm" alt="npm version"></a>
   <!-- Release automation pins PyPI: the default endpoint still selects legacy 4.x. -->
-  <a href="https://pypi.org/project/aria-code/0.155.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.155.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/aria-code/0.156.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.156.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
   <a href="https://github.com/artheras/aria-code/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/artheras/aria-code/ci.yml?branch=main&style=flat-square&logo=githubactions&label=CI" alt="CI status"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-Apache%202.0-64748b?style=flat-square" alt="Apache License 2.0"></a>
 </p>

{aria_code-0.155.0 → aria_code-0.156.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aria-code"
-version = "0.155.0"
+version = "0.156.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.155.0 → aria_code-0.156.0}/src/aria_code/_version.py

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

{aria_code-0.155.0 → aria_code-0.156.0}/src/aria_code/apps/cli/commands/model_cmds.py

@@ -182,9 +182,37 @@ logger = logging.getLogger(__name__)
 class ModelCommandsMixin:
     """Mixin: Model/config commands: model, apikey, providers, cloud, config, tools, skills."""
 
+    def _listed_model_ids(self) -> list:
+        """The ids behind the numbers /models prints, in the order it prints them.
+
+        The number is the most prominent thing on each row, so it has to mean
+        something; the printer and this must therefore read the same order
+        from the same place. Community models are keyed by name, which is
+        their id.
+        """
+        ids = [m["id"] for m in _get_MODELS().values()]
+        for name in getattr(self, "_last_listed_community", []) or []:
+            if name not in ids:
+                ids.append(name)
+        return ids
+
     async def cmd_model(self, args: str):
         name = args.strip()
 
+        # /models prints a numbered list, so a number is the obvious thing to
+        # type back. It used to be taken as a model name and matched nothing,
+        # which left the numbers on screen decorative.
+        if name.isdigit():
+            listed = self._listed_model_ids()
+            index = int(name)
+            if 1 <= index <= len(listed):
+                name = listed[index - 1]
+            else:
+                msg = ui_text(self, f"没有第 {index} 个模型；用 /models 查看当前列表。",
+                              f"There is no model {index}; run /models for the current list.")
+                self.context.console.print(f"[yellow]{msg}[/yellow]") if self.context.has_rich else print(msg)
+                return
+
         # A model Google has retired would fail on the next request; switch
         # to its replacement and say so (model_retirement.py).
         from aria_code.apps.cli.model_retirement import retired_replacement
@@ -429,6 +457,8 @@ class ModelCommandsMixin:
             # Non-interactive (-p mode): show static numbered list then return.
             # The arrow picker cannot run without a TTY.
             community_list = [cm for cm in rich_models if cm["name"] not in aria_ids]
+            # What /model <number> will count through, recorded as it is printed.
+            self._last_listed_community = [cm["name"] for cm in community_list]
             for key, m in _get_MODELS().items():
                 mid    = m["id"]
                 is_cur = mid == current_id
@@ -450,7 +480,8 @@ class ModelCommandsMixin:
                     self.context.console.print(line) if self.context.has_rich else print(line)
                     idx_counter += 1
             self.context.console.print() if self.context.has_rich else print()
-            self.context.console.print("  [dim]Use /model <id> to switch. E.g. /model deepseek/deepseek-chat[/dim]") if self.context.has_rich else print("  Use /model <id> to switch.")
+            hint = "Use /model <number> from this list, or /model <id>. E.g. /model 1 or /model deepseek/deepseek-chat"
+            self.context.console.print(f"  [dim]{hint}[/dim]") if self.context.has_rich else print(f"  {hint}")
             return
 
         # ── Build compact options for _arrow_select ────────────────────────

aria_code-0.156.0/src/aria_code/packages/aria_services/research_protocol.py

@@ -0,0 +1,134 @@
+"""Financial-research grounding policy shared by Aria adapters."""
+
+from __future__ import annotations
+
+import re
+from collections.abc import Iterable, Mapping
+
+
+_MARKET_IDENTIFIER = re.compile(
+    r"\b[A-Z]{1,5}(?:\.[A-Z]{1,4})?\b|\b\d{6}(?:\.(?:SH|SZ|SS|HK))?\b"
+)
+# Words that name a financial subject on their own. Everyday words used to be
+# in this list too — market, share, option, volume, price, holding, fund, and
+# 市场, 组合, 加密 — so "analyze the performance of the share button", "how is
+# option parsing doing", "分析一下这个项目的市场定位" and "分析当前加密算法的风险"
+# (加密 alone is encryption) were all gated as market questions, and refused
+# after the turn had already done its real work, because no market-data tool
+# had run. Only terms that mean finance wherever they appear belong here.
+_FINANCIAL_SUBJECTS = (
+    "stock", "equity", "equities", "portfolio", "crypto", "bitcoin", "forex",
+    "etf", "dividend", "earnings", "ticker", "share price", "stock market",
+    "market cap", "trading volume", "bond yield", "mutual fund", "index fund",
+    "hedge fund", "options chain", "call option", "put option", "price target",
+    "股票", "股价", "股市", "大盘", "a股", "美股", "港股", "行情", "持仓", "基金",
+    "期权", "债券", "外汇", "加密货币", "数字货币", "币价", "成交量", "市值",
+    "财报", "基本面", "涨跌", "收盘价", "开盘价", "目标价",
+)
+
+# Words that are financial only some of the time: "how is the market doing
+# today" is a market question, "这个项目的市场定位" is not. They count when the
+# message asks about now and says nothing that places it in the user's own
+# product or code.
+_AMBIGUOUS_SUBJECTS = ("market", "price", "valuation", "市场", "价格", "估值")
+_TIME_INTENTS = (
+    "today", "current", "latest", "this week", "right now", "how's", "how is",
+    "doing", "今天", "当前", "最新", "本周", "这周", "近期", "现在", "走势", "怎么样",
+)
+_WORKSPACE_CONTEXT = (
+    "project", "product", "code", "app", "function", "button", "feature",
+    "repo", "file", "api", "cli", "page", "plan", "pricing", "subscription",
+    "competitor", "positioning", "user",
+    "项目", "产品", "代码", "函数", "功能", "页面", "应用", "仓库", "文件",
+    "定位", "竞品", "用户", "方案", "套餐",
+)
+
+_EVIDENCE_INTENTS = (
+    "analyze", "analysis", "forecast", "predict", "recommend", "buy", "sell",
+    "current", "latest", "today", "trend", "risk", "target", "outlook",
+    "this week", "how's", "how is", "doing", "performance",
+    "分析", "预测", "推荐", "买入", "卖出", "最新", "当前", "今天", "走势",
+    "趋势", "风险", "目标价", "前景", "成交量", "全面分析",
+    "本周", "这周", "近期", "怎么样", "表现", "看法", "值得",
+)
+_GROUNDING_TOOL_FRAGMENTS = (
+    "market", "quote", "price", "ohlcv", "fundamental", "filing", "news",
+    "factor", "risk", "backtest", "portfolio", "prediction", "signal",
+    "regime", "financial", "earnings", "valuation", "macro", "flow",
+)
+
+
+def _mentions(lowered: str, terms: Iterable[str]) -> bool:
+    """Whether any term occurs — as a word for English, as a substring for CJK.
+
+    Substrings are wrong for short English words in both directions: "market"
+    is inside "marketing" and "equity" inside "inequity", while "app" is inside
+    "happening" and "api" inside "capital". Chinese has no spaces to bound a
+    word, so it keeps substring matching.
+    """
+    for term in terms:
+        if term.isascii():
+            if re.search(rf"(?<![a-z]){re.escape(term)}(?![a-z])", lowered):
+                return True
+        elif term in lowered:
+            return True
+    return False
+
+
+def _names_an_instrument(text: str) -> bool:
+    """A ticker, code or company the finance pack recognises, or a named company.
+
+    This used to be a bare regex, \\b[A-Z]{1,5}\\b: it took "API" and "CSV"
+    for tickers ("analyze the API latency trend" was gated as a market
+    question and refused without market data), and missed "分析AAPL" — no word
+    boundary between 析 and A — and every company named in Chinese.
+    """
+    try:
+        from aria_code.packs import activate_packs, load_builtin_packs
+
+        load_builtin_packs()
+        if any(a.pack == "finance" for a in activate_packs(text)):
+            return True
+    except Exception:
+        if _MARKET_IDENTIFIER.search(text):
+            return True
+    try:
+        from aria_code.apps.cli.intent_signals import _ENTITY_TERMS
+    except Exception:
+        return False
+    lowered = text.lower()
+    return any(term in lowered for term in _ENTITY_TERMS)
+
+
+def requires_financial_evidence(query: str) -> bool:
+    """Return whether a request needs current external financial evidence."""
+    text = str(query or "").strip()
+    if not text:
+        return False
+    lowered = text.lower()
+    if not any(term in lowered for term in _EVIDENCE_INTENTS):
+        return False
+    if _mentions(lowered, _FINANCIAL_SUBJECTS) or _names_an_instrument(text):
+        return True
+    return (
+        _mentions(lowered, _AMBIGUOUS_SUBJECTS)
+        and _mentions(lowered, _TIME_INTENTS)
+        and not _mentions(lowered, _WORKSPACE_CONTEXT)
+    )
+
+
+def grounding_tool_names(tool_schemas: Iterable[Mapping]) -> frozenset[str]:
+    """Select registered tools whose outputs can ground a financial conclusion."""
+    names: set[str] = set()
+    for schema in tool_schemas:
+        function = schema.get("function") if isinstance(schema, Mapping) else None
+        name = str(
+            (function or {}).get("name")
+            if isinstance(function, Mapping)
+            else schema.get("name", "")
+        )
+        canonical = name.rsplit("__", 1)[-1].lower()
+        if name and any(fragment in canonical for fragment in _GROUNDING_TOOL_FRAGMENTS):
+            names.add(name)
+    return frozenset(names)
+

{aria_code-0.155.0 → aria_code-0.156.0}/src/aria_code/providers/llm/arthera_gateway.py

@@ -32,7 +32,19 @@ DEFAULT_MODEL = "google/gemini-3.5-flash"
 
 
 def _config_path() -> Path:
-    return Path(os.getenv("ARIA_CONFIG_DIR", Path.home() / ".arthera")) / "config.json"
+    """Where /login saved the session: the same file, by the same rule.
+
+    This used to be ~/.arthera/config.json, or ARIA_CONFIG_DIR. /login writes
+    through aria_home(), which is ~/.arthera only when that legacy folder
+    already exists and ~/.aria-code otherwise, and is overridden by ARIA_HOME.
+    So on a new install — the people signing in to reach a model without a
+    key — the session was saved in one folder and looked for in another, and
+    signing in never gave them a model. Asking aria_home() is what the 38
+    files that once hardcoded ~/.arthera were moved over to for this reason.
+    """
+    from aria_code.packages.aria_core.paths import aria_home
+
+    return aria_home() / "config.json"
 
 
 def session_token() -> Optional[str]:

{aria_code-0.155.0 → aria_code-0.156.0/src/aria_code.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.155.0
+Version: 0.156.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
@@ -172,7 +172,7 @@ Numbers you can check, client data kept apart, actions a person approves.</p>
 <p align="center">
   <a href="https://www.npmjs.com/package/@artheras/aria-code"><img src="https://img.shields.io/npm/v/@artheras/aria-code/latest?style=flat-square&logo=npm&label=npm" alt="npm version"></a>
   <!-- Release automation pins PyPI: the default endpoint still selects legacy 4.x. -->
-  <a href="https://pypi.org/project/aria-code/0.155.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.155.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/aria-code/0.156.0/"><img src="https://img.shields.io/badge/dynamic/json?url=https%3A%2F%2Fpypi.org%2Fpypi%2Faria-code%2F0.156.0%2Fjson&amp;query=%24.info.version&amp;prefix=v&amp;style=flat-square&amp;logo=pypi&amp;label=PyPI&amp;color=blue&amp;cacheSeconds=300" alt="PyPI version"></a>
   <a href="https://github.com/artheras/aria-code/actions/workflows/ci.yml"><img src="https://img.shields.io/github/actions/workflow/status/artheras/aria-code/ci.yml?branch=main&style=flat-square&logo=githubactions&label=CI" alt="CI status"></a>
   <a href="LICENSE"><img src="https://img.shields.io/badge/license-Apache%202.0-64748b?style=flat-square" alt="Apache License 2.0"></a>
 </p>

{aria_code-0.155.0 → aria_code-0.156.0}/src/aria_code.egg-info/SOURCES.txt

@@ -706,6 +706,7 @@ tests/test_mcp_client.py
 tests/test_mentions.py
 tests/test_model_capability_registry.py
 tests/test_model_choice_is_the_users.py
+tests/test_model_number_selection.py
 tests/test_model_probe.py
 tests/test_model_selection_state.py
 tests/test_narrow_chinese_output.py

{aria_code-0.155.0 → aria_code-0.156.0}/tests/test_arthera_gateway_provider.py

@@ -30,29 +30,52 @@ class SessionTest(unittest.TestCase):
 
     def test_the_token_login_stored_is_what_authorises_calls(self):
         self.write_config({"auth_token": "firebase-token"})
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": str(self.dir)}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": str(self.dir)}):
             self.assertEqual(gw.session_token(), "firebase-token")
 
     def test_signed_out_is_no_token_rather_than_an_empty_one(self):
         # An empty credential reaches the server and comes back 401, which
         # tells the user nothing they can act on.
         self.write_config({"auth_token": None})
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": str(self.dir)}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": str(self.dir)}):
             self.assertIsNone(gw.session_token())
             self.assertFalse(gw.ArtheraGatewayProvider.available())
 
     def test_a_missing_config_is_not_an_error(self):
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": str(self.dir / "absent")}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": str(self.dir / "absent")}):
             self.assertIsNone(gw.session_token())
 
     def test_a_corrupt_config_is_not_an_error(self):
         (self.dir / "config.json").write_text("{ not json", encoding="utf-8")
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": str(self.dir)}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": str(self.dir)}):
             self.assertIsNone(gw.session_token())
 
+    def test_a_new_install_reads_the_session_where_login_wrote_it(self):
+        # With no legacy ~/.arthera, /login saves under ~/.aria-code. The
+        # provider used to look only in ~/.arthera, so on a new install signing
+        # in never gave the user a model.
+        home = self.dir / "home"
+        (home / ".aria-code").mkdir(parents=True)
+        (home / ".aria-code" / "config.json").write_text(
+            json.dumps({"auth_token": "fresh-install"}), encoding="utf-8")
+        env = {k: v for k, v in os.environ.items() if k != "ARIA_HOME"}
+        env["HOME"] = str(home)
+        with mock.patch.dict(os.environ, env, clear=True):
+            self.assertEqual(gw.session_token(), "fresh-install")
+
+    def test_an_existing_install_keeps_reading_its_legacy_folder(self):
+        home = self.dir / "home"
+        (home / ".arthera").mkdir(parents=True)
+        (home / ".arthera" / "config.json").write_text(
+            json.dumps({"auth_token": "legacy"}), encoding="utf-8")
+        env = {k: v for k, v in os.environ.items() if k != "ARIA_HOME"}
+        env["HOME"] = str(home)
+        with mock.patch.dict(os.environ, env, clear=True):
+            self.assertEqual(gw.session_token(), "legacy")
+
     def test_a_signed_in_user_is_available(self):
         self.write_config({"auth_token": "t"})
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": str(self.dir)}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": str(self.dir)}):
             self.assertTrue(gw.ArtheraGatewayProvider.available())
 
 
@@ -77,13 +100,13 @@ class ProviderTest(unittest.TestCase):
         Path(self.tmp.name, "config.json").write_text(json.dumps({"auth_token": "t"}), encoding="utf-8")
 
     def test_it_builds_itself_from_the_session_with_nothing_supplied(self):
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": self.tmp.name, "ARIA_API_URL": ""}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": self.tmp.name, "ARIA_API_URL": ""}):
             provider = gw.ArtheraGatewayProvider(ProviderConfig(name="arthera", model=""))
         self.assertEqual(provider.config.api_key, "t")
         self.assertEqual(provider.config.base_url, "https://api.arthera.finance/api/v2/llm/v1")
 
     def test_an_explicit_base_url_is_respected(self):
-        with mock.patch.dict(os.environ, {"ARIA_CONFIG_DIR": self.tmp.name}):
+        with mock.patch.dict(os.environ, {"ARIA_HOME": self.tmp.name}):
             provider = gw.ArtheraGatewayProvider(
                 ProviderConfig(name="arthera", model="", base_url="https://elsewhere.example/v1"))
         self.assertEqual(provider.config.base_url, "https://elsewhere.example/v1")

aria_code-0.156.0/tests/test_model_number_selection.py

@@ -0,0 +1,129 @@
+"""The numbers /models prints are the numbers /model accepts.
+
+Under the Rust frontend the Python worker's stdin is a pipe, so it can never
+read keystrokes and the arrow picker cannot run there — the numbered list is
+the right output, not a fallback to be removed. What was wrong is that the
+list numbered every row with something that did nothing, showed a display
+name you could not type, and then told you to use an id it never printed.
+
+So the number has to resolve, and the printer and the resolver have to agree
+about what it counts.
+"""
+from __future__ import annotations
+
+import unittest
+from unittest import mock
+
+from aria_code.apps.cli.commands import model_cmds
+
+
+class _Console:
+    def __init__(self):
+        self.lines = []
+
+    def print(self, text=""):
+        self.lines.append(str(text))
+
+
+class _Context:
+    def __init__(self):
+        self.console = _Console()
+        # True so messages route to the console above rather than stdout.
+        self.has_rich = True
+
+
+class _Mixin(model_cmds.ModelCommandsMixin):
+    """The mixin alone, with only what the number path touches."""
+
+    def __init__(self):
+        self.context = _Context()
+        self.lang = "en"
+
+
+ARIA = {
+    "a": {"id": "google/gemini-3.5-flash", "name": "Gemini"},
+    "b": {"id": "deepseek/deepseek-chat", "name": "DeepSeek"},
+}
+
+
+class ListedOrderTest(unittest.TestCase):
+    def setUp(self):
+        self.mixin = _Mixin()
+
+    def test_the_order_is_the_aria_models_then_the_community_ones(self):
+        with mock.patch.object(model_cmds, "_get_MODELS", lambda: ARIA):
+            self.mixin._last_listed_community = ["qwen3:8b"]
+            self.assertEqual(
+                self.mixin._listed_model_ids(),
+                ["google/gemini-3.5-flash", "deepseek/deepseek-chat", "qwen3:8b"],
+            )
+
+    def test_a_community_model_already_listed_is_not_counted_twice(self):
+        # Two rows for one model would make every number after it wrong.
+        with mock.patch.object(model_cmds, "_get_MODELS", lambda: ARIA):
+            self.mixin._last_listed_community = ["deepseek/deepseek-chat", "qwen3:8b"]
+            self.assertEqual(self.mixin._listed_model_ids().count("deepseek/deepseek-chat"), 1)
+
+    def test_without_a_listing_it_is_just_the_aria_models(self):
+        with mock.patch.object(model_cmds, "_get_MODELS", lambda: ARIA):
+            self.assertEqual(len(self.mixin._listed_model_ids()), 2)
+
+
+class NumberResolvesTest(unittest.IsolatedAsyncioTestCase):
+    def setUp(self):
+        self.mixin = _Mixin()
+        self.mixin._last_listed_community = ["qwen3:8b"]
+
+    async def resolve(self, typed: str):
+        """Run cmd_model far enough to see what the number became."""
+        seen = {}
+
+        async def stop_at_name(_self, args):
+            seen["name"] = args
+            return None
+
+        with mock.patch.object(model_cmds, "_get_MODELS", lambda: ARIA), \
+             mock.patch.object(model_cmds, "retired_replacement", lambda _n: None, create=True):
+            # Everything past resolution is provider plumbing; capture the
+            # resolved name by letting the retirement lookup observe it.
+            with mock.patch("aria_code.apps.cli.model_retirement.retired_replacement",
+                            side_effect=lambda n: seen.setdefault("name", n) and None):
+                try:
+                    await self.mixin.cmd_model(typed)
+                except Exception:
+                    pass
+        return seen.get("name")
+
+    async def test_a_number_becomes_the_id_on_that_row(self):
+        self.assertEqual(await self.resolve("2"), "deepseek/deepseek-chat")
+
+    async def test_the_first_row_is_one_not_zero(self):
+        self.assertEqual(await self.resolve("1"), "google/gemini-3.5-flash")
+
+    async def test_a_community_row_resolves_too(self):
+        self.assertEqual(await self.resolve("3"), "qwen3:8b")
+
+    async def test_a_number_past_the_end_says_so_and_changes_nothing(self):
+        await self.mixin.cmd_model("99")
+        printed = " ".join(self.mixin.context.console.lines)
+        self.assertIn("99", printed)
+        self.assertIn("/models", printed)
+
+    async def test_zero_is_out_of_range_rather_than_the_last_row(self):
+        # Negative indexing would silently pick something.
+        await self.mixin.cmd_model("0")
+        self.assertIn("/models", " ".join(self.mixin.context.console.lines))
+
+
+class HintTest(unittest.TestCase):
+    def test_the_hint_offers_what_the_screen_actually_shows(self):
+        source = model_cmds.__file__
+        with open(source, encoding="utf-8") as handle:
+            text = handle.read()
+        self.assertIn("Use /model <number> from this list", text)
+        # The old hint named only an id the list never printed.
+        self.assertNotIn("Use /model <id> to switch. E.g.", text)
+
+
+if __name__ == "__main__":
+    unittest.main()

aria_code-0.156.0/tests/test_research_protocol.py

@@ -0,0 +1,70 @@
+from aria_code.packages.aria_services.research_protocol import (
+    grounding_tool_names,
+    requires_financial_evidence,
+)
+
+
+def test_financial_evidence_intent_is_conservative():
+    assert requires_financial_evidence("分析苹果股票走势和成交量")
+    assert requires_financial_evidence("What is the current AAPL price?")
+    assert not requires_financial_evidence("Explain what a Sharpe ratio is")
+
+
+def test_grounding_tool_names_support_local_and_mcp_schemas():
+    schemas = [
+        {"name": "get_market_data"},
+        {"name": "read_file"},
+        {
+            "type": "function",
+            "function": {"name": "mcp__arthera_quant_engine__run_backtest"},
+        },
+    ]
+
+    assert grounding_tool_names(schemas) == frozenset({
+        "get_market_data",
+        "mcp__arthera_quant_engine__run_backtest",
+    })
+
+
+# ── Ordinary words are not financial subjects ────────────────────────────────
+# The subject list used to hold everyday words — market, share, option, volume,
+# price, holding, fund, 市场, 组合, 加密 — so a question about the user's own
+# product or code was gated as a market question and refused after the turn
+# had done its real work, because no market-data tool had run. A turn that read
+# the project structure and was then refused for 179 seconds is how this
+# surfaced.
+
+import pytest  # noqa: E402
+
+
+@pytest.mark.parametrize("question", [
+    "分析一下这个项目的市场定位",
+    "帮我写一份产品的市场分析报告",
+    "analyze the performance of the share button",
+    "what is the current volume setting in the player",
+    "how is option parsing doing in the CLI",
+    "分析当前加密算法的风险",          # 加密 alone is encryption
+    "分析一下用户组合功能的表现",      # 组合 is any combination
+    "how is our marketing doing this week",   # market is inside marketing
+    "analyze the inequity in current data",   # equity is inside inequity
+])
+def test_a_question_about_the_users_own_work_is_not_a_market_question(question):
+    assert not requires_financial_evidence(question)
+
+
+@pytest.mark.parametrize("question", [
+    "how is the market doing today",
+    "今天市场怎么样",
+    "大盘走势",
+    "美股今天表现怎么样",
+])
+def test_a_question_about_the_market_itself_still_needs_evidence(question):
+    # Narrowing the list must not open the door the gate exists to close: a
+    # conclusion about today's market with no data behind it.
+    assert requires_financial_evidence(question)
+
+
+def test_a_word_inside_a_longer_word_does_not_count_as_context():
+    # "app" is inside "happening"; matched as a substring it would veto a real
+    # market question as if it were about the user's app.
+    assert requires_financial_evidence("what is happening in the market today")

aria_code-0.155.0/src/aria_code/packages/aria_services/research_protocol.py

@@ -1,83 +0,0 @@
-"""Financial-research grounding policy shared by Aria adapters."""
-
-from __future__ import annotations
-
-import re
-from collections.abc import Iterable, Mapping
-
-
-_MARKET_IDENTIFIER = re.compile(
-    r"\b[A-Z]{1,5}(?:\.[A-Z]{1,4})?\b|\b\d{6}(?:\.(?:SH|SZ|SS|HK))?\b"
-)
-_FINANCIAL_SUBJECTS = (
-    "stock", "share", "equity", "market", "portfolio", "holding", "fund",
-    "option", "bond", "forex", "crypto", "price", "volume", "valuation",
-    "股票", "股价", "行情", "市场", "持仓", "组合", "基金", "期权", "债券",
-    "外汇", "加密", "成交量", "估值", "市值", "财报", "基本面",
-)
-_EVIDENCE_INTENTS = (
-    "analyze", "analysis", "forecast", "predict", "recommend", "buy", "sell",
-    "current", "latest", "today", "trend", "risk", "target", "outlook",
-    "this week", "how's", "how is", "doing", "performance",
-    "分析", "预测", "推荐", "买入", "卖出", "最新", "当前", "今天", "走势",
-    "趋势", "风险", "目标价", "前景", "成交量", "全面分析",
-    "本周", "这周", "近期", "怎么样", "表现", "看法", "值得",
-)
-_GROUNDING_TOOL_FRAGMENTS = (
-    "market", "quote", "price", "ohlcv", "fundamental", "filing", "news",
-    "factor", "risk", "backtest", "portfolio", "prediction", "signal",
-    "regime", "financial", "earnings", "valuation", "macro", "flow",
-)
-
-
-def _names_an_instrument(text: str) -> bool:
-    """A ticker, code or company the finance pack recognises, or a named company.
-
-    This used to be a bare regex, \\b[A-Z]{1,5}\\b: it took "API" and "CSV"
-    for tickers ("analyze the API latency trend" was gated as a market
-    question and refused without market data), and missed "分析AAPL" — no word
-    boundary between 析 and A — and every company named in Chinese.
-    """
-    try:
-        from aria_code.packs import activate_packs, load_builtin_packs
-
-        load_builtin_packs()
-        if any(a.pack == "finance" for a in activate_packs(text)):
-            return True
-    except Exception:
-        if _MARKET_IDENTIFIER.search(text):
-            return True
-    try:
-        from aria_code.apps.cli.intent_signals import _ENTITY_TERMS
-    except Exception:
-        return False
-    lowered = text.lower()
-    return any(term in lowered for term in _ENTITY_TERMS)
-
-
-def requires_financial_evidence(query: str) -> bool:
-    """Return whether a request needs current external financial evidence."""
-    text = str(query or "").strip()
-    if not text:
-        return False
-    lowered = text.lower()
-    if not any(term in lowered for term in _EVIDENCE_INTENTS):
-        return False
-    return any(term in lowered for term in _FINANCIAL_SUBJECTS) or _names_an_instrument(text)
-
-
-def grounding_tool_names(tool_schemas: Iterable[Mapping]) -> frozenset[str]:
-    """Select registered tools whose outputs can ground a financial conclusion."""
-    names: set[str] = set()
-    for schema in tool_schemas:
-        function = schema.get("function") if isinstance(schema, Mapping) else None
-        name = str(
-            (function or {}).get("name")
-            if isinstance(function, Mapping)
-            else schema.get("name", "")
-        )
-        canonical = name.rsplit("__", 1)[-1].lower()
-        if name and any(fragment in canonical for fragment in _GROUNDING_TOOL_FRAGMENTS):
-            names.add(name)
-    return frozenset(names)
-