athena-intelligence 0.1.212__py3-none-any.whl → 0.1.214__py3-none-any.whl

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.212",
+            "User-Agent": "athena-intelligence/0.1.214",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.212",
+            "X-Fern-SDK-Version": "0.1.214",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-KEY"] = self.api_key

athena/tools/__init__.py

@@ -4,5 +4,14 @@
 
 from .types import ToolsDataFrameRequestColumnsItem
 from . import calendar, email, sheets, structured_data_extractor, tasks
+from .sheets import InsertTableRowRequestRowDataItemValue
 
-__all__ = ["ToolsDataFrameRequestColumnsItem", "calendar", "email", "sheets", "structured_data_extractor", "tasks"]
+__all__ = [
+    "InsertTableRowRequestRowDataItemValue",
+    "ToolsDataFrameRequestColumnsItem",
+    "calendar",
+    "email",
+    "sheets",
+    "structured_data_extractor",
+    "tasks",
+]

athena/tools/sheets/__init__.py

@@ -2,3 +2,6 @@
 
 # isort: skip_file
 
+from .types import InsertTableRowRequestRowDataItemValue
+
+__all__ = ["InsertTableRowRequestRowDataItemValue"]

athena/tools/sheets/client.py

@@ -8,6 +8,7 @@ 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
+from .types.insert_table_row_request_row_data_item_value import InsertTableRowRequestRowDataItemValue
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -884,7 +885,7 @@ class SheetsClient:
         )
         return _response.data
 
-    def get_table_rows(
+    def get_table(
         self,
         *,
         asset_id: str,
@@ -921,12 +922,12 @@ class SheetsClient:
         client = Athena(
             api_key="YOUR_API_KEY",
         )
-        client.tools.sheets.get_table_rows(
+        client.tools.sheets.get_table(
             asset_id="asset_id",
             table_name="table_name",
         )
         """
-        _response = self._raw_client.get_table_rows(
+        _response = self._raw_client.get_table(
             asset_id=asset_id, table_name=table_name, table_id=table_id, request_options=request_options
         )
         return _response.data
@@ -935,9 +936,9 @@ class SheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]],
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SheetOperationResponse:
         """
@@ -948,15 +949,15 @@ class SheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]]
             Array of row objects where keys are column names and values are cell values
 
+        table_name : str
+            Table name to insert row into
+
         table_id : typing.Optional[str]
             Table ID to insert row into
 
-        table_name : typing.Optional[str]
-            Table name to insert row into
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -975,13 +976,14 @@ class SheetsClient:
         client.tools.sheets.insert_table_row(
             asset_id="asset_id",
             row_data=[{}],
+            table_name="table_name",
         )
         """
         _response = self._raw_client.insert_table_row(
             asset_id=asset_id,
             row_data=row_data,
-            table_id=table_id,
             table_name=table_name,
+            table_id=table_id,
             request_options=request_options,
         )
         return _response.data
@@ -2050,7 +2052,7 @@ class AsyncSheetsClient:
         )
         return _response.data
 
-    async def get_table_rows(
+    async def get_table(
         self,
         *,
         asset_id: str,
@@ -2092,7 +2094,7 @@ class AsyncSheetsClient:
 
 
         async def main() -> None:
-            await client.tools.sheets.get_table_rows(
+            await client.tools.sheets.get_table(
                 asset_id="asset_id",
                 table_name="table_name",
             )
@@ -2100,7 +2102,7 @@ class AsyncSheetsClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.get_table_rows(
+        _response = await self._raw_client.get_table(
             asset_id=asset_id, table_name=table_name, table_id=table_id, request_options=request_options
         )
         return _response.data
@@ -2109,9 +2111,9 @@ class AsyncSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]],
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SheetOperationResponse:
         """
@@ -2122,15 +2124,15 @@ class AsyncSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]]
             Array of row objects where keys are column names and values are cell values
 
+        table_name : str
+            Table name to insert row into
+
         table_id : typing.Optional[str]
             Table ID to insert row into
 
-        table_name : typing.Optional[str]
-            Table name to insert row into
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -2154,6 +2156,7 @@ class AsyncSheetsClient:
             await client.tools.sheets.insert_table_row(
                 asset_id="asset_id",
                 row_data=[{}],
+                table_name="table_name",
             )
 
 
@@ -2162,8 +2165,8 @@ class AsyncSheetsClient:
         _response = await self._raw_client.insert_table_row(
             asset_id=asset_id,
             row_data=row_data,
-            table_id=table_id,
             table_name=table_name,
+            table_id=table_id,
             request_options=request_options,
         )
         return _response.data

