forked from protectai/llm-guard
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
102 lines (93 loc) · 2.92 KB
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
[project]
name = "llm-guard"
description = "LLM-Guard is a comprehensive tool designed to fortify the security of Large Language Models (LLMs). By offering sanitization, detection of harmful language, prevention of data leakage, and resistance against prompt injection attacks, LLM-Guard ensures that your interactions with LLMs remain safe and secure."
authors = [
{ name = "Protect AI", email = "[email protected]"}
]
keywords = ["llm", "language model", "security", "adversarial attacks", "prompt injection", "prompt leakage", "PII detection", "self-hardening", "firewall"]
license = { file = "LICENSE" }
readme = "README.md"
dynamic = ["version"]
classifiers = [
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
]
requires-python = ">=3.9"
dependencies = [
"detect-secrets==1.4.0",
"faker>=22,<23",
"fuzzysearch==0.7.3",
"json-repair==0.8.1",
"nltk>=3.8,<4",
"presidio-analyzer>=2.2,<3",
"presidio-anonymizer>=2.2,<3",
"protobuf>=4",
"regex==2023.12.25",
"sentencepiece==0.1.99",
"tiktoken>=0.5,<0.6",
"torch==2.0.1",
"transformers==4.36.2",
"xformers==0.0.22",
"span-marker==1.5.0",
"structlog>=24"
]
[project.optional-dependencies]
onnxruntime = [
"optimum[onnxruntime]",
]
onnxruntime-gpu = [
"optimum[onnxruntime-gpu]",
]
docs-dev = [
"mkdocs>=1.5,<2",
"mkdocs-autorefs==0.5.0",
"mkdocs-git-revision-date-localized-plugin==1.2.2",
"mkdocs-jupyter==0.24.6",
"mkdocs-material==9.5.6",
"mkdocs-material-extensions==1.3.1",
"mkdocs-swagger-ui-tag==0.6.8",
"mkdocstrings==0.24.0",
"mkdocstrings-python==1.8.0"
]
dev = [
"llm_guard[docs-dev]",
"autoflake>=2,<3",
"black>=23,<24",
"isort>5,<6",
"pytest>=7.4,<9",
"pytest-cov>=4.1,<5",
"pre-commit>=3.6,<4",
]
[project.urls]
homepage = "https://github.com/protectai/llm-guard"
documentation = "https://llm-guard.com/"
repository = "https://github.com/protectai/llm-guard"
issues = "https://github.com/protectai/llm-guard/issues"
changelog = "https://llm-guard.com/changelog/"
[tool.setuptools]
packages = {find = {where=["."], include=["llm_guard", "llm_guard.*"]}}
license-files = [
"LICENSE"
]
[tool.setuptools.dynamic]
version = {attr = "llm_guard.version.__version__"}
[tool.setuptools.package-data]
llm_guard = [
"**/*.json"
]
[tool.pytest.ini_options]
addopts = "-p no:warnings"
log-level = "DEBUG"
# The flag below should only be activated in special debug sessions
# i.e. the test hangs and we need to see what happened up to that point.
# There are some race conditions with how the logging streams are closed in the teardown
# phase, which will cause tests to fail or "magically" ignored.
log_cli = "False"
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"