[package]
name = "sz-orm-ai"
version.workspace = true
edition.workspace = true
authors.workspace = true
license.workspace = true
repository.workspace = true
homepage.workspace = true
keywords.workspace = true
categories.workspace = true
description = "SZ-ORM AI Extension - Embedding, Vector Store, RAG Engine"
[features]
default = []
real = ["dep:reqwest", "dep:base64"]
llm-optimizer = ["real"]
plan-cache = ["dep:sz-orm-core", "sz-orm-core/plan-cache"]
ai-nl2sql-enhanced = []
ai-index-advisor = ["dep:sqlparser"]
ai-rewrite-advisor = ["dep:sqlparser"]
[dependencies]
async-trait = "0.1"
tokio = { version = "1.40", features = ["full"] }
thiserror = "2.0"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
reqwest = { version = "0.12", default-features = false, features = ["json", "rustls-tls"], optional = true }
base64 = { version = "0.22", optional = true }
parking_lot.workspace = true
sz-orm-core = { version = "3.5.0", path = "../sz-orm-core", optional = true }
sqlparser = { workspace = true, optional = true }
[dev-dependencies]
tokio-test = "0.4"
[[test]]
name = "ai_tests"
path = "tests/ai_tests.rs"
[[test]]
name = "llm_integration"
path = "tests/llm_integration.rs"
required-features = ["llm-optimizer"]
[[test]]
name = "ai_m4"
path = "tests/ai_m4.rs"
required-features = ["ai-nl2sql-enhanced", "ai-index-advisor", "ai-rewrite-advisor"]
[lints]
workspace = true