[package]
edition = "2024"
rust-version = "1.88.0"
name = "ck-embed"
version = "0.7.2"
authors = ["Mike Renwick"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Text embedding providers for ck semantic search"
readme = false
keywords = [
"embedding",
"nlp",
"semantic",
]
categories = ["science"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/BeaconBay/ck"
resolver = "2"
[features]
default = [
"fastembed",
"mixedbread",
]
fastembed = ["dep:fastembed"]
mixedbread = [
"dep:hf-hub",
"dep:tokenizers",
"dep:ort",
"dep:once_cell",
"dep:ndarray",
"dep:num_cpus",
]
[lib]
name = "ck_embed"
path = "src/lib.rs"
[[example]]
name = "list_models"
path = "examples/list_models.rs"
[[example]]
name = "test_mixedbread"
path = "examples/test_mixedbread.rs"
[[example]]
name = "test_nomic"
path = "examples/test_nomic.rs"
[[example]]
name = "test_reranking"
path = "examples/test_reranking.rs"
[dependencies.anyhow]
version = "1.0"
[dependencies.ck-core]
version = "0.7.2"
[dependencies.ck-models]
version = "0.7.2"
[dependencies.fastembed]
version = "5.1"
features = [
"hf-hub-rustls-tls",
"ort-download-binaries",
]
optional = true
default-features = false
[dependencies.hf-hub]
version = "0.4.3"
features = ["ureq"]
optional = true
default-features = false
[dependencies.ndarray]
version = "0.17"
features = ["std"]
optional = true
default-features = false
[dependencies.num_cpus]
version = "1.16"
optional = true
[dependencies.once_cell]
version = "1.19"
optional = true
[dependencies.ort]
version = "2.0.0-rc.11"
features = [
"download-binaries",
"tls-native",
"ndarray",
"std",
]
optional = true
default-features = false
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.tokenizers]
version = "0.20.1"
optional = true
[dependencies.tokio]
version = "1.35"
features = ["full"]