athena-intelligence 0.1.112__tar.gz → 0.1.114__tar.gz

{athena_intelligence-0.1.112 → athena_intelligence-0.1.114}/PKG-INFO

@@ -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 → athena_intelligence-0.1.114}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.112"
+version = "0.1.114"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.112 → athena_intelligence-0.1.114}/src/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_intelligence-0.1.112 → athena_intelligence-0.1.114}/src/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