langchain/libs/partners/ibm/tests/unit_tests/test_imports.py
Mateusz Szewczyk 75ffe51bbe
ibm: Add support for Embedding Models (#20647)
---------

Co-authored-by: Erick Friis <erick@langchain.dev>
2024-04-19 20:56:24 +00:00

8 lines
171 B
Python

from langchain_ibm import __all__
EXPECTED_ALL = ["WatsonxLLM", "WatsonxEmbeddings"]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)