torch>=2.1.0
numpy>=1.24.0
tokenizers>=0.15.0
PyYAML>=6.0
portalocker>=2.8
requests>=2.28

[ann]
faiss-cpu>=1.7.0

[bench]
soma[ann,dev,metrics,sbert]
chromadb>=0.5

[chroma]
chromadb>=0.5

[data]
datasets>=2.14.0

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.4.0
mypy>=1.8.0
types-PyYAML>=6.0

[dev-chat]
transformers>=4.40
accelerate>=0.30
safetensors>=0.4

[gcs]
google-cloud-storage>=2.10

[gcs-test]
gcp-storage-emulator>=2024.8

[gguf]
llama-cpp-python>=0.2.70

[lancedb]
lancedb>=0.11

[langchain]
langchain-core>=0.1.0

[llamaindex]
llama-index-core>=0.10.0

[metrics]
prometheus-fastapi-instrumentator>=7.1
prometheus-client>=0.20

[multimodal]
torchvision>=0.16.0
torchaudio>=2.1.0

[otel]
opentelemetry-instrumentation-fastapi>=0.50
opentelemetry-sdk>=1.30
opentelemetry-exporter-otlp>=1.30

[pgvector]
psycopg[binary]>=3
pgvector>=0.3

[pgvector-test]
testcontainers>=4
psycopg[binary]>=3
pgvector>=0.3

[qdrant]
qdrant-client>=1.10

[qdrant-test]
testcontainers>=4
qdrant-client>=1.10

[quant]
bitsandbytes>=0.43.0

[redis-revocation]
redis>=5.0

[s3]
boto3>=1.34

[s3-test]
moto[s3]>=5

[sbert]
sentence-transformers>=2.2.0

[serve]
fastapi>=0.100
uvicorn>=0.20
sentence-transformers>=2.2.0
pyjwt[crypto]>=2.8

[ui]
dearpygui>=1.10.0
matplotlib>=3.7.0
networkx>=3.1

[viz]
matplotlib>=3.7.0
networkx>=3.1
