chore(deps): add pyatr (ATR rail) as optional extra + dev dependency

Installs pyatr so tests/test_atr_rail.py runs in CI (previously ModuleNotFoundError: No module named 'pyatr'). Mirrors the yara-python jailbreak-rail pattern: optional dependency, atr extra, included in all, and dev group.

Signed-off-by: eeee2345 <217509886+eeee2345@users.noreply.github.com>

Author: eeee2345

poetry.lock

@@ -114,6 +114,9 @@ chainlit = { version = "^2.11.0", optional = true }
 transformers = { version = ">=4.35.0", optional = true }
 torch = { version = ">=2.0.0", optional = true }
 
+# agent threat rules (ATR) detection rail
+pyatr = { version = ">=0.2.6", optional = true }
+
 [tool.poetry.extras]
 sdd = ["presidio-analyzer", "presidio-anonymizer"]
 eval = ["tqdm", "numpy", "streamlit", "tornado"]
@@ -124,6 +127,7 @@ multilingual = ["fast-langdetect"]
 server = ["aiofiles", "openai", "fastapi", "starlette", "uvicorn", "watchdog"]
 chat-ui = ["chainlit"]
 hf-classifier = ["transformers", "torch"]
+atr = ["pyatr"]
 # Poetry does not support recursive dependencies, so we need to add all the dependencies here.
 # I also support their decision. There is no PEP for recursive dependencies, but it has been supported in pip since version 21.2.
 # It is here for backward compatibility.
@@ -146,6 +150,7 @@ all = [
   "chainlit",
   "transformers",
   "torch",
+  "pyatr",
 ]
 
 [tool.poetry.group.dev]
@@ -166,6 +171,7 @@ yara-python = "^4.5.1"
 opentelemetry-api = "^1.34.1"
 opentelemetry-sdk = "^1.34.1"
 fast-langdetect = ">=1.0.0"
+pyatr = ">=0.2.6"
 openai = ">=1.0.0,<3.0.0"
 # server (optional at runtime, required for tests/server/)
 fastapi = ">=0.103.0"