Fix bug in logging (#47)

run-llama · Jan 24, 2025 · e049c74 · e049c74
1 parent fcb8d4b
commit e049c74
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/llama_extract/extract.py b/llama_extract/extract.py
@@ -19,7 +19,6 @@
     StatusEnum,
     Project,
 )
-from llama_cloud.core.api_error import ApiError
 from llama_cloud.client import AsyncLlamaCloud
 from llama_extract.utils import JSONObjectType, augment_async_errors
 from llama_index.core.schema import BaseComponent
@@ -242,7 +241,7 @@ async def queue_extraction(
             )
 
         if self._verbose:
-            for file, job in zip(files, results):
+            for file, job in zip(files, extract_jobs):
                 file_repr = (
                     str(file) if isinstance(file, (str, Path)) else "<bytes/buffer>"
                 )

diff --git a/tests/test_extract_api.py b/tests/test_extract_api.py
@@ -38,6 +38,7 @@ def llama_extract():
         api_key=LLAMA_CLOUD_API_KEY,
         base_url=LLAMA_CLOUD_BASE_URL,
         project_id=LLAMA_CLOUD_PROJECT_ID,
+        verbose=True,
     )
 
 

diff --git a/tests/test_extract_e2e.py b/tests/test_extract_e2e.py
@@ -82,6 +82,7 @@ def extractor():
         api_key=LLAMA_CLOUD_API_KEY,
         base_url=LLAMA_CLOUD_BASE_URL,
         project_id=LLAMA_CLOUD_PROJECT_ID,
+        verbose=True,
     )
     yield extract
     # Cleanup thread pool at end of session