INTPYTHON-828 - Make Langchain-MongoDB compatible with LangChain >= 1.0

Author: NoahStapp

libs/langchain-mongodb/langchain_mongodb/retrievers/parent_document.py

@@ -3,7 +3,15 @@
 from typing import Any, List, Optional
 
 import pymongo
-from langchain.retrievers.parent_document_retriever import ParentDocumentRetriever
+
+try:
+    from langchain.retrievers.parent_document_retriever import (  # type: ignore[import-not-found]
+        ParentDocumentRetriever,
+    )
+except ImportError:
+    from langchain_classic.retrievers.parent_document_retriever import (  # type: ignore[import-not-found]
+        ParentDocumentRetriever,
+    )
 from langchain_core.callbacks import (
     AsyncCallbackManagerForRetrieverRun,
     CallbackManagerForRetrieverRun,

libs/langchain-mongodb/langchain_mongodb/retrievers/self_querying.py

@@ -1,7 +1,21 @@
 from typing import Any, Dict, Sequence, Tuple, Union
 
-from langchain.chains.query_constructor.schema import AttributeInfo
-from langchain.retrievers.self_query.base import SelfQueryRetriever
+try:
+    from langchain.chains.query_constructor.schema import (  # type: ignore[import-not-found]
+        AttributeInfo,
+    )
+except ImportError:
+    from langchain_classic.chains.query_constructor.schema import (  # type: ignore[import-not-found]
+        AttributeInfo,
+    )
+try:
+    from langchain.retrievers.self_query.base import (  # type: ignore[import-not-found]
+        SelfQueryRetriever,
+    )
+except ImportError:
+    from langchain_classic.retrievers.self_query.base import (  # type: ignore[import-not-found]
+        SelfQueryRetriever,
+    )
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.runnables import Runnable
 from langchain_core.structured_query import (

libs/langchain-mongodb/pyproject.toml

@@ -10,20 +10,21 @@ description = "An integration package connecting MongoDB and LangChain"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "langchain-core>=0.3,<1.0",
-    "langchain>=0.3,<1.0",
+    "langchain-core>=1.0",
+    "langchain>=1.0",
+    "langchain-classic>=1.0",
     "pymongo>=4.6.1",
-    "langchain-text-splitters>=0.3,<1.0",
+    "langchain-text-splitters>=1.0",
     "numpy>=1.26",
     "lark<2.0.0,>=1.1.9",
 ]
 
 [dependency-groups]
 dev = [
     "freezegun>=1.2.2",
-    "langchain>=0.3.14,<1.0",
-    "langchain-core>=0.3.29,<1.0",
-    "langchain-text-splitters>=0.3.5,<1.0",
+    "langchain>=1.0",
+    "langchain-core>=1.0",
+    "langchain-text-splitters>=1.0",
     "pytest-mock>=3.10.0",
     "pytest>=7.3.0",
     "syrupy>=4.0.2",
@@ -33,13 +34,13 @@ dev = [
     "pre-commit>=4.0",
     "mypy>=1.10",
     "simsimd>=6.5.0",
-    "langchain-ollama>=0.2.2,<1.0",
-    "langchain-openai>=0.2.14,<1.0",
-    "langchain-community>=0.3.27,<1.0",
+    "langchain-ollama>=1.0",
+    "langchain-openai>=1.0",
+    "langchain-community>=0.3.27",
     "pypdf>=5.0.1",
     "langgraph>=0.2.72",
     "flaky>=3.8.1",
-    "langchain-tests==0.3.22,<1.0",
+    "langchain-tests==0.3.22",
     "pip>=25.0.1",
     "typing-extensions>=4.12.2",
 ]

libs/langchain-mongodb/tests/integration_tests/test_agent_toolkit.py

@@ -5,7 +5,15 @@
 import requests
 from flaky import flaky  # type:ignore[import-untyped]
 from langchain_openai import AzureChatOpenAI, ChatOpenAI
-from langgraph.prebuilt import create_react_agent
+
+try:
+    from langgraph.prebuilt import create_react_agent as create_agent
+
+    _HAVE_V1 = False
+except ImportError:
+    from langchain.agents import create_agent  # type: ignore[assignment]
+
+    _HAVE_V1 = True
 from pymongo import MongoClient
 
 from langchain_mongodb.agent_toolkit import (
@@ -64,7 +72,10 @@ def test_toolkit_response(db):
     prompt = MONGODB_AGENT_SYSTEM_PROMPT.format(top_k=5)
 
     test_query = "Which country's customers spent the most?"
-    agent = create_react_agent(llm, toolkit.get_tools(), prompt=prompt)
+    if _HAVE_V1:
+        agent = create_agent(llm, toolkit.get_tools(), system_prompt=prompt)
+    else:
+        agent = create_agent(llm, toolkit.get_tools(), prompt=prompt)
     agent.step_timeout = 60
     events = agent.stream(
         {"messages": [("user", test_query)]},

libs/langchain-mongodb/tests/integration_tests/test_chat_message_histories.py

@@ -1,7 +1,14 @@
 import json
 import warnings
 
-from langchain.memory import ConversationBufferMemory  # type: ignore[import-not-found]
+try:
+    from langchain.memory import (  # type: ignore[import-not-found]
+        ConversationBufferMemory,
+    )
+except ImportError:
+    from langchain_classic.memory import (  # type: ignore[import-not-found]
+        ConversationBufferMemory,
+    )
 from langchain_core.messages import message_to_dict
 
 from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory

libs/langchain-mongodb/tests/integration_tests/test_retriever_selfquerying.py

@@ -4,8 +4,21 @@
 from typing import Generator, Sequence, Union
 
 import pytest
-from langchain.chains.query_constructor.schema import AttributeInfo
-from langchain.retrievers.self_query.base import SelfQueryRetriever
+
+try:
+    from langchain.chains.query_constructor.schema import (  # type: ignore[import-not-found]
+        AttributeInfo,
+    )
+    from langchain.retrievers.self_query.base import (  # type: ignore[import-not-found]
+        SelfQueryRetriever,
+    )
+except ImportError:
+    from langchain_classic.chains.query_constructor.schema import (  # type: ignore[import-not-found]
+        AttributeInfo,
+    )
+    from langchain_classic.retrievers.self_query.base import (  # type: ignore[import-not-found]
+        SelfQueryRetriever,
+    )
 from langchain_core.documents import Document
 from langchain_openai import AzureChatOpenAI, ChatOpenAI
 from langchain_openai.chat_models.base import BaseChatOpenAI

libs/langchain-mongodb/tests/unit_tests/test_chat_message_histories.py

@@ -3,7 +3,15 @@
 
 import mongomock
 import pytest
-from langchain.memory import ConversationBufferMemory  # type: ignore[import-not-found]
+
+try:
+    from langchain.memory import (  # type: ignore[import-not-found]
+        ConversationBufferMemory,
+    )
+except ImportError:
+    from langchain_classic.memory import (  # type: ignore[import-not-found]
+        ConversationBufferMemory,
+    )
 from langchain_core.messages import message_to_dict
 from pytest_mock import MockerFixture