From eff1336b821c07c7094522f639263d5a94cc34bf Mon Sep 17 00:00:00 2001 From: rennokki Date: Fri, 21 Jun 2024 18:52:54 +0300 Subject: [PATCH] update: VoyageAI Models (#2688) * Updated VoyageAI Models * Update VoyageAI Rerankers * Update VoyageAIRerankRetriever.ts --------- Co-authored-by: Henry Heng --- packages/components/models.json | 26 ++++++++++++++++--- .../VoyageAIRerankRetriever.ts | 4 +++ 2 files changed, 27 insertions(+), 3 deletions(-) diff --git a/packages/components/models.json b/packages/components/models.json index 55f83d410..ff3983495 100644 --- a/packages/components/models.json +++ b/packages/components/models.json @@ -1047,22 +1047,42 @@ { "label": "voyage-2", "name": "voyage-2", - "description": "Base generalist embedding model optimized for both latency and quality" + "description": "General-purpose embedding model optimized for a balance between cost, latency, and retrieval quality." }, { "label": "voyage-code-2", "name": "voyage-code-2", - "description": "Optimized for code retrieval" + "description": "Optimized for code retrieval." + }, + { + "label": "voyage-finance-2", + "name": "voyage-finance-2", + "description": "Optimized for finance retrieval and RAG." }, { "label": "voyage-large-2", "name": "voyage-large-2", - "description": "Powerful generalist embedding model" + "description": "General-purpose embedding model that is optimized for retrieval quality." + }, + { + "label": "voyage-large-2-instruct", + "name": "voyage-large-2-instruct", + "description": "Instruction-tuned general-purpose embedding model optimized for clustering, classification, and retrieval." + }, + { + "label": "voyage-law-2", + "name": "voyage-law-2", + "description": "Optimized for legal and long-context retrieval and RAG. Also improved performance across all domains." }, { "label": "voyage-lite-02-instruct", "name": "voyage-lite-02-instruct", "description": "Instruction-tuned for classification, clustering, and sentence textual similarity tasks" + }, + { + "label": "voyage-multilingual-2", + "name": "voyage-multilingual-2", + "description": "Optimized for multilingual retrieval and RAG." } ] }, diff --git a/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts b/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts index b5687b4a8..4fa480314 100644 --- a/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts +++ b/packages/components/nodes/retrievers/VoyageAIRetriever/VoyageAIRerankRetriever.ts @@ -49,6 +49,10 @@ class VoyageAIRerankRetriever_Retrievers implements INode { { label: 'rerank-lite-1', name: 'rerank-lite-1' + }, + { + label: 'rerank-1', + name: 'rerank-1' } ], default: 'rerank-lite-1',