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", ]