Skip to content

Commit 3553ff1

Browse files
committed
Refactor
1 parent e28fe19 commit 3553ff1

File tree

7 files changed

+27
-70
lines changed

7 files changed

+27
-70
lines changed

libs/langchain-mongodb/CHANGELOG.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@
22

33
---
44

5+
## Changes in version 0.8.0 (2025/11/18)
6+
7+
- Update dependencies to align with LangChain 1.0, which is now the minimum supported version of LangChain.
8+
59
## Changes in version 0.7.2 (2025/10/30)
610

711
- Pin LangChain version to < 1.0 to avoid compatibility issues while in pre-1.0.

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

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,9 @@
33
from typing import Any, List, Optional
44

55
import pymongo
6-
7-
try:
8-
from langchain.retrievers.parent_document_retriever import ( # type: ignore[import-not-found]
9-
ParentDocumentRetriever,
10-
)
11-
except ImportError:
12-
from langchain_classic.retrievers.parent_document_retriever import ( # type: ignore[import-not-found]
13-
ParentDocumentRetriever,
14-
)
6+
from langchain_classic.retrievers.parent_document_retriever import (
7+
ParentDocumentRetriever,
8+
)
159
from langchain_core.callbacks import (
1610
AsyncCallbackManagerForRetrieverRun,
1711
CallbackManagerForRetrieverRun,

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

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,11 @@
11
from typing import Any, Dict, Sequence, Tuple, Union
22

3-
try:
4-
from langchain.chains.query_constructor.schema import ( # type: ignore[import-not-found]
5-
AttributeInfo,
6-
)
7-
except ImportError:
8-
from langchain_classic.chains.query_constructor.schema import ( # type: ignore[import-not-found]
9-
AttributeInfo,
10-
)
11-
try:
12-
from langchain.retrievers.self_query.base import ( # type: ignore[import-not-found]
13-
SelfQueryRetriever,
14-
)
15-
except ImportError:
16-
from langchain_classic.retrievers.self_query.base import ( # type: ignore[import-not-found]
17-
SelfQueryRetriever,
18-
)
3+
from langchain_classic.chains.query_constructor.schema import (
4+
AttributeInfo,
5+
)
6+
from langchain_classic.retrievers.self_query.base import (
7+
SelfQueryRetriever,
8+
)
199
from langchain_core.language_models import BaseLanguageModel
2010
from langchain_core.runnables import Runnable
2111
from langchain_core.structured_query import (

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

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,8 @@
44
import pytest
55
import requests
66
from flaky import flaky # type:ignore[import-untyped]
7+
from langchain.agents import create_agent # type: ignore[assignment]
78
from langchain_openai import AzureChatOpenAI, ChatOpenAI
8-
9-
try:
10-
from langgraph.prebuilt import create_react_agent as create_agent
11-
12-
_HAVE_V1 = False
13-
except ImportError:
14-
from langchain.agents import create_agent # type: ignore[assignment]
15-
16-
_HAVE_V1 = True
179
from pymongo import MongoClient
1810

1911
from langchain_mongodb.agent_toolkit import (
@@ -72,10 +64,7 @@ def test_toolkit_response(db):
7264
prompt = MONGODB_AGENT_SYSTEM_PROMPT.format(top_k=5)
7365

7466
test_query = "Which country's customers spent the most?"
75-
if _HAVE_V1:
76-
agent = create_agent(llm, toolkit.get_tools(), system_prompt=prompt)
77-
else:
78-
agent = create_agent(llm, toolkit.get_tools(), prompt=prompt)
67+
agent = create_agent(llm, toolkit.get_tools(), system_prompt=prompt)
7968
agent.step_timeout = 60
8069
events = agent.stream(
8170
{"messages": [("user", test_query)]},

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

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,9 @@
11
import json
22
import warnings
33

4-
try:
5-
from langchain.memory import ( # type: ignore[import-not-found]
6-
ConversationBufferMemory,
7-
)
8-
except ImportError:
9-
from langchain_classic.memory import ( # type: ignore[import-not-found]
10-
ConversationBufferMemory,
11-
)
4+
from langchain_classic.memory import (
5+
ConversationBufferMemory,
6+
)
127
from langchain_core.messages import message_to_dict
138

149
from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory

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

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -4,21 +4,12 @@
44
from typing import Generator, Sequence, Union
55

66
import pytest
7-
8-
try:
9-
from langchain.chains.query_constructor.schema import ( # type: ignore[import-not-found]
10-
AttributeInfo,
11-
)
12-
from langchain.retrievers.self_query.base import ( # type: ignore[import-not-found]
13-
SelfQueryRetriever,
14-
)
15-
except ImportError:
16-
from langchain_classic.chains.query_constructor.schema import ( # type: ignore[import-not-found]
17-
AttributeInfo,
18-
)
19-
from langchain_classic.retrievers.self_query.base import ( # type: ignore[import-not-found]
20-
SelfQueryRetriever,
21-
)
7+
from langchain_classic.chains.query_constructor.schema import (
8+
AttributeInfo,
9+
)
10+
from langchain_classic.retrievers.self_query.base import (
11+
SelfQueryRetriever,
12+
)
2213
from langchain_core.documents import Document
2314
from langchain_openai import AzureChatOpenAI, ChatOpenAI
2415
from langchain_openai.chat_models.base import BaseChatOpenAI

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

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,9 @@
33

44
import mongomock
55
import pytest
6-
7-
try:
8-
from langchain.memory import ( # type: ignore[import-not-found]
9-
ConversationBufferMemory,
10-
)
11-
except ImportError:
12-
from langchain_classic.memory import ( # type: ignore[import-not-found]
13-
ConversationBufferMemory,
14-
)
6+
from langchain_classic.memory import (
7+
ConversationBufferMemory,
8+
)
159
from langchain_core.messages import message_to_dict
1610
from pytest_mock import MockerFixture
1711

0 commit comments

Comments
 (0)