athena/tools/sheets/raw_client.py

@@ -8,10 +8,12 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.http_response import AsyncHttpResponse, HttpResponse
 from ...core.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
+from ...core.serialization import convert_and_respect_annotation_metadata
 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
+from .types.insert_table_row_request_row_data_item_value import InsertTableRowRequestRowDataItemValue
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -1141,7 +1143,7 @@ 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_rows(
+    def get_table(
         self,
         *,
         asset_id: str,
@@ -1172,7 +1174,7 @@ class RawSheetsClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/v0/tools/sheets/table/get_rows",
+            "api/v0/tools/sheets/table/get",
             method="POST",
             json={
                 "asset_id": asset_id,
@@ -1215,9 +1217,9 @@ class RawSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]],
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
         """
@@ -1228,15 +1230,15 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]]
             Array of row objects where keys are column names and values are cell values
 
+        table_name : str
+            Table name to insert row into
+
         table_id : typing.Optional[str]
             Table ID to insert row into
 
-        table_name : typing.Optional[str]
-            Table name to insert row into
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1250,7 +1252,13 @@ class RawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
-                "row_data": row_data,
+                "row_data": convert_and_respect_annotation_metadata(
+                    object_=row_data,
+                    annotation=typing.Sequence[
+                        typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]
+                    ],
+                    direction="write",
+                ),
                 "table_id": table_id,
                 "table_name": table_name,
             },
@@ -2506,7 +2514,7 @@ 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_rows(
+    async def get_table(
         self,
         *,
         asset_id: str,
@@ -2537,7 +2545,7 @@ class AsyncRawSheetsClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/tools/sheets/table/get_rows",
+            "api/v0/tools/sheets/table/get",
             method="POST",
             json={
                 "asset_id": asset_id,
@@ -2580,9 +2588,9 @@ class AsyncRawSheetsClient:
         self,
         *,
         asset_id: str,
-        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]],
+        table_name: str,
         table_id: typing.Optional[str] = OMIT,
-        table_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
         """
@@ -2593,15 +2601,15 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]]
             Array of row objects where keys are column names and values are cell values
 
+        table_name : str
+            Table name to insert row into
+
         table_id : typing.Optional[str]
             Table ID to insert row into
 
-        table_name : typing.Optional[str]
-            Table name to insert row into
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -2615,7 +2623,13 @@ class AsyncRawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
-                "row_data": row_data,
+                "row_data": convert_and_respect_annotation_metadata(
+                    object_=row_data,
+                    annotation=typing.Sequence[
+                        typing.Dict[str, typing.Optional[InsertTableRowRequestRowDataItemValue]]
+                    ],
+                    direction="write",
+                ),
                 "table_id": table_id,
                 "table_name": table_name,
             },

athena/tools/sheets/types/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .insert_table_row_request_row_data_item_value import InsertTableRowRequestRowDataItemValue
+
+__all__ = ["InsertTableRowRequestRowDataItemValue"]

athena/tools/sheets/types/insert_table_row_request_row_data_item_value.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+InsertTableRowRequestRowDataItemValue = typing.Union[str, int, float, bool]

athena/types/get_table_response.py

@@ -12,9 +12,9 @@ class GetTableResponse(UniversalBaseModel):
     The ID of the spreadsheet asset
     """
 
-    data: typing.List[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field()
+    columns: typing.List[typing.Dict[str, str]] = pydantic.Field()
     """
-    Array of row objects with column names as keys
+    Array of column objects with name and other metadata
     """
 
     message: str = pydantic.Field()
@@ -22,6 +22,11 @@ class GetTableResponse(UniversalBaseModel):
     Success message or error description
     """
 
+    rows: typing.List[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field()
+    """
+    Array of row objects with column names as keys
+    """
+
     success: bool = pydantic.Field()
     """
     Whether the operation was successful

{athena_intelligence-0.1.212.dist-info → athena_intelligence-0.1.214.dist-info}/METADATA

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

{athena_intelligence-0.1.212.dist-info → athena_intelligence-0.1.214.dist-info}/RECORD

@@ -24,7 +24,7 @@ athena/base_client.py,sha256=9OfmXHlNIFE_4Udzxn0Ue-H-sC2ps0AIm8ZRyuaDwXA,6738
 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=CZNPAITFjyqBe0YR7Ps3bDMpB8ipQi6IZ3L2kYN3kTE,2392
+athena/core/client_wrapper.py,sha256=IoVfTAwho3WdndjWuN8fGOqacPIssXEv8NMXRCaVQ2I,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
@@ -54,7 +54,7 @@ athena/query/types/query_execute_request_database_asset_ids.py,sha256=aoVl5Xb34Q
 athena/threads/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 athena/threads/client.py,sha256=gLjlEgLuEMduyPlK54GhGxWu6Vto-_TKFHcYcu9ZoiY,3328
 athena/threads/raw_client.py,sha256=1-y1q2M5UyoIV6kuH-zkTGDKNdGUWMWUALsKJhtPPvE,8072
-athena/tools/__init__.py,sha256=Lq5sY-WpodI9cuRtzoJXXPiK7oQ-JAopxo2-M6ugJEU,326
+athena/tools/__init__.py,sha256=TxIs_v3YkPfqPV95RUx-qVpZ55mQrD5qbxGoV0I_P1g,456
 athena/tools/calendar/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 athena/tools/calendar/client.py,sha256=NMiwSaBD-JYcAoudGyBMtWVIWPm6ChFqOFSndxYupY0,4385
 athena/tools/calendar/raw_client.py,sha256=6HmAXGcOxXuRMdxA1_U1oROkouJebOXVFAnz5CDHgto,6858
@@ -63,9 +63,11 @@ athena/tools/email/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa
 athena/tools/email/client.py,sha256=dOidOOOLHdfz2c3nykzyOa7nTftS91d2_aws0LTg8DU,5921
 athena/tools/email/raw_client.py,sha256=GkSxb-RFdhGgFNghnwH6i0cPI_gfWGLUmWvBNlBlhE4,9962
 athena/tools/raw_client.py,sha256=iiAKpFTyBoYLzL5G0BCV6GM69NxCkksB_30XAAmyc24,53049
-athena/tools/sheets/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
-athena/tools/sheets/client.py,sha256=eG8uWf5-cJC5vTDH4NBvwsJB6ljMj4JNAzGc7nfaZYQ,58607
-athena/tools/sheets/raw_client.py,sha256=ULwZbn_Fqg8D62IYalIJ-SSYJKoycdLvzGUeGZf4aJc,95473
+athena/tools/sheets/__init__.py,sha256=vxLg5xXUBSXe3GY09QR-m_5T-Poi9r_9of8Q4sFrSgI,195
+athena/tools/sheets/client.py,sha256=cGPDqEyhMp9DTLxc1L-M3JXvOMqHj1jc6xEgBh19p7Y,58811
+athena/tools/sheets/raw_client.py,sha256=oxT8AM88pzI1WmY7U0zKURfF4T1UZ9RFPJygGCq7imc,96273
+athena/tools/sheets/types/__init__.py,sha256=6REU5mirjPqkpLFUQ_ZvQrypqCjisoWi3wQbHZF5krM,234
+athena/tools/sheets/types/insert_table_row_request_row_data_item_value.py,sha256=B-ZBD4ma6DlXrGUH2rxJbB9BqHKXQ7vfs2t5mot4pCM,156
 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
@@ -106,7 +108,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/get_table_response.py,sha256=KztIDpd3J4jiUUOz0DAH_SLMA-2uLGWL_lZ345jYdbw,1084
 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
@@ -123,6 +125,6 @@ athena/types/text_content.py,sha256=tcVCPj3tHh5zQcTElr2tdCIjjfx3ZI63rKIlaG8vo64,
 athena/types/thread_status_response_out.py,sha256=UuSAvs9woL1i8RwvVRKsFUufN4A9jO3jsV47YMckvQU,1219
 athena/types/type.py,sha256=Gvs56nvBMPcQpOZkfPocGNNb7S05PuINianbT309QAQ,146
 athena/version.py,sha256=tnXYUugs9zF_pkVdem-QBorKSuhEOOuetkR57dADDxE,86
-athena_intelligence-0.1.212.dist-info/METADATA,sha256=dsk2nguBxRB3NQASypenaIZNlwky9BqhO1UDun2sw0I,5440
-athena_intelligence-0.1.212.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.212.dist-info/RECORD,,
+athena_intelligence-0.1.214.dist-info/METADATA,sha256=72gFuJ76Ewqa-_sEWxCkn--m-j4hBPB2GQPDuT-f0jg,5440
+athena_intelligence-0.1.214.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.214.dist-info/RECORD,,