diff --git a/setup.py b/setup.py index e43bf3ed0..1d4301113 100644 --- a/setup.py +++ b/setup.py @@ -40,6 +40,9 @@ extras_require = { "llama-index-vector-stores-faiss==0.1.1", "llama-index-vector-stores-elasticsearch==0.1.6", "llama-index-vector-stores-chroma==0.1.6", + "llama-index-postprocessor-cohere-rerank==0.1.4", + "llama-index-postprocessor-colbert-rerank==0.1.1", + "llama-index-postprocessor-flag-embedding-reranker==0.1.2", "docx2txt==0.8", ], "android_assistant": ["pyshine==0.0.9", "opencv-python==4.6.0.66"],