diff --git a/tests/rust_core_unit_tests/__init__.py b/tests/rust_core_unit_tests/__init__.py new file mode 100644 index 00000000000..e69de29bb2d diff --git a/tests/rust_core_unit_tests/test_pattern_prefilter.py b/tests/rust_core_unit_tests/test_pattern_prefilter.py new file mode 100644 index 00000000000..393d7619cec --- /dev/null +++ b/tests/rust_core_unit_tests/test_pattern_prefilter.py @@ -0,0 +1,51 @@ +import pytest + +litellm_core = pytest.importorskip( + "litellm_core", + reason="litellm_core Rust extension not built; run `maturin develop` in rust/", +) +build_pattern_prefilter = litellm_core.build_pattern_prefilter + +SSN_PATTERN = r"\b\d{3}-\d{2}-\d{4}\b" +API_KEY_PATTERN = r"sk-[A-Za-z0-9]{20,}" +LOOKBEHIND_PATTERN = r"(?