Merge pull request #97 from PyAutoLabs/feature/assistant-release-baseline

feat: auto-refresh autolens_assistant API baseline in release pipeline

Author: Jammy2211

.github/workflows/release.yml

@@ -837,6 +837,12 @@ jobs:
             package: PyAutoLens
             generate_notebooks: false
             bump_colab_urls: false
+          - repository: PyAutoLabs/autolens_assistant
+            name: autolens
+            package: PyAutoLens
+            generate_notebooks: false
+            bump_colab_urls: false
+            write_api_baseline: true
     steps:
     - uses: actions/checkout@v2
       if: "${{ github.event.inputs.skip_release != 'true' }}"
@@ -905,6 +911,39 @@ jobs:
         PY
         git add version.txt config/general.yaml 2>/dev/null || git add version.txt
         git commit -m "Release $VERSION: pin workspace version" || true
+    - name: Regenerate API audit baseline
+      if: "${{ github.event.inputs.skip_release != 'true' && matrix.workspace.write_api_baseline == true }}"
+      env:
+        NUMBA_CACHE_DIR: /tmp/numba_cache
+        MPLCONFIGDIR: /tmp/matplotlib
+        PYAUTO_SKIP_WORKSPACE_VERSION_CHECK: "1"
+      run: |
+        VERSION="${{ needs.version_number.outputs.version_number }}"
+        # Install the just-released wheels so the baseline hashes the published
+        # API surface, then let the assistant regenerate its own baseline (it
+        # owns the hash format — no reimplementation here).
+        install_cnt=0
+        until python3 -m pip install \
+              --index-url https://test.pypi.org/simple/ \
+              --extra-index-url https://pypi.org/simple \
+              "autoconf[optional]==$VERSION" \
+              "autoarray[optional]==$VERSION" \
+              "autofit[optional]==$VERSION" \
+              "autogalaxy[optional]==$VERSION" \
+              "autolens[optional]==$VERSION"; do
+          ((install_cnt=install_cnt+1))
+          if [[ $install_cnt -ge 30 ]]; then
+            echo "ERROR: Install failed after 30 attempts"
+            exit 1
+          fi
+          sleep 10
+        done
+        pip install matplotlib==3.6.0 pynufft==2025.1.1 numba
+        pushd workspace
+        python3 work/audit_skill_apis.py --write-baseline
+        git add wiki/core/api_audit_baseline.json
+        git commit -m "Release $VERSION: regenerate API audit baseline" || true
+        popd
     - name: Bump Colab URL tag refs
       if: "${{ github.event.inputs.skip_release != 'true' && matrix.workspace.bump_colab_urls == true }}"
       run: |

pre_build.sh

@@ -99,6 +99,10 @@ run_workspace "HowToLens"                            "howtolens"    true   false
 run_workspace "HowToFit"                             "howtofit"     true   false  PyAutoFit
 run_workspace "autofit_workspace_developer"          ""             false  false  ""
 run_workspace "autolens_workspace_developer"         ""             false  false  ""
+# The AI assistant repo. No notebook generation or README pin; release.yml's
+# release_workspaces job stamps its workspace version and regenerates
+# wiki/core/api_audit_baseline.json against the released wheels.
+run_workspace "autolens_assistant"                   "autolens"     false  false  ""
 
 # Commit and push PyAutoBuild itself
 echo ""

verify_workspace_versions.sh

@@ -16,9 +16,9 @@
 # If both exist and disagree, the script fails — they must be kept in sync
 # by the release pipeline.
 #
-# Compares each of the 7 workspaces (3 main + 3 HowTo + euclid_pipeline)
-# against its installed library version, parsed as a YYYY.M.D.B 4-tuple of
-# ints. Exits 1 if any workspace.version > library.version.
+# Compares each of the 8 workspaces (3 main + 3 HowTo + euclid_pipeline +
+# autolens_assistant) against its installed library version, parsed as a
+# YYYY.M.D.B 4-tuple of ints. Exits 1 if any workspace.version > library.version.
 #
 # Workspace → library mapping mirrors the release_workspaces matrix in
 # .github/workflows/release.yml.
@@ -37,6 +37,7 @@ WORKSPACES=(
     "HowToGalaxy|autogalaxy"
     "HowToLens|autolens"
     "euclid_strong_lens_modeling_pipeline|autolens"
+    "autolens_assistant|autolens"
 )
 
 # Compare two YYYY.M.D.B versions. Echoes AHEAD / BEHIND / MATCH / BAD.