athena-intelligence 0.1.107__py3-none-any.whl → 0.1.109__py3-none-any.whl

athena/client.py

@@ -48,7 +48,7 @@ class AthenaAsset:
     data: bytes
     media_type: str
 
-    def _repr_mimebundle_(self):
+    def _repr_mimebundle_(self, include=None, exclude=None):
         if self.media_type == "application/sql":
             # it is safe to import IPython in `_repr_mimebundle_`
             # as this is only intended to be invoked by IPython.
@@ -213,9 +213,8 @@ class WrappedToolsClient(ToolsClient):
             "application/sql",
         }
 
-        data = file_bytes.read()
-
         if media_type in supported_media_types:
+            data = file_bytes.read()
             return AthenaAsset(
                 asset_id=asset_id,
                 data=data,
@@ -223,7 +222,7 @@ class WrappedToolsClient(ToolsClient):
             )
 
         if media_type in _pandas_media_types:
-            return _to_pandas_df(file_bytes)
+            return _to_pandas_df(file_bytes, media_type=media_type)
 
         raise NotImplementedError("Assets of `{media_type}` type are not yet supported")
     
@@ -488,21 +487,27 @@ _pandas_media_types = {
 }
 
 
-def _to_pandas_df(bytes_io: io.BytesIO, *args, **kwargs):
+def _to_pandas_df(
+    bytes_io: io.BytesIO, *args, media_type: Union[str, None] = None, **kwargs
+):
     import pandas as pd
 
-    media_type = _infer_media_type(bytes_io)
+    if media_type is None:
+        media_type = _infer_media_type(bytes_io)
 
-    if media_type == 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet':
-        return pd.read_excel(bytes_io, *args, engine='openpyxl', **kwargs)
-    elif media_type in {'application/vnd.apache.parquet', 'application/octet-stream'}:
+    if (
+        media_type
+        == "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    ):
+        return pd.read_excel(bytes_io, *args, engine="openpyxl", **kwargs)
+    elif media_type in {"application/vnd.apache.parquet", "application/octet-stream"}:
         return pd.read_parquet(bytes_io, *args, **kwargs)
-    elif media_type == 'application/vnd.ms-excel':
+    elif media_type == "application/vnd.ms-excel":
         return pd.read_excel(bytes_io, *args, **kwargs)
-    elif media_type in {'text/csv', 'application/csv'}:
+    elif media_type in {"text/csv", "application/csv"}:
         return pd.read_csv(bytes_io, *args, **kwargs)
     else:
-        raise Exception("Unknown media type")
+        raise Exception(f"Unknown media type: {media_type}")
 
 
 def _convert_asset_object(

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

{athena_intelligence-0.1.107.dist-info → athena_intelligence-0.1.109.dist-info}/METADATA

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

{athena_intelligence-0.1.107.dist-info → athena_intelligence-0.1.109.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=OhZA99pgQ2kEy18GoyUt25AZdqjQ49YdXvaewz3llLU,17939
+athena/client.py,sha256=ftMZDJDpMR-2l-cpLmbDsSHN8jha6BSmpbUWukoLg28,18106
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=A07xONnQhSHb3pTSjJC_rKQUFDhHJycf4wvFEh5XkfI,1806
+athena/core/client_wrapper.py,sha256=1URZtdMR1zPYGotw65fW_iGp0HwLh4fm3nk9JgzpZa4,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
@@ -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.107.dist-info/METADATA,sha256=wuYxYlnBqbPOgg6rVwFafRSoIWxRjvn8QnbhenZExgg,5274
-athena_intelligence-0.1.107.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.107.dist-info/RECORD,,
+athena_intelligence-0.1.109.dist-info/METADATA,sha256=WOKB_OigPPFU18nCgFU1SHBgBhnFH9FLGJqzJlI4eXw,5274
+athena_intelligence-0.1.109.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.109.dist-info/RECORD,,