llama-cloud 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

llama_cloud/resources/llama_extract/types/__init__.py

@@ -1,6 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .extract_agent_create_data_schema_value import ExtractAgentCreateDataSchemaValue
-from .extract_agent_update_data_schema_value import ExtractAgentUpdateDataSchemaValue
+from .extract_agent_create_data_schema import ExtractAgentCreateDataSchema
+from .extract_agent_create_data_schema_zero_value import ExtractAgentCreateDataSchemaZeroValue
+from .extract_agent_update_data_schema import ExtractAgentUpdateDataSchema
+from .extract_agent_update_data_schema_zero_value import ExtractAgentUpdateDataSchemaZeroValue
+from .extract_schema_validate_request_data_schema import ExtractSchemaValidateRequestDataSchema
+from .extract_schema_validate_request_data_schema_zero_value import ExtractSchemaValidateRequestDataSchemaZeroValue
 
-__all__ = ["ExtractAgentCreateDataSchemaValue", "ExtractAgentUpdateDataSchemaValue"]
+__all__ = [
+    "ExtractAgentCreateDataSchema",
+    "ExtractAgentCreateDataSchemaZeroValue",
+    "ExtractAgentUpdateDataSchema",
+    "ExtractAgentUpdateDataSchemaZeroValue",
+    "ExtractSchemaValidateRequestDataSchema",
+    "ExtractSchemaValidateRequestDataSchemaZeroValue",
+]

llama_cloud/resources/llama_extract/types/extract_agent_create_data_schema.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .extract_agent_create_data_schema_zero_value import ExtractAgentCreateDataSchemaZeroValue
+
+ExtractAgentCreateDataSchema = typing.Union[
+    typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaZeroValue]], str
+]

llama_cloud/resources/llama_extract/types/{extract_agent_create_data_schema_value.py → extract_agent_create_data_schema_zero_value.py}

@@ -2,6 +2,6 @@
 
 import typing
 
-ExtractAgentCreateDataSchemaValue = typing.Union[
+ExtractAgentCreateDataSchemaZeroValue = typing.Union[
     typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
 ]

llama_cloud/resources/llama_extract/types/extract_agent_update_data_schema.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .extract_agent_update_data_schema_zero_value import ExtractAgentUpdateDataSchemaZeroValue
+
+ExtractAgentUpdateDataSchema = typing.Union[
+    typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaZeroValue]], str
+]

llama_cloud/resources/llama_extract/types/{extract_agent_update_data_schema_value.py → extract_agent_update_data_schema_zero_value.py}

@@ -2,6 +2,6 @@
 
 import typing
 
