[project]
name = "pool"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
   "accelerate>=1.13.0",
   "bitsandbytes>=0.49.2; sys_platform == 'linux' and platform_machine == 'x86_64'",
   "chromadb>=0.5.0",
   "datasets>=4.8.4",
   "fastapi>=0.135.2",
   "gradio>=4.0.0",
   "huggingface-hub>=1.8.0",
   "peft>=0.18.1",
   "redis>=5.0.0",
   "pymongo>=4.6.0",
   "motor>=3.4.0",
   "minio>=7.2.0",
   "python-multipart>=0.0.9",
   "sentence-transformers>=3.0.0",
   "torch>=2.5.0",
   "torchaudio>=2.5.0",
   "transformers>=5.4.0",
   "trl>=0.29.1",
   "uvicorn>=0.42.0",
   "torchvision>=0.20.0",
   "wrapt>=2.1.2",
]

[tool.uv]
# Resolve and lock for both:
# - local dev on macOS (Apple Silicon)
# - Hugging Face Spaces runtime (Linux x86_64)
environments = [
   "sys_platform == 'darwin' and platform_machine == 'arm64'",
   "sys_platform == 'linux' and platform_machine == 'x86_64'",
]

[tool.uv.sources]
# Use CUDA-enabled PyTorch wheels on Linux; other platforms will use the default index (PyPI).
torch = [{ index = "pytorch-cu121", marker = "sys_platform == 'linux'" }]
torchvision = [{ index = "pytorch-cu121", marker = "sys_platform == 'linux'" }]
torchaudio = [{ index = "pytorch-cu121", marker = "sys_platform == 'linux'" }]

[[tool.uv.index]]
name = "pytorch-cu121"
url = "https://download.pytorch.org/whl/cu121"
explicit = true