Adding VoyageAI's text_embedding to 'integrations/text_embedding/'

langchain-ai · Dec 19, 2024 · d3e02ed · d3e02ed
1 parent a9782e1
commit d3e02ed
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/docs/src/theme/EmbeddingTabs.js b/docs/src/theme/EmbeddingTabs.js
@@ -25,11 +25,13 @@ export default function EmbeddingTabs(props) {
       hideNomic,
       nvidiaParams,
       hideNvidia,
+      voyageaiParams,
+      hideVoyageai,
       fakeEmbeddingParams,
       hideFakeEmbedding,
       customVarName,
     } = props;
-  
+
     const openAIParamsOrDefault = openaiParams ?? `model="text-embedding-3-large"`;
     const azureParamsOrDefault =
       azureOpenaiParams ??
@@ -42,10 +44,11 @@ export default function EmbeddingTabs(props) {
     const mistralParamsOrDefault = mistralParams ?? `model="mistral-embed"`;
     const nomicsParamsOrDefault = nomicParams ?? `model="nomic-embed-text-v1.5"`;
     const nvidiaParamsOrDefault = nvidiaParams ?? `model="NV-Embed-QA"`;
+    const voyageaiParamsOrDefault = voyageaiParams ?? `model="voyage-3"`;
     const fakeEmbeddingParamsOrDefault = fakeEmbeddingParams ?? `size=4096`;
-  
+
     const embeddingVarName = customVarName ?? "embeddings";
-  
+
     const tabItems = [
       {
         value: "OpenAI",
@@ -137,6 +140,15 @@ export default function EmbeddingTabs(props) {
         default: false,
         shouldHide: hideNvidia,
       },
+      {
+        value: "Voyage AI",
+        label: "Voyage AI",
+        text: `from langchain-voyageai import VoyageAIEmbeddings\n\n${embeddingVarName} = VoyageAIEmbeddings(${voyageaiParamsOrDefault})`,
+        apiKeyName: "VOYAGE_API_KEY",
+        packageName: "langchain-voyageai",
+        default: false,
+        shouldHide: hideVoyageai,
+      },
       {
         value: "Fake",
         label: "Fake",