[package]
edition = "2021"
rust-version = "1.95"
name = "fathomdb-embedder"
version = "0.8.20"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "FathomDB embedder runtime — built-in embedder implementations for the fathomdb-embedder-api trait."
readme = "README.md"
keywords = [
"embeddings",
"embedder",
"candle",
"onnx",
"reranker",
]
license = "MIT"
license-file = "LICENSE"
repository = "https://github.com/coreyt/fathomdb"
[features]
default = []
default-embedder = [
"dep:ureq",
"dep:sha2",
"dep:dirs",
"dep:fs2",
"dep:serde_json",
"dep:thiserror",
"dep:candle-core",
"dep:candle-nn",
"dep:candle-transformers",
"dep:tokenizers",
]
default-reranker = [
"dep:ureq",
"dep:sha2",
"dep:dirs",
"dep:fs2",
"dep:serde_json",
"dep:thiserror",
"dep:candle-core",
"dep:candle-nn",
"dep:candle-transformers",
"dep:tokenizers",
]
embed-cuda = [
"default-embedder",
"candle-core/cuda",
"candle-nn/cuda",
"candle-transformers/cuda",
]
embed-metal = [
"default-embedder",
"candle-core/metal",
"candle-nn/metal",
"candle-transformers/metal",
]
loader-test-hooks = ["default-embedder"]
onnx-embedder = [
"dep:ort",
"dep:tokenizers",
"dep:sha2",
]
rerank-cuda = [
"default-reranker",
"candle-core/cuda",
"candle-nn/cuda",
"candle-transformers/cuda",
]
rerank-metal = [
"default-reranker",
"candle-core/metal",
"candle-nn/metal",
"candle-transformers/metal",
]
[lib]
name = "fathomdb_embedder"
path = "src/lib.rs"
[[example]]
name = "gpu_speedup"
path = "examples/gpu_speedup.rs"
required-features = ["default-embedder"]
[[test]]
name = "candle_bge"
path = "tests/candle_bge.rs"
[[test]]
name = "candle_dump"
path = "tests/candle_dump.rs"
[[test]]
name = "candle_onnx_equivalence"
path = "tests/candle_onnx_equivalence.rs"
[[test]]
name = "cross_backend_calibration"
path = "tests/cross_backend_calibration.rs"
[[test]]
name = "loader"
path = "tests/loader.rs"
[[test]]
name = "no_engine_dep"
path = "tests/no_engine_dep.rs"
[dependencies.candle-core]
version = "0.10"
optional = true
[dependencies.candle-nn]
version = "0.10"
optional = true
[dependencies.candle-transformers]
version = "0.10"
optional = true
default-features = false
[dependencies.dirs]
version = "5"
optional = true
[dependencies.fathomdb-embedder-api]
version = "0.6.1"
[dependencies.fs2]
version = "0.4"
optional = true
[dependencies.ort]
version = "=2.0.0-rc.10"
features = ["load-dynamic"]
optional = true
default-features = false
[dependencies.serde_json]
version = "1"
optional = true
[dependencies.sha2]
version = "0.11"
optional = true
[dependencies.thiserror]
version = "1"
optional = true
[dependencies.tokenizers]
version = "0.20"
features = ["onig"]
optional = true
default-features = false
[dependencies.ureq]
version = "2"
features = ["tls"]
optional = true
default-features = false
[dev-dependencies.httpmock]
version = "0.7"
[dev-dependencies.sha2]
version = "0.11"
[dev-dependencies.tempfile]
version = "3"