langchain/libs/partners/ai21/tests/unit_tests/test_imports.py
miri-bar 55db737302
ai21[minor]: AI21 Labs Semantic Text Splitter support (#19510)
Description: Added support for AI21 Labs model - Segmentation, as a Text
Splitter
Dependencies: ai21, langchain-text-splitter
Twitter handle: https://github.com/AI21Labs

---------

Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
Co-authored-by: Bagatur <baskaryan@gmail.com>
2024-03-26 01:39:37 +00:00

14 lines
254 B
Python

from langchain_ai21 import __all__
EXPECTED_ALL = [
"AI21LLM",
"ChatAI21",
"AI21Embeddings",
"AI21ContextualAnswers",
"AI21SemanticTextSplitter",
]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)