diff --git a/langchain/retrievers/document_compressors/cohere_rerank.py b/langchain/retrievers/document_compressors/cohere_rerank.py index 7774c339..63060914 100644 --- a/langchain/retrievers/document_compressors/cohere_rerank.py +++ b/langchain/retrievers/document_compressors/cohere_rerank.py @@ -50,6 +50,8 @@ class CohereRerank(BaseDocumentCompressor): def compress_documents( self, documents: Sequence[Document], query: str ) -> Sequence[Document]: + if len(documents) == 0: # to avoid empty api call + return [] doc_list = list(documents) _docs = [d.page_content for d in doc_list] results = self.client.rerank(