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":