aria-code 0.57.0__tar.gz → 0.58.0__tar.gz

{aria_code-0.57.0 → aria_code-0.58.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.57.0
+Version: 0.58.0
 Summary: AI coding and research workspace with optional finance and connected-domain tools
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/artheras/aria-code

{aria_code-0.57.0 → aria_code-0.58.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aria-code"
-version = "0.57.0"
+version = "0.58.0"
 description = "AI coding and research workspace with optional finance and connected-domain tools"
 readme = "README.md"
 license = { text = "Apache-2.0" }

{aria_code-0.57.0 → aria_code-0.58.0}/src/aria_code/_version.py

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

{aria_code-0.57.0 → aria_code-0.58.0}/src/aria_code/tools/local_finance_tools.py

@@ -69,7 +69,10 @@ except ImportError:
 try:
     import akshare as ak
     _HAS_AK = True
-except ImportError:
+# Not just ImportError: akshare reads bundled data files and calendars on
+# import, and a partial install (or a frozen binary without its data) raises
+# FileNotFoundError or worse. Optional means optional — never crash the CLI.
+except Exception:  # noqa: BLE001
     _HAS_AK = False
 
 

{aria_code-0.57.0 → aria_code-0.58.0}/src/aria_code/tools/macro_tools.py

@@ -36,7 +36,10 @@ except ImportError:
 try:
     import akshare as ak
     _HAS_AK = True
-except ImportError:
+# Not just ImportError: akshare reads bundled data files and calendars on
+# import, and a partial install (or a frozen binary without its data) raises
+# FileNotFoundError or worse. Optional means optional — never crash the CLI.
+except Exception:  # noqa: BLE001
     _HAS_AK = False
 
 try:

{aria_code-0.57.0 → aria_code-0.58.0}/src/aria_code/tools/realty_data_tools.py

@@ -26,7 +26,10 @@ logger = logging.getLogger(__name__)
 try:
     import akshare as ak
     _HAS_AK = True
-except ImportError:
+# Not just ImportError: akshare reads bundled data files and calendars on
+# import, and a partial install (or a frozen binary without its data) raises
+# FileNotFoundError or worse. Optional means optional — never crash the CLI.
+except Exception:  # noqa: BLE001
     _HAS_AK = False
 
 try:

{aria_code-0.57.0 → aria_code-0.58.0}/src/aria_code.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aria-code
-Version: 0.57.0
+Version: 0.58.0
 Summary: AI coding and research workspace with optional finance and connected-domain tools
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/artheras/aria-code

{aria_code-0.57.0 → aria_code-0.58.0}/src/aria_code.egg-info/SOURCES.txt

@@ -497,6 +497,8 @@ tests/test_ashare_prediction_cmds.py
 tests/test_backtest_cmds.py
 tests/test_backtest_engine.py
 tests/test_backtest_report.py
+tests/test_binary_bundles_runtime_imports.py
+tests/test_broken_akshare_is_optional.py
 tests/test_broker_automation.py
 tests/test_broker_chat_confirm_gate.py
 tests/test_broker_planning.py

aria_code-0.58.0/tests/test_binary_bundles_runtime_imports.py

@@ -0,0 +1,142 @@
+"""Every module the program loads by name at runtime must be in the frozen binary.
+
+PyInstaller follows import statements. A module loaded by name —
+
+    sys.modules[__name__] = import_module("clients.market_data_client")
+    importlib.import_module(module_path)        # from a registry table
+
+— is invisible to it. Inspected, the v0.56.0 macOS binary contained:
+
+    compatibility shim targets    0 of 26
+    broker adapters               1 of 10   (paper trading only)
+    agents loaded by name         8 of 31   (no warehouse, realty, coder, supervisor)
+
+Every one of those loaders catches ImportError and logs at debug level, so
+nothing failed. `quote` fell back to a backend at localhost:8000; the broker
+list was shorter; whole agent teams were absent.
+
+scripts/pyinstaller_collect_args.py writes a PyInstaller hook listing every
+first-party module under both import roots. These tests keep it that way
+without building anything: each runtime target must be listed under the exact
+name it is imported by (the bare and packaged roots are different modules —
+listing only one is how --collect-submodules failed), and every PyInstaller
+invocation must use the collector.
+"""
+
+from __future__ import annotations
+
+import ast
+import importlib.util
+import pathlib
+import re
+import tempfile
+import unittest
+
+ROOT = pathlib.Path(__file__).resolve().parents[1]
+SRC = ROOT / "src" / "aria_code"
+WORKFLOW = ROOT / ".github" / "workflows" / "build-native-binaries.yml"
+MAC_SCRIPT = ROOT / "scripts" / "build_native_binary.sh"
+
+
+def _collector():
+    spec = importlib.util.spec_from_file_location(
+        "pyinstaller_collect_args", ROOT / "scripts" / "pyinstaller_collect_args.py")
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+    return module
+
+
+def _runtime_targets() -> dict[str, list[str]]:
+    shims = [m.group(1) for f in sorted(SRC.glob("*.py"))
+             if (m := re.search(r'_import_module\("([\w.]+)"\)', f.read_text(encoding="utf-8")))]
+    brokers = re.findall(r'\("(brokers\.[\w.]+)",', (SRC / "brokers" / "registry.py").read_text(encoding="utf-8"))
+    agents = re.findall(r'_try_import_builtin\(\s*"([\w.]+)"', (SRC / "agents" / "registry.py").read_text(encoding="utf-8"))
+    return {"shim": shims, "broker": brokers, "agent": agents}
+
+
+class EveryRuntimeImportIsListed(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls) -> None:
+        cls.hidden = set(_collector().hidden_imports())
+
+    def test_the_tables_were_found(self) -> None:
+        # Guards against the parsing below silently matching nothing, which
+        # would make the next test pass vacuously.
+        targets = _runtime_targets()
+        self.assertGreaterEqual(len(targets["shim"]), 20)
+        self.assertGreaterEqual(len(targets["broker"]), 5)
+        self.assertGreaterEqual(len(targets["agent"]), 20)
+
+    def test_each_target_is_listed_under_its_exact_name(self) -> None:
+        for kind, modules in _runtime_targets().items():
+            for module in modules:
+                with self.subTest(kind=kind, module=module):
+                    self.assertIn(
+                        module, self.hidden,
+                        f"{module} is loaded by name at runtime but is not in the hook, so "
+                        f"the binary will not contain it — and the loader will swallow "
+                        f"the ImportError",
+                    )
+
+    def test_both_roots_are_listed(self) -> None:
+        self.assertIn("clients.market_data_client", self.hidden)
+        self.assertIn("aria_code.clients.market_data_client", self.hidden)
+        self.assertIn("aria_code", self.hidden)
+
+    def test_each_target_exists_on_disk(self) -> None:
+        # Listing a module still cannot supply one that is not there.
+        for kind, modules in _runtime_targets().items():
+            for module in modules:
+                parts = module.split(".")
+                if parts[0] == "aria_code":
+                    parts = parts[1:]
+                base = SRC.joinpath(*parts)
+                with self.subTest(kind=kind, module=module):
+                    self.assertTrue(
+                        base.with_suffix(".py").is_file() or (base / "__init__.py").is_file(),
+                        f"{module} does not exist",
+                    )
+
+
+class TheHookLoads(unittest.TestCase):
+    def test_written_hook_is_valid_python_with_every_name(self) -> None:
+        collector = _collector()
+        with tempfile.TemporaryDirectory() as tmp:
+            hook = collector.write_hook(pathlib.Path(tmp)) / "hook-aria_code.py"
+            tree = ast.parse(hook.read_text(encoding="utf-8"))
+        (assign,) = [n for n in tree.body if isinstance(n, ast.Assign)]
+        self.assertEqual(assign.targets[0].id, "hiddenimports")
+        self.assertEqual(ast.literal_eval(assign.value), collector.hidden_imports())
+
+    def test_hook_fires_even_if_the_entry_never_imports_aria_code(self) -> None:
+        # A hook runs only when its module is imported; without this a build
+        # whose entry skipped aria_code silently got none of the list.
+        args = _collector().collect_args()
+        self.assertIn("--additional-hooks-dir", args)
+        i = args.index("--hidden-import")
+        self.assertEqual(args[i + 1], "aria_code")
+
+    def test_printed_args_have_no_spaces(self) -> None:
+        # Spliced in by an unquoted $(...): a space would split an argument.
+        for arg in _collector().collect_args():
+            self.assertNotIn(" ", arg)
+
+
+class EveryBuildUsesTheCollector(unittest.TestCase):
+    def test_workflow_builds(self) -> None:
+        text = WORKFLOW.read_text(encoding="utf-8")
+        builds = text.count("--distpath dist-native/dist")
+        self.assertGreater(builds, 0)
+        self.assertEqual(text.count("pyinstaller_collect_args.py"), builds,
+                         "a PyInstaller build in the workflow does not use the collector")
+
+    def test_macos_script_builds(self) -> None:
+        text = MAC_SCRIPT.read_text(encoding="utf-8")
+        builds = text.count("--onefile --name")
+        self.assertGreater(builds, 0)
+        self.assertEqual(text.count("pyinstaller_collect_args.py"), builds,
+                         "a PyInstaller build in the macOS script does not use the collector")
+
+
+if __name__ == "__main__":
+    unittest.main()

aria_code-0.58.0/tests/test_broken_akshare_is_optional.py

@@ -0,0 +1,52 @@
+"""A broken akshare must disable China data, not crash the CLI.
+
+akshare reads bundled data files on import (futures/cons.py loads a calendar
+JSON). Once the native binary started bundling every first-party module,
+tools/local_finance_tools.py was in it, and with akshare present but its data
+missing every command — `quote AAPL` included — died with FileNotFoundError
+before doing anything. The guards caught only ImportError.
+"""
+
+from __future__ import annotations
+
+import importlib
+import importlib.abc
+import sys
+import unittest
+
+MODULES = (
+    "aria_code.tools.local_finance_tools",
+    "aria_code.tools.macro_tools",
+    "aria_code.tools.realty_data_tools",
+)
+
+
+class _BrokenAkshare(importlib.abc.MetaPathFinder):
+    def find_spec(self, name, path=None, target=None):
+        if name == "akshare" or name.startswith("akshare."):
+            raise FileNotFoundError("akshare/file_fold/calendar.json")
+        return None
+
+
+class BrokenAkshareIsOptional(unittest.TestCase):
+    def test_modules_import_with_china_data_disabled(self) -> None:
+        saved = {k: v for k, v in sys.modules.items()
+                 if k == "akshare" or k.startswith("akshare.") or k in MODULES}
+        for name in saved:
+            del sys.modules[name]
+        finder = _BrokenAkshare()
+        sys.meta_path.insert(0, finder)
+        try:
+            for name in MODULES:
+                with self.subTest(module=name):
+                    module = importlib.import_module(name)
+                    self.assertFalse(module._HAS_AK)
+        finally:
+            sys.meta_path.remove(finder)
+            for name in MODULES:
+                sys.modules.pop(name, None)
+            sys.modules.update(saved)
+
+
+if __name__ == "__main__":
+    unittest.main()