-
Notifications
You must be signed in to change notification settings - Fork 329
/
pyproject.toml
135 lines (120 loc) · 3.49 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
[tool.poetry]
name = "guardrails-ai"
version = "0.6.2"
description = "Adding guardrails to large language models."
authors = ["Guardrails AI <[email protected]>"]
license = "Apache License 2.0"
homepage = "https://www.guardrailsai.com/"
documentation = "https://www.guardrailsai.com/docs"
readme = "README.md"
packages = [
{ include = "guardrails", from = "." }
]
[tool.poetry.scripts]
guardrails = "guardrails.cli:cli"
[tool.poetry.dependencies]
python = "^3.9"
lxml = "^4.9.3"
openai = "^1.30.1"
rich = "^13.6.0"
pydantic = ">=2.0.0, <3.0"
typer = {extras = ["all"], version = ">=0.9.0, <0.13"}
griffe = "^0.36.9"
tenacity = ">=8.1.0"
rstr = "^3.2.2"
typing-extensions = "^4.8.0"
python-dateutil = "^2.8.2"
tiktoken = ">=0.5.1"
litellm = "^1.37.14"
sqlvalidator = {version = "^0.0.20", optional = true}
sqlalchemy = {version = ">=2.0.9", optional = true}
sqlglot = {version = "^19.0.3", optional = true}
faiss-cpu = {version = "^1.7.4", optional = true}
numpy = {version = ">=1.25, <2.0", optional = true}
manifest-ml = {version = "^0.1.8", optional = true}
transformers = {version = "^4.38.0", optional = true}
anthropic = {version = "^0.7.2", optional = true}
torch = {version = "^2.1.1", optional = true}
nbdoc = {version = "^0.0.82", optional = true}
pydash = "^7.0.6"
docspec_python = "2.2.1"
pydoc-markdown = "4.8.2"
langchain-core = ">=0.1,<0.4"
coloredlogs = "^15.0.1"
requests = "^2.31.0"
faker = "^25.2.0"
jsonref = "^1.1.0"
jsonformer = {version = "0.12.0", optional = true}
jsonschema = {version = "^4.22.0", extras = ["format"]}
pip = ">=22"
pyjwt = "^2.8.0"
opentelemetry-sdk = "^1.24.0"
opentelemetry-exporter-otlp-proto-grpc = "^1.24.0"
opentelemetry-exporter-otlp-proto-http = "^1.24.0"
guardrails-hub-types = "^0.0.4"
guardrails-api-client = "^0.4.0a1"
diff-match-patch = "^20230430"
guardrails-api = ">=0.1.0a1,<0.2.0"
mlflow = {version = "^2.0.1", optional = true}
uvloop = {version = "^0.20.0", optional = true}
semver = "^3.0.2"
llama-index = {version = ">=0.11.0", optional = true}
boto3 = {version = ">1, <2", optional = true}
[tool.poetry.extras]
sql = ["sqlvalidator", "sqlalchemy", "sqlglot"]
vectordb = ["faiss-cpu", "numpy"]
manifest = ["manifest-ml"]
anthropic = ["anthropic"]
docs-build = ["nbdoc", "docspec_python", "pydoc-markdown"]
huggingface = ["transformers", "torch", "jsonformer"]
api = ["guardrails-api"]
databricks = ["mlflow"]
uv = ["uvloop"]
llama = ["llama-index"]
[tool.poetry.group.dev.dependencies]
pytest = "^7.4.3"
pytest-asyncio = "^0.21.1"
docformatter = ">=1.4"
pytest-cov = ">=2.10.1"
pre-commit = ">=2.9.3"
twine = "^4.0.2"
pytest-mock = "^3.12.0"
pypdfium2 = "^4.23.1"
pyright = "1.1.334"
lxml-stubs = "^0.4.0"
ruff = ">=0.4.1"
[tool.poetry.group.api]
optional = true
[tool.poetry.group.api.dependencies]
guardrails-api = ">=0.0.0a0"
[tool.poetry.group.docs]
optional = true
[tool.poetry.group.docs.dependencies]
mkdocs = "^1.5.3"
mkdocstrings = {extras = ["python"], version = "^0.23.0"}
mkdocs-jupyter = ">=0.23"
mkdocs-material = "^9.4.8"
mknotebooks = "^0.8.0"
griffe = "^0.36.9"
pillow = "^10.1.0"
cairosvg = "^2.7.1"
mkdocs-glightbox = "^0.3.4"
[[tool.poetry.source]]
name = "PyPI"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.ruff.lint]
select = ["E4", "E7", "E9", "F", "E501"]
ignore = ["E731", "E203", "E741"]
[tool.pytest.ini_options]
python_classes = ["Test"]
python_functions = ["test_"]
python_files = ["test_*.py"]
testpaths = ["tests"]
markers = [
"no_hub_telemetry_mock"
]
[pytest]
log_cli = "True"
log_cli_level = "DEBUG"