-ExtractAgentUpdateDataSchemaValue = typing.Union[
+ExtractAgentUpdateDataSchemaZeroValue = typing.Union[
     typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
 ]

llama_cloud/resources/llama_extract/types/extract_schema_validate_request_data_schema.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .extract_schema_validate_request_data_schema_zero_value import ExtractSchemaValidateRequestDataSchemaZeroValue
+
+ExtractSchemaValidateRequestDataSchema = typing.Union[
+    typing.Dict[str, typing.Optional[ExtractSchemaValidateRequestDataSchemaZeroValue]], str
+]

llama_cloud/resources/llama_extract/types/extract_schema_validate_request_data_schema_zero_value.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ExtractSchemaValidateRequestDataSchemaZeroValue = typing.Union[
+    typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
+]

llama_cloud/resources/organizations/client.py

@@ -229,14 +229,14 @@ class OrganizationsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def update_organization(self, organization_id: str, *, name: typing.Optional[str] = OMIT) -> Organization:
+    def update_organization(self, organization_id: str, *, name: str) -> Organization:
         """
         Update an existing organization.
 
         Parameters:
             - organization_id: str.
 
-            - name: typing.Optional[str].
+            - name: str. A name for the organization.
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -245,15 +245,13 @@ class OrganizationsClient:
         )
         client.organizations.update_organization(
             organization_id="string",
+            name="string",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {}
-        if name is not OMIT:
-            _request["name"] = name
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
-            json=jsonable_encoder(_request),
+            json=jsonable_encoder({"name": name}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -922,14 +920,14 @@ class AsyncOrganizationsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def update_organization(self, organization_id: str, *, name: typing.Optional[str] = OMIT) -> Organization:
+    async def update_organization(self, organization_id: str, *, name: str) -> Organization:
         """
         Update an existing organization.
 
         Parameters:
             - organization_id: str.
 
-            - name: typing.Optional[str].
+            - name: str. A name for the organization.
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -938,15 +936,13 @@ class AsyncOrganizationsClient:
         )
         await client.organizations.update_organization(
             organization_id="string",
+            name="string",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {}
-        if name is not OMIT:
-            _request["name"] = name
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
-            json=jsonable_encoder(_request),
+            json=jsonable_encoder({"name": name}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )

llama_cloud/resources/parsing/client.py

@@ -100,6 +100,96 @@ class ParsingClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def screenshot(
+        self,
+        *,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        file: typing.Optional[str] = OMIT,
+        do_not_cache: bool,
+        http_proxy: str,
+        input_s_3_path: str,
+        input_s_3_region: str,
+        input_url: str,
+        invalidate_cache: bool,
+        max_pages: typing.Optional[int] = OMIT,
+        output_s_3_path_prefix: str,
+        output_s_3_region: str,
+        target_pages: str,
+        webhook_url: str,
+        job_timeout_in_seconds: float,
+        job_timeout_extra_time_per_page_in_seconds: float,
+    ) -> ParsingJob:
+        """
+        Parameters:
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+
+            - file: typing.Optional[str].
+
+            - do_not_cache: bool.
+
+            - http_proxy: str.
+
+            - input_s_3_path: str.
+
+            - input_s_3_region: str.
+
+            - input_url: str.
+
+            - invalidate_cache: bool.
+
+            - max_pages: typing.Optional[int].
+
+            - output_s_3_path_prefix: str.
+
+            - output_s_3_region: str.
+
+            - target_pages: str.
+
+            - webhook_url: str.
+
+            - job_timeout_in_seconds: float.
+
+            - job_timeout_extra_time_per_page_in_seconds: float.
+        """
+        _request: typing.Dict[str, typing.Any] = {
+            "do_not_cache": do_not_cache,
+            "http_proxy": http_proxy,
+            "input_s3_path": input_s_3_path,
+            "input_s3_region": input_s_3_region,
+            "input_url": input_url,
+            "invalidate_cache": invalidate_cache,
+            "output_s3_path_prefix": output_s_3_path_prefix,
+            "output_s3_region": output_s_3_region,
+            "target_pages": target_pages,
+            "webhook_url": webhook_url,
+            "job_timeout_in_seconds": job_timeout_in_seconds,
+            "job_timeout_extra_time_per_page_in_seconds": job_timeout_extra_time_per_page_in_seconds,
+        }
+        if file is not OMIT:
+            _request["file"] = file
+        if max_pages is not OMIT:
+            _request["max_pages"] = max_pages
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/parsing/screenshot"),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ParsingJob, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def upload_file(
         self,
         *,
@@ -166,6 +256,14 @@ class ParsingClient:
         bounding_box: str,
         gpt_4_o_mode: bool,
         gpt_4_o_api_key: str,
+        job_timeout_in_seconds: float,
+        job_timeout_extra_time_per_page_in_seconds: float,
+        strict_mode_image_extraction: bool,
+        strict_mode_image_ocr: bool,
+        strict_mode_reconstruction: bool,
+        strict_mode_buggy_font: bool,
+        ignore_document_elements_for_layout_detection: bool,
+        output_tables_as_html: bool,
     ) -> ParsingJob:
         """
         Upload a file to s3 and create a job. return a job id
@@ -296,6 +394,22 @@ class ParsingClient:
             - gpt_4_o_mode: bool.
 
             - gpt_4_o_api_key: str.
+
+            - job_timeout_in_seconds: float.
+
+            - job_timeout_extra_time_per_page_in_seconds: float.
+
+            - strict_mode_image_extraction: bool.
+
+            - strict_mode_image_ocr: bool.
+
+            - strict_mode_reconstruction: bool.
+
+            - strict_mode_buggy_font: bool.
+
+            - ignore_document_elements_for_layout_detection: bool.
+
+            - output_tables_as_html: bool.
         """
         _request: typing.Dict[str, typing.Any] = {
             "annotate_links": annotate_links,
@@ -357,6 +471,14 @@ class ParsingClient:
             "bounding_box": bounding_box,
             "gpt4o_mode": gpt_4_o_mode,
             "gpt4o_api_key": gpt_4_o_api_key,
+            "job_timeout_in_seconds": job_timeout_in_seconds,
+            "job_timeout_extra_time_per_page_in_seconds": job_timeout_extra_time_per_page_in_seconds,
+            "strict_mode_image_extraction": strict_mode_image_extraction,
+            "strict_mode_image_ocr": strict_mode_image_ocr,
+            "strict_mode_reconstruction": strict_mode_reconstruction,
+            "strict_mode_buggy_font": strict_mode_buggy_font,
+            "ignore_document_elements_for_layout_detection": ignore_document_elements_for_layout_detection,
+            "output_tables_as_HTML": output_tables_as_html,
         }
         if file is not OMIT:
             _request["file"] = file
