From e4d7b1a482a640bc50c363bf6826eaab42b5ea5b Mon Sep 17 00:00:00 2001 From: Jaid Date: Thu, 28 Mar 2024 12:22:55 +0545 Subject: [PATCH] voyageai[patch]: top level reranker import (#19645) The previous version didn't had Voyage rerank in the init file - [ ] **PR title**: langchain_voyageai reranker is not working - [ ] **PR message**: - **Description:** This fix let you run reranker from voyage - **Issue:** Was not able to run reranker from voyage @efriis --- libs/partners/voyageai/langchain_voyageai/__init__.py | 5 ++--- libs/partners/voyageai/tests/unit_tests/test_imports.py | 1 + 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/partners/voyageai/langchain_voyageai/__init__.py b/libs/partners/voyageai/langchain_voyageai/__init__.py index d79c441280..38dae75181 100644 --- a/libs/partners/voyageai/langchain_voyageai/__init__.py +++ b/libs/partners/voyageai/langchain_voyageai/__init__.py @@ -1,5 +1,4 @@ from langchain_voyageai.embeddings import VoyageAIEmbeddings +from langchain_voyageai.rerank import VoyageAIRerank -__all__ = [ - "VoyageAIEmbeddings", -] +__all__ = ["VoyageAIEmbeddings", "VoyageAIRerank"] diff --git a/libs/partners/voyageai/tests/unit_tests/test_imports.py b/libs/partners/voyageai/tests/unit_tests/test_imports.py index cfcdd93507..9b8977126a 100644 --- a/libs/partners/voyageai/tests/unit_tests/test_imports.py +++ b/libs/partners/voyageai/tests/unit_tests/test_imports.py @@ -2,6 +2,7 @@ from langchain_voyageai import __all__ EXPECTED_ALL = [ "VoyageAIEmbeddings", + "VoyageAIRerank", ]