feat: support ai extension (#88)

Author: Cuiyus

.github/workflows/linux_arm64_docker_ci.yml

@@ -69,9 +69,9 @@ jobs:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         shell: bash
 
-      - name: Install Ruff
+      - name: Install dependencies
         run: |
-          ${{ env.PIP_BIN }} install --upgrade pip ruff
+          ${{ env.PIP_BIN }} install --upgrade pip ruff==v0.14.4 clang-format==18.1.8 pybind11==3.0 pytest pytest-cov
         shell: bash
 
       - name: Run Ruff Linter
@@ -88,7 +88,6 @@ jobs:
 
       - name: Run clang-format Check
         run: |
-          ${{ env.PIP_BIN }} install clang-format==18.1.8
           cd "$CLEAN_WORKSPACE"
           
 
@@ -120,11 +119,6 @@ jobs:
           ${{ env.PIP_BIN }} install -v . --config-settings='cmake.define.BUILD_TOOLS="ON"'
         shell: bash
 
-      - name: Install test dependencies
-        run: |
-          ${{ env.PIP_BIN }} install pytest pytest-cov
-        shell: bash
-
       - name: Run Python Tests with Coverage
         run: |
           cd "$CLEAN_WORKSPACE"
@@ -133,7 +127,6 @@ jobs:
 
       - name: Run Cpp Tests
         run: |
-          ${{ env.PIP_BIN }} install pybind11==3.0
           cd "$CLEAN_WORKSPACE/build"
           make unittest -j$(nproc)
         shell: bash

.github/workflows/linux_x64_docker_ci.yml

@@ -69,9 +69,9 @@ jobs:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         shell: bash
 
-      - name: Install Ruff
+      - name: Install dependencies
         run: |
-          ${{ env.PIP_BIN }} install --upgrade pip ruff
+          ${{ env.PIP_BIN }} install --upgrade pip ruff==v0.14.4 clang-format==18.1.8 pybind11==3.0 pytest pytest-cov
         shell: bash
 
       - name: Run Ruff Linter
@@ -88,7 +88,6 @@ jobs:
 
       - name: Run clang-format Check
         run: |
-          ${{ env.PIP_BIN }} install clang-format==18.1.8 
           cd "$CLEAN_WORKSPACE"
           
           
@@ -120,11 +119,6 @@ jobs:
           ${{ env.PIP_BIN }} install -v . --config-settings='cmake.define.BUILD_TOOLS="ON"'
         shell: bash
 
-      - name: Install test dependencies
-        run: |
-          ${{ env.PIP_BIN }} install pytest pytest-cov
-        shell: bash
-
       - name: Run Python Tests with Coverage
         run: |
           cd "$CLEAN_WORKSPACE"
@@ -133,7 +127,6 @@ jobs:
 
       - name: Run Cpp Tests
         run: |
-          ${{ env.PIP_BIN }} install pybind11==3.0
           cd "$CLEAN_WORKSPACE/build"
           make unittest -j$(nproc)
         shell: bash

pyproject.toml

@@ -217,10 +217,21 @@ ignore = [
     "E731",     # Lambda assignment (used in callbacks)
     "B019",     # `functools.lru_cache` on methods (handled manually)
     "PLR0912",  # Too many branches
+    "PLC0105",  # Ignore contravariant
+    "RUF002",   # Ignore Unicode
 ]
 fixable = ["ALL"]
 unfixable = []
 
+# Ignore all errors in docstrings
+[tool.ruff.lint.pydocstyle]
+convention = "google"  # or "numpy", "pep257"
+ignore-decorators = ["typing.overload"]
+
+[tool.ruff.lint.flake8-type-checking]
+# Don't check code examples in docstrings
+quote-annotations = true
+
 [tool.ruff.lint.isort]
 required-imports = ["from __future__ import annotations"]
 known-first-party = ["zvec"]
@@ -237,6 +248,9 @@ known-first-party = ["zvec"]
 "python/zvec/model/doc.py" = [
     "RUF023",   # Unused sort (for __slot__)
 ]
+"python/zvec/extension/**" = [
+    "PLC0415",  # Import outside top-level (dynamic imports in _get_model)
+]
 
 [tool.ruff.format]
 indent-style = "space"