From 42a98f921ab3a7b4557085ea7dc9db05e4c842ff Mon Sep 17 00:00:00 2001 From: Pierre-Loic Doulcet Date: Tue, 10 Sep 2024 10:30:48 -0700 Subject: [PATCH] Update llama_parse/base.py Co-authored-by: Logan --- llama_parse/base.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) 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: