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

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.104",
+            "X-Fern-SDK-Version": "0.1.105",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/tools/client.py

@@ -179,6 +179,7 @@ class ToolsClient:
             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.
@@ -373,6 +374,7 @@ class AsyncToolsClient:
             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.

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_intelligence-0.1.104.dist-info → athena_intelligence-0.1.105.dist-info}/METADATA

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

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

@@ -1,9 +1,9 @@
 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=CVCLfzte1vJ41WQFSEMvtOxD-Tdhu47d61bYsisHkgw,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
@@ -27,9 +27,9 @@ 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=yjDglwWsGjbeznHhpKCqBagx8Or5k_bM_i59Vz7ZX5g,15966
+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/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
@@ -41,6 +41,6 @@ athena/types/file_too_large_error.py,sha256=AinkrcgR7lcTILAD8RX0x48P3GlSoAh1Oihx
 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.104.dist-info/METADATA,sha256=YGcU_KVOxR5QrY9nZ4z2iWbM3T9sjcdm7-qkFFkY9XQ,5274
-athena_intelligence-0.1.104.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.104.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,,