diff --git a/llama_parse/base.py b/llama_parse/base.py index 8db4161..3ebf0ad 100644 --- 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: