[workspace]
members = [
    ".",
    "crates/infer-models",
    "crates/infer-compute",
    "crates/infer-vindex",
    "crates/infer-core",
    "crates/infer-inference",
    "crates/infer-client",
]
resolver = "2"

[workspace.package]
version = "0.1.3-alpha"
edition = "2021"
rust-version = "1.80"
authors = ["Greg Burd <greg@burd.me>"]
license = "Apache-2.0"

[workspace.dependencies]
serde = { version = "1", features = ["derive"] }
serde_json = "1"
thiserror = "2"

[package]
name = "pg_infer"
version.workspace = true
edition.workspace = true
rust-version.workspace = true
authors.workspace = true
license.workspace = true
description = "Query transformer model weights as PostgreSQL relations"
repository = "https://codeberg.org/gregburd/pg_infer"
homepage = "https://codeberg.org/gregburd/pg_infer"
keywords = ["postgresql", "transformer", "interpretability", "knowledge-graph", "pgrx"]
categories = ["database", "science"]

[lib]
crate-type = ["cdylib", "lib"]

[[bin]]
name = "pgrx_embed_pg_infer"
path = "./src/bin/pgrx_embed.rs"

[features]
default = ["pg18"]
pg18 = ["pgrx/pg18", "pgrx-tests/pg18"]
pg_test = []
# Enable the infer() function.  Pulls in infer-inference which depends on
# wasmtime + protobuf-src (heavy C++ build).
inference = ["dep:infer-inference"]

[dependencies]
pgrx = "0.17.0"
serde = { workspace = true }
serde_json = { workspace = true }
thiserror = { workspace = true }
once_cell = "1.19"
tracing = "0.1"
tracing-subscriber = { version = "0.3", features = ["env-filter"] }
ndarray = "0.16"
half = "2"
memmap2 = "0.9"
rayon = "1.10"

# Internal crates
infer-vindex = { path = "crates/infer-vindex" }
infer-compute = { path = "crates/infer-compute" }
infer-models = { path = "crates/infer-models" }
infer-client = { path = "crates/infer-client" }

# Optional: full inference (forward pass).  Heavy dependency chain.
infer-inference = { path = "crates/infer-inference", optional = true }

[dev-dependencies]
pgrx-tests = "0.17.0"

[profile.release]
opt-level = 3
lto = "fat"
codegen-units = 1

[lints.clippy]
unwrap_used = "deny"
panic = "deny"
panic_in_result_fn = "deny"