pydantic>=2.0.0
pyyaml>=6.0
typer>=0.9.0
rich>=13.0.0
numpy>=1.24.0
aiohttp>=3.9.0

[all]
bns-nlp-engine[api,cohere,faiss,huggingface,openai,pinecone,qdrant]

[api]
fastapi>=0.104.0
uvicorn>=0.24.0

[cohere]
cohere>=4.0.0

[dev]
pytest>=7.4.0
pytest-asyncio>=0.21.0
pytest-cov>=4.1.0
pytest-mock>=3.12.0
black>=23.0.0
isort>=5.12.0
ruff>=0.1.0
mypy>=1.7.0
pre-commit>=3.5.0
tox>=4.11.0

[docs]
mkdocs>=1.5.0
mkdocs-material>=9.4.0
mkdocstrings[python]>=0.24.0

[faiss]
faiss-cpu>=1.7.4

[huggingface]
transformers>=4.30.0
torch>=2.0.0
sentence-transformers>=2.2.0

[openai]
openai>=1.0.0

[pinecone]
pinecone-client>=3.0.0

[qdrant]
qdrant-client>=1.7.0
