allow for .jpg images (#195)

run-llama · May 23, 2024 · d19b35c · d19b35c
1 parent 0c83fbd
commit d19b35c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llama_parse/base.py b/llama_parse/base.py
@@ -341,7 +341,8 @@ def get_images(self, json_result: List[dict], download_path: str) -> List[dict]:
 
                         # get a valid image path
                         if not image_path.endswith(".png"):
-                            image_path += ".png"
+                            if not image_path.endswith(".jpg"):
+                                image_path += ".png"
 
                         image["path"] = image_path
                         image["job_id"] = job_id