| from bertopic._utils import NotInstalled |
| from bertopic.representation._cohere import Cohere |
| from bertopic.representation._base import BaseRepresentation |
| from bertopic.representation._keybert import KeyBERTInspired |
| from bertopic.representation._mmr import MaximalMarginalRelevance |
|
|
|
|
| |
| try: |
| from bertopic.representation._llamacpp import LlamaCPP |
| except ModuleNotFoundError: |
| msg = "`pip install llama-cpp-python` \n\n" |
| LlamaCPP = NotInstalled("llama.cpp", "llama-cpp-python", custom_msg=msg) |
|
|
| |
| try: |
| from bertopic.representation._textgeneration import TextGeneration |
| except ModuleNotFoundError: |
| msg = "`pip install bertopic` without `--no-deps` \n\n" |
| TextGeneration = NotInstalled("TextGeneration", "transformers", custom_msg=msg) |
|
|
| |
| try: |
| from bertopic.representation._zeroshot import ZeroShotClassification |
| except ModuleNotFoundError: |
| msg = "`pip install bertopic` without `--no-deps` \n\n" |
| ZeroShotClassification = NotInstalled("ZeroShotClassification", "transformers", custom_msg=msg) |
|
|
| |
| try: |
| from bertopic.representation._openai import OpenAI |
| except ModuleNotFoundError: |
| msg = "`pip install openai` \n\n" |
| OpenAI = NotInstalled("OpenAI", "openai", custom_msg=msg) |
|
|
| |
| try: |
| from bertopic.representation._langchain import LangChain |
| except ModuleNotFoundError: |
| msg = "`pip install langchain` \n\n" |
| LangChain = NotInstalled("langchain", "langchain", custom_msg=msg) |
|
|
| |
| try: |
| from bertopic.representation._pos import PartOfSpeech |
| except ModuleNotFoundError: |
| PartOfSpeech = NotInstalled("Part of Speech with Spacy", "spacy") |
|
|
| |
| try: |
| from bertopic.representation._visual import VisualRepresentation |
| except ModuleNotFoundError: |
| VisualRepresentation = NotInstalled("a visual representation model", "vision") |
|
|
|
|
| __all__ = [ |
| "BaseRepresentation", |
| "TextGeneration", |
| "ZeroShotClassification", |
| "KeyBERTInspired", |
| "PartOfSpeech", |
| "MaximalMarginalRelevance", |
| "Cohere", |
| "OpenAI", |
| "LangChain", |
| "LlamaCPP", |
| "VisualRepresentation" |
| ] |
|
|