@@ -925,6 +1047,96 @@ class AsyncParsingClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def screenshot(
+        self,
+        *,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        file: typing.Optional[str] = OMIT,
+        do_not_cache: bool,
+        http_proxy: str,
+        input_s_3_path: str,
+        input_s_3_region: str,
+        input_url: str,
+        invalidate_cache: bool,
+        max_pages: typing.Optional[int] = OMIT,
+        output_s_3_path_prefix: str,
+        output_s_3_region: str,
+        target_pages: str,
+        webhook_url: str,
+        job_timeout_in_seconds: float,
+        job_timeout_extra_time_per_page_in_seconds: float,
+    ) -> ParsingJob:
+        """
+        Parameters:
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+
+            - file: typing.Optional[str].
+
+            - do_not_cache: bool.
+
+            - http_proxy: str.
+
+            - input_s_3_path: str.
+
+            - input_s_3_region: str.
+
+            - input_url: str.
+
+            - invalidate_cache: bool.
+
+            - max_pages: typing.Optional[int].
+
+            - output_s_3_path_prefix: str.
+
+            - output_s_3_region: str.
+
+            - target_pages: str.
+
+            - webhook_url: str.
+
+            - job_timeout_in_seconds: float.
+
+            - job_timeout_extra_time_per_page_in_seconds: float.
+        """
+        _request: typing.Dict[str, typing.Any] = {
+            "do_not_cache": do_not_cache,
+            "http_proxy": http_proxy,
+            "input_s3_path": input_s_3_path,
+            "input_s3_region": input_s_3_region,
+            "input_url": input_url,
+            "invalidate_cache": invalidate_cache,
+            "output_s3_path_prefix": output_s_3_path_prefix,
+            "output_s3_region": output_s_3_region,
+            "target_pages": target_pages,
+            "webhook_url": webhook_url,
+            "job_timeout_in_seconds": job_timeout_in_seconds,
+            "job_timeout_extra_time_per_page_in_seconds": job_timeout_extra_time_per_page_in_seconds,
+        }
+        if file is not OMIT:
+            _request["file"] = file
+        if max_pages is not OMIT:
+            _request["max_pages"] = max_pages
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/parsing/screenshot"),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ParsingJob, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def upload_file(
         self,
         *,
@@ -991,6 +1203,14 @@ class AsyncParsingClient:
         bounding_box: str,
         gpt_4_o_mode: bool,
         gpt_4_o_api_key: str,
+        job_timeout_in_seconds: float,
+        job_timeout_extra_time_per_page_in_seconds: float,
+        strict_mode_image_extraction: bool,
+        strict_mode_image_ocr: bool,
+        strict_mode_reconstruction: bool,
+        strict_mode_buggy_font: bool,
+        ignore_document_elements_for_layout_detection: bool,
+        output_tables_as_html: bool,
     ) -> ParsingJob:
         """
         Upload a file to s3 and create a job. return a job id
@@ -1121,6 +1341,22 @@ class AsyncParsingClient:
             - gpt_4_o_mode: bool.
 
             - gpt_4_o_api_key: str.
+
+            - job_timeout_in_seconds: float.
+
+            - job_timeout_extra_time_per_page_in_seconds: float.
+
+            - strict_mode_image_extraction: bool.
+
+            - strict_mode_image_ocr: bool.
+
+            - strict_mode_reconstruction: bool.
+
+            - strict_mode_buggy_font: bool.
+
+            - ignore_document_elements_for_layout_detection: bool.
+
+            - output_tables_as_html: bool.
         """
         _request: typing.Dict[str, typing.Any] = {
             "annotate_links": annotate_links,
@@ -1182,6 +1418,14 @@ class AsyncParsingClient:
             "bounding_box": bounding_box,
             "gpt4o_mode": gpt_4_o_mode,
             "gpt4o_api_key": gpt_4_o_api_key,
+            "job_timeout_in_seconds": job_timeout_in_seconds,
+            "job_timeout_extra_time_per_page_in_seconds": job_timeout_extra_time_per_page_in_seconds,
+            "strict_mode_image_extraction": strict_mode_image_extraction,
+            "strict_mode_image_ocr": strict_mode_image_ocr,
+            "strict_mode_reconstruction": strict_mode_reconstruction,
+            "strict_mode_buggy_font": strict_mode_buggy_font,
+            "ignore_document_elements_for_layout_detection": ignore_document_elements_for_layout_detection,
+            "output_tables_as_HTML": output_tables_as_html,
         }
         if file is not OMIT:
             _request["file"] = file

