Add new Voyage AI embedding models (#2193)

new voyageai embedding models
Mintplex-Labs · Aug 29, 2024 · 86ba16a · 86ba16a
1 parent bc0cc20
commit 86ba16a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git 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
@@ -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":