llama-cloud 0.1.7__py3-none-any.whl → 0.1.8__py3-none-any.whl

llama_cloud/__init__.py

@@ -50,6 +50,7 @@ from .types import (
     CompositeRetrievalMode,
     CompositeRetrievalResult,
     CompositeRetrievedTextNode,
+    CompositeRetrievedTextNodeWithScore,
     ConfigurableDataSinkNames,
     ConfigurableDataSourceNames,
     ConfigurableTransformationDefinition,
@@ -109,6 +110,9 @@ from .types import (
     ExtractResultsetDataItemValue,
     ExtractResultsetDataZeroValue,
     ExtractResultsetExtractionMetadataValue,
+    ExtractRun,
+    ExtractRunDataSchemaValue,
+    ExtractState,
     ExtractionJob,
     ExtractionResult,
     ExtractionResultDataValue,
@@ -133,6 +137,7 @@ from .types import (
     JobNames,
     JobRecord,
     JobRecordWithUsageMetrics,
+    LlamaExtractSettings,
     LlamaIndexCoreBaseLlmsTypesChatMessage,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image,
@@ -165,6 +170,7 @@ from .types import (
     OpenAiEmbeddingConfig,
     Organization,
     OrganizationCreate,
+    PageFigureMetadata,
     PageScreenshotMetadata,
     PageScreenshotNodeWithScore,
     PageSegmentationConfig,
@@ -383,6 +389,7 @@ __all__ = [
     "CompositeRetrievalMode",
     "CompositeRetrievalResult",
     "CompositeRetrievedTextNode",
+    "CompositeRetrievedTextNodeWithScore",
     "ConfigurableDataSinkNames",
     "ConfigurableDataSourceNames",
     "ConfigurableTransformationDefinition",
@@ -455,6 +462,9 @@ __all__ = [
     "ExtractResultsetDataItemValue",
     "ExtractResultsetDataZeroValue",
     "ExtractResultsetExtractionMetadataValue",
+    "ExtractRun",
+    "ExtractRunDataSchemaValue",
+    "ExtractState",
     "ExtractionJob",
     "ExtractionResult",
     "ExtractionResultDataValue",
@@ -485,6 +495,7 @@ __all__ = [
     "JobRecord",
     "JobRecordWithUsageMetrics",
     "LlamaCloudEnvironment",
+    "LlamaExtractSettings",
     "LlamaIndexCoreBaseLlmsTypesChatMessage",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image",
@@ -517,6 +528,7 @@ __all__ = [
     "OpenAiEmbeddingConfig",
     "Organization",
     "OrganizationCreate",
+    "PageFigureMetadata",
     "PageScreenshotMetadata",
     "PageScreenshotNodeWithScore",
     "PageSegmentationConfig",

llama_cloud/resources/files/client.py

@@ -12,6 +12,7 @@ from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.file import File
 from ...types.http_validation_error import HttpValidationError
+from ...types.page_figure_metadata import PageFigureMetadata
 from ...types.page_screenshot_metadata import PageScreenshotMetadata
 from ...types.presigned_url import PresignedUrl
 from .types.file_create_from_url_resource_info_value import FileCreateFromUrlResourceInfoValue
@@ -495,6 +496,141 @@ class FilesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_file_pages_figures(
+        self, id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[PageFigureMetadata]:
+        """
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.files.list_file_pages_figures(
+            id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page-figures"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[PageFigureMetadata], _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 list_file_page_figures(
+        self,
+        id: str,
+        page_index: int,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+    ) -> typing.List[PageFigureMetadata]:
+        """
+        Parameters:
+            - id: str.
+
+            - page_index: int.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.files.list_file_page_figures(
+            id="string",
+            page_index=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page-figures/{page_index}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[PageFigureMetadata], _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 get_file_page_figure(
+        self,
+        id: str,
+        page_index: int,
+        figure_name: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+    ) -> typing.Any:
+        """
+        Parameters:
+            - id: str.
+
+            - page_index: int.
+
+            - figure_name: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.files.get_file_page_figure(
+            id="string",
+            page_index=1,
+            figure_name="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page-figures/{page_index}/{figure_name}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _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)
+
 
 class AsyncFilesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -960,3 +1096,138 @@ class AsyncFilesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def list_file_pages_figures(
+        self, id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[PageFigureMetadata]:
+        """
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.files.list_file_pages_figures(
+            id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page-figures"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[PageFigureMetadata], _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 list_file_page_figures(
+        self,
+        id: str,
+        page_index: int,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+    ) -> typing.List[PageFigureMetadata]:
+        """
+        Parameters:
+            - id: str.
+
+            - page_index: int.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.files.list_file_page_figures(
+            id="string",
+            page_index=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page-figures/{page_index}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[PageFigureMetadata], _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 get_file_page_figure(
+        self,
+        id: str,
+        page_index: int,
+        figure_name: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+    ) -> typing.Any:
+        """
+        Parameters:
+            - id: str.
+
+            - page_index: int.
+
+            - figure_name: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.files.get_file_page_figure(
+            id="string",
+            page_index=1,
+            figure_name="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}/page-figures/{page_index}/{figure_name}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _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)