langchain[patch]: Migrate chat loaders to optional community imports (#21089)

Migrate chat loaders to optional community imports

libs/langchain/langchain/chat_loaders/gmail.py

@@ -1,5 +1,23 @@
-from langchain_community.chat_loaders.gmail import (
+from typing import TYPE_CHECKING, Any
-    GMailLoader,
-)
 
-__all__ = ["GMailLoader"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.chat_loaders.gmail import GMailLoader
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {"GMailLoader": "langchain_community.chat_loaders.gmail"}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "GMailLoader",
+]

libs/langchain/langchain/chat_loaders/imessage.py

@@ -1,3 +1,23 @@
+from typing import TYPE_CHECKING, Any
+
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
     from langchain_community.chat_loaders.imessage import IMessageChatLoader
 
-__all__ = ["IMessageChatLoader"]
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {"IMessageChatLoader": "langchain_community.chat_loaders.imessage"}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "IMessageChatLoader",
+]

libs/langchain/langchain/chat_loaders/langsmith.py

@@ -1,6 +1,30 @@
+from typing import TYPE_CHECKING, Any
+
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
     from langchain_community.chat_loaders.langsmith import (
         LangSmithDatasetChatLoader,
         LangSmithRunChatLoader,
     )
 
-__all__ = ["LangSmithRunChatLoader", "LangSmithDatasetChatLoader"]
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "LangSmithRunChatLoader": "langchain_community.chat_loaders.langsmith",
+    "LangSmithDatasetChatLoader": "langchain_community.chat_loaders.langsmith",
+}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "LangSmithRunChatLoader",
+    "LangSmithDatasetChatLoader",
+]

libs/langchain/langchain/chat_loaders/slack.py

@@ -1,3 +1,23 @@
+from typing import TYPE_CHECKING, Any
+
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
     from langchain_community.chat_loaders.slack import SlackChatLoader
 
-__all__ = ["SlackChatLoader"]
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {"SlackChatLoader": "langchain_community.chat_loaders.slack"}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "SlackChatLoader",
+]

libs/langchain/langchain/chat_loaders/telegram.py

@@ -1,3 +1,23 @@
+from typing import TYPE_CHECKING, Any
+
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
     from langchain_community.chat_loaders.telegram import TelegramChatLoader
 
-__all__ = ["TelegramChatLoader"]
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {"TelegramChatLoader": "langchain_community.chat_loaders.telegram"}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "TelegramChatLoader",
+]

libs/langchain/langchain/chat_loaders/utils.py

@@ -1,3 +1,8 @@
+from typing import TYPE_CHECKING, Any
+
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
     from langchain_community.chat_loaders.utils import (
         map_ai_messages,
         map_ai_messages_in_session,
@@ -5,6 +10,24 @@ from langchain_community.chat_loaders.utils import (
         merge_chat_runs_in_session,
     )
 
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "merge_chat_runs_in_session": "langchain_community.chat_loaders.utils",
+    "merge_chat_runs": "langchain_community.chat_loaders.utils",
+    "map_ai_messages_in_session": "langchain_community.chat_loaders.utils",
+    "map_ai_messages": "langchain_community.chat_loaders.utils",
+}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
 __all__ = [
     "merge_chat_runs_in_session",
     "merge_chat_runs",

libs/langchain/langchain/chat_loaders/whatsapp.py

@@ -1,3 +1,23 @@
+from typing import TYPE_CHECKING, Any
+
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
     from langchain_community.chat_loaders.whatsapp import WhatsAppChatLoader
 
-__all__ = ["WhatsAppChatLoader"]
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {"WhatsAppChatLoader": "langchain_community.chat_loaders.whatsapp"}
+
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "WhatsAppChatLoader",
+]