llama_cloud/resources/reports/client.py

@@ -47,9 +47,10 @@ class ReportsClient:
         organization_id: typing.Optional[str] = None,
         name: str,
         template_text: str,
-        template_instructions: str,
+        template_instructions: typing.Optional[str] = OMIT,
+        existing_retriever_id: typing.Optional[str] = OMIT,
         files: typing.List[str],
-        template_file: typing.IO,
+        template_file: typing.Optional[str] = OMIT,
     ) -> ReportCreateResponse:
         """
         Create a new report.
@@ -63,25 +64,26 @@ class ReportsClient:
 
             - template_text: str.
 
-            - template_instructions: str.
+            - template_instructions: typing.Optional[str].
+
+            - existing_retriever_id: typing.Optional[str].
 
             - files: typing.List[str].
 
-            - template_file: typing.IO.
+            - template_file: typing.Optional[str].
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name, "template_text": template_text, "files": files}
+        if template_instructions is not OMIT:
+            _request["template_instructions"] = template_instructions
+        if existing_retriever_id is not OMIT:
+            _request["existing_retriever_id"] = existing_retriever_id
+        if template_file is not OMIT:
+            _request["template_file"] = template_file
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/reports"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            data=jsonable_encoder(
-                {
-                    "name": name,
-                    "template_text": template_text,
-                    "template_instructions": template_instructions,
-                    "files": files,
-                }
-            ),
-            files={"template_file": template_file},
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -628,9 +630,10 @@ class AsyncReportsClient:
         organization_id: typing.Optional[str] = None,
         name: str,
         template_text: str,
-        template_instructions: str,
+        template_instructions: typing.Optional[str] = OMIT,
+        existing_retriever_id: typing.Optional[str] = OMIT,
         files: typing.List[str],
-        template_file: typing.IO,
+        template_file: typing.Optional[str] = OMIT,
     ) -> ReportCreateResponse:
         """
         Create a new report.
@@ -644,25 +647,26 @@ class AsyncReportsClient:
 
             - template_text: str.
 
-            - template_instructions: str.
+            - template_instructions: typing.Optional[str].
+
+            - existing_retriever_id: typing.Optional[str].
 
             - files: typing.List[str].
 
-            - template_file: typing.IO.
+            - template_file: typing.Optional[str].
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name, "template_text": template_text, "files": files}
+        if template_instructions is not OMIT:
+            _request["template_instructions"] = template_instructions
+        if existing_retriever_id is not OMIT:
+            _request["existing_retriever_id"] = existing_retriever_id
+        if template_file is not OMIT:
+            _request["template_file"] = template_file
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/reports"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            data=jsonable_encoder(
-                {
-                    "name": name,
-                    "template_text": template_text,
-                    "template_instructions": template_instructions,
-                    "files": files,
-                }
-            ),
-            files={"template_file": template_file},
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )

llama_cloud/resources/retrievers/client.py

@@ -33,12 +33,18 @@ class RetrieversClient:
         self._client_wrapper = client_wrapper
 
     def list_retrievers(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        name: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
     ) -> typing.List[Retriever]:
         """
         List Retrievers for a project.
 
         Parameters:
+            - name: typing.Optional[str].
+
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
@@ -53,7 +59,7 @@ class RetrieversClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            params=remove_none_from_dict({"name": name, "project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -346,12 +352,18 @@ class AsyncRetrieversClient:
         self._client_wrapper = client_wrapper
 
     async def list_retrievers(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        name: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
     ) -> typing.List[Retriever]:
         """
         List Retrievers for a project.
 
         Parameters:
+            - name: typing.Optional[str].
+
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
@@ -366,7 +378,7 @@ class AsyncRetrieversClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            params=remove_none_from_dict({"name": name, "project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )