agentx-python 0.8.7__tar.gz → 0.8.9__tar.gz

{agentx_python-0.8.7 → agentx_python-0.8.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.8.7
+Version: 0.8.9
 Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python
 Author: Robin Wang and AgentX Team

{agentx_python-0.8.7 → agentx_python-0.8.9}/agentx/cli.py

@@ -4,29 +4,43 @@ Evaluate, Monitor), published separately at github.com/AgentX-ai/AgentX-trace-ev
 wrapping a Bun-compiled TypeScript engine, not Python). That compiled engine binary is tens of
 megabytes; most `pip install agentx-python` installs are just this SDK talking to the hosted
 AgentX SaaS and would never touch it, so it isn't bundled in this package. Instead, this command
-downloads the matching release into ~/.agentx/bin the first time it's needed (mirroring
+downloads the release into ~/.agentx/bin the first time it's needed (mirroring
 AgentX-trace-eval's own install.sh) and then hands off to the real `agentx-server` binary.
 
+Versioning: each SDK release pins the engine release it was tested against
+(agentx.version.ENGINE_VERSION); the launcher installs that pin and converges to it - if the
+stamped install (~/.agentx/bin/.version) differs from the pin, it re-installs, so upgrading the
+SDK upgrades the engine to the matching pair. `--update` force-reinstalls the resolved version
+(recovering a broken or pre-stamp install). Set AGENTX_TRACE_EVAL_VERSION to a tag or `latest`
+to override the pin; in `latest` mode the launcher trusts whatever is installed, prints a
+fail-open notice when GitHub has something newer, and `--update` fetches it.
+
 Usage:
     agentx-trace-eval --dev
+    agentx-trace-eval --update --dev
+    AGENTX_TRACE_EVAL_VERSION=latest agentx-trace-eval --update --dev
     agentx-trace-eval --port 5000 --db-url postgres://...
 """
 
 import os
 import platform
+import re
 import shutil
 import stat
 import sys
 import tarfile
 import tempfile
 from pathlib import Path
-from typing import Tuple
+from typing import Optional, Tuple
 
 import requests
 
+from agentx.version import ENGINE_VERSION
+
 REPO = "AgentX-ai/AgentX-trace-eval"
 INSTALL_DIR = Path(os.environ.get("AGENTX_INSTALL_DIR", str(Path.home() / ".agentx" / "bin")))
 _BIN_NAMES = ("agentx", "agentx-server", "agentx-engine")
+_VERSION_STAMP = ".version"
 
 
 def _platform_tag() -> Tuple[str, str]:
@@ -57,12 +71,22 @@ def _release_url(asset: str, version: str) -> str:
     return f"https://github.com/{REPO}/releases/download/{version}/{asset}"
 
 
-def _download(url: str, dest: Path) -> None:
+def _tag_from_release_url(url: str) -> Optional[str]:
+    # GitHub redirects releases/latest/download/... to releases/download/<tag>/...; the final
+    # URL is the one place the resolved tag shows up without a second API call.
+    match = re.search(r"/releases/download/([^/]+)/", url)
+    return match.group(1) if match else None
+
+
+def _download(url: str, dest: Path) -> Optional[str]:
+    """Downloads url to dest; returns the release tag resolved from the final (post-redirect)
+    URL, or None if it can't be determined."""
     response = requests.get(url, stream=True, timeout=60)
     response.raise_for_status()
     with open(dest, "wb") as f:
         for chunk in response.iter_content(chunk_size=1 << 16):
             f.write(chunk)
+    return _tag_from_release_url(response.url)
 
 
 def _extract_tar(archive: Path, dest_dir: Path) -> None:
@@ -76,17 +100,42 @@ def _extract_tar(archive: Path, dest_dir: Path) -> None:
             tar.extractall(dest_dir)  # Python < 3.12: filter kwarg doesn't exist yet
 
 
