From f13a1a2fc34beb60b4fa84d56f310f7015906bf6 Mon Sep 17 00:00:00 2001 From: Simon Suo Date: Wed, 28 Aug 2024 22:24:14 -0700 Subject: [PATCH] Support `take_screenshot` (#372) * wip * wip] * wip --- llama_parse/base.py | 5 +++++ pyproject.toml | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/llama_parse/base.py b/llama_parse/base.py index 32a1352..ae2ee8d 100644 --- a/llama_parse/base.py +++ b/llama_parse/base.py @@ -137,6 +137,10 @@ class LlamaParse(BasePydanticReader): default=None, description="The model name for the vendor multimodal API.", ) + take_screenshot: bool = Field( + default=False, + description="Whether to take screenshot of each page of the document.", + ) @field_validator("api_key", mode="before", check_fields=True) @classmethod @@ -206,6 +210,7 @@ async def _create_job( "vendor_multimodal_api_key": self.vendor_multimodal_api_key, "use_vendor_multimodal_model": self.use_vendor_multimodal_model, "vendor_multimodal_model_name": self.vendor_multimodal_model_name, + "take_screenshot": self.take_screenshot, } # only send page separator to server if it is not None diff --git a/pyproject.toml b/pyproject.toml index 507b657..6a2450e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "llama-parse" -version = "0.5.0" +version = "0.5.1" description = "Parse files into RAG-Optimized formats." authors = ["Logan Markewich "] license = "MIT"