fix(cohere,langchain): handle exceptions in compressDocuments and for…

…matDocuments methods due to empty documents (#7372)
langchain-ai · Dec 24, 2024 · 9039484 · 9039484
1 parent a51bb4a
commit 9039484
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/langchain/src/chains/combine_documents/base.ts b/langchain/src/chains/combine_documents/base.ts
@@ -21,6 +21,9 @@ export async function formatDocuments({
   documents: Document[];
   config?: RunnableConfig;
 }) {
+  if (documents == null || documents.length === 0) {
+    return "";
+  }
   const formattedDocs = await Promise.all(
     documents.map((document) =>
       documentPrompt

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,