pip>=25.3
pydantic
openai
jinja2>=3.1.6
transformers>=4.30.0
scikit-learn
tritonclient[grpc]>=2.60.0
gector-triton
fastapi>=0.128.0
errant>=3.0.0
uvicorn
grpcio
grpcio-tools
protobuf>=5.29.5
torch
numpy
sentencepiece>=0.2.1

[all]
grammared-language[dev,test,triton]

[dev]
pytest>=7.0.0
pytest-asyncio>=0.21.0
pytest-cov>=4.0.0
pytest-mock>=3.10.0
black>=23.0.0
isort>=5.12.0
mypy>=1.0.0
ruff>=0.1.0
pre-commit>=3.0.0
datasets

[docker]
pydantic==2.12.5
openai==2.15.0
jinja2==3.1.6
transformers==4.57.6
scikit-learn==1.8.0
tritonclient[grpc]==2.64.0
gector-triton
fastapi==0.128.0
errant==3.0.0
spacy==3.8.11
uvicorn==0.40.0
grpcio==1.67.1
grpcio-tools==1.67.1
numpy==1.26.4

[test]
pytest>=7.0.0
pytest-asyncio>=0.21.0
pytest-mock>=3.10.0
requests>=2.28.0

[triton]
torch
transformers[torch]==4.57.6
optimum[onnxruntime-gpu]
bitsandbytes
accelerate

[triton-arm]
transformers[torch]==4.57.6
optimum[onnxruntime]
bitsandbytes
accelerate
torch==2.10.0+cpu