+def _stamped_version() -> Optional[str]:
+    try:
+        return (INSTALL_DIR / _VERSION_STAMP).read_text().strip() or None
+    except OSError:
+        return None
+
+
+def _latest_release_tag() -> Optional[str]:
+    """The newest published tag, or None when it can't be determined (offline, rate-limited).
+    Fail-open by design: an update NOTICE must never break launching the engine."""
+    try:
+        response = requests.get(f"https://api.github.com/repos/{REPO}/releases/latest", timeout=3)
+        response.raise_for_status()
+        tag = response.json().get("tag_name")
+        return tag if isinstance(tag, str) and tag else None
+    except Exception:
+        return None
+
+
+def resolve_version() -> str:
+    """The engine version this launcher should be running: the AGENTX_TRACE_EVAL_VERSION
+    override (a tag, or `latest`) when set, else the SDK's tested pin."""
+    return os.environ.get("AGENTX_TRACE_EVAL_VERSION") or ENGINE_VERSION
+
+
 def _install(version: str = "latest") -> None:
     os_name, arch = _platform_tag()
     INSTALL_DIR.mkdir(parents=True, exist_ok=True)
 
-    print(f"agentx-trace-eval: downloading agentx ({os_name}/{arch})...", file=sys.stderr)
+    print(f"agentx-trace-eval: downloading agentx {version} ({os_name}/{arch})...", file=sys.stderr)
     with tempfile.TemporaryDirectory() as tmp:
         tmp_dir = Path(tmp)
         archive = tmp_dir / "agentx.tar.gz"
         url = _release_url(f"agentx_{os_name}_{arch}.tar.gz", version)
         try:
