Update llama_parse/base.py

Co-authored-by: Logan <[email protected]>
run-llama · Sep 10, 2024 · 42a98f9 · 42a98f9
1 parent 1c45339
commit 42a98f9
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/llama_parse/base.py b/llama_parse/base.py
@@ -508,10 +508,7 @@ async def aget_images(
                         image["path"] = image_path
                         image["job_id"] = job_id
 
-                        if isinstance(result["file_path"], str):
-                            image["original_file_path"] = result["file_path"]
-                        else:
-                            image["original_file_path"] = None
+                        image["original_file_path"] = result.get("file_path", None)
 
                         image["page_number"] = page["page"]
                         with open(image_path, "wb") as f: