athena-intelligence 0.1.202__py3-none-any.whl → 0.1.203__py3-none-any.whl

athena/__init__.py

@@ -32,6 +32,7 @@ from .types import (
     GeneralAgentResponse,
     GeneralAgentResponseMessage,
     GeneralAgentResponseMessageKwargs,
+    GetTableResponse,
     Id,
     ImageUrlContent,
     InputMessage,
@@ -101,6 +102,7 @@ __all__ = [
     "GeneralAgentResponse",
     "GeneralAgentResponseMessage",
     "GeneralAgentResponseMessageKwargs",
+    "GetTableResponse",
     "Id",
     "ImageUrlContent",
     "InputMessage",

athena/core/client_wrapper.py

@@ -22,10 +22,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "athena-intelligence/0.1.202",
+            "User-Agent": "athena-intelligence/0.1.203",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.202",
+            "X-Fern-SDK-Version": "0.1.203",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-KEY"] = self.api_key

athena/tools/sheets/client.py

@@ -5,6 +5,7 @@ import typing
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.create_new_sheet_tab_response import CreateNewSheetTabResponse
+from ...types.get_table_response import GetTableResponse
 from ...types.sheet_operation_response import SheetOperationResponse
 from .raw_client import AsyncRawSheetsClient, RawSheetsClient
 
@@ -883,6 +884,52 @@ class SheetsClient:
         )
         return _response.data
 
+    def get_table(
+        self,
+        *,
+        asset_id: str,
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> GetTableResponse:
+        """
+        Retrieve table data from an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        table_id : typing.Optional[str]
+            Table ID to retrieve
+
+        table_name : typing.Optional[str]
+            Table name to retrieve
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GetTableResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.sheets.get_table(
+            asset_id="asset_id",
+        )
+        """
+        _response = self._raw_client.get_table(
+            asset_id=asset_id, table_id=table_id, table_name=table_name, request_options=request_options
+        )
+        return _response.data
+
     def update_table(
         self,
         *,
@@ -1947,6 +1994,60 @@ class AsyncSheetsClient:
         )
         return _response.data
 
+    async def get_table(
+        self,
+        *,
+        asset_id: str,
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> GetTableResponse:
+        """
+        Retrieve table data from an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        table_id : typing.Optional[str]
+            Table ID to retrieve
+
+        table_name : typing.Optional[str]
+            Table name to retrieve
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GetTableResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.sheets.get_table(
+                asset_id="asset_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_table(
+            asset_id=asset_id, table_id=table_id, table_name=table_name, request_options=request_options
+        )
+        return _response.data
+
     async def update_table(
         self,
         *,

athena/tools/sheets/raw_client.py

@@ -10,6 +10,7 @@ from ...core.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.create_new_sheet_tab_response import CreateNewSheetTabResponse
+from ...types.get_table_response import GetTableResponse
 from ...types.sheet_operation_response import SheetOperationResponse
 
 # this is used as the default value for optional parameters
@@ -1140,6 +1141,76 @@ class RawSheetsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    def get_table(
+        self,
+        *,
+        asset_id: str,
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[GetTableResponse]:
+        """
+        Retrieve table data from an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        table_id : typing.Optional[str]
+            Table ID to retrieve
+
+        table_name : typing.Optional[str]
+            Table name to retrieve
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[GetTableResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/sheets/table/get",
+            method="POST",
+            json={
+                "asset_id": asset_id,
+                "table_id": table_id,
+                "table_name": table_name,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    GetTableResponse,
+                    parse_obj_as(
+                        type_=GetTableResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
     def update_table(
         self,
         *,
@@ -2360,6 +2431,76 @@ class AsyncRawSheetsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    async def get_table(
+        self,
+        *,
+        asset_id: str,
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[GetTableResponse]:
+        """
+        Retrieve table data from an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        table_id : typing.Optional[str]
+            Table ID to retrieve
+
+        table_name : typing.Optional[str]
+            Table name to retrieve
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[GetTableResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/sheets/table/get",
+            method="POST",
+            json={
+                "asset_id": asset_id,
+                "table_id": table_id,
+                "table_name": table_name,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    GetTableResponse,
+                    parse_obj_as(
+                        type_=GetTableResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
     async def update_table(
         self,
         *,

athena/types/__init__.py

@@ -29,6 +29,7 @@ from .general_agent_request import GeneralAgentRequest
 from .general_agent_response import GeneralAgentResponse
 from .general_agent_response_message import GeneralAgentResponseMessage
 from .general_agent_response_message_kwargs import GeneralAgentResponseMessageKwargs
+from .get_table_response import GetTableResponse
 from .id import Id
 from .image_url_content import ImageUrlContent
 from .input_message import InputMessage
@@ -79,6 +80,7 @@ __all__ = [
     "GeneralAgentResponse",
     "GeneralAgentResponseMessage",
     "GeneralAgentResponseMessageKwargs",
+    "GetTableResponse",
     "Id",
     "ImageUrlContent",
     "InputMessage",

athena/types/get_table_response.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class GetTableResponse(UniversalBaseModel):
+    asset_id: str = pydantic.Field()
+    """
+    The ID of the spreadsheet asset
+    """
+
+    data: typing.List[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field()
+    """
+    Array of row objects with column names as keys
+    """
+
+    message: str = pydantic.Field()
+    """
+    Success message or error description
+    """
+
+    success: bool = pydantic.Field()
+    """
+    Whether the operation was successful
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

{athena_intelligence-0.1.202.dist-info → athena_intelligence-0.1.203.dist-info}/METADATA

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

{athena_intelligence-0.1.202.dist-info → athena_intelligence-0.1.203.dist-info}/RECORD

@@ -1,4 +1,4 @@
-athena/__init__.py,sha256=Z5VOxDLNVeRpmNJz19W7I1YXlgSDE5fXkLb5UBl7QBQ,3409
+athena/__init__.py,sha256=nQ-oM62OoQr7TrcL7YqLGCieFFxYmw0Dbf5dFORbyhI,3455
 athena/agents/__init__.py,sha256=dg7IOwE6-BQSx20JEhdc1VDHlbIHPy08x5fuww_Tvko,180
 athena/agents/client.py,sha256=A70jdG6spqLkPriU8-NCn0vOJvdc5f4SKoVZLOebZjQ,5975
 athena/agents/drive/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
@@ -21,7 +21,7 @@ athena/base_client.py,sha256=dWl0sEjfhGQ_Jj_wq0ypDrGnOoi1s5YwCXinP6Qnyjk,6336
 athena/client.py,sha256=lK3vVU3TF3YjPpiohpxcuRl8x_sSw8HmQ-uuDDeAT6I,22161
 athena/core/__init__.py,sha256=lTcqUPXcx4112yLDd70RAPeqq6tu3eFMe1pKOqkW9JQ,1562
 athena/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-athena/core/client_wrapper.py,sha256=SlHIUqPWwBdggDUEFehVAu6se1IPHTkSMe-G6CZRhVc,2392
+athena/core/client_wrapper.py,sha256=ncxABrvUNAyODlc4ybw9Oo3AGD-s3RhDojHTAUh609s,2392
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 athena/core/force_multipart.py,sha256=awxh5MtcRYe74ehY8U76jzv6fYM_w_D3Rur7KQQzSDk,429
@@ -58,8 +58,8 @@ athena/tools/email/client.py,sha256=dOidOOOLHdfz2c3nykzyOa7nTftS91d2_aws0LTg8DU,
 athena/tools/email/raw_client.py,sha256=GkSxb-RFdhGgFNghnwH6i0cPI_gfWGLUmWvBNlBlhE4,9962
 athena/tools/raw_client.py,sha256=gUwmZV08uBGmTTNoWFuYGPMnFxPEhswKPDEbLIcVHPo,53027
 athena/tools/sheets/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
-athena/tools/sheets/client.py,sha256=FeZDM-grtK34oiH0_60fggD1bo9ddojDwr1n58vBnHA,52680
-athena/tools/sheets/raw_client.py,sha256=l2P7Iv-V_AEMLkXnzrOvkwZYQRyY6VGkjNicW3S5ST4,84972
+athena/tools/sheets/client.py,sha256=PFN_qTfL2sNXkSyU8rDbwEqrHW1ACI9hVkGdfzbDdOE,55273
+athena/tools/sheets/raw_client.py,sha256=bi2koBw2Ggct_9ZWZ3kRS50a9sVbT-ZWVtpU9XfMS9k,89950
 athena/tools/structured_data_extractor/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 athena/tools/structured_data_extractor/client.py,sha256=V1FcGZTPbrlz2d9oQZhsj3UIN1ZlZfnqRdDXj16xiPs,10623
 athena/tools/structured_data_extractor/raw_client.py,sha256=1ZYZBssmf1jDomopeJ3PMRLql3zT4c7ssRNNLa1YrGE,11245
@@ -68,7 +68,7 @@ athena/tools/tasks/client.py,sha256=c_YZ7OjQNJmPKbeeXJznXj3zo5CRFSv02dLupAdHAyo,
 athena/tools/tasks/raw_client.py,sha256=Mexzuf_HcRXWNlESDGQkHHv5tC2tAo-AX3PBSuSRO3U,3812
 athena/tools/types/__init__.py,sha256=ZI6REilQ0Xuocjd8iyNVfLvOsA2Ltlb8--px07wRSPg,217
 athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
-athena/types/__init__.py,sha256=TH-zHkr7p_GvtBIbuGBGQPSv71jwsooaaLzczFfVas4,3777
+athena/types/__init__.py,sha256=yxNEamC14ufn5CUS3y57Qda5qvauEEXEKEY9LruwSDo,3850
 athena/types/asset_content_request_out.py,sha256=RYlcY6j6Ju5EQL7UquDwyTe7uqxyuO8Bg8LCsv1YiUE,652
 athena/types/asset_node.py,sha256=3l7CUK2c_h4QT8ktSq0rFP9veF9G_V9mNe3NZlGl-xQ,804
 athena/types/asset_not_found_error.py,sha256=lIQpdTXCgbXRs21XCMhd_kB7fR6Y6Ep112mK4rejNx0,528
@@ -96,6 +96,7 @@ athena/types/general_agent_request.py,sha256=ltPew4FLKx795M9ktnsWbArLJHqTqhpP-R4
 athena/types/general_agent_response.py,sha256=G5LYQnZJeFdU3COK9Syyuk8rGvRYd1F_NeiDiu1zIFg,690
 athena/types/general_agent_response_message.py,sha256=P9JpetwgY2z_iSjFbiLZwuSv2Ez7g_hDvRkQo-31Gok,3071
 athena/types/general_agent_response_message_kwargs.py,sha256=XWMwp6WLAPiArBKU7NuuebTkq0jM8QVlTVfEfoldJOk,2170
+athena/types/get_table_response.py,sha256=vO2W6_rkFrbr3ykA10bB0fJBCprteyai5lNHbbhSyS4,943
 athena/types/id.py,sha256=6fnF__LI2KYyDjZwbs8y5sik8HoYPMRF7Ljihn1MBYY,121
 athena/types/image_url_content.py,sha256=SIlScFxuyfXpSU-xMLypitQK5gu_6ITjZYt64Dq9RYQ,589
 athena/types/input_message.py,sha256=MwgLCWH9mfUM5NFEV_AVjhS12ruNOxZxSXLcYKQ0rww,854
@@ -112,6 +113,6 @@ athena/types/structured_data_extractor_response.py,sha256=yFQ0CiFDdlZIq2X8UprEAw
 athena/types/text_content.py,sha256=tcVCPj3tHh5zQcTElr2tdCIjjfx3ZI63rKIlaG8vo64,592
 athena/types/type.py,sha256=Gvs56nvBMPcQpOZkfPocGNNb7S05PuINianbT309QAQ,146
 athena/version.py,sha256=tnXYUugs9zF_pkVdem-QBorKSuhEOOuetkR57dADDxE,86
-athena_intelligence-0.1.202.dist-info/METADATA,sha256=IGQbQdAlTge-fog01SloSBkJRd6zshGxowApC9ypG8A,5440
-athena_intelligence-0.1.202.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.202.dist-info/RECORD,,
+athena_intelligence-0.1.203.dist-info/METADATA,sha256=DYlLNE2F5bhceg8hu9GNhoGAeT4wYUx4WAOMJfRhA-4,5440
+athena_intelligence-0.1.203.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.203.dist-info/RECORD,,