debug

Author: blink1073

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

@@ -8,6 +8,7 @@
 from langchain.retrievers.self_query.base import SelfQueryRetriever
 from langchain_core.documents import Document
 from langchain_openai import AzureChatOpenAI, ChatOpenAI
+from langchain_openai.chat_models.base import BaseChatOpenAI
 
 from langchain_mongodb import MongoDBAtlasVectorSearch, index
 from langchain_mongodb.retrievers import MongoDBAtlasSelfQueryRetriever
@@ -161,7 +162,7 @@ def vectorstore(
 
 
 @pytest.fixture
-def llm() -> ChatOpenAI:
+def llm() -> BaseChatOpenAI:
     """Model used for interpreting query."""
     if "AZURE_OPENAI_ENDPOINT" in os.environ:
         return AzureChatOpenAI(model="o4-mini", cache=False)

libs/langgraph-checkpoint-mongodb/tests/integration_tests/conftest.py

@@ -9,13 +9,13 @@
 @pytest.fixture(scope="session")
 def embedding() -> Embeddings:
     if os.environ.get("AZURE_OPENAI_ENDPOINT"):
-        return AzureOpenAIEmbeddings(model="o4-mini")
+        return AzureOpenAIEmbeddings(model="text-embedding-3-small")
     if os.environ.get("OPENAI_API_KEY"):
         return OpenAIEmbeddings(
             openai_api_key=os.environ["OPENAI_API_KEY"],  # type: ignore # noqa
             model="text-embedding-3-small",
         )
-
+    raise ValueError("REMOVE BEFORE MERGING")
     return OllamaEmbeddings(model="all-minilm:l6-v2")