merge main

run-llama · Apr 14, 2024 · dee309d · dee309d
2 parents 399c94a + f679e1c
commit dee309d
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/tests/test_reader.py b/tests/test_reader.py
@@ -1,10 +1,14 @@
 import os
-
+import pytest
 from fsspec.implementations.local import LocalFileSystem
 
 from llama_parse import LlamaParse
 
 
+@pytest.mark.skipif(
+    os.environ.get("LLAMA_CLOUD_API_KEY", "") == "",
+    reason="LLAMA_CLOUD_API_KEY not set",
+)
 def test_simple_page_text() -> None:
     parser = LlamaParse(result_type="text")
 
@@ -16,6 +20,10 @@ def test_simple_page_text() -> None:
     assert len(result[0].text) > 0
 
 
+@pytest.mark.skipif(
+    os.environ.get("LLAMA_CLOUD_API_KEY", "") == "",
+    reason="LLAMA_CLOUD_API_KEY not set",
+)
 def test_simple_page_markdown() -> None:
     parser = LlamaParse(result_type="markdown")
 
@@ -27,6 +35,10 @@ def test_simple_page_markdown() -> None:
     assert len(result[0].text) > 0
 
 
+@pytest.mark.skipif(
+    os.environ.get("LLAMA_CLOUD_API_KEY", "") == "",
+    reason="LLAMA_CLOUD_API_KEY not set",
+)
 def test_simple_page_with_custom_fs() -> None:
     parser = LlamaParse(result_type="markdown")
     fs = LocalFileSystem()
@@ -37,6 +49,10 @@ def test_simple_page_with_custom_fs() -> None:
     assert len(result) == 1
 
 
+@pytest.mark.skipif(
+    os.environ.get("LLAMA_CLOUD_API_KEY", "") == "",
+    reason="LLAMA_CLOUD_API_KEY not set",
+)
 def test_simple_page_progress_workers() -> None:
     parser = LlamaParse(result_type="markdown", show_progress=True, verbose=True)