-            _download(url, archive)
+            resolved_tag = _download(url, archive)
         except requests.HTTPError as exc:
             raise SystemExit(
                 f"agentx-trace-eval: failed to download {url} ({exc}).\n"
@@ -108,6 +157,11 @@ def _install(version: str = "latest") -> None:
         if not found_any:
             raise SystemExit(f"agentx-trace-eval: downloaded archive from {url} didn't contain any of {_BIN_NAMES}")
 
+    stamp = resolved_tag or (version if version != "latest" else None)
+    if stamp:
+        (INSTALL_DIR / _VERSION_STAMP).write_text(stamp + "\n")
+        print(f"agentx-trace-eval: installed {stamp}", file=sys.stderr)
+
     if os.environ.get("AGENTX_TRACE_EVAL_SKIP_WEB"):
         return
 
@@ -118,7 +172,9 @@ def _install(version: str = "latest") -> None:
     web_dir = INSTALL_DIR / "web"
     with tempfile.TemporaryDirectory() as tmp:
         web_archive = Path(tmp) / "agentx-web.tar.gz"
-        web_url = _release_url("agentx-web.tar.gz", version)
+        # Pin the dashboard to the tag the engine actually resolved to, so the two halves of one
+        # install can't skew (releases/latest assets are updated independently).
+        web_url = _release_url("agentx-web.tar.gz", stamp or version)
         try:
             _download(web_url, web_archive)
         except requests.HTTPError:
@@ -133,25 +189,61 @@ def _install(version: str = "latest") -> None:
         _extract_tar(web_archive, web_dir)
 
 
-def ensure_installed(version: str = "latest") -> Path:
-    """Downloads agentx-server (+ its engine) into ~/.agentx/bin if not already present there.
+def ensure_installed(version: Optional[str] = None, force: bool = False) -> Path:
+    """Installs agentx-server (+ its engine and dashboard) into ~/.agentx/bin when it's missing
+    there, when `force` is set, or when `version` is a specific tag that differs from the
+    stamped install - the convergence that makes an SDK upgrade carry its tested engine along.
+    `version` defaults to resolve_version() (the AGENTX_TRACE_EVAL_VERSION override, else the
+    SDK's pin); pass "latest" to trust whatever is installed and only fetch when missing.
     Returns the path to the agentx-server executable. Set AGENTX_INSTALL_DIR to change where
-    this looks/installs; set AGENTX_TRACE_EVAL_VERSION to pin a release tag instead of latest."""
+    this looks/installs."""
+    if version is None:
+        version = resolve_version()
     server_path = INSTALL_DIR / "agentx-server"
-    if not server_path.exists():
+    stamped = _stamped_version()
+    pin_mismatch = version != "latest" and stamped != version
+    if force or pin_mismatch or not server_path.exists():
         _install(version=version)
     return server_path
 
 
+def _maybe_print_update_notice(installed: Optional[str]) -> None:
+    """Only relevant in `latest` mode, where nothing converges automatically - a quick fail-open
+    check tells the user when the world has moved on. (Pinned mode needs no notice: a pin
+    mismatch re-installs instead of nagging.)"""
+    latest = _latest_release_tag()
+    if installed is None:
+        print(
+            "agentx-trace-eval: installed engine version unknown"
+            + (f" (newest release is {latest})" if latest else "")
+            + "; run with --update to fetch the newest release",
+            file=sys.stderr,
+        )
+    elif latest and latest != installed:
+        print(
+            f"agentx-trace-eval: engine {installed} is installed but {latest} is available; "
+            "run with --update to upgrade",
+            file=sys.stderr,
+        )
+
+
 def main() -> None:
-    version = os.environ.get("AGENTX_TRACE_EVAL_VERSION", "latest")
-    server_path = ensure_installed(version=version)
+    args = sys.argv[1:]
+    force_update = "--update" in args
+    # --update belongs to this launcher, not to agentx-server - consume it before the handoff.
+    args = [a for a in args if a != "--update"]
+
+    version = resolve_version()
+    server_path = ensure_installed(version=version, force=force_update)
     if not server_path.exists():
         raise SystemExit(f"agentx-trace-eval: {server_path} still missing after install, giving up")
 
+    if not force_update and version == "latest":
+        _maybe_print_update_notice(_stamped_version())
+
     # os.execv replaces this process rather than spawning a subprocess: signals, stdio, and the
     # exit code all pass straight through to agentx-server, same as invoking it directly.
-    os.execv(str(server_path), [str(server_path)] + sys.argv[1:])
+    os.execv(str(server_path), [str(server_path)] + args)
 
 
 if __name__ == "__main__":

agentx_python-0.8.9/agentx/version.py

@@ -0,0 +1,7 @@
+VERSION = "0.8.9"
+
+# The AgentX-trace-eval release this SDK version is tested against - what `agentx-trace-eval`
+# installs and converges to (see agentx/cli.py). Bump together with VERSION when releasing, so
+# every published SDK names a known-good engine+dashboard pair. Users can override with
+# AGENTX_TRACE_EVAL_VERSION=<tag|latest>.
+ENGINE_VERSION = "v0.3.1"

{agentx_python-0.8.7 → agentx_python-0.8.9}/agentx_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.8.7
+Version: 0.8.9
 Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python
 Author: Robin Wang and AgentX Team

{agentx_python-0.8.7 → agentx_python-0.8.9}/agentx_python.egg-info/SOURCES.txt

@@ -73,6 +73,7 @@ agentx_python.egg-info/entry_points.txt
 agentx_python.egg-info/not-zip-safe
 agentx_python.egg-info/requires.txt
 agentx_python.egg-info/top_level.txt
+tests/test_cli_launcher.py
 tests/test_deep_dive_fixes.py
 tests/test_docs_match_sdk.py
 tests/test_eval_scope.py

agentx_python-0.8.9/tests/test_cli_launcher.py

@@ -0,0 +1,117 @@
+"""The agentx-trace-eval launcher's version pinning: each SDK release names the engine release
+it was tested against (agentx.version.ENGINE_VERSION) and the launcher converges the local
+install to it, so `pip install -U agentx-python` upgrades the engine to the matching pair.
+Before this, the launcher downloaded "latest" once and then ran that binary forever - a July
+install silently served a July engine months later, with no stamp and no way to update."""
+
+import pytest
+
+from agentx import cli
+from agentx.version import ENGINE_VERSION
+
+
+@pytest.fixture
+def install_dir(tmp_path, monkeypatch):
+    monkeypatch.setattr(cli, "INSTALL_DIR", tmp_path)
+    return tmp_path
+
+
+def _fake_install(calls):
+    def fake(version="latest"):
+        calls.append(version)
+
+    return fake
+
+
+def test_engine_pin_is_a_release_tag():
+    # The pin the SDK ships must look like a real trace-eval release tag (v<series>.<n>), not
+    # "latest" - "latest" as a pin would silently disable the whole tested-pair contract.
+    assert cli.ENGINE_VERSION == ENGINE_VERSION
+    import re
+
+    assert re.fullmatch(r"v\d+\.\d+\.\d+", ENGINE_VERSION)
+
+
+def test_resolve_version_prefers_env_override(monkeypatch):
+    monkeypatch.delenv("AGENTX_TRACE_EVAL_VERSION", raising=False)
+    assert cli.resolve_version() == ENGINE_VERSION
+    monkeypatch.setenv("AGENTX_TRACE_EVAL_VERSION", "latest")
+    assert cli.resolve_version() == "latest"
+    monkeypatch.setenv("AGENTX_TRACE_EVAL_VERSION", "v9.9.9")
+    assert cli.resolve_version() == "v9.9.9"
+
+
+def test_tag_resolved_from_redirected_release_url():
+    url = "https://github.com/AgentX-ai/AgentX-trace-eval/releases/download/v0.3.1/agentx_darwin_arm64.tar.gz"
+    assert cli._tag_from_release_url(url) == "v0.3.1"
+    assert cli._tag_from_release_url("https://example.com/not-a-release") is None
+
+
+def test_install_matching_pin_is_not_redownloaded(install_dir, monkeypatch):
+    (install_dir / "agentx-server").write_text("bin")
+    (install_dir / ".version").write_text(ENGINE_VERSION + "\n")
+    calls = []
+    monkeypatch.setattr(cli, "_install", _fake_install(calls))
+    cli.ensure_installed(version=ENGINE_VERSION)
+    assert calls == []
+
+
+def test_stamp_differing_from_pin_converges(install_dir, monkeypatch):
+    # The SDK-upgrade path: binaries exist but were installed for an older pin (or pre-stamp,
+    # where .version is missing entirely) - the launcher re-installs the pinned pair.
+    (install_dir / "agentx-server").write_text("bin")
+    (install_dir / ".version").write_text("v0.2.0\n")
+    calls = []
+    monkeypatch.setattr(cli, "_install", _fake_install(calls))
+    cli.ensure_installed(version=ENGINE_VERSION)
+    assert calls == [ENGINE_VERSION]
+
+    (install_dir / ".version").unlink()
+    calls.clear()
+    cli.ensure_installed(version=ENGINE_VERSION)
+    assert calls == [ENGINE_VERSION]
+
+
+def test_latest_mode_trusts_existing_install(install_dir, monkeypatch):
+    (install_dir / "agentx-server").write_text("bin")
+    (install_dir / ".version").write_text("v0.2.0\n")
+    calls = []
+    monkeypatch.setattr(cli, "_install", _fake_install(calls))
+    cli.ensure_installed(version="latest")
+    assert calls == []
+
+
+def test_force_reinstalls_over_existing(install_dir, monkeypatch):
+    (install_dir / "agentx-server").write_text("bin")
+    (install_dir / ".version").write_text(ENGINE_VERSION + "\n")
+    calls = []
+    monkeypatch.setattr(cli, "_install", _fake_install(calls))
+    cli.ensure_installed(version=ENGINE_VERSION, force=True)
+    assert calls == [ENGINE_VERSION]
+
+
+def test_update_notice_names_both_versions(install_dir, monkeypatch, capsys):
+    monkeypatch.setattr(cli, "_latest_release_tag", lambda: "v0.9.0")
+    cli._maybe_print_update_notice("v0.3.1")
+    err = capsys.readouterr().err
+    assert "v0.3.1" in err and "v0.9.0" in err and "--update" in err
+
+
+def test_update_notice_handles_unstamped_install(install_dir, monkeypatch, capsys):
+    monkeypatch.setattr(cli, "_latest_release_tag", lambda: None)
+    cli._maybe_print_update_notice(None)
+    assert "--update" in capsys.readouterr().err
+
+
+def test_update_notice_is_silent_when_current(install_dir, monkeypatch, capsys):
+    monkeypatch.setattr(cli, "_latest_release_tag", lambda: "v0.3.1")
+    cli._maybe_print_update_notice("v0.3.1")
+    assert capsys.readouterr().err == ""
+
+
+def test_latest_release_tag_fails_open(monkeypatch):
+    def boom(*args, **kwargs):
+        raise OSError("offline")
+
+    monkeypatch.setattr(cli.requests, "get", boom)
+    assert cli._latest_release_tag() is None

agentx_python-0.8.7/agentx/version.py

@@ -1 +0,0 @@
-VERSION = "0.8.7"