From bd37ee26f4a20f7d73922ef61a030634eb7fdd2a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?B=C5=82a=C5=BCej=20Owczarczyk?= <blazeyy@gmail.com>
Date: Sun, 29 Sep 2024 23:39:54 +0200
Subject: [PATCH] Added voyage-3 and voyage-3-lite.

---
 .../components/EmbeddingSelection/VoyageAiOptions/index.jsx   | 2 ++
 server/utils/EmbeddingEngines/voyageAi/index.js               | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx b/frontend/src/components/EmbeddingSelection/VoyageAiOptions/index.jsx
index 252cb0a7b86..b55fc674365 100644
--- 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
index 7f284fb4959..f1527294308 100644
--- 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":