langchain/libs/partners/ollama/tests/unit_tests/test_imports.py
Isaac Francisco 838464de25
ollama: init package (#23615)
Co-authored-by: Erick Friis <erick@langchain.dev>
2024-07-20 00:43:29 +00:00

13 lines
220 B
Python

from langchain_ollama import __all__
EXPECTED_ALL = [
"OllamaLLM",
"ChatOllama",
"OllamaEmbeddings",
"__version__",
]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)