pydantic<3.0.0,>=2.5.0
pydantic-settings<3.0.0,>=2.1.0
aiofiles<24.0.0,>=23.2.0
httpx<1.0.0,>=0.25.0
redis<6.0.0,>=5.0.0
prometheus-client<1.0.0,>=0.19.0
python-dotenv<2.0.0,>=1.0.0
tenacity<9.0.0,>=8.2.0
cryptography<44.0.0,>=41.0.0
orjson<4.0.0,>=3.9.0

[all]
deepcompress[aws,dev,gpu,llm,vector-db]

[all-flash]
deepcompress[aws,dev,gpu-flash,llm,vector-db]

[aws]
boto3<2.0.0,>=1.34.0
aioboto3<13.0.0,>=12.3.0

[colab]
tokenizers<1.0.0,>=0.19.0
pillow>=10.0.0
pdf2image>=1.17.0
accelerate>=0.20.0
sentencepiece>=0.1.99
addict>=2.4.0
tqdm>=4.65.0

[dev]
pytest>=7.4.0
pytest-asyncio>=0.21.0
pytest-cov>=4.1.0
pytest-mock>=3.12.0
black>=23.12.0
ruff>=0.1.8
mypy>=1.7.0
pre-commit>=3.6.0

[gpu]
torch<3.0.0,>=2.0.0
transformers<5.0.0,>=4.38.0
tokenizers<1.0.0,>=0.19.0
pillow>=10.0.0
pdf2image>=1.17.0
accelerate>=0.20.0
sentencepiece>=0.1.99
addict>=2.4.0
tqdm>=4.65.0

[gpu-flash]
deepcompress[gpu]

[gpu-flash:platform_machine == "x86_64"]
flash-attn<2.8.0,>=2.3.0

[llm]
openai<2.0.0,>=1.6.0
anthropic<1.0.0,>=0.8.0

[vector-db]
pinecone-client<4.0.0,>=3.0.0
weaviate-client<5.0.0,>=4.4.0
