Skip to content

Commit 54c2150

Browse files
authored
INTPYTHON-869 - [LangChain] Update tests to use supported gpt model (langchain-ai#312)
Passing tests: https://spruce.mongodb.com/version/696f91e258ffd000075e35e7/tasks?sorts=STATUS%3AASC%3BBASE_STATUS%3ADESC.
1 parent 0d637a6 commit 54c2150

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,9 @@ def test_toolkit_response(db):
5555
CONNECTION_STRING, database=DB_NAME
5656
)
5757
if "AZURE_OPENAI_ENDPOINT" in os.environ:
58-
llm = AzureChatOpenAI(model="gpt-4o-mini", timeout=60, seed=12345)
58+
llm = AzureChatOpenAI(model="gpt-5-mini", timeout=60, seed=12345)
5959
else:
60-
llm = ChatOpenAI(model="gpt-4o-mini", timeout=60, seed=12345)
60+
llm = ChatOpenAI(model="gpt-5-mini", timeout=60, seed=12345)
6161

6262
toolkit = MongoDBDatabaseToolkit(db=db_wrapper, llm=llm)
6363

libs/langchain-mongodb/tests/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def create_llm() -> BaseChatModel:
5252
if os.environ.get("AZURE_OPENAI_ENDPOINT"):
5353
return AzureChatOpenAI(model="o4-mini", timeout=60, cache=False, seed=12345)
5454
if os.environ.get("OPENAI_API_KEY"):
55-
return ChatOpenAI(model="gpt-4o-mini", timeout=60, cache=False, seed=12345)
55+
return ChatOpenAI(model="gpt-5-mini", timeout=60, cache=False, seed=12345)
5656
return ChatOllama(model="llama3:8b", cache=False, seed=12345)
5757

5858

0 commit comments

Comments
 (0)