athena-intelligence 0.1.111__py3-none-any.whl → 0.1.113__py3-none-any.whl

athena/__init__.py

@@ -7,6 +7,8 @@ from .types import (
     DataFrameRequestOutDataItemItem,
     DataFrameRequestOutIndexItem,
     DataFrameUnknownFormatError,
+    DocumentChunk,
+    FileChunkRequestOut,
     FileTooLargeError,
     ParentFolderError,
     SaveAssetRequestOut,
@@ -36,6 +38,8 @@ __all__ = [
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
+    "DocumentChunk",
+    "FileChunkRequestOut",
     "FileTooLargeError",
     "InternalServerError",
     "NotFoundError",

athena/client.py

@@ -131,7 +131,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 +515,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 +528,18 @@ 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:
+                return (
+                    name or f"Untitled {label}",
+                    data[media_type],
+                    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.111",
+            "X-Fern-SDK-Version": "0.1.113",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/tools/client.py

@@ -18,6 +18,7 @@ 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_chunk_request_out import FileChunkRequestOut
 from ..types.file_too_large_error import FileTooLargeError
 from ..types.parent_folder_error import ParentFolderError
 from ..types.save_asset_request_out import SaveAssetRequestOut
@@ -31,6 +32,60 @@ class ToolsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    def get_file_chunks(
+        self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> FileChunkRequestOut:
+        """
+        Get the chunks of a file.
+
+        Parameters
+        ----------
+        asset_ids : typing.Sequence[str]
+            Identifiers of the assets
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        FileChunkRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.get_file_chunks(
+            asset_ids=[
+                "asset_9249292-d118-42d3-96b4-00eccfe0754f",
+                "asset_9249292-d118-42d3-95b4-01eccfe0754f",
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/file/chunks",
+            method="POST",
+            json={"asset_ids": asset_ids},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(FileChunkRequestOut, _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(AssetNotFoundError, _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)
+
     def data_frame(
         self,
         *,
@@ -226,6 +281,60 @@ class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    async def get_file_chunks(
+        self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> FileChunkRequestOut:
+        """
+        Get the chunks of a file.
+
+        Parameters
+        ----------
+        asset_ids : typing.Sequence[str]
+            Identifiers of the assets
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        FileChunkRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.get_file_chunks(
+            asset_ids=[
+                "asset_9249292-d118-42d3-96b4-00eccfe0754f",
+                "asset_9249292-d118-42d3-95b4-01eccfe0754f",
+            ],
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/file/chunks",
+            method="POST",
+            json={"asset_ids": asset_ids},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(FileChunkRequestOut, _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(AssetNotFoundError, _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)
+
     async def data_frame(
         self,
         *,

athena/types/__init__.py

@@ -6,6 +6,8 @@ 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 .document_chunk import DocumentChunk
+from .file_chunk_request_out import FileChunkRequestOut
 from .file_too_large_error import FileTooLargeError
 from .parent_folder_error import ParentFolderError
 from .save_asset_request_out import SaveAssetRequestOut
@@ -17,6 +19,8 @@ __all__ = [
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
+    "DocumentChunk",
+    "FileChunkRequestOut",
     "FileTooLargeError",
     "ParentFolderError",
     "SaveAssetRequestOut",

athena/types/document_chunk.py

@@ -0,0 +1,34 @@
+# 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 DocumentChunk(pydantic_v1.BaseModel):
+    """
+    A document chunk.
+    """
+
+    metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
+    page_content: 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/file_chunk_request_out.py

@@ -0,0 +1,34 @@
+# 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
+from .document_chunk import DocumentChunk
+
+
+class FileChunkRequestOut(pydantic_v1.BaseModel):
+    """
+    Response model for a file chunk.
+    """
+
+    chunks: typing.Dict[str, typing.List[DocumentChunk]]
+
+    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.111.dist-info → athena_intelligence-0.1.113.dist-info}/METADATA

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

{athena_intelligence-0.1.111.dist-info → athena_intelligence-0.1.113.dist-info}/RECORD

@@ -1,9 +1,9 @@
-athena/__init__.py,sha256=EtDa2cZbHfqJnBxCrjErWF1oJWH0cbJbFB7bkxeOFnE,1392
+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=em5UYFrE-2NMO391Stf6p7n1vTDqK-qfyOl1ajuo5aI,18765
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=-qxfDxMRONxgjZ6ROPWhwyi-ZzXHntkQQ5NhIPTuQAo,1806
+athena/core/client_wrapper.py,sha256=gLeaSAPiG-nCTmstAt7nS4I_wTVggpdEg54ShE5Y8w8,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,20 +27,22 @@ 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=5oFszmOEfdzZJNV5QHjkdyThhEswWAnWXObmK8halkE,16114
+athena/tools/client.py,sha256=1l9MrzkTlZdszyuLLrqF_kREt11X2DjXjsp2pCIsBMo,20203
 athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
 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/__init__.py,sha256=863IuasnsjdZ4LFKjfQhczKesHHp_Ey68A9AWuNcDlc,1085
 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/document_chunk.py,sha256=deXiiMA_U5EabUh1Fg2AB4ElYuc5OsTnrU7JwLApJmA,1227
+athena/types/file_chunk_request_out.py,sha256=Ju0I_UpSitjQ-XqSIRvvg2XA6QtfCLZClPq5PUqmPNg,1258
 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.111.dist-info/METADATA,sha256=VXeitKtL5DVBMte_QGa3DeGV4LUL9aiRvjCTsEoe1Pw,5274
-athena_intelligence-0.1.111.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.111.dist-info/RECORD,,
+athena_intelligence-0.1.113.dist-info/METADATA,sha256=XegkvOoDvfCcAxFft1S8_vtcBxwuox-4aTOZABdHUvo,5274
+athena_intelligence-0.1.113.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.113.dist-info/RECORD,,