"""
llm-taint — lightweight taint tracking for LLM pipelines.
"""
from .taint import (
    TaintViolationError,
    TaintedStr,
    _env_taint_registry,
    add_safe_sink,
    add_secret_env_key,
    add_unsafe_sink,
    check_sink,
    get_label,
    get_tainted_env,
    is_tainted,
    scrub,
    scrub_dict,
    taint,
    taint_env_secrets,
)

__all__ = [
    "TaintedStr",
    "TaintViolationError",
    "taint",
    "is_tainted",
    "get_label",
    "check_sink",
    "scrub",
    "scrub_dict",
    "taint_env_secrets",
    "get_tainted_env",
    "add_safe_sink",
    "add_unsafe_sink",
    "add_secret_env_key",
    "_env_taint_registry",
]

__version__ = "0.1.0"
