cleanup

Author: blink1073

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

@@ -35,7 +35,7 @@ def embedding() -> Embeddings:
             model="text-embedding-3-small",
         )
     if os.environ.get("AZURE_OPENAI_ENDPOINT"):
-        return AzureOpenAIEmbeddings(model="o4-mini")
+        return AzureOpenAIEmbeddings(model="text-embedding-3-small")
 
     return OllamaEmbeddings(model="all-minilm:l6-v2")
 

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

@@ -164,7 +164,7 @@ def vectorstore(
 def llm() -> ChatOpenAI:
     """Model used for interpreting query."""
     if "AZURE_OPENAI_ENDPOINT" in os.environ:
-        return AzureChatOpenAI(model="o4-mini", temperature=0.0, cache=False)
+        return AzureChatOpenAI(model="o4-mini", cache=False)
     return ChatOpenAI(model="gpt-4o", temperature=0.0, cache=False)