mteb[beir]==1.1.1
pydantic>=2.1.0
catalogue>=2.0.8
radicli>=0.0.25
datasets<2.20.0
fsspec<=2023.9.2

[arctic]
xformers>=0.0.10

[cohere]
cohere>=4.34

[dev]
cruft>=2.0.0
pyright==1.1.348
ruff==0.6.0

[docs]
mkdocs==1.5.1
mkdocs-jupyter==0.24.6
mkdocs-material==9.1.21
mkdocstrings[python-legacy]==0.22.0
datawrapper<0.6.0,>=0.5.3
jupyter>=1.0.0

[fasttext]
pybind11>=2.11.0
fasttext-wheel>=0.9.0

[jina]
einops
numpy<2.0.0

[llm2vec]
ninja>=1.10.0
llm2vec>=0.2.2

[llm2vec:sys_platform == "linux"]
flash-attn>=2.0.1

[mistral]
transformers>=4.31.0

[openai]
openai>=0.27.4

[sonar]
fairseq2>=0.1.0
sonar-space>=0.2.1

[tests]
pytest>=7.1.3
pytest-cov>=3.0.0

[voyageai]
voyageai>=0.2.3
