fix test

langchain-ai · Dec 13, 2024 · 9b4bdf4 · 9b4bdf4
1 parent 8c34cc1
commit 9b4bdf4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/libs/mongodb/tests/integration_tests/test_parent_document.py b/libs/mongodb/tests/integration_tests/test_parent_document.py
@@ -87,6 +87,11 @@ def test_1clxn_retriever(
     question = "What percentage of the Uniform Bar Examination can GPT4 pass?"
     responses = retriever.invoke(question)
 
+    if not os.environ.get("OPENAI_API_KEY"):
+        assert len(responses) == 1
+        assert responses[0].metadata["page"] == 99
+        return
+
     assert len(responses) == 3
     assert all("GPT-4" in doc.page_content for doc in responses)
     assert {4, 5, 29} == set(doc.metadata["page"] for doc in responses)