[package]
edition = "2021"
name = "codemem-embeddings"
version = "0.9.1"
authors = ["Codemem Contributors"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Candle-based embedding service for Codemem using BAAI/bge-base-en-v1.5"
homepage = "https://github.com/cogniplex/codemem"
readme = "README.md"
license = "Apache-2.0"
repository = "https://github.com/cogniplex/codemem"
[features]
accelerate = [
"candle-core/accelerate",
"candle-nn/accelerate",
"candle-transformers/accelerate",
]
cuda = [
"candle-core/cuda",
"candle-nn/cuda",
"candle-transformers/cuda",
]
default = []
metal = [
"candle-core/metal",
"candle-nn/metal",
"candle-transformers/metal",
]
[lib]
name = "codemem_embeddings"
path = "src/lib.rs"
[dependencies.anyhow]
version = "1"
[dependencies.candle-core]
version = "0.9"
default-features = false
[dependencies.candle-nn]
version = "0.9"
default-features = false
[dependencies.candle-transformers]
version = "0.9"
default-features = false
[dependencies.codemem-core]
version = "0.10.1"
[dependencies.dirs]
version = "6"
[dependencies.hf-hub]
version = "0.5"
[dependencies.lru]
version = "0.16"
[dependencies.reqwest]
version = "0.13"
features = [
"blocking",
"json",
]
[dependencies.serde]
version = "1"
features = ["derive"]
[dependencies.serde_json]
version = "1"
[dependencies.thiserror]
version = "2"
[dependencies.tokenizers]
version = "0.22"
features = ["http"]
[dependencies.tracing]
version = "0.1"
[dev-dependencies.mockito]
version = "1"