merge stable to main

.github/workflows/rust.yml

@@ -5,9 +5,9 @@ on:
     paths-ignore:
       - 'examples/**'
       - 'README.md'
-    branches: [ "main" ]
+    branches: [ "main", "stable" ]
   pull_request:
-    branches: [ "main" ]
+    branches: [ "main", "stable" ]
 
 env:
   CARGO_TERM_COLOR: always

.github/workflows/test.yml

@@ -5,9 +5,9 @@ on:
     paths-ignore:
       - 'examples/**'
       - 'README.md'
-    branches: [ "main" ]
+    branches: [ "main", "stable" ]
   pull_request:
-    branches: [ "main" ]
+    branches: [ "main", "stable" ]
 
 env:
   CARGO_TERM_COLOR: always
@@ -21,25 +21,13 @@ jobs:
     - uses: actions/checkout@v4
     - name: Install dependencies
       run: |
-        rustup toolchain install nightly
+        rustup update
         mkdir all-MiniLM-L6-v2_onnx
-        cd all-MiniLM-L6-v2_onnx && wget https://huggingface.co/nigel-christian/all-MiniLM-L6-v2_onnx/resolve/main/config.json
-        wget https://huggingface.co/nigel-christian/all-MiniLM-L6-v2_onnx/resolve/main/model.onnx
-        wget https://huggingface.co/nigel-christian/all-MiniLM-L6-v2_onnx/resolve/main/special_tokens_map.json
-        wget https://huggingface.co/nigel-christian/all-MiniLM-L6-v2_onnx/resolve/main/tokenizer_config.json
-        wget https://huggingface.co/nigel-christian/all-MiniLM-L6-v2_onnx/resolve/main/tokenizer.json
-        wget https://huggingface.co/nigel-christian/all-MiniLM-L6-v2_onnx/resolve/main/vocab.txt
-    - name: database test
-      run: |
-        export LMDB_USER=$USER
-        RUST_LOG=DEBUG cargo test database::tests::environment_test -- --exact
-    - name: etl test
-      run: |
-        export LMDB_USER=$USER
-        ONNX_PARALLEL_THREADS=2
-        cargo test embeddings::tests::cosine_etl_test -- --exact
-    - name: nearest test
-      run: |
-        export LMDB_USER=$USER
-        ONNX_PARALLEL_THREADS=2
-        cargo test embeddings::tests::nearest_test -- --exact
+        cd all-MiniLM-L6-v2_onnx && wget https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/resolve/main/config.json
+        wget https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/resolve/main/onnx/model.onnx
+        wget https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/resolve/main/special_tokens_map.json
+        wget https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/resolve/main/tokenizer_config.json
+        wget https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/resolve/main/tokenizer.json
+        wget https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2/resolve/main/vocab.txt
+    - name: test
+      run: RUST_TEST_THREADS=1 cargo test