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 langchain_community.chat_loaders.imessage import IMessageChatLoader
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["IMessageChatLoader"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.chat_loaders.imessage import 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 langchain_community.chat_loaders.langsmith import (
+from typing import TYPE_CHECKING, Any
-    LangSmithDatasetChatLoader,
-    LangSmithRunChatLoader,
-)
 
-__all__ = ["LangSmithRunChatLoader", "LangSmithDatasetChatLoader"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.chat_loaders.langsmith import (
+        LangSmithDatasetChatLoader,
+        LangSmithRunChatLoader,
+    )
+
+# 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 langchain_community.chat_loaders.slack import SlackChatLoader
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["SlackChatLoader"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.chat_loaders.slack import 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 langchain_community.chat_loaders.telegram import TelegramChatLoader
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["TelegramChatLoader"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.chat_loaders.telegram import 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,9 +1,32 @@
-from langchain_community.chat_loaders.utils import (
+from typing import TYPE_CHECKING, Any
-    map_ai_messages,
+
-    map_ai_messages_in_session,
+from langchain._api import create_importer
-    merge_chat_runs,
+
-    merge_chat_runs_in_session,
+if TYPE_CHECKING:
-)
+    from langchain_community.chat_loaders.utils import (
+        map_ai_messages,
+        map_ai_messages_in_session,
+        merge_chat_runs,
+        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",

libs/langchain/langchain/chat_loaders/whatsapp.py

@@ -1,3 +1,23 @@
-from langchain_community.chat_loaders.whatsapp import WhatsAppChatLoader
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["WhatsAppChatLoader"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.chat_loaders.whatsapp import 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",
+]