Fixes issue #7370: return empty array from compressDocuments if the d…

…ocuments parameter is null or empty
langchain-ai · Dec 16, 2024 · e94d615 · e94d615
1 parent 538ea0d
commit e94d615
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/langchain-cohere/src/rerank.ts b/libs/langchain-cohere/src/rerank.ts
@@ -60,6 +60,9 @@ export class CohereRerank extends BaseDocumentCompressor {
     documents: Array<DocumentInterface>,
     query: string
   ): Promise<Array<DocumentInterface>> {
+    if (documents == null || documents.length == 0) {
+      return [];
+    }
     const _docs = documents.map((doc) => doc.pageContent);
     const { results } = await this.client.rerank({
       model: this.model,