Build win arm64 (#1944)

<!-- ELLIPSIS_HIDDEN -->


> [!IMPORTANT]
> Add support for building Python wheels for Windows ARM64 in GitHub
Actions workflows and update `pyproject.toml` for compatibility.
> 
>   - **Build System**:
> - Add ARM64 Windows target to `build-python-release.reusable.yaml`.
> - Setup build environment for ARM64 Windows, including Rust and Python
3.11, 3.12, 3.13.
> - Update `pyproject.toml` to include `pyo3/generate-import-lib`
feature for Windows ARM64.
>   - **Workflows**:
> - Modify `release.yml` to expect at least 8 wheels, reflecting the new
ARM64 build.
>     - Minor formatting changes in `release.yml`.
> 
> <sup>This description was created by </sup>[<img alt="Ellipsis"
src="https://img.shields.io/badge/Ellipsis-blue?color=175173">](https://www.ellipsis.dev?ref=BoundaryML%2Fbaml&utm_source=github&utm_medium=referral)<sup>
for 63d12a4. You can
[customize](https://app.ellipsis.dev/BoundaryML/settings/summaries) this
summary. It will automatically update as commits are pushed.</sup>


<!-- ELLIPSIS_HIDDEN -->

Author: aaronvg

.github/workflows/build-python-release.reusable.yaml

@@ -2,12 +2,19 @@ name: BAML Release - Build Python
 
 on:
   workflow_call: {}
+  push:
+    branches:
+      - baml-py-win
 
 concurrency:
   # suffix is important to prevent a concurrency deadlock with the calling workflow
   group: ${{ github.workflow }}-${{ github.ref }}-build-python
   cancel-in-progress: true
 
+defaults:
+  run:
+    shell: bash
+
 jobs:
   build:
     strategy:
@@ -49,24 +56,86 @@ jobs:
           - target: x86_64-pc-windows-msvc
             runs_on: windows-latest
 
+          - target: aarch64-pc-windows-msvc
+            runs_on: windows-11-arm
+            architecture: arm64
+
     name: ${{ matrix._.target }}
     runs-on: ${{ matrix._.runs_on }}
+    # See also https://github.com/mcrumiller/polars/.github/workflows/release-python.yml#L282
     steps:
+      - name: Setup build environment (ARM64 Windows)
+        if: matrix._.target == 'aarch64-pc-windows-msvc'
+        shell:
+          powershell
+          # Notes
+          # * We update `Expand-Archive` to avoid "" is not a supported archive file format when extracting
+          #   files that don't end in `.zip`
+        run: |
+          # rustup is not installed in aarch64
+          if (-not (Get-Command rustup -ErrorAction SilentlyContinue)) {
+              Invoke-WebRequest -Uri "https://static.rust-lang.org/rustup/dist/aarch64-pc-windows-msvc/rustup-init.exe" -OutFile "rustup-init.exe"
+              .\rustup-init.exe --default-toolchain stable -y
+              Remove-Item "rustup-init.exe"
+              "$env:USERPROFILE/.cargo/bin" | Out-File -FilePath "$env:GITHUB_PATH" -Append
+          }
+
+      - name: Check build environment (ARM64 Windows)
+        if: matrix._.target == 'aarch64-pc-windows-msvc'
+        run: |
+          set -x
+          bash --version
+          rustup show
+          clang -v
+          cmake --version
+
       - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+
+      # Setup Python for non-ARM64 Windows targets and other OS
+      - name: Setup Python (default)
+        if: matrix._.target != 'aarch64-pc-windows-msvc'
+        uses: actions/setup-python@v5
         with:
           python-version: "3.8"
-          architecture: ${{ matrix._.runs_on == 'windows-latest' && 'x64' || null }}
+          architecture: ${{ matrix._.architecture }}
+
+      # Setup Python versions for ARM64 Windows
+      - name: Setup Python 3.11 (ARM64 Windows)
+        if: matrix._.target == 'aarch64-pc-windows-msvc'
+        uses: actions/setup-python@v5
+        id: py311
+        with:
+          python-version: "3.11"
+          architecture: "arm64"
+          allow-prereleases: true
+      - name: Setup Python 3.12 (ARM64 Windows)
+        if: matrix._.target == 'aarch64-pc-windows-msvc'
+        uses: actions/setup-python@v5
+        id: py312
+        with:
+          python-version: "3.12"
+          architecture: "arm64"
+          allow-prereleases: true
+      - name: Setup Python 3.13 (ARM64 Windows)
+        if: matrix._.target == 'aarch64-pc-windows-msvc'
+        uses: actions/setup-python@v5
+        id: py313
+        with:
+          python-version: "3.13"
+          architecture: "arm64"
+          allow-prereleases: true
 
       - name: Build wheels
         uses: PyO3/maturin-action@v1
-        env: ${{ matrix._.env || fromJSON('{}') }}
+        env:
+          # see https://github.com/PyO3/maturin/issues/2110
+          XWIN_VERSION: "16"
         with:
           target: ${{ matrix._.target }}
           command: build
           # building in engine/ ensures that we pick up .cargo/config.toml
           working-directory: engine
-          args: --release --out language_client_python/dist --manifest-path language_client_python/Cargo.toml
+          args: --release --out language_client_python/dist --manifest-path language_client_python/Cargo.toml ${{ (matrix._.target == 'aarch64-pc-windows-msvc' && format('--interpreter {0} {1} {2}', steps.py311.outputs.python-path, steps.py312.outputs.python-path, steps.py313.outputs.python-path)) || '' }}
           manylinux: ${{ matrix._.manylinux }}
           before-script-linux: |
             if command -v yum &> /dev/null; then

.github/workflows/release.yml

@@ -69,7 +69,7 @@ jobs:
 
   build-typescript-release:
     uses: ./.github/workflows/build-typescript-release.reusable.yaml
-  
+
   build-cli:
     needs: determine-version
     uses: ./.github/workflows/build-cli-release.reusable.yaml
@@ -79,14 +79,14 @@ jobs:
 
   # This job now calls the reusable workflow which contains the matrix strategy
   build-vscode-reusable: # Renamed job for clarity
-    needs: [ determine-version, build-cli ]
+    needs: [determine-version, build-cli]
     uses: ./.github/workflows/build-vscode-release.reusable.yaml
     with:
       # No artifact_name needed here anymore
       version: ${{ needs.determine-version.outputs.version_string }}
       # Pass the boolean output directly
       is_release_build: ${{ needs.determine-version.outputs.is_release_tag == 'true' }}
-  
+
   # Kick off integration tests when we build everything.
   # This does not yet succeed/pass consistently, or even run everything, but it's a good start.
   integ-tests:
@@ -105,7 +105,6 @@ jobs:
       - build-vscode-reusable # Depends on the job calling the reusable workflow
     steps:
       - run: echo "::do-nothing::" >/dev/null
-      
 
   publish-to-pypi:
     environment: release
@@ -129,8 +128,8 @@ jobs:
           set -euo pipefail
           ls dist/
           wheel_count=$(ls dist/*.whl 2>/dev/null | wc -l)
-          if [ "$wheel_count" -lt 7 ]; then
-            echo "Error: Expected at least 7 wheels, but found $wheel_count"
+          if [ "$wheel_count" -lt 8 ]; then
+            echo "Error: Expected at least 8 wheels, but found $wheel_count"
             exit 1
           fi
           echo "Found $wheel_count wheels"
@@ -223,8 +222,6 @@ jobs:
             gem push $i
           done
 
-
-
   publish-to-open-vsx:
     environment: release
     needs: [determine-version, all-builds] # Also need determine-version for the condition
@@ -308,19 +305,19 @@ jobs:
         uses: actions/download-artifact@v4
         with:
           # Match artifacts starting with 'baml-cli-'
-          pattern: baml-cli-* 
+          pattern: baml-cli-*
           # Download into this directory
-          path: cli-artifacts 
+          path: cli-artifacts
           merge-multiple: true # Merge artifacts from different OS/arch into the path
 
       # Download all CFFI library artifacts produced by the build-cli job
       - name: Download all libbaml-cffi artifacts
         uses: actions/download-artifact@v4
         with:
           # Match artifacts starting with 'libbaml-cffi-'
-          pattern: libbaml-cffi-* 
+          pattern: libbaml-cffi-*
           # Download into a separate directory
-          path: cffi-artifacts 
+          path: cffi-artifacts
           merge-multiple: true # Merge artifacts from different OS into the path
 
       - name: List downloaded CLI artifacts
@@ -369,4 +366,4 @@ jobs:
           # Use default GITHUB_TOKEN
           token: ${{ secrets.GITHUB_TOKEN }}
           # Do not auto-generate release notes, rely on changelog or empty body
-          generate_release_notes: false
+          generate_release_notes: false

engine/language_client_python/pyproject.toml

@@ -15,7 +15,8 @@ build-backend = "maturin"
 [tool.maturin]
 python-source = "python_src"
 # "extension-module" tells pyo3 we want to build an extension module (skips linking against libpython.so).
-features = ["pyo3/extension-module"]
+# The second feature is to fix windows aarch64 https://github.com/samuelcolvin/rtoml/pull/73/files
+features = ["pyo3/extension-module", "pyo3/generate-import-lib"]
 
 [project.scripts]
 baml-cli = "baml_py:invoke_runtime_cli"