tqdm>=4.64.0
numpy>=2.0.0

[all]
tokenizers>=0.16.0
tiktoken>=0.5.0
rich>=13.0.0
tree-sitter>=0.20.0
tree-sitter-language-pack>=0.7.0
magika<1.1.0,>=0.6.0
sentence-transformers>=3.0.0
openai>=1.0.0
model2vec>=0.3.0
cohere>=5.13.0
voyageai>=0.3.2
accelerate>=1.6.0
huggingface-hub>=0.24.0
jsonschema>=4.23.0
pydantic>=2.0.0
google-genai>=1.0.0
transformers>=4.0.0
torch<3.0,>=2.0.0
chromadb>=1.0.0
qdrant-client>=1.0.0
turbopuffer[fast]>=0.2.0
weaviate-client>=4.16.7
azure-identity>=1.23.0
pinecone
pymongo
vecs>=0.4.0
pandas
tabulate
openpyxl
elasticsearch>=8.0.0
pymilvus>=2.1.0

[azure-openai]
azure-identity>=1.23.0
tiktoken>=0.5.0
openai>=1.0.0
pydantic>=2.0.0

[chroma]
chromadb>=1.0.0

[code]
tree-sitter>=0.20.0
tree-sitter-language-pack>=0.7.0
magika<1.1.0,>=0.6.0

[cohere]
tokenizers>=0.16.0
cohere>=5.13.0

[datasets]
datasets>=4.0.0

[dev]
datasets>=1.14.0
transformers>=4.0.0
pytest>=6.2.0
pytest-cov>=4.0.0
pytest-xdist>=2.5.0
pytest-asyncio>=0.26.0
coverage
ruff>=0.0.265
mypy>=1.11.0
cython>=3.0.0
pandas
tabulate
openpyxl

[elastic]
elasticsearch>=8.0.0

[gemini]
pydantic>=2.0.0
google-genai>=1.0.0

[genie]
pydantic>=2.0.0
google-genai>=1.0.0

[hub]
huggingface-hub>=0.24.0
jsonschema>=4.23.0

[jina]
tokenizers>=0.16.0

[milvus]
pymilvus>=2.1.0

[model2vec]
tokenizers>=0.16.0
model2vec>=0.3.0

[mongodb]
pymongo

[neural]
transformers>=4.0.0
torch<3.0,>=2.0.0

[openai]
tiktoken>=0.5.0
openai>=1.0.0
pydantic>=2.0.0

[pgvector]
vecs>=0.4.0

[pinecone]
pinecone

[qdrant]
qdrant-client>=1.0.0

[semantic]
tokenizers>=0.16.0
model2vec>=0.3.0

[st]
tokenizers>=0.16.0
sentence-transformers>=3.0.0
accelerate>=1.6.0

[table]
pandas
tabulate
openpyxl

[tiktoken]
tiktoken>=0.5.0

[tokenizers]
tokenizers>=0.16.0

[tpuf]
turbopuffer[fast]>=0.2.0

[viz]
rich>=13.0.0

[voyageai]
tokenizers>=0.16.0
voyageai>=0.3.2

[weaviate]
weaviate-client>=4.16.7
