mirror of
https://github.com/hwchase17/langchain
synced 2024-11-02 09:40:22 +00:00
36813d2f00
Fixes the __init__ files in community to use __all__ which is statically defined.
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
"""**Chat Loaders** load chat messages from common communications platforms.
|
|
|
|
Load chat messages from various
|
|
communications platforms such as Facebook Messenger, Telegram, and
|
|
WhatsApp. The loaded chat messages can be used for fine-tuning models.
|
|
|
|
**Class hierarchy:**
|
|
|
|
.. code-block::
|
|
|
|
BaseChatLoader --> <name>ChatLoader # Examples: WhatsAppChatLoader, IMessageChatLoader
|
|
|
|
**Main helpers:**
|
|
|
|
.. code-block::
|
|
|
|
ChatSession
|
|
|
|
""" # noqa: E501
|
|
|
|
import importlib
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
if TYPE_CHECKING:
|
|
from langchain_community.chat_loaders.base import (
|
|
BaseChatLoader,
|
|
)
|
|
from langchain_community.chat_loaders.facebook_messenger import (
|
|
FolderFacebookMessengerChatLoader,
|
|
SingleFileFacebookMessengerChatLoader,
|
|
)
|
|
from langchain_community.chat_loaders.gmail import (
|
|
GMailLoader,
|
|
)
|
|
from langchain_community.chat_loaders.imessage import (
|
|
IMessageChatLoader,
|
|
)
|
|
from langchain_community.chat_loaders.langsmith import (
|
|
LangSmithDatasetChatLoader,
|
|
LangSmithRunChatLoader,
|
|
)
|
|
from langchain_community.chat_loaders.slack import (
|
|
SlackChatLoader,
|
|
)
|
|
from langchain_community.chat_loaders.telegram import (
|
|
TelegramChatLoader,
|
|
)
|
|
from langchain_community.chat_loaders.whatsapp import (
|
|
WhatsAppChatLoader,
|
|
)
|
|
|
|
__all__ = [
|
|
"BaseChatLoader",
|
|
"FolderFacebookMessengerChatLoader",
|
|
"GMailLoader",
|
|
"IMessageChatLoader",
|
|
"LangSmithDatasetChatLoader",
|
|
"LangSmithRunChatLoader",
|
|
"SingleFileFacebookMessengerChatLoader",
|
|
"SlackChatLoader",
|
|
"TelegramChatLoader",
|
|
"WhatsAppChatLoader",
|
|
]
|
|
|
|
_module_lookup = {
|
|
"BaseChatLoader": "langchain_core.chat_loaders",
|
|
"FolderFacebookMessengerChatLoader": "langchain_community.chat_loaders.facebook_messenger", # noqa: E501
|
|
"GMailLoader": "langchain_community.chat_loaders.gmail",
|
|
"IMessageChatLoader": "langchain_community.chat_loaders.imessage",
|
|
"LangSmithDatasetChatLoader": "langchain_community.chat_loaders.langsmith",
|
|
"LangSmithRunChatLoader": "langchain_community.chat_loaders.langsmith",
|
|
"SingleFileFacebookMessengerChatLoader": "langchain_community.chat_loaders.facebook_messenger", # noqa: E501
|
|
"SlackChatLoader": "langchain_community.chat_loaders.slack",
|
|
"TelegramChatLoader": "langchain_community.chat_loaders.telegram",
|
|
"WhatsAppChatLoader": "langchain_community.chat_loaders.whatsapp",
|
|
}
|
|
|
|
|
|
def __getattr__(name: str) -> Any:
|
|
if name in _module_lookup:
|
|
module = importlib.import_module(_module_lookup[name])
|
|
return getattr(module, name)
|
|
raise AttributeError(f"module {__name__} has no attribute {name}")
|