Added voyage-3 and voyage-3-lite. (#2394)

Mintplex-Labs · Sep 30, 2024 · e075354 · e075354
1 parent cadc09d
commit e075354
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx b/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx
@@ -36,6 +36,8 @@ export default function VoyageAiOptions({ settings }) {
                 "voyage-code-2",
                 "voyage-large-2",
                 "voyage-2",
+                "voyage-3",
+                "voyage-3-lite",
               ].map((model) => {
                 return (
                   <option key={model} value={model}>

diff --git a/server/utils/EmbeddingEngines/voyageAi/index.js b/server/utils/EmbeddingEngines/voyageAi/index.js
@@ -11,7 +11,7 @@ class VoyageAiEmbedder {
     });
 
     this.voyage = voyage;
-    this.model = process.env.EMBEDDING_MODEL_PREF || "voyage-large-2-instruct";
+    this.model = process.env.EMBEDDING_MODEL_PREF || "voyage-3-lite";
 
     // Limit of how many strings we can process in a single pass to stay with resource or network limits
     this.batchSize = 128; // Voyage AI's limit per request is 128 https://docs.voyageai.com/docs/rate-limits#use-larger-batches
@@ -23,6 +23,8 @@ class VoyageAiEmbedder {
     switch (this.model) {
       case "voyage-finance-2":
       case "voyage-multilingual-2":
+      case "voyage-3":
+      case "voyage-3-lite":
         return 32_000;
       case "voyage-large-2-instruct":
       case "voyage-law-2":