VoyageAI Reranker optional API Key (#17310)

run-llama · Dec 18, 2024 · 8ecc02a · 8ecc02a
1 parent a909a65
commit 8ecc02a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/...ama-index-postprocessor-voyageai-rerank/llama_index/postprocessor/voyageai_rerank/base.py b/...ama-index-postprocessor-voyageai-rerank/llama_index/postprocessor/voyageai_rerank/base.py
@@ -26,8 +26,8 @@ class VoyageAIRerank(BaseNodePostprocessor):
 
     def __init__(
         self,
-        api_key: str,
         model: str,
+        api_key: Optional[str] = None,
         top_n: Optional[int] = None,
         truncation: Optional[bool] = None,
         # deprecated

diff --git a/...index-integrations/postprocessor/llama-index-postprocessor-voyageai-rerank/pyproject.toml b/...index-integrations/postprocessor/llama-index-postprocessor-voyageai-rerank/pyproject.toml
@@ -30,11 +30,11 @@ license = "MIT"
 name = "llama-index-postprocessor-voyageai-rerank"
 packages = [{include = "llama_index/"}]
 readme = "README.md"
-version = "0.3.0"
+version = "0.3.1"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
-voyageai = "^0.2.1"
+voyageai = {python = ">=3.9,<3.13", version = ">=0.3.2,<0.4.0"}
 llama-index-core = "^0.12.0"
 
 [tool.poetry.group.dev.dependencies]