pyproject.toml

[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.backends.legacy:build"

[project]
name = "prompt-injection-triage-env"
version = "1.0.0"
description = "OpenEnv RL environment for training AI agents to detect, classify, and mitigate prompt injection attacks."
readme = "README.md"
license = { text = "Apache-2.0" }
requires-python = ">=3.11"
authors = [
  { name = "Muhammed Al Hameed M" },
  { name = "Vignesh Anbalagan" },
  { name = "A. Shameer" },
]
keywords = ["prompt-injection", "ai-safety", "reinforcement-learning", "openenv", "cybersecurity"]

# ── Dependencies ─────────────────────────────────────────────────────────────
dependencies = [
  "openenv-core>=0.2.0",           # required by openenv validate
  "fastapi==0.135.3",
  "uvicorn[standard]==0.43.0",
  "pydantic==2.12.5",
  "openai==2.30.0",
  "httpx==0.28.1",
  "gradio==6.11.0",
]

[project.optional-dependencies]
test = [
  "pytest==9.0.2",
  "pytest-asyncio==1.3.0",
  "anyio==4.13.0",
]

# ── Entry points ─────────────────────────────────────────────────────────────
# "server" script is required by openenv validate multi-mode check
[project.scripts]
server = "server.app:main"

[tool.setuptools.packages.find]
where = ["."]
include = ["server*"]

# ── OpenEnv multi-mode deployment ─────────────────────────────────────────────
[tool.openenv]
name = "prompt-injection-triage-env"
version = "1.0.0"

[tool.openenv.modes.server]
description = "FastAPI environment server (RL loop endpoints: /reset /step /state /health)"
command = "uvicorn server.app:app --host 0.0.0.0 --port 7860"
port = 7860
health_check = "/health"

[tool.openenv.modes.ui]
description = "Gradio visual interface for judges"
command = "python app.py"
port = 7862

[tool.openenv.modes.inference]
description = "Baseline inference agent"
command = "python inference.py"

# ── Pytest ────────────────────────────────────────────────────────────────────
[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]