llama-cloud 0.0.1__py3-none-any.whl

llama_cloud/resources/files/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .types import FileCreateResourceInfoValue
+
+__all__ = ["FileCreateResourceInfoValue"]

llama_cloud/resources/files/client.py

@@ -0,0 +1,560 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+import urllib.parse
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.jsonable_encoder import jsonable_encoder
+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.presigned_url import PresignedUrl
+from .types.file_create_resource_info_value import FileCreateResourceInfoValue
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class FilesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def read_file(self, id: str, *, project_id: typing.Optional[str] = None) -> File:
+        """
+        Read File metadata objects.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import PlatformApi
+
+        client = PlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.files.read_file(
+            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}"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 delete_file(self, id: str, *, project_id: typing.Optional[str] = None) -> None:
+        """
+        Delete the file from S3.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import PlatformApi
+
+        client = PlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.files.delete_file(
+            id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        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 read_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
+        """
+        Read File metadata objects.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import PlatformApi
+
+        client = PlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.files.read_files()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[File], _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, *, project_id: typing.Optional[str] = None, upload_file: typing.IO) -> File:
+        """
+        Upload a file to S3.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - upload_file: typing.IO.
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            data=jsonable_encoder({}),
+            files={"upload_file": upload_file},
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 generate_presigned_url(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        name: str,
+        file_size: typing.Optional[int] = OMIT,
+        last_modified_at: typing.Optional[dt.datetime] = OMIT,
+        resource_info: typing.Optional[typing.Dict[str, FileCreateResourceInfoValue]] = OMIT,
+        data_source_id: typing.Optional[str] = OMIT,
+    ) -> PresignedUrl:
+        """
+        Create a presigned url for uploading a file.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - name: str.
+
+            - file_size: typing.Optional[int]. Size of the file in bytes
+
+            - last_modified_at: typing.Optional[dt.datetime]. The last modified time of the file
+
+            - resource_info: typing.Optional[typing.Dict[str, FileCreateResourceInfoValue]]. Resource information for the file
+
+            - data_source_id: typing.Optional[str]. The ID of the data source that the file belongs to
+        ---
+        from platform.client import PlatformApi
+
+        client = PlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.files.generate_presigned_url(
+            name="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"name": name}
+        if file_size is not OMIT:
+            _request["file_size"] = file_size
+        if last_modified_at is not OMIT:
+            _request["last_modified_at"] = last_modified_at
+        if resource_info is not OMIT:
+            _request["resource_info"] = resource_info
+        if data_source_id is not OMIT:
+            _request["data_source_id"] = data_source_id
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files"),
+            params=remove_none_from_dict({"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(PresignedUrl, _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 sync_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
+        """
+        Sync Files API against file contents uploaded via S3 presigned urls.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import PlatformApi
+
+        client = PlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.files.sync_files()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files/sync"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[File], _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 read_file_content(self, id: str, *, project_id: typing.Optional[str] = None) -> PresignedUrl:
+        """
+        Returns a presigned url to read the file content.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import PlatformApi
+
+        client = PlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.files.read_file_content(
+            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}/content"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PresignedUrl, _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):
+        self._client_wrapper = client_wrapper
+
+    async def read_file(self, id: str, *, project_id: typing.Optional[str] = None) -> File:
+        """
+        Read File metadata objects.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import AsyncPlatformApi
+
+        client = AsyncPlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.files.read_file(
+            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}"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 delete_file(self, id: str, *, project_id: typing.Optional[str] = None) -> None:
+        """
+        Delete the file from S3.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import AsyncPlatformApi
+
+        client = AsyncPlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.files.delete_file(
+            id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/files/{id}"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        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 read_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
+        """
+        Read File metadata objects.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import AsyncPlatformApi
+
+        client = AsyncPlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.files.read_files()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[File], _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, *, project_id: typing.Optional[str] = None, upload_file: typing.IO) -> File:
+        """
+        Upload a file to S3.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - upload_file: typing.IO.
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            data=jsonable_encoder({}),
+            files={"upload_file": upload_file},
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 generate_presigned_url(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        name: str,
+        file_size: typing.Optional[int] = OMIT,
+        last_modified_at: typing.Optional[dt.datetime] = OMIT,
+        resource_info: typing.Optional[typing.Dict[str, FileCreateResourceInfoValue]] = OMIT,
+        data_source_id: typing.Optional[str] = OMIT,
+    ) -> PresignedUrl:
+        """
+        Create a presigned url for uploading a file.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - name: str.
+
+            - file_size: typing.Optional[int]. Size of the file in bytes
+
+            - last_modified_at: typing.Optional[dt.datetime]. The last modified time of the file
+
+            - resource_info: typing.Optional[typing.Dict[str, FileCreateResourceInfoValue]]. Resource information for the file
+
+            - data_source_id: typing.Optional[str]. The ID of the data source that the file belongs to
+        ---
+        from platform.client import AsyncPlatformApi
+
+        client = AsyncPlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.files.generate_presigned_url(
+            name="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"name": name}
+        if file_size is not OMIT:
+            _request["file_size"] = file_size
+        if last_modified_at is not OMIT:
+            _request["last_modified_at"] = last_modified_at
+        if resource_info is not OMIT:
+            _request["resource_info"] = resource_info
+        if data_source_id is not OMIT:
+            _request["data_source_id"] = data_source_id
+        _response = await self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files"),
+            params=remove_none_from_dict({"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(PresignedUrl, _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 sync_files(self, *, project_id: typing.Optional[str] = None) -> typing.List[File]:
+        """
+        Sync Files API against file contents uploaded via S3 presigned urls.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import AsyncPlatformApi
+
+        client = AsyncPlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.files.sync_files()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/files/sync"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[File], _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 read_file_content(self, id: str, *, project_id: typing.Optional[str] = None) -> PresignedUrl:
+        """
+        Returns a presigned url to read the file content.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from platform.client import AsyncPlatformApi
+
+        client = AsyncPlatformApi(
+            token="YOUR_TOKEN",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        await client.files.read_file_content(
+            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}/content"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PresignedUrl, _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)

llama_cloud/resources/files/types/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .file_create_resource_info_value import FileCreateResourceInfoValue
+
+__all__ = ["FileCreateResourceInfoValue"]

llama_cloud/resources/files/types/file_create_resource_info_value.py

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

llama_cloud/resources/parsing/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+