athena-intelligence 0.1.212__tar.gz → 0.1.214__tar.gz

{athena_intelligence-0.1.212 → athena_intelligence-0.1.214}/PKG-INFO

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

@@ -3,7 +3,7 @@ name = "athena-intelligence"
 
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.212"
+version = "0.1.214"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.212 → athena_intelligence-0.1.214}/src/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_intelligence-0.1.214/src/athena/tools/__init__.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import ToolsDataFrameRequestColumnsItem
+from . import calendar, email, sheets, structured_data_extractor, tasks
+from .sheets import InsertTableRowRequestRowDataItemValue
+
+__all__ = [
+    "InsertTableRowRequestRowDataItemValue",
+    "ToolsDataFrameRequestColumnsItem",
+    "calendar",
+    "email",
+    "sheets",
+    "structured_data_extractor",
+    "tasks",
+]

athena_intelligence-0.1.214/src/athena/tools/sheets/__init__.py

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

{athena_intelligence-0.1.212 → athena_intelligence-0.1.214}/src/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_intelligence-0.1.212 → athena_intelligence-0.1.214}/src/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_intelligence-0.1.214/src/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_intelligence-0.1.214/src/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_intelligence-0.1.212 → athena_intelligence-0.1.214}/src/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/src/athena/tools/__init__.py

@@ -1,8 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-# isort: skip_file
-
-from .types import ToolsDataFrameRequestColumnsItem
-from . import calendar, email, sheets, structured_data_extractor, tasks
-
-__all__ = ["ToolsDataFrameRequestColumnsItem", "calendar", "email", "sheets", "structured_data_extractor", "tasks"]

athena_intelligence-0.1.212/src/athena/tools/tasks/__init__.py

@@ -1,4 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-# isort: skip_file
-