From 9cceba2240e5a6de2e988c4e94c9e6f4ee23ea0f Mon Sep 17 00:00:00 2001 From: Hans Date: Thu, 2 Jan 2025 21:51:59 +0800 Subject: [PATCH] Add `rerank-2` and `rerank-2-lite` for Voyage AI Rerank Retriever (#3776) Add `rerank-2` --- .../VoyageAIRetriever/VoyageAIRerankRetriever.ts | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts b/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts index 8acf4dd6e..53f02187f 100644 --- a/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts +++ b/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts @@ -49,9 +49,17 @@ class VoyageAIRerankRetriever_Retrievers implements INode { label: 'rerank-lite-1', name: 'rerank-lite-1' }, + { + label: 'rerank-lite-2', + name: 'rerank-lite-2' + }, { label: 'rerank-1', name: 'rerank-1' + }, + { + label: 'rerank-2', + name: 'rerank-2' } ], default: 'rerank-lite-1',