athena-intelligence 0.1.126__py3-none-any.whl → 0.1.184__py3-none-any.whl

athena/tools/client.py

@@ -1,37 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
-from .calendar.client import CalendarClient
-from .email.client import EmailClient
-from .structured_data_extractor.client import StructuredDataExtractorClient
-from .tasks.client import TasksClient
+
+from .. import core
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from ..types.asset_content_request_out import AssetContentRequestOut
+from ..types.asset_screenshot_response_out import AssetScreenshotResponseOut
+from ..types.data_frame_request_out import DataFrameRequestOut
 from ..types.file_chunk_request_out import FileChunkRequestOut
-from ..core.pydantic_utilities import parse_obj_as
-from ..errors.unauthorized_error import UnauthorizedError
-from ..errors.not_found_error import NotFoundError
-from ..types.asset_not_found_error import AssetNotFoundError
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
 from ..types.folder_response import FolderResponse
-from ..errors.bad_request_error import BadRequestError
-from ..types.parent_folder_error import ParentFolderError
-from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
-from ..types.data_frame_request_out import DataFrameRequestOut
-from ..errors.unsupported_media_type_error import UnsupportedMediaTypeError
-from ..types.data_frame_unknown_format_error import DataFrameUnknownFormatError
-from ..errors.internal_server_error import InternalServerError
-from .. import core
 from ..types.save_asset_request_out import SaveAssetRequestOut
-from ..errors.content_too_large_error import ContentTooLargeError
-from ..types.file_too_large_error import FileTooLargeError
-from ..core.client_wrapper import AsyncClientWrapper
-from .calendar.client import AsyncCalendarClient
-from .email.client import AsyncEmailClient
-from .structured_data_extractor.client import AsyncStructuredDataExtractorClient
-from .tasks.client import AsyncTasksClient
+from .calendar.client import AsyncCalendarClient, CalendarClient
+from .email.client import AsyncEmailClient, EmailClient
+from .raw_client import AsyncRawToolsClient, RawToolsClient
+from .structured_data_extractor.client import AsyncStructuredDataExtractorClient, StructuredDataExtractorClient
+from .tasks.client import AsyncTasksClient, TasksClient
+from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -39,11 +24,25 @@ OMIT = typing.cast(typing.Any, ...)
 
 class ToolsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-        self.calendar = CalendarClient(client_wrapper=self._client_wrapper)
-        self.email = EmailClient(client_wrapper=self._client_wrapper)
-        self.structured_data_extractor = StructuredDataExtractorClient(client_wrapper=self._client_wrapper)
-        self.tasks = TasksClient(client_wrapper=self._client_wrapper)
+        self._raw_client = RawToolsClient(client_wrapper=client_wrapper)
+        self.calendar = CalendarClient(client_wrapper=client_wrapper)
+
+        self.email = EmailClient(client_wrapper=client_wrapper)
+
+        self.structured_data_extractor = StructuredDataExtractorClient(client_wrapper=client_wrapper)
+
+        self.tasks = TasksClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawToolsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawToolsClient
+        """
+        return self._raw_client
 
     def get_asset_chunks(
         self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
@@ -78,75 +77,97 @@ class ToolsClient:
             ],
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/tools/asset/chunks",
-            method="POST",
-            json={
-                "asset_ids": asset_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.get_asset_chunks(asset_ids=asset_ids, request_options=request_options)
+        return _response.data
+
+    def get_asset_content(
+        self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AssetContentRequestOut:
+        """
+        Get the content of an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetContentRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    FileChunkRequestOut,
-                    parse_obj_as(
-                        type_=FileChunkRequestOut,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        AssetNotFoundError,
-                        parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        client.tools.get_asset_content(
+            asset_id="asset_id",
+        )
+        """
+        _response = self._raw_client.get_asset_content(asset_id=asset_id, request_options=request_options)
+        return _response.data
+
+    def get_asset_screenshot(
+        self,
+        *,
+        asset_id: str,
+        page_number: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AssetScreenshotResponseOut:
+        """
+        Get a screenshot of a specific page from an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        page_number : typing.Optional[int]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetScreenshotResponseOut
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.get_asset_screenshot(
+            asset_id="asset_id",
+        )
+        """
+        _response = self._raw_client.get_asset_screenshot(
+            asset_id=asset_id, page_number=page_number, request_options=request_options
+        )
+        return _response.data
 
     def list_contents(
         self,
         *,
+        asset_id: typing.Optional[str] = None,
         folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> FolderResponse:
         """
-        List contents of a folder or entire workspace in a tree structure.
+        List contents of an asset (Folder, Collection, Project) or entire workspace in a tree structure.
 
         Parameters
         ----------
+        asset_id : typing.Optional[str]
+
         folder_id : typing.Optional[str]
 
         include_asset_details : typing.Optional[bool]
@@ -170,69 +191,14 @@ class ToolsClient:
         )
         client.tools.list_contents()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/tools/contents",
