[package]
edition = "2021"
rust-version = "1.80"
name = "embedd"
version = "0.1.0"
authors = ["Arc <attobop@gmail.com>"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Embedding interfaces + local backends (Candle/HF)."
homepage = "https://github.com/arclabs561/embedd"
documentation = "https://docs.rs/embedd"
readme = "README.md"
keywords = [
"embeddings",
"nlp",
"inference",
"candle",
]
categories = [
"science",
"text-processing",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/arclabs561/embedd"
[features]
all = [
"candle-hf",
"openai",
"tei",
"hf-inference",
"fastembed",
"ort-tokenizers",
"burn-backend",
"siglip",
"serde",
"async-openai",
"async-tei",
"async-hf-inference",
"qdrant",
]
async-hf-inference = [
"dep:reqwest",
"dep:tokio",
"dep:serde",
]
async-openai = [
"dep:reqwest",
"dep:tokio",
"dep:serde",
]
async-tei = [
"dep:reqwest",
"dep:tokio",
]
burn-backend = ["dep:burn"]
candle-hf = [
"dep:candle-core",
"dep:candle-nn",
"dep:candle-transformers",
"dep:hf-hub",
"dep:tokenizers",
]
cli = ["dep:clap"]
default = []
fastembed = ["dep:fastembed"]
hf-inference = [
"dep:ureq",
"dep:serde",
]
openai = [
"dep:ureq",
"dep:serde",
]
ort-tokenizers = [
"dep:ort",
"dep:tokenizers",
]
qdrant = [
"dep:qdrant-client",
"dep:tokio",
]
serde = ["dep:serde"]
siglip = []
tei = ["dep:ureq"]
[lib]
name = "embedd"
path = "src/lib.rs"
[[bin]]
name = "embedd"
path = "src/bin/embedd.rs"
required-features = ["cli"]
[[example]]
name = "backend_compare"
path = "examples/backend_compare.rs"
[[example]]
name = "backend_matrix"
path = "examples/backend_matrix.rs"
[[example]]
name = "batched_embed"
path = "examples/batched_embed.rs"
required-features = ["fastembed"]
[[example]]
name = "hello_embed"
path = "examples/hello_embed.rs"
required-features = ["fastembed"]
[[example]]
name = "semantic_search"
path = "examples/semantic_search.rs"
required-features = ["fastembed"]
[[example]]
name = "sparse_retrieval"
path = "examples/sparse_retrieval.rs"
required-features = ["fastembed"]
[[test]]
name = "arch_detection"
path = "tests/arch_detection.rs"
[[test]]
name = "batching"
path = "tests/batching.rs"
[[test]]
name = "cli_contract"
path = "tests/cli_contract.rs"
[[test]]
name = "e2e_qdrant"
path = "tests/e2e_qdrant.rs"
[[test]]
name = "integration_hf_inference"
path = "tests/integration_hf_inference.rs"
[[test]]
name = "integration_http"
path = "tests/integration_http.rs"
[[test]]
name = "normalization_policy"
path = "tests/normalization_policy.rs"
[[test]]
name = "output_dim_policy"
path = "tests/output_dim_policy.rs"
[[test]]
name = "prompted_embedder"
path = "tests/prompted_embedder.rs"
[[test]]
name = "property_core"
path = "tests/property_core.rs"
[[test]]
name = "property_modalities"
path = "tests/property_modalities.rs"
[[test]]
name = "property_safetensors"
path = "tests/property_safetensors.rs"
[[bench]]
name = "embedd_bench"
path = "benches/embedd_bench.rs"
harness = false
[dependencies.anyhow]
version = "1.0"
[dependencies.burn]
version = "0.20.0"
optional = true
[dependencies.candle-core]
version = "0.9"
optional = true
[dependencies.candle-nn]
version = "0.9"
optional = true
[dependencies.candle-transformers]
version = "0.9"
optional = true
[dependencies.clap]
version = "4.5"
features = ["derive"]
optional = true
[dependencies.fastembed]
version = "5.8.1"
optional = true
[dependencies.hf-hub]
version = "0.4"
optional = true
[dependencies.innr]
version = "0.1.8"
[dependencies.once_cell]
version = "1.19"
[dependencies.ort]
version = "2.0.0-rc.11"
optional = true
[dependencies.qdrant-client]
version = "1.16"
optional = true
[dependencies.reqwest]
version = "0.12"
features = [
"json",
"rustls-tls",
]
optional = true
default-features = false
[dependencies.serde]
version = "1.0"
features = [
"derive",
"derive",
]
optional = true
[dependencies.serde_json]
version = "1.0"
[dependencies.tokenizers]
version = "0.22"
optional = true
[dependencies.tokio]
version = "1"
features = ["rt"]
optional = true
[dependencies.ureq]
version = "2.10"
features = ["json"]
optional = true
[dev-dependencies.assert_cmd]
version = "2.0"
[dev-dependencies.criterion]
version = "0.7"
[dev-dependencies.predicates]
version = "3.0"
[dev-dependencies.proptest]
version = "1.9"
[dev-dependencies.tokio]
version = "1"
features = [
"rt",
"rt",
"macros",
]