aider-vision-core 0.90.9.dev0__tar.gz → 0.90.10.dev0__tar.gz

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aider-vision-core
-Version: 0.90.9.dev0
+Version: 0.90.10.dev0
 Summary: aider-vision-core: Aider fork with headless API, submodule git, and web integration
 Project-URL: Homepage, https://github.com/Digital-Defiance/aider-vision-core
 Classifier: Development Status :: 4 - Beta

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.90.9.dev0'
-__version_tuple__ = version_tuple = (0, 90, 9, 'dev0')
+__version__ = version = '0.90.10.dev0'
+__version_tuple__ = version_tuple = (0, 90, 10, 'dev0')
 
-__commit_id__ = commit_id = 'gdf5587afc'
+__commit_id__ = commit_id = 'g7d7163ed7'

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core/gui_progress.py

@@ -13,7 +13,9 @@ T = TypeVar("T")
 
 
 def headless_enabled() -> bool:
-    return os.environ.get("AIDER_VISION_HEADLESS") == "1"
+    from aider_vision_core.headless_stdio import headless_enabled as _headless_enabled
+
+    return _headless_enabled()
 
 
 def emit_progress(

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core/http_api.py

@@ -25,9 +25,9 @@ from pydantic import BaseModel, Field
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.requests import Request
 
-from aider_vision_core.headless_stdio import install_headless_stdio
+from aider_vision_core.vision_runtime import configure_vision_runtime
 
-install_headless_stdio()
+configure_vision_runtime()
 
 from aider_vision_core.git_undo import undo_last_aider_commit_for_coder
 from aider_vision_core.http_auth import auth_enabled, configure_auth, get_token_from_env, verify_bearer

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core/repomap.py

@@ -20,6 +20,13 @@ from tree_sitter import Query
 from aider_vision_core.dump import dump
 from aider_vision_core.gui_progress import progress_iter
 from aider_vision_core.special import filter_important_files
+from aider_vision_core.vision_runtime import (
+    CACHE_LOAD_ERRORS,
+    REPO_MAP_CACHE_DIRNAME,
+    REPO_MAP_CACHE_VERSION,
+    purge_legacy_tag_caches,
+    safe_cache_len,
+)
 from aider_vision_core.waiting import Spinner
 
 # tree_sitter is throwing a FutureWarning
@@ -30,17 +37,14 @@ Tag = namedtuple("Tag", "rel_fname fname line name kind".split())
 
 
 SQLITE_ERRORS = (sqlite3.OperationalError, sqlite3.DatabaseError, OSError)
-
-
-CACHE_VERSION = 3
-if USING_TSL_PACK:
-    CACHE_VERSION = 4
+# Re-export for tests / callers
+CACHE_VERSION = REPO_MAP_CACHE_VERSION
 
 UPDATING_REPO_MAP_MESSAGE = "Updating repo map"
 
 
 class RepoMap:
-    TAGS_CACHE_DIR = f".aider.tags.cache.v{CACHE_VERSION}"
+    TAGS_CACHE_DIR = REPO_MAP_CACHE_DIRNAME
 
     warned_files = set()
 
@@ -215,6 +219,7 @@ class RepoMap:
         self.TAGS_CACHE = dict()
 
     def load_tags_cache(self):
+        purge_legacy_tag_caches(self.root)
         path = Path(self.root) / self.TAGS_CACHE_DIR
         try:
             self.TAGS_CACHE = Cache(path)
@@ -239,14 +244,14 @@ class RepoMap:
         cache_key = fname
         try:
             val = self.TAGS_CACHE.get(cache_key)  # Issue #1308
-        except SQLITE_ERRORS as e:
+        except CACHE_LOAD_ERRORS as e:
             self.tags_cache_error(e)
             val = self.TAGS_CACHE.get(cache_key)
 
         if val is not None and val.get("mtime") == file_mtime:
             try:
                 return self.TAGS_CACHE[cache_key]["data"]
-            except SQLITE_ERRORS as e:
+            except CACHE_LOAD_ERRORS as e:
                 self.tags_cache_error(e)
                 return self.TAGS_CACHE[cache_key]["data"]
 
@@ -382,11 +387,10 @@ class RepoMap:
         # https://networkx.org/documentation/stable/_modules/networkx/algorithms/link_analysis/pagerank_alg.html#pagerank
         personalize = 100 / len(fnames)
 
-        try:
-            cache_size = len(self.TAGS_CACHE)
-        except SQLITE_ERRORS as e:
-            self.tags_cache_error(e)
-            cache_size = len(self.TAGS_CACHE)
+        cache_size = safe_cache_len(self.TAGS_CACHE)
+        if cache_size < 0:
+            self.tags_cache_error()
+            cache_size = safe_cache_len(self.TAGS_CACHE)
 
         if len(fnames) - cache_size > 100:
             self.io.tool_output(

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core/session.py

@@ -52,6 +52,11 @@ class Session:
         if not workspace.is_dir():
             raise FileNotFoundError(f"Workspace not found: {workspace}")
 
+        from aider_vision_core.vision_runtime import configure_vision_runtime, purge_legacy_tag_caches
+
+        configure_vision_runtime()
+        purge_legacy_tag_caches(workspace)
+
         prev_cwd = os.getcwd()
         os.chdir(workspace)
         try:

aider_vision_core-0.90.10.dev0/aider_vision_core/vision_runtime.py

@@ -0,0 +1,127 @@
+"""
+Vision / headless runtime setup and guards against legacy ``aider`` rename issues.
+
+Call :func:`configure_vision_runtime` when starting the HTTP API or any headless
+session so TUI output and stale on-disk caches do not leak into the desktop GUI.
+"""
+
+from __future__ import annotations
+
+import os
+import pickle
+import re
+import shutil
+import sqlite3
+from pathlib import Path
+from typing import Any
+
+# Bump when repo-map tag cache schema or Python package name changes.
+REPO_MAP_CACHE_VERSION = 5
+REPO_MAP_CACHE_DIRNAME = f".aider.tags.cache.v{REPO_MAP_CACHE_VERSION}"
+LEGACY_PYTHON_PACKAGE = "aider"
+LEGACY_TAG_CACHE_PATTERN = re.compile(r"^\.aider\.tags\.cache\.v(\d+)$")
+
+SQLITE_ERRORS = (sqlite3.OperationalError, sqlite3.DatabaseError, OSError)
+CACHE_LOAD_ERRORS = SQLITE_ERRORS + (
+    ModuleNotFoundError,
+    AttributeError,
+    pickle.UnpicklingError,
+)
+
+_tqdm_patched = False
+_runtime_configured = False
+
+
+def headless_enabled() -> bool:
+    from aider_vision_core.headless_stdio import headless_enabled as _he
+
+    return _he()
+
+
+def purge_legacy_tag_caches(root: str | Path) -> list[str]:
+    """
+    Remove repo-map cache dirs from older package names / cache versions.
+
+    Pickled entries may reference the pre-rename ``aider`` module and raise
+    ``ModuleNotFoundError`` when loaded under ``aider_vision_core``.
+    """
+    root = Path(root)
+    removed: list[str] = []
+    if not root.is_dir():
+        return removed
+    for path in sorted(root.iterdir()):
+        if not path.is_dir():
+            continue
+        m = LEGACY_TAG_CACHE_PATTERN.match(path.name)
+        if not m:
+            continue
+        if path.name == REPO_MAP_CACHE_DIRNAME:
+            continue
+        try:
+            shutil.rmtree(path)
+            removed.append(str(path))
+        except OSError:
+            pass
+    return removed
+
+
+def safe_cache_len(cache: Any) -> int:
+    """``len(diskcache.Cache)`` can unpickle entries; treat failures as empty."""
+    try:
+        return len(cache)
+    except CACHE_LOAD_ERRORS:
+        return -1
+
+
+def configure_vision_runtime(*, force: bool = False) -> None:
+    """
+    One-time setup for Aider Vision desktop / API child processes.
+
+    - Redirect stdio (headless)
+    - Disable tqdm terminal bars
+    - Patch tqdm → GUI progress when still invoked from legacy call sites
+    """
+    global _runtime_configured, _tqdm_patched
+    if _runtime_configured and not force:
+        return
+    _runtime_configured = True
+
+    if not headless_enabled():
+        return
+
+    os.environ.setdefault("TQDM_DISABLE", "1")
+
+    from aider_vision_core.headless_stdio import install_headless_stdio
+
+    install_headless_stdio()
+    _patch_tqdm_for_headless()
+
+
+def _patch_tqdm_for_headless() -> None:
+    global _tqdm_patched
+    if _tqdm_patched:
+        return
+    try:
+        import tqdm as tqdm_mod
+    except ImportError:
+        return
+
+    _orig = tqdm_mod.tqdm
+
+    def _vision_tqdm(iterable=None, *args, **kwargs):
+        if iterable is None:
+            return _orig(*args, **kwargs)
+        from aider_vision_core.gui_progress import progress_iter
+
+        desc = kwargs.get("desc")
+        if desc is None and args and isinstance(args[0], str):
+            desc = args[0]
+        return progress_iter(
+            iterable,
+            desc=str(desc or "Working"),
+            io=kwargs.get("io"),
+            total=kwargs.get("total"),
+        )
+
+    tqdm_mod.tqdm = _vision_tqdm
+    _tqdm_patched = True

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core/vision_serve.py

@@ -33,10 +33,10 @@ def run(argv: list[str] | None = None) -> None:
     validate_listen_address(args.host)
     configure_auth(args.host)
 
-    if os.environ.get("AIDER_VISION_HEADLESS") == "1":
-        from aider_vision_core.headless_stdio import install_headless_stdio
+    from aider_vision_core.vision_runtime import configure_vision_runtime
 
-        install_headless_stdio()
+    if os.environ.get("AIDER_VISION_HEADLESS") == "1":
+        configure_vision_runtime()
     else:
         print(startup_message(args.host))
 

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aider-vision-core
-Version: 0.90.9.dev0
+Version: 0.90.10.dev0
 Summary: aider-vision-core: Aider fork with headless API, submodule git, and web integration
 Project-URL: Homepage, https://github.com/Digital-Defiance/aider-vision-core
 Classifier: Development Status :: 4 - Beta

{aider_vision_core-0.90.9.dev0 → aider_vision_core-0.90.10.dev0}/aider_vision_core.egg-info/SOURCES.txt

@@ -8,6 +8,7 @@ HISTORY.md
 LICENSE.txt
 MANIFEST.in
 README.md
+build.sh
 pyproject.toml
 pytest.ini
 requirements.txt
@@ -69,6 +70,7 @@ aider_vision_core/special.py
 aider_vision_core/urls.py
 aider_vision_core/utils.py
 aider_vision_core/versioncheck.py
+aider_vision_core/vision_runtime.py
 aider_vision_core/vision_serve.py
 aider_vision_core/voice.py
 aider_vision_core/waiting.py
@@ -308,6 +310,7 @@ requirements/tree-sitter.in
 scripts/30k-image.py
 scripts/Dockerfile.jekyll
 scripts/__init__.py
+scripts/audit_rename_compat.py
 scripts/blame.py
 scripts/clean_metadata.py
 scripts/dl_icons.py
@@ -363,6 +366,7 @@ tests/basic/test_ssl_verification.py
 tests/basic/test_udiff.py
 tests/basic/test_urls.py
 tests/basic/test_utils.py
+tests/basic/test_vision_runtime.py
 tests/basic/test_voice.py
 tests/basic/test_watch.py
 tests/basic/test_wholefile.py

aider_vision_core-0.90.10.dev0/build.sh

@@ -0,0 +1,244 @@
+#!/usr/bin/env bash
+# Release helper: commit release metadata, tag, push, build sdist/wheel, upload to PyPI.
+#
+# Usage:
+#   ./build.sh v0.90.10
+#   ./build.sh v0.90.10 --no-upload    # tag + push + build only
+#   ./build.sh v0.90.10 --yes          # non-interactive (CI); fails if tree is dirty
+#
+# Requires: git, python3.14 (or PYTHON=python3.12), build, twine; PyPI credentials for upload.
+
+set -euo pipefail
+
+ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+cd "$ROOT"
+
+VERSION=""
+UPLOAD=1
+ASSUME_YES=0
+
+# setuptools-scm release metadata (allowed dirty paths without prompting)
+VERSION_FILES=(
+  "aider_vision_core.egg-info/PKG-INFO"
+  "aider_vision_core.egg-info/SOURCES.txt"
+  "aider_vision_core/_version.py"
+)
+
+usage() {
+  echo "Usage: $0 <version-tag> [--no-upload] [--yes]" >&2
+  echo "  version-tag   e.g. v0.90.10 or v0.90.9.dev0" >&2
+  exit 1
+}
+
+die() {
+  echo "error: $*" >&2
+  exit 1
+}
+
+confirm() {
+  local prompt="$1"
+  if (( ASSUME_YES )); then
+    return 0
+  fi
+  local ans
+  read -r -p "${prompt} [y/N] " ans
+  ans="$(printf '%s' "$ans" | tr '[:upper:]' '[:lower:]')"
+  [[ "$ans" == "y" || "$ans" == "yes" ]]
+}
+
+# First path from a porcelain line (handles rename "old -> new").
+porcelain_path() {
+  local rest="${1:3}"
+  rest="${rest#\"}"
+  rest="${rest%\"}"
+  if [[ "$rest" == *" -> "* ]]; then
+    echo "${rest%% -> *}"
+  else
+    echo "${rest%% *}"
+  fi
+}
+
+is_version_file() {
+  local path="$1"
+  local f
+  for f in "${VERSION_FILES[@]}"; do
+    [[ "$path" == "$f" ]] && return 0
+  done
+  return 1
+}
+
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    --no-upload) UPLOAD=0; shift ;;
+    --yes) ASSUME_YES=1; shift ;;
+    -h|--help) usage ;;
+    -*)
+      die "unknown option: $1"
+      ;;
+    *)
+      [[ -z "$VERSION" ]] || die "unexpected extra argument: $1 (version already set to ${VERSION})"
+      VERSION="$1"
+      shift
+      ;;
+  esac
+done
+
+[[ -n "$VERSION" ]] || usage
+
+if [[ ! "$VERSION" =~ ^v[0-9]+\.[0-9]+\.[0-9]+([.-][0-9A-Za-z.]+)?$ ]]; then
+  die "version must look like v0.90.10 or v0.90.9.dev0 (got: ${VERSION})"
+fi
+
+PEP440_VERSION="${VERSION#v}"
+
+command -v git >/dev/null 2>&1 || die "git not found"
+git rev-parse --is-inside-work-tree >/dev/null 2>&1 || die "not a git repository"
+
+PYTHON="${PYTHON:-python3.14}"
+if ! command -v "$PYTHON" >/dev/null 2>&1; then
+  PYTHON=python3
+fi
+command -v "$PYTHON" >/dev/null 2>&1 || die "python not found (set PYTHON=...)"
+
+if git rev-parse "$VERSION" >/dev/null 2>&1; then
+  die "tag ${VERSION} already exists locally"
+fi
+
+if git ls-remote --exit-code --tags origin "$VERSION" >/dev/null 2>&1; then
+  die "tag ${VERSION} already exists on origin"
+fi
+
+BRANCH="$(git rev-parse --abbrev-ref HEAD)"
+if [[ "$BRANCH" != "main" && "$BRANCH" != "master" ]]; then
+  echo "warning: on branch '${BRANCH}', not main/master" >&2
+  confirm "Continue anyway?" || die "aborted"
+fi
+
+echo "Checking for uncommitted changes..."
+OTHER_LINES=()
+while IFS= read -r line; do
+  [[ -z "$line" ]] && continue
+  path="$(porcelain_path "$line")"
+  if is_version_file "$path"; then
+    continue
+  fi
+  OTHER_LINES+=("$line")
+done < <(git status --porcelain 2>/dev/null || true)
+
+if ((${#OTHER_LINES[@]} > 0)); then
+  echo "Uncommitted changes outside release metadata:" >&2
+  printf '  %s\n' "${OTHER_LINES[@]}" >&2
+  echo >&2
+  if (( ASSUME_YES )); then
+    die "working tree is dirty; commit or stash other changes, or run without --yes"
+  fi
+  if confirm "Commit these changes now before release?"; then
+    git add -A
+    DEFAULT_MSG="chore: prepare ${VERSION}"
+    if ! confirm "Use default commit message: \"${DEFAULT_MSG}\"?"; then
+      read -r -p "Commit message: " COMMIT_MSG
+      [[ -n "${COMMIT_MSG:-}" ]] || die "empty commit message"
+    else
+      COMMIT_MSG="$DEFAULT_MSG"
+    fi
+    git commit -m "$COMMIT_MSG"
+    echo "Committed other changes."
+  else
+    die "aborted — commit or stash changes before releasing"
+  fi
+fi
+
+echo "Regenerating version metadata for ${VERSION}..."
+export SETUPTOOLS_SCM_PRETEND_VERSION="$PEP440_VERSION"
+
+# setuptools-scm is a build dependency only — not installed by pip install -e .
+"$PYTHON" -m pip install -q "setuptools-scm[toml]>=8" build wheel
+
+"$PYTHON" - <<'PY' || die "failed to write aider_vision_core/_version.py"
+import os
+from setuptools_scm import dump_version
+
+ver = os.environ["SETUPTOOLS_SCM_PRETEND_VERSION"]
+dump_version(root=".", write_to="aider_vision_core/_version.py", version=ver)
+print(f"wrote aider_vision_core/_version.py -> {ver}")
+PY
+
+# Refresh egg-info so PKG-INFO / SOURCES.txt match the intended release.
+"$PYTHON" -m pip install -q -e . --no-deps
+unset SETUPTOOLS_SCM_PRETEND_VERSION
+
+echo "Committing release metadata..."
+for f in "${VERSION_FILES[@]}"; do
+  if [[ -e "$f" ]]; then
+    git add "$f"
+  else
+    echo "warning: ${f} not found, skipping" >&2
+  fi
+done
+
+if git diff --cached --quiet; then
+  echo "Release metadata already committed; nothing to commit."
+else
+  RELEASE_MSG="chore: release ${VERSION}"
+  if (( ASSUME_YES )); then
+    git commit -m "$RELEASE_MSG"
+  elif confirm "Commit release metadata with message: \"${RELEASE_MSG}\"?"; then
+    git commit -m "$RELEASE_MSG"
+  else
+    die "aborted before release metadata commit"
+  fi
+fi
+
+# Final sanity check — only ignored version files may remain dirty.
+REMAINING=()
+while IFS= read -r line; do
+  [[ -z "$line" ]] && continue
+  path="$(porcelain_path "$line")"
+  is_version_file "$path" && continue
+  REMAINING+=("$line")
+done < <(git status --porcelain 2>/dev/null || true)
+
+if ((${#REMAINING[@]} > 0)); then
+  echo "Still dirty after release commit:" >&2
+  printf '  %s\n' "${REMAINING[@]}" >&2
+  die "resolve remaining changes before tagging"
+fi
+
+echo "Tagging ${VERSION}..."
+if (( ASSUME_YES )); then
+  git tag "$VERSION"
+else
+  confirm "Create git tag ${VERSION}?" || die "aborted before tag"
+  git tag "$VERSION"
+fi
+
+echo "Pushing to origin..."
+if (( ASSUME_YES )); then
+  git push origin HEAD
+  git push origin "$VERSION"
+else
+  confirm "Push branch ${BRANCH} and tag ${VERSION} to origin?" || die "aborted before push"
+  git push origin HEAD
+  git push origin "$VERSION"
+fi
+
+echo "Cleaning build artifacts..."
+rm -rf dist build *.egg-info aider_vision_core.egg-info
+
+echo "Building sdist and wheel..."
+"$PYTHON" -m pip install -q build
+"$PYTHON" -m build
+
+if (( UPLOAD )); then
+  command -v twine >/dev/null 2>&1 || "$PYTHON" -m pip install -q twine
+  echo "Uploading to PyPI..."
+  if (( ASSUME_YES )); then
+    "$PYTHON" -m twine upload dist/*
+  else
+    confirm "Upload dist/* to PyPI?" || die "aborted before upload (artifacts in dist/)"
+    "$PYTHON" -m twine upload dist/*
+  fi
+  echo "Done. Released ${VERSION} to PyPI."
+else
+  echo "Done. Built dist/ for ${VERSION} (--no-upload)."
+fi

aider_vision_core-0.90.10.dev0/scripts/audit_rename_compat.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python3
+"""
+CI / dev audit: catch regressions from the aider → aider_vision_core rename.
+
+Usage:
+    python scripts/audit_rename_compat.py
+"""
+
+from __future__ import annotations
+
+import re
+import sys
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+PKG = ROOT / "aider_vision_core"
+SKIP_DIRS = {
+    "website",
+    "__pycache__",
+    ".git",
+}
+SKIP_FILES = {
+    "audit_rename_compat.py",
+}
+
+# `from aider.` / `import aider` but not aider_vision_core / aiderignore / …
+BAD_IMPORT = re.compile(
+    r"^\s*(?:from|import)\s+aider(?!_vision_core|ignore|_edited|_commit|@)",
+)
+# tqdm( in production code — allow gui_progress + search_replace __main__
+BAD_TQDM = re.compile(r"\btqdm\s*\(")
+ALLOW_TQDM = {
+    PKG / "gui_progress.py",
+    PKG / "coders" / "search_replace.py",
+}
+# resources.files(__package__) under repomap breaks query paths
+BAD_RESOURCES = re.compile(r"resources\.files\(\s*__package__\s*\)")
+
+
+def iter_py_files() -> list[Path]:
+    out: list[Path] = []
+    for path in PKG.rglob("*.py"):
+        if any(part in SKIP_DIRS for part in path.parts):
+            continue
+        if path.name in SKIP_FILES:
+            continue
+        out.append(path)
+    return sorted(out)
+
+
+def audit() -> list[str]:
+    errors: list[str] = []
+    for path in iter_py_files():
+        rel = path.relative_to(ROOT)
+        text = path.read_text(encoding="utf-8", errors="replace")
+        for i, line in enumerate(text.splitlines(), start=1):
+            if BAD_IMPORT.search(line):
+                errors.append(f"{rel}:{i}: legacy import of package 'aider': {line.strip()}")
+            if BAD_RESOURCES.search(line):
+                errors.append(f"{rel}:{i}: use resources.files('aider_vision_core'), not __package__")
+        if path in ALLOW_TQDM:
+            continue
+        if path.name == "repomap.py":
+            continue  # uses progress_iter
+        for i, line in enumerate(text.splitlines(), start=1):
+            if BAD_TQDM.search(line) and "progress_iter" not in line:
+                errors.append(f"{rel}:{i}: use progress_iter() or gui_progress in headless paths: {line.strip()}")
+
+    repomap = PKG / "repomap.py"
+    body = repomap.read_text(encoding="utf-8")
+    if "vision_runtime" not in body or "REPO_MAP_CACHE_VERSION" not in body:
+        errors.append("repomap.py: must use vision_runtime.REPO_MAP_CACHE_VERSION and purge_legacy_tag_caches")
+
+    return errors
+
+
+def main() -> int:
+    errors = audit()
+    if errors:
+        print("audit_rename_compat FAILED:\n", file=sys.stderr)
+        for err in errors:
+            print(f"  - {err}", file=sys.stderr)
+        return 1
+    print("audit_rename_compat OK")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

aider_vision_core-0.90.10.dev0/tests/basic/test_vision_runtime.py

@@ -0,0 +1,43 @@
+import pickle
+import shutil
+import tempfile
+import unittest
+from pathlib import Path
+
+from aider_vision_core.vision_runtime import (
+    CACHE_LOAD_ERRORS,
+    LEGACY_PYTHON_PACKAGE,
+    REPO_MAP_CACHE_DIRNAME,
+    purge_legacy_tag_caches,
+    safe_cache_len,
+)
+
+
+class TestVisionRuntime(unittest.TestCase):
+    def test_purge_legacy_tag_caches(self):
+        root = Path(tempfile.mkdtemp())
+        try:
+            (root / ".aider.tags.cache.v3").mkdir()
+            (root / ".aider.tags.cache.v4").mkdir()
+            (root / REPO_MAP_CACHE_DIRNAME).mkdir()
+            (root / "src").mkdir()
+            removed = purge_legacy_tag_caches(root)
+            self.assertEqual(len(removed), 2)
+            self.assertFalse((root / ".aider.tags.cache.v3").exists())
+            self.assertTrue((root / REPO_MAP_CACHE_DIRNAME).exists())
+        finally:
+            shutil.rmtree(root, ignore_errors=True)
+
+    def test_safe_cache_len_on_bad_pickle(self):
+        class BadCache:
+            def __len__(self):
+                raise ModuleNotFoundError(f"No module named '{LEGACY_PYTHON_PACKAGE}'")
+
+        self.assertEqual(safe_cache_len(BadCache()), -1)
+
+    def test_cache_load_errors_includes_unpickle(self):
+        self.assertIn(pickle.UnpicklingError, CACHE_LOAD_ERRORS)
+
+
+if __name__ == "__main__":
+    unittest.main()