langchain[patch]: Update docstore module to use optional imports from community (#21091)

libs/langchain/langchain/docstore/__init__.py

@@ -14,29 +14,34 @@ The **Docstore** is a simplified version of the Document Loader.
 
     Document, AddableMixin
 """
-import warnings
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from langchain_core._api import LangChainDeprecationWarning
+from langchain._api import create_importer
 
-from langchain.utils.interactive_env import is_interactive_env
+if TYPE_CHECKING:
+    from langchain_community.docstore.arbitrary_fn import DocstoreFn
+    from langchain_community.docstore.in_memory import InMemoryDocstore
+    from langchain_community.docstore.wikipedia import Wikipedia
 
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "DocstoreFn": "langchain_community.docstore.arbitrary_fn",
+    "InMemoryDocstore": "langchain_community.docstore.in_memory",
+    "Wikipedia": "langchain_community.docstore.wikipedia",
+}
 
-def __getattr__(name: str) -> Any:
-    from langchain_community import docstore
+_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
 
-    # If not in interactive env, raise warning.
-    if not is_interactive_env():
-        warnings.warn(
-            "Importing docstores from langchain is deprecated. Importing from "
-            "langchain will no longer be supported as of langchain==0.2.0. "
-            "Please import from langchain-community instead:\n\n"
-            f"`from langchain_community.docstore import {name}`.\n\n"
-            "To install langchain-community run `pip install -U langchain-community`.",
-            category=LangChainDeprecationWarning,
-        )
 
-    return getattr(docstore, name)
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
 
 
-__all__ = ["DocstoreFn", "InMemoryDocstore", "Wikipedia"]
+__all__ = [
+    "DocstoreFn",
+    "InMemoryDocstore",
+    "Wikipedia",
+]

libs/langchain/langchain/docstore/arbitrary_fn.py

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

libs/langchain/langchain/docstore/base.py

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

libs/langchain/langchain/docstore/in_memory.py

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

libs/langchain/langchain/docstore/wikipedia.py

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