Add timeout to the image request using httpx (#378)

llama_parse/base.py

@@ -481,7 +481,11 @@ def get_images(self, json_result: List[dict], download_path: str) -> List[dict]:
                         image["page_number"] = page["page"]
                         with open(image_path, "wb") as f:
                             image_url = f"{self.base_url}/api/parsing/job/{job_id}/result/image/{image_name}"
-                            f.write(httpx.get(image_url, headers=headers).content)
+                            f.write(
+                                httpx.get(
+                                    image_url, headers=headers, timeout=self.max_timeout
+                                ).content
+                            )
                         images.append(image)
             return images
         except Exception as e:

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "llama-parse"
-version = "0.5.1"
+version = "0.5.2"
 description = "Parse files into RAG-Optimized formats."
 authors = ["Logan Markewich <[email protected]>"]
 license = "MIT"