supabase / edge-runtime / 13515208152

52%

main: 52%

Build Type

push

github

Committed by web-flow

Commit Message

feat: onnx runtime upgrade (#478)

* chore(sb_ai): update `Cargo.toml`

* chore(sb_ai): update `ONNXRUNTIME_VERSION`

* stamp: solving `ort` api breaking changes

- `ndarray_linalg`: Since `ort` deps to `ndarray=0.16`, decided to
remove this extra dependency. Normalization can be done in a simple
small funtion instead of deriving it from an external crate.

* stamp: converting `from_ort_tensor` without clonning

using the `rc-9` apis to get the same results from older `rc-2`

* test(sb_ai): removing snapshots from onnx `v1.19.2`

Since `ONNX`version has been upgraded, the snapshots need to be
recreated

* stamp: cleaning

* test(sb_ai): adding `macos_aarch64` snapshots with onnx `v1.20`

* feat: apply `normalize` without external deps

- Since `ndarray_linalg` has been removed we need to manually apply L2
normalization

* test(sb_ai): ensuring `mean_pool` and `normalize`

- Applying `dot product` between two embeddings to ensures that
`mean_pool` and `normalize` is working properly

* chore: add aarch64 (linux) snapshots for integration tests

* chore: add x86_64 (macos) snapshots for integration tests

* chore: add x86_64 (linux) snapshots for integration tests

* feat: add `Tensor<String>` support

Co-authored-by: Nyannyacha <meow@nnc.gg>
Co-authored-by: kallebysantos <kalleby_santos@hotmail.com>

* test(sb_ai): adding `Tensor<string>` integration test

* fix(sb_ai): adding 'bool' alias to `JsTensorData`

* fix(sb_ai): parsing input tensors correctly

* test(sb_ai): updating `extract_tensor_ref` to match `JsTensorData` enum

* fix(ai): reflecting module rename

* stamp: enabling cache adapter on `ort-rust-backend` example

* stamp: applying format

* chore: update `Cargo.lock`

---------

Co-authored-by: Nyannyacha <meow@nnc.gg>
Co-authored-by: EC2 Default User <ec2-user@ip-172-31-6-80.ap-northeast-2.compute.internal>

Run Details

46 of 56 new or added lines in 3 files covered. (82.14%)

6 existing lines in 5 files now uncovered.

16801 of 34571 relevant lines covered (48.6%)

1807.66 hits per line