-            method="GET",
-            params={
-                "folder_id": folder_id,
-                "include_asset_details": include_asset_details,
-                "include_system_files": include_system_files,
-            },
+        _response = self._raw_client.list_contents(
+            asset_id=asset_id,
+            folder_id=folder_id,
+            include_asset_details=include_asset_details,
+            include_system_files=include_system_files,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    FolderResponse,
-                    parse_obj_as(
-                        type_=FolderResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        ParentFolderError,
-                        parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        AssetNotFoundError,
-                        parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data
 
     def data_frame(
         self,
@@ -243,7 +209,7 @@ class ToolsClient:
         columns: typing.Optional[
             typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
         ] = None,
-        sheet_name: typing.Optional[int] = None,
+        sheet_name: typing.Optional[str] = None,
         separator: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DataFrameRequestOut:
@@ -259,7 +225,7 @@ class ToolsClient:
         columns : typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]
             should be a list of strings or a list of integers
 
-        sheet_name : typing.Optional[int]
+        sheet_name : typing.Optional[str]
             only for excel files
 
         separator : typing.Optional[str]
@@ -284,82 +250,16 @@ class ToolsClient:
             asset_id="asset_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/tools/file/data-frame",
-            method="GET",
-            params={
-                "asset_id": asset_id,
-                "row_limit": row_limit,
-                "index_column": index_column,
-                "columns": columns,
-                "sheet_name": sheet_name,
-                "separator": separator,
-            },
+        _response = self._raw_client.data_frame(
+            asset_id=asset_id,
+            row_limit=row_limit,
+            index_column=index_column,
+            columns=columns,
+            sheet_name=sheet_name,
+            separator=separator,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    DataFrameRequestOut,
-                    parse_obj_as(
-                        type_=DataFrameRequestOut,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        AssetNotFoundError,
-                        parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 415:
-                raise UnsupportedMediaTypeError(
-                    typing.cast(
-                        DataFrameUnknownFormatError,
-                        parse_obj_as(
-                            type_=DataFrameUnknownFormatError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data
 
     def raw_data(
         self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
@@ -374,70 +274,13 @@ class ToolsClient:
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
-        Yields
-        ------
+        Returns
+        -------
         typing.Iterator[bytes]
             Stream the file in original format.
         """
-        with self._client_wrapper.httpx_client.stream(
-            "api/v0/tools/file/raw-data",
-            method="GET",
-            params={
-                "asset_id": asset_id,
-            },
-            request_options=request_options,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
-                    for _chunk in _response.iter_bytes(chunk_size=_chunk_size):
-                        yield _chunk
-                    return
-                _response.read()
-                if _response.status_code == 401:
-                    raise UnauthorizedError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                if _response.status_code == 404:
-                    raise NotFoundError(
-                        typing.cast(
-                            AssetNotFoundError,
-                            parse_obj_as(
-                                type_=AssetNotFoundError,  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                if _response.status_code == 422:
-                    raise UnprocessableEntityError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                if _response.status_code == 500:
-                    raise InternalServerError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                _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)
+        with self._raw_client.raw_data(asset_id=asset_id, request_options=request_options) as r:
+            yield from r.data
 
     def save_asset(
         self,
@@ -472,71 +315,33 @@ class ToolsClient:
         )
         client.tools.save_asset()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/tools/file/save",
-            method="POST",
-            params={
-                "parent_folder_id": parent_folder_id,
-            },
-            data={},
-            files={
-                "file": file,
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.save_asset(
+            file=file, parent_folder_id=parent_folder_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    SaveAssetRequestOut,
-                    parse_obj_as(
-                        type_=SaveAssetRequestOut,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        ParentFolderError,
-                        parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 413:
-                raise ContentTooLargeError(
-                    typing.cast(
-                        FileTooLargeError,
-                        parse_obj_as(
-                            type_=FileTooLargeError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data
 
 
 class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-        self.calendar = AsyncCalendarClient(client_wrapper=self._client_wrapper)
-        self.email = AsyncEmailClient(client_wrapper=self._client_wrapper)
-        self.structured_data_extractor = AsyncStructuredDataExtractorClient(client_wrapper=self._client_wrapper)
-        self.tasks = AsyncTasksClient(client_wrapper=self._client_wrapper)
+        self._raw_client = AsyncRawToolsClient(client_wrapper=client_wrapper)
+        self.calendar = AsyncCalendarClient(client_wrapper=client_wrapper)
+
+        self.email = AsyncEmailClient(client_wrapper=client_wrapper)
+
+        self.structured_data_extractor = AsyncStructuredDataExtractorClient(client_wrapper=client_wrapper)
+
+        self.tasks = AsyncTasksClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawToolsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawToolsClient
+        """
+        return self._raw_client
 
     async def get_asset_chunks(
         self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
@@ -579,75 +384,113 @@ class AsyncToolsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/tools/asset/chunks",
-            method="POST",
-            json={
-                "asset_ids": asset_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.get_asset_chunks(asset_ids=asset_ids, request_options=request_options)
+        return _response.data
+
+    async def get_asset_content(
+        self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AssetContentRequestOut:
+        """
+        Get the content of an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetContentRequestOut
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.get_asset_content(
+                asset_id="asset_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_asset_content(asset_id=asset_id, request_options=request_options)
+        return _response.data
+
+    async def get_asset_screenshot(
+        self,
+        *,
+        asset_id: str,
+        page_number: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AssetScreenshotResponseOut:
+        """
+        Get a screenshot of a specific page from an asset.
+
+        Parameters
+        ----------
+        asset_id : str
+
+        page_number : typing.Optional[int]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AssetScreenshotResponseOut
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.get_asset_screenshot(
+                asset_id="asset_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_asset_screenshot(
+            asset_id=asset_id, page_number=page_number, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    FileChunkRequestOut,
-                    parse_obj_as(
-                        type_=FileChunkRequestOut,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        AssetNotFoundError,
-                        parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data
 
     async def list_contents(
         self,
         *,
+        asset_id: typing.Optional[str] = None,
         folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> FolderResponse:
         """
-        List contents of a folder or entire workspace in a tree structure.
+        List contents of an asset (Folder, Collection, Project) or entire workspace in a tree structure.
 
         Parameters
         ----------
+        asset_id : typing.Optional[str]
+
         folder_id : typing.Optional[str]
 
         include_asset_details : typing.Optional[bool]
@@ -679,69 +522,14 @@ class AsyncToolsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/tools/contents",
-            method="GET",
-            params={
-                "folder_id": folder_id,
-                "include_asset_details": include_asset_details,
-                "include_system_files": include_system_files,
-            },
+        _response = await self._raw_client.list_contents(
+            asset_id=asset_id,
+            folder_id=folder_id,
+            include_asset_details=include_asset_details,
+            include_system_files=include_system_files,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    FolderResponse,
-                    parse_obj_as(
-                        type_=FolderResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        ParentFolderError,
-                        parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        AssetNotFoundError,
-                        parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data
 
     async def data_frame(
         self,
@@ -752,7 +540,7 @@ class AsyncToolsClient:
         columns: typing.Optional[
             typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
         ] = None,
-        sheet_name: typing.Optional[int] = None,
+        sheet_name: typing.Optional[str] = None,
         separator: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DataFrameRequestOut:
@@ -768,7 +556,7 @@ class AsyncToolsClient:
         columns : typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]
             should be a list of strings or a list of integers
 
-        sheet_name : typing.Optional[int]
+        sheet_name : typing.Optional[str]
             only for excel files
 
         separator : typing.Optional[str]
@@ -801,82 +589,16 @@ class AsyncToolsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/tools/file/data-frame",
-            method="GET",
-            params={
-                "asset_id": asset_id,
-                "row_limit": row_limit,
-                "index_column": index_column,
-                "columns": columns,
-                "sheet_name": sheet_name,
-                "separator": separator,
-            },
+        _response = await self._raw_client.data_frame(
+            asset_id=asset_id,
+            row_limit=row_limit,
+            index_column=index_column,
+            columns=columns,
+            sheet_name=sheet_name,
+            separator=separator,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    DataFrameRequestOut,
-                    parse_obj_as(
-                        type_=DataFrameRequestOut,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        AssetNotFoundError,
-                        parse_obj_as(
-                            type_=AssetNotFoundError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 415:
-                raise UnsupportedMediaTypeError(
-                    typing.cast(
-                        DataFrameUnknownFormatError,
-                        parse_obj_as(
-                            type_=DataFrameUnknownFormatError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data
 
     async def raw_data(
         self, *, asset_id: str, request_options: typing.Optional[RequestOptions] = None
@@ -891,70 +613,14 @@ class AsyncToolsClient:
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
-        Yields
-        ------
+        Returns
+        -------
         typing.AsyncIterator[bytes]
             Stream the file in original format.
         """
-        async with self._client_wrapper.httpx_client.stream(
-            "api/v0/tools/file/raw-data",
-            method="GET",
-            params={
-                "asset_id": asset_id,
-            },
-            request_options=request_options,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
-                    async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size):
-                        yield _chunk
-                    return
-                await _response.aread()
-                if _response.status_code == 401:
-                    raise UnauthorizedError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                if _response.status_code == 404:
-                    raise NotFoundError(
-                        typing.cast(
-                            AssetNotFoundError,
-                            parse_obj_as(
-                                type_=AssetNotFoundError,  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                if _response.status_code == 422:
-                    raise UnprocessableEntityError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                if _response.status_code == 500:
-                    raise InternalServerError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
-                        )
-                    )
-                _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 with self._raw_client.raw_data(asset_id=asset_id, request_options=request_options) as r:
+            async for _chunk in r.data:
+                yield _chunk
 
     async def save_asset(
         self,
@@ -997,59 +663,7 @@ class AsyncToolsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/tools/file/save",
-            method="POST",
-            params={
-                "parent_folder_id": parent_folder_id,
-            },
-            data={},
-            files={
-                "file": file,
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.save_asset(
+            file=file, parent_folder_id=parent_folder_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    SaveAssetRequestOut,
-                    parse_obj_as(
-                        type_=SaveAssetRequestOut,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        ParentFolderError,
-                        parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 413:
-                raise ContentTooLargeError(
-                    typing.cast(
-                        FileTooLargeError,
-                        parse_obj_as(
-                            type_=FileTooLargeError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _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)
+        return _response.data