[project]
name = "vrag"
version = "0.1.0"
description = "Multimodal video RAG over YouTube videos with EmbeddingGemma 2 + Gemma 4 12B"
readme = "README.md"
requires-python = ">=3.12"
dependencies = ["fastapi==0.143.0", "uvicorn==0.54.0", "numpy==1.26.4", "pillow==12.3.0", "python-multipart==0.0.32", "httpx==0.28.1", "pydantic==2.14.0", "PyYAML==6.0.3"]

[project.optional-dependencies]
offline = ["av", "soundfile", "librosa", "yt-dlp[default,curl-cffi]", "webvtt-py", "jiwer", "openai>=1.40", "rich", "typer"]
legacy-local = ["torch>=2.11", "torchvision", "torchaudio", "transformers>=5.19.0", "sentence-transformers>=6.1.0", "torchcodec", "accelerate"]

[tool.uv]
package = true

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

[tool.uv.sources]
torch = { index = "pytorch-cu130" }
torchvision = { index = "pytorch-cu130" }
torchaudio = { index = "pytorch-cu130" }
torchcodec = { index = "pytorch-cu130" }

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["vrag", "config"]

[project.scripts]
vrag = "vrag.cli:app"

[dependency-groups]
dev = ["pytest==9.1.1", "pytest-asyncio==1.4.0", "psutil==7.2.2"]