athena-intelligence 0.1.103__py3-none-any.whl → 0.1.105__py3-none-any.whl

athena/__init__.py

@@ -1,15 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AssetNotFoundError,
     DataFrameRequestOut,
     DataFrameRequestOutColumnsItem,
     DataFrameRequestOutDataItemItem,
     DataFrameRequestOutIndexItem,
     DataFrameUnknownFormatError,
-    FileFetchError,
-    SqlUnauthorizedError,
+    FileTooLargeError,
+    ParentFolderError,
+    SaveAssetRequestOut,
 )
 from .errors import (
+    BadRequestError,
+    ContentTooLargeError,
     InternalServerError,
     NotFoundError,
     UnauthorizedError,
@@ -23,17 +27,21 @@ from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
+    "AssetNotFoundError",
     "AthenaEnvironment",
+    "BadRequestError",
+    "ContentTooLargeError",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
-    "FileFetchError",
+    "FileTooLargeError",
     "InternalServerError",
     "NotFoundError",
+    "ParentFolderError",
     "QueryExecuteRequestDatabaseAssetIds",
-    "SqlUnauthorizedError",
+    "SaveAssetRequestOut",
     "ToolsDataFrameRequestColumnsItem",
     "UnauthorizedError",
     "UnprocessableEntityError",

athena/client.py

@@ -8,11 +8,13 @@ import httpx
 from typing import cast, List, Union
 from typing_extensions import TypeVar, ParamSpec
 
+from . import core
 from .base_client import BaseAthena, AsyncBaseAthena
 from .environment import AthenaEnvironment
 from .tools.client import AsyncToolsClient, ToolsClient
 from .query.client import AsyncQueryClient, QueryClient
 from .types.data_frame_request_out import DataFrameRequestOut
+from .types.save_asset_request_out import SaveAssetRequestOut
 
 if typing.TYPE_CHECKING:
     import pandas as pd
@@ -96,6 +98,43 @@ class WrappedToolsClient(ToolsClient):
         return _to_pandas_df(file_bytes, *args, **kwargs)
 
 
+    def save_asset(    # type: ignore[override]
+        self,
+        asset_object: Union["pd.DataFrame", "pd.Series", core.File],
+        *,
+        parent_folder_id: Union[str, None] = None,
+        name: Union[str, None] = None,
+        **kwargs,
+    ) -> SaveAssetRequestOut:
+        """
+        Parameters
+        ----------
+        asset_object : pd.DataFrame | pd.Series | core.File
+            A pandas data frame, series, or core.File
+
+        parent_folder_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SaveAssetRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(api_key="YOUR_API_KEY")
+        client.tools.save_asset(df)
+        """
+        asset_object = _convert_asset_object(asset_object=asset_object, name=name)
+        return super().save_asset(
+            file=asset_object, parent_folder_id=parent_folder_id, **kwargs
+        )
+    
+
 class WrappedQueryClient(QueryClient):
 
     @_inherit_signature_and_doc(QueryClient.execute, {'DataFrameRequestOut': 'pd.DataFrame'})
@@ -148,6 +187,20 @@ class WrappedAsyncToolsClient(AsyncToolsClient):
         file_bytes = await self.get_file(asset_id)
         return _to_pandas_df(file_bytes, *args, **kwargs)
 
+    @_add_docs_for_async_variant(WrappedToolsClient.save_asset)
+    async def save_asset(
+        self,
+        asset_object: Union["pd.DataFrame", "pd.Series", core.File],
+        *,
+        parent_folder_id: Union[str, None] = None,
+        name: Union[str, None] = None,
+        **kwargs,
+    ) -> SaveAssetRequestOut:
+        asset_object = _convert_asset_object(asset_object=asset_object, name=name)
+        return await super().save_asset(
+            file=asset_object, parent_folder_id=parent_folder_id, **kwargs
+        )
+
 
 class WrappedAsyncQueryClient(AsyncQueryClient):
 
@@ -333,9 +386,28 @@ def _to_pandas_df(bytes_io: io.BytesIO, *args, **kwargs):
 
     if media_type == 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet':
         return pd.read_excel(bytes_io, *args, engine='openpyxl', **kwargs)
+    elif media_type == 'application/vnd.apache.parquet':
+        return pd.read_parquet(bytes_io, *args, **kwargs)
     elif media_type == 'application/vnd.ms-excel':
         return pd.read_excel(bytes_io, *args, **kwargs)
-    elif media_type == 'text/csv':
+    elif media_type in {'text/csv', 'application/csv'}:
         return pd.read_csv(bytes_io, *args, **kwargs)
     else:
         raise Exception("Unknown media type")
+
+
+def _convert_asset_object(
+    asset_object: Union["pd.DataFrame", "pd.Series", core.File],
+    name: Union[str, None] = None,
+) -> core.File:
+    import pandas as pd
+
+    if isinstance(asset_object, pd.Series):
+        asset_object = asset_object.to_frame()
+    if isinstance(asset_object, pd.DataFrame):
+        return (
+            name or "Uploaded data frame",
+            asset_object.to_parquet(path=None),
+            "application/vnd.apache.parquet",
+        )
+    return asset_object    # type: ignore[return-value]

athena/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.103",
+            "X-Fern-SDK-Version": "0.1.105",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/errors/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .bad_request_error import BadRequestError
+from .content_too_large_error import ContentTooLargeError
 from .internal_server_error import InternalServerError
 from .not_found_error import NotFoundError
 from .unauthorized_error import UnauthorizedError
@@ -7,6 +9,8 @@ from .unprocessable_entity_error import UnprocessableEntityError
 from .unsupported_media_type_error import UnsupportedMediaTypeError
 
 __all__ = [
+    "BadRequestError",
+    "ContentTooLargeError",
     "InternalServerError",
     "NotFoundError",
     "UnauthorizedError",

athena/errors/bad_request_error.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.api_error import ApiError
+from ..types.parent_folder_error import ParentFolderError
+
+
+class BadRequestError(ApiError):
+    def __init__(self, body: ParentFolderError):
+        super().__init__(status_code=400, body=body)

athena/errors/content_too_large_error.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.api_error import ApiError
+from ..types.file_too_large_error import FileTooLargeError
+
+
+class ContentTooLargeError(ApiError):
+    def __init__(self, body: FileTooLargeError):
+        super().__init__(status_code=413, body=body)

athena/errors/not_found_error.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.api_error import ApiError
-from ..types.file_fetch_error import FileFetchError
+from ..types.asset_not_found_error import AssetNotFoundError
 
 
 class NotFoundError(ApiError):
-    def __init__(self, body: FileFetchError):
+    def __init__(self, body: AssetNotFoundError):
         super().__init__(status_code=404, body=body)

athena/errors/unauthorized_error.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import typing
+
 from ..core.api_error import ApiError
-from ..types.sql_unauthorized_error import SqlUnauthorizedError
 
 
 class UnauthorizedError(ApiError):
-    def __init__(self, body: SqlUnauthorizedError):
+    def __init__(self, body: typing.Any):
         super().__init__(status_code=401, body=body)

athena/query/client.py

@@ -12,7 +12,6 @@ from ..errors.internal_server_error import InternalServerError
 from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.data_frame_request_out import DataFrameRequestOut
-from ..types.sql_unauthorized_error import SqlUnauthorizedError
 from .types.query_execute_request_database_asset_ids import QueryExecuteRequestDatabaseAssetIds
 
 
@@ -67,7 +66,7 @@ class QueryClient:
         if 200 <= _response.status_code < 300:
             return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
         if _response.status_code == 401:
-            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
@@ -116,7 +115,7 @@ class QueryClient:
         if 200 <= _response.status_code < 300:
             return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
         if _response.status_code == 401:
-            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
@@ -179,7 +178,7 @@ class AsyncQueryClient:
         if 200 <= _response.status_code < 300:
             return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
         if _response.status_code == 401:
-            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
@@ -228,7 +227,7 @@ class AsyncQueryClient:
         if 200 <= _response.status_code < 300:
             return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
         if _response.status_code == 401:
-            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:

athena/tools/client.py

@@ -3,19 +3,29 @@
 import typing
 from json.decoder import JSONDecodeError
 
+from .. import core
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.pydantic_utilities import pydantic_v1
 from ..core.request_options import RequestOptions
+from ..errors.bad_request_error import BadRequestError
+from ..errors.content_too_large_error import ContentTooLargeError
 from ..errors.internal_server_error import InternalServerError
 from ..errors.not_found_error import NotFoundError
+from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..errors.unsupported_media_type_error import UnsupportedMediaTypeError
+from ..types.asset_not_found_error import AssetNotFoundError
 from ..types.data_frame_request_out import DataFrameRequestOut
 from ..types.data_frame_unknown_format_error import DataFrameUnknownFormatError
-from ..types.file_fetch_error import FileFetchError
+from ..types.file_too_large_error import FileTooLargeError
+from ..types.parent_folder_error import ParentFolderError
+from ..types.save_asset_request_out import SaveAssetRequestOut
 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, ...)
+
 
 class ToolsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -86,8 +96,10 @@ class ToolsClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 404:
-            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+            raise NotFoundError(pydantic_v1.parse_obj_as(AssetNotFoundError, _response.json()))  # type: ignore
         if _response.status_code == 415:
             raise UnsupportedMediaTypeError(
                 pydantic_v1.parse_obj_as(DataFrameUnknownFormatError, _response.json())  # type: ignore
@@ -139,16 +151,76 @@ class ToolsClient:
                     yield _chunk
                 return
             _response.read()
+            if _response.status_code == 401:
+                raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 404:
-                raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+                raise NotFoundError(pydantic_v1.parse_obj_as(AssetNotFoundError, _response.json()))  # type: ignore
             if _response.status_code == 422:
                 raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            if _response.status_code == 500:
+                raise InternalServerError(pydantic_v1.parse_obj_as(typing.Any, _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 save_asset(
+        self,
+        *,
+        file: core.File,
+        parent_folder_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> SaveAssetRequestOut:
+        """
+        Parameters
+        ----------
+        file : core.File
+            See core.File for more documentation
+
+        parent_folder_id : typing.Optional[str]
+            Identifier of the folder into which the asset should be saved
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SaveAssetRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        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,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(SaveAssetRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic_v1.parse_obj_as(ParentFolderError, _response.json()))  # type: ignore
+        if _response.status_code == 413:
+            raise ContentTooLargeError(pydantic_v1.parse_obj_as(FileTooLargeError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _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 AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -219,8 +291,10 @@ class AsyncToolsClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 404:
-            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+            raise NotFoundError(pydantic_v1.parse_obj_as(AssetNotFoundError, _response.json()))  # type: ignore
         if _response.status_code == 415:
             raise UnsupportedMediaTypeError(
                 pydantic_v1.parse_obj_as(DataFrameUnknownFormatError, _response.json())  # type: ignore
@@ -272,12 +346,72 @@ class AsyncToolsClient:
                     yield _chunk
                 return
             await _response.aread()
+            if _response.status_code == 401:
+                raise UnauthorizedError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 404:
-                raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+                raise NotFoundError(pydantic_v1.parse_obj_as(AssetNotFoundError, _response.json()))  # type: ignore
             if _response.status_code == 422:
                 raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            if _response.status_code == 500:
+                raise InternalServerError(pydantic_v1.parse_obj_as(typing.Any, _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 save_asset(
+        self,
+        *,
+        file: core.File,
+        parent_folder_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> SaveAssetRequestOut:
+        """
+        Parameters
+        ----------
+        file : core.File
+            See core.File for more documentation
+
+        parent_folder_id : typing.Optional[str]
+            Identifier of the folder into which the asset should be saved
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SaveAssetRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.save_asset()
+        """
+        _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,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(SaveAssetRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic_v1.parse_obj_as(ParentFolderError, _response.json()))  # type: ignore
+        if _response.status_code == 413:
+            raise ContentTooLargeError(pydantic_v1.parse_obj_as(FileTooLargeError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _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)

athena/tools/types/tools_data_frame_request_columns_item.py

@@ -2,4 +2,4 @@
 
 import typing
 
-ToolsDataFrameRequestColumnsItem = typing.Union[str, int]
+ToolsDataFrameRequestColumnsItem = typing.Union[int, str]

athena/types/__init__.py

@@ -1,19 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .asset_not_found_error import AssetNotFoundError
 from .data_frame_request_out import DataFrameRequestOut
 from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
 from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
 from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
 from .data_frame_unknown_format_error import DataFrameUnknownFormatError
-from .file_fetch_error import FileFetchError
-from .sql_unauthorized_error import SqlUnauthorizedError
+from .file_too_large_error import FileTooLargeError
+from .parent_folder_error import ParentFolderError
+from .save_asset_request_out import SaveAssetRequestOut
 
 __all__ = [
+    "AssetNotFoundError",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
-    "FileFetchError",
-    "SqlUnauthorizedError",
+    "FileTooLargeError",
+    "ParentFolderError",
+    "SaveAssetRequestOut",
 ]

athena/types/{sql_unauthorized_error.py → asset_not_found_error.py}

@@ -7,7 +7,7 @@ from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
-class SqlUnauthorizedError(pydantic_v1.BaseModel):
+class AssetNotFoundError(pydantic_v1.BaseModel):
     message: str
 
     def json(self, **kwargs: typing.Any) -> str:

athena/types/{file_fetch_error.py → file_too_large_error.py}

@@ -7,9 +7,8 @@ from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
-class FileFetchError(pydantic_v1.BaseModel):
+class FileTooLargeError(pydantic_v1.BaseModel):
     message: str
-    status_code: typing.Optional[int] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena/types/parent_folder_error.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class ParentFolderError(pydantic_v1.BaseModel):
+    message: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena/types/save_asset_request_out.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class SaveAssetRequestOut(pydantic_v1.BaseModel):
+    """
+    Response model asset information representation.
+    """
+
+    asset_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{athena_intelligence-0.1.103.dist-info → athena_intelligence-0.1.105.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.103
+Version: 0.1.105
 Summary: Athena Intelligence Python Library
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{athena_intelligence-0.1.103.dist-info → athena_intelligence-0.1.105.dist-info}/RECORD

@@ -1,9 +1,9 @@
-athena/__init__.py,sha256=lGo6ASUYoueZCu_oWrD7ruCsa0nLRNo9qZ_-2RV4WxY,1192
+athena/__init__.py,sha256=EtDa2cZbHfqJnBxCrjErWF1oJWH0cbJbFB7bkxeOFnE,1392
 athena/base_client.py,sha256=TlP599mOBvj7Tk8IpFe5dgrDM98GJu3lEQh_zwl4vtA,5439
-athena/client.py,sha256=2iXOf_xzfAqy_QcBWHaILM4RqmrF-TuZUULoA6wEtJw,12374
+athena/client.py,sha256=9hj9SzW7OhAIY8yKHA0WObOtx8-1W4aLSHduqkBHJT0,14722
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=Ef1urlkCpXDULFqlxmLoLIKgOvEH8E4WfiCfrIB5J-0,1806
+athena/core/client_wrapper.py,sha256=ijXhms74o95_1sms0mC87d_xv0uBTKXzxH7pmz9YEKE,1806
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=Z4NuAsJD-51yqmoME17O5sxwx5orSp1wsnd6bPyKcgA,17768
@@ -13,30 +13,34 @@ athena/core/query_encoder.py,sha256=sI6XiwFby-WRIk4MVN_5wlAlm30aCHE73Uvm09fb9qQ,
 athena/core/remove_none_from_dict.py,sha256=EU9SGgYidWq7SexuJbNs4-PZ-5Bl3Vppd864mS6vQZw,342
 athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
 athena/environment.py,sha256=_e7YwByXALEk7fNJNJvjF81caYTbsx4c7wwftCJHQ7g,214
-athena/errors/__init__.py,sha256=3hOHgz6h6rQZ_8URX6GpKLhKIhMVxYHANI-X8jTO1t0,499
+athena/errors/__init__.py,sha256=WlRIo4aGYcnxtY2DtoWS5YzmaKQHFp4Uh3mXxrakU_c,655
+athena/errors/bad_request_error.py,sha256=wfNVQgYe-5jsfeex-IBMrqn05oArDz0PT0ddj01S9aE,298
+athena/errors/content_too_large_error.py,sha256=_Jacqup8-YwalVeCv6NWDo9bqZhByXoUaApAxfFc9tU,304
 athena/errors/internal_server_error.py,sha256=E0rgqJC0-LcetLi1HmSi92KpvNkGSRCIdBeEqT_ln1s,252
-athena/errors/not_found_error.py,sha256=uBK3JapPPgTkMoSCX9j22C6MDtSuTuSPmjT0lDRigpA,287
-athena/errors/unauthorized_error.py,sha256=6qGByG08vA3iFHmK5ReHUK2AiGeCsK3Vf50C5LQNILM,309
+athena/errors/not_found_error.py,sha256=fIMBz7ubqcX3HpLIoqmF7-kFCoN6-Uj1lW4OuBnaK6s,300
+athena/errors/unauthorized_error.py,sha256=T-UWUDnIEmguQei9CYcRJ2SCt4DhZVxpuZRNAwK0Ru4,250
 athena/errors/unprocessable_entity_error.py,sha256=OztAzRntOJVxbZtXwoR4epahhG6chKOB5chVLUC8D9Q,257
 athena/errors/unsupported_media_type_error.py,sha256=fQ7TYQ3QYcT_YzFzO4f8_bLger7UQfZmuF22I9jYxFA,340
 athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 athena/query/__init__.py,sha256=EIeVs3aDVNAqLyWetSrQldzc2yUdJkf_cJXyrLdMDj0,171
-athena/query/client.py,sha256=3i_yiGD7JM204TsTwY7R1CvGRs0odCh6QkYeNggHbc8,9147
+athena/query/client.py,sha256=GYvBV7YFYjL3gyskeF6C8BJZlLvU7R045gJ5kGiYkvg,9043
 athena/query/types/__init__.py,sha256=WX-Or2h5NY2sv93ojrZsHcmiFHGYdqd0yxNo-5iGHR4,206
 athena/query/types/query_execute_request_database_asset_ids.py,sha256=aoVl5Xb34Q27hYGuVTnByGIxtHkL67wAwzXh7eJctew,154
 athena/tools/__init__.py,sha256=3n7oOoMebo06MAQqYRE2CX9Q0fTNnKBYE0cTlh1MPkM,165
-athena/tools/client.py,sha256=sMV_--PzUReXXILfKJeS8g9GQzYxxPDZI59wlos_rSo,10565
+athena/tools/client.py,sha256=5oFszmOEfdzZJNV5QHjkdyThhEswWAnWXObmK8halkE,16114
 athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
-athena/tools/types/tools_data_frame_request_columns_item.py,sha256=4rve98HoxzGpDhfEcNwqkH1mUQQXL8vAnX4T04PTTLk,138
-athena/types/__init__.py,sha256=9pojMYMfD3Bk4w4V8FXZ_Apwf1DFDCDsbKtqsMR6EK0,775
+athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
+athena/types/__init__.py,sha256=rN0nC7uMZooYmwp7F7rSIb4tIaH2bIr_yd5P1c7e-XY,939
+athena/types/asset_not_found_error.py,sha256=ZcgqRuzvO4Z8vVVxwtDB-QtKhpVIVV3hqQuJeUoOoJE,1121
 athena/types/data_frame_request_out.py,sha256=1CEBe-baDQi0uz_EgMw0TKGYXGj6KV44cL3ViRTZLKM,1669
 athena/types/data_frame_request_out_columns_item.py,sha256=9cjzciFv6C8n8Griytt_q_8ovkzHViS5tvUcMDfkfKE,143
 athena/types/data_frame_request_out_data_item_item.py,sha256=KMTJRr-1bdKDNMbUericCliwRoPHLGRV-n2bJtxdRW0,144
 athena/types/data_frame_request_out_index_item.py,sha256=bW7oe912trpkYKodj-I_AiTXXy61yWzliekcsUZkZE0,141
 athena/types/data_frame_unknown_format_error.py,sha256=lbgAUArEgIYZt3P5Ji4Clp-xyXnKJR7-v5VzXwKu0J8,1168
-athena/types/file_fetch_error.py,sha256=iR7IXjj2C4lJFcWV3aS2gfhmT3CE-_hhBDSWFaCktHU,1162
-athena/types/sql_unauthorized_error.py,sha256=QSQbBlOBVYYjvcmSLZbsr4twcanPeMxIl74sKUU3Hac,1123
+athena/types/file_too_large_error.py,sha256=AinkrcgR7lcTILAD8RX0x48P3GlSoAh1OihxMvSvRuo,1120
+athena/types/parent_folder_error.py,sha256=ZMF-i3mZY6Mu1n5uQ60Q3mIIfehlWuXtgFUkSYspkx8,1120
+athena/types/save_asset_request_out.py,sha256=5bpBaUV3oeuL_hz4s07c-6MQHkn4cBsyxgT_SD5oi6I,1193
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
-athena_intelligence-0.1.103.dist-info/METADATA,sha256=MWCffRWIKQNIUK97iEqSuliS9IJLadf_RPKEQ7h6Co4,5274
-athena_intelligence-0.1.103.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.103.dist-info/RECORD,,
+athena_intelligence-0.1.105.dist-info/METADATA,sha256=0-dSyqtK9SICscX0vlve617c46HjDvYdj6NdWuZcn9k,5274
+athena_intelligence-0.1.105.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.105.dist-info/RECORD,,