[package]
name = "hallucination-detection"
version = "0.1.5"
edition = "2021"
license = "MIT"
repository = "https://github.com/devflowinc/trieve"
authors = ["Devflow Inc. <humans@trieve.ai>"]
description = "Extremely fast Hallucination Detection for RAG using BERT NER, noun, and number analysis"

[profile.test]
inherits = "release"

[features]
default = ["ner"]
ner = ["rust-bert"]
download-onnx = ["ort?/download-binaries"]
onnx = ["ort", "rust-bert?/onnx"]

[dependencies]
# Core dependencies
reqwest = "0.12.9"
regex = "1.11.1"
serde = { version = "1.0.215", features = ["derive"] }
tokio = { version = "1.42.0", features = ["full"] }
once_cell = "1.18"
rust-bert = { version = "0.23.0", optional = true }
ort = { version = "1.16.3", features = [
    "load-dynamic",
], optional = true, default-features = false }


[dev-dependencies]
ort = { version = "1.16.3", features = ["download-binaries", "load-dynamic"] }
csv = "1.3.1"
dotenvy = "0.15.7"
openai_dive = "0.7.0"
serde_json = "1.0.133"
rstest = "0.23.0"
futures = "0.3.31"