athena-intelligence 0.1.112__py3-none-any.whl → 0.1.114__py3-none-any.whl

athena/client.py

@@ -1,3 +1,4 @@
+import base64
 import dataclasses
 import enum
 import io
@@ -131,7 +132,7 @@ class WrappedToolsClient(ToolsClient):
         Parameters
         ----------
         asset_object : pd.DataFrame | pd.Series | core.File
-            A pandas data frame, series, or core.File
+            A pandas data frame, series, matplotlib figure, or core.File
 
         parent_folder_id : typing.Optional[str]
             Identifier of the folder into which the asset should be saved
@@ -515,6 +516,10 @@ def _convert_asset_object(
     name: Union[str, None] = None,
 ) -> core.File:
     import pandas as pd
+    try:
+        from IPython.core.formatters import format_display_data    # type: ignore[import]
+    except ImportError:
+        format_display_data = None
 
     if isinstance(asset_object, pd.Series):
         asset_object = asset_object.to_frame()
@@ -524,4 +529,20 @@ def _convert_asset_object(
             asset_object.to_parquet(path=None),
             "application/vnd.apache.parquet",
         )
+    if format_display_data:
+        data, _metadata = format_display_data(asset_object)
+        image_types = {
+            "image/png": "Plot",
+            "image/jpeg": "Image",
+            "image/gif": "Gif",
+        }
+        for media_type, label in image_types.items():
+            if media_type in data:
+                extension = media_type.split("/")[-1]
+                image_bytes = base64.b64decode(data[media_type])
+                return (
+                    name or f"Untitled {label}.{extension}",
+                    image_bytes,
+                    media_type,
+                )
     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.112",
+            "X-Fern-SDK-Version": "0.1.114",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.112.dist-info → athena_intelligence-0.1.114.dist-info}/METADATA

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

{athena_intelligence-0.1.112.dist-info → athena_intelligence-0.1.114.dist-info}/RECORD

@@ -1,9 +1,9 @@
 athena/__init__.py,sha256=5ayB3YWH8AYc_ahG2b8ovLPX2u5xugauR2luprNXi0E,1484
 athena/base_client.py,sha256=TlP599mOBvj7Tk8IpFe5dgrDM98GJu3lEQh_zwl4vtA,5439
-athena/client.py,sha256=8CW7e_aBNkSnrMfMwbZx1BRpwp1MPONa0Uv97D-WY7w,18115
+athena/client.py,sha256=2KT03Oi_7-qEgPY8D9r-Ek5KoyEfl0c-AT3PIMj-IPA,18905
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=uzmbiYLp0guW6cRqRO65YdU1UiV6w4D1tssJFEvmbIM,1806
+athena/core/client_wrapper.py,sha256=D5OjJciHKkoas1wU78BjbtSB60puDT6QZy8hpfiwqYA,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
@@ -43,6 +43,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.112.dist-info/METADATA,sha256=wzsKc_psT7edVTZlvNOEvaB5sxdfXFElDOKpHs3-hmI,5274
-athena_intelligence-0.1.112.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.112.dist-info/RECORD,,
+athena_intelligence-0.1.114.dist-info/METADATA,sha256=b4O9CEbZ_R78QswvVNnsLCY51MdJ2JUOUSpstKCMxoo,5274
+athena_intelligence-0.1.114.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.114.dist-info/RECORD,,