From 86ba16acdc38efaaba0479cd4d969674977ab01c Mon Sep 17 00:00:00 2001
From: Sean Hatfield <seanhatfield5@gmail.com>
Date: Thu, 29 Aug 2024 14:11:00 -0700
Subject: [PATCH] Add new Voyage AI embedding models (#2193)

new voyageai embedding models
---
 .../components/EmbeddingSelection/VoyageAiOptions/index.jsx    | 2 ++
 server/utils/EmbeddingEngines/voyageAi/index.js                | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx b/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx
index ca5b53f31..252cb0a7b 100644
--- a/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx
+++ b/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx
@@ -30,6 +30,8 @@ export default function VoyageAiOptions({ settings }) {
             <optgroup label="Available embedding models">
               {[
                 "voyage-large-2-instruct",
+                "voyage-finance-2",
+                "voyage-multilingual-2",
                 "voyage-law-2",
                 "voyage-code-2",
                 "voyage-large-2",
diff --git a/server/utils/EmbeddingEngines/voyageAi/index.js b/server/utils/EmbeddingEngines/voyageAi/index.js
index 65126613b..7f284fb49 100644
--- a/server/utils/EmbeddingEngines/voyageAi/index.js
+++ b/server/utils/EmbeddingEngines/voyageAi/index.js
@@ -21,6 +21,9 @@ class VoyageAiEmbedder {
   // https://docs.voyageai.com/docs/embeddings
   #getMaxEmbeddingLength() {
     switch (this.model) {
+      case "voyage-finance-2":
+      case "voyage-multilingual-2":
+        return 32_000;
       case "voyage-large-2-instruct":
       case "voyage-law-2":
       case "voyage-code-2":