Introducing new VoyageAI models (#16150)

run-llama · Sep 22, 2024 · 6a8a441 · 6a8a441
1 parent 01da176
commit 6a8a441
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 6 deletions.
diff --git a/docs/docs/examples/embeddings/voyageai.ipynb b/docs/docs/examples/embeddings/voyageai.ipynb
@@ -59,7 +59,7 @@
    "source": [
     "# get API key and create embeddings\n",
     "\n",
-    "model_name = \"voyage-law-2\"  # Please check https://docs.voyageai.com/docs/embeddings for the available models\n",
+    "model_name = \"voyage-3\"  # Please check https://docs.voyageai.com/docs/embeddings for the available models\n",
     "voyage_api_key = os.environ.get(\"VOYAGE_API_KEY\", \"your-api-key\")\n",
     "\n",
     "embed_model = VoyageEmbedding(\n",

diff --git a/docs/docs/examples/node_postprocessor/VoyageAIRerank.ipynb b/docs/docs/examples/node_postprocessor/VoyageAIRerank.ipynb
@@ -102,7 +102,7 @@
     "\n",
     "api_key = os.environ[\"VOYAGE_API_KEY\"]\n",
     "voyageai_embeddings = VoyageEmbedding(\n",
-    "    voyage_api_key=api_key, model_name=\"voyage-large-2\"\n",
+    "    voyage_api_key=api_key, model_name=\"voyage-3\"\n",
     ")\n",
     "\n",
     "# load documents\n",
@@ -130,7 +130,7 @@
     "from llama_index.postprocessor.voyageai_rerank import VoyageAIRerank\n",
     "\n",
     "voyageai_rerank = VoyageAIRerank(\n",
-    "    api_key=api_key, top_k=2, model=\"rerank-lite-1\", truncation=True\n",
+    "    api_key=api_key, top_k=2, model=\"rerank-2\", truncation=True\n",
     ")"
    ]
   },

diff --git a/...ations/embeddings/llama-index-embeddings-voyageai/llama_index/embeddings/voyageai/base.py b/...ations/embeddings/llama-index-embeddings-voyageai/llama_index/embeddings/voyageai/base.py
@@ -36,9 +36,16 @@ def __init__(
         callback_manager: Optional[CallbackManager] = None,
         **kwargs: Any,
     ):
-        if model_name == "voyage-01":
+        if model_name in [
+            "voyage-01",
+            "voyage-lite-01",
+            "voyage-lite-01-instruct",
+            "voyage-02",
+            "voyage-2",
+            "voyage-lite-02-instruct",
+        ]:
             logger.warning(
-                "voyage-01 is not the latest model by Voyage AI. Please note that `model_name` "
+                f"{model_name} is not the latest model by Voyage AI. Please note that `model_name` "
                 "will be a required argument in the future. We recommend setting it explicitly. Please see "
                 "https://docs.voyageai.com/docs/embeddings for the latest models offered by Voyage AI."
             )

diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-voyageai/pyproject.toml b/llama-index-integrations/embeddings/llama-index-embeddings-voyageai/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-embeddings-voyageai"
 readme = "README.md"
-version = "0.2.1"
+version = "0.2.2"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"