File size: 958 Bytes
19b102a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
from ._base import BaseEmbedder
from ._word_doc import WordDocEmbedder
from ._utils import languages
from bertopic._utils import NotInstalled
# OpenAI Embeddings
try:
from bertopic.backend._openai import OpenAIBackend
except ModuleNotFoundError:
msg = "`pip install openai` \n\n"
OpenAIBackend = NotInstalled("OpenAI", "OpenAI", custom_msg=msg)
# Cohere Embeddings
try:
from bertopic.backend._cohere import CohereBackend
except ModuleNotFoundError:
msg = "`pip install cohere` \n\n"
CohereBackend = NotInstalled("Cohere", "Cohere", custom_msg=msg)
# Multimodal Embeddings
try:
from bertopic.backend._multimodal import MultiModalBackend
except ModuleNotFoundError:
msg = "`pip install bertopic[vision]` \n\n"
MultiModalBackend = NotInstalled("Vision", "Vision", custom_msg=msg)
__all__ = [
"BaseEmbedder",
"WordDocEmbedder",
"OpenAIBackend",
"CohereBackend",
"MultiModalBackend",
"languages"
]
|