fix tests

langchain-ai · Dec 13, 2024 · eeff780 · eeff780
1 parent 90c1cc6
commit eeff780
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/libs/mongodb/tests/integration_tests/test_parent_document.py b/libs/mongodb/tests/integration_tests/test_parent_document.py
@@ -32,13 +32,12 @@
 def embedding_model() -> Embeddings:
     from langchain_openai import OpenAIEmbeddings
 
-    try:
-        return OpenAIEmbeddings(
-            openai_api_key=os.environ["OPENAI_API_KEY"],  # type: ignore # noqa
-            model="text-embedding-3-small",
-        )
-    except Exception:
+    if not os.environ.get("OPENAI_API_KEY"):
         return ConsistentFakeEmbeddings(DIMENSIONS)
+    return OpenAIEmbeddings(
+        openai_api_key=os.environ["OPENAI_API_KEY"],  # type: ignore # noqa
+        model="text-embedding-3-small",
+    )
 
 
 def test_1clxn_retriever(

diff --git a/libs/mongodb/tests/integration_tests/test_retrievers.py b/libs/mongodb/tests/integration_tests/test_retrievers.py
@@ -47,8 +47,8 @@ def example_documents() -> List[Document]:
 def embedding_openai() -> Embeddings:
     from langchain_openai import OpenAIEmbeddings
 
-    if os.environ.get("OPENAI_API_KEY") is None:
-         pytest.skip("test_retrievers expects OPENAI_API_KEY in os.environ")
+    if not os.environ.get("OPENAI_API_KEY"):
+        pytest.skip("test_retrievers expects OPENAI_API_KEY in os.environ")
 
     from langchain_openai import OpenAIEmbeddings