[package]
name = "rtx-nlg"
version = "1.0.0"
edition.workspace = true
rust-version = "1.92"
description = "Advanced Natural Language Generation for RTX - Production-ready text generation with state-of-the-art algorithms"
license = "MIT OR Apache-2.0"
keywords = ["nlp", "nlg", "text-generation", "transformers", "machine-learning"]
categories = ["science", "text-processing"]
authors = ["RTX Team"]

[dependencies]
rtx-tensor = { path = "../../core/rtx-tensor" }
rtx-transformers = { path = "../../training/rtx-transformers" }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
thiserror = "2"
anyhow = "1.0"
rayon = "1.8"
tokio = { version = "1.0", features = ["full"] }
futures = "0.3"
dashmap = "5.5"
indexmap = "2.0"
regex = "1.10"
unicode-segmentation = "1.10"
rand = "0.8"
rand_distr = "0.4"
ndarray = "0.15"
# candle-core and candle-nn disabled - incompatible with Rust 2024 edition due to `gen` keyword
# candle-core = { version = "0.3", default-features = false }
# candle-nn = { version = "0.3", default-features = false }
# tch = "0.13" # Removed: PyTorch dependency - using pure Rust implementation
tracing = "0.1"
tracing-subscriber = "0.3"
metrics = "0.22"
lru = "0.12"
num_cpus = "1.16"
chrono = { version = "0.4", features = ["serde"] }
async-trait = "0.1"
tokio-stream = "0.1"
uuid = { version = "1.6", features = ["v4"] }

[dev-dependencies]
criterion = { version = "0.5", features = ["html_reports"] }
proptest = "1.4"
tempfile = "3.8"
env_logger = "0.10"

[[bench]]
name = "generation_benchmarks"
harness = false

[[bench]]
name = "translation_benchmarks"
harness = false

[[bench]]
name = "summarization_benchmarks"
harness = false

[lib]
name = "rtx_nlg"
path = "src/lib.rs"

[features]
default = ["production", "cuda"]
production = []
cuda = []
metal = []
distributed = []
streaming = []
[lints]
workspace = true