athena-intelligence 0.1.205__tar.gz → 0.1.206__tar.gz

{athena_intelligence-0.1.205 → athena_intelligence-0.1.206}/PKG-INFO

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

{athena_intelligence-0.1.205 → athena_intelligence-0.1.206}/pyproject.toml

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

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

{athena_intelligence-0.1.205 → athena_intelligence-0.1.206}/src/athena/tools/sheets/client.py

@@ -930,6 +930,61 @@ class SheetsClient:
         )
         return _response.data
 
+    def insert_table_row(
+        self,
+        *,
+        asset_id: str,
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SheetOperationResponse:
+        """
+        Insert rows into a table in an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+            Array of row objects where keys are column names and values are cell values
+
+        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.
+
+        Returns
+        -------
+        SheetOperationResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.sheets.insert_table_row(
+            asset_id="asset_id",
+            row_data=[{}],
+        )
+        """
+        _response = self._raw_client.insert_table_row(
+            asset_id=asset_id,
+            row_data=row_data,
+            table_id=table_id,
+            table_name=table_name,
+            request_options=request_options,
+        )
+        return _response.data
+
     def update_table(
         self,
         *,
@@ -2048,6 +2103,69 @@ class AsyncSheetsClient:
         )
         return _response.data
 
+    async def insert_table_row(
+        self,
+        *,
+        asset_id: str,
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SheetOperationResponse:
+        """
+        Insert rows into a table in an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+            Array of row objects where keys are column names and values are cell values
+
+        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.
+
+        Returns
+        -------
+        SheetOperationResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.sheets.insert_table_row(
+                asset_id="asset_id",
+                row_data=[{}],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.insert_table_row(
+            asset_id=asset_id,
+            row_data=row_data,
+            table_id=table_id,
+            table_name=table_name,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def update_table(
         self,
         *,

{athena_intelligence-0.1.205 → athena_intelligence-0.1.206}/src/athena/tools/sheets/raw_client.py

@@ -1211,6 +1211,81 @@ 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 insert_table_row(
+        self,
+        *,
+        asset_id: str,
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[SheetOperationResponse]:
+        """
+        Insert rows into a table in an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+            Array of row objects where keys are column names and values are cell values
+
+        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.
+
+        Returns
+        -------
+        HttpResponse[SheetOperationResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/sheets/table/insert-row",
+            method="POST",
+            json={
+                "asset_id": asset_id,
+                "row_data": row_data,
+                "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(
+                    SheetOperationResponse,
+                    parse_obj_as(
+                        type_=SheetOperationResponse,  # 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,
         *,
@@ -2501,6 +2576,81 @@ 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 insert_table_row(
+        self,
+        *,
+        asset_id: str,
+        row_data: typing.Sequence[typing.Dict[str, typing.Optional[str]]],
+        table_id: typing.Optional[str] = OMIT,
+        table_name: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[SheetOperationResponse]:
+        """
+        Insert rows into a table in an Athena spreadsheet.
+
+        Parameters
+        ----------
+        asset_id : str
+            The ID of the spreadsheet asset
+
+        row_data : typing.Sequence[typing.Dict[str, typing.Optional[str]]]
+            Array of row objects where keys are column names and values are cell values
+
+        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.
+
+        Returns
+        -------
+        AsyncHttpResponse[SheetOperationResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/sheets/table/insert-row",
+            method="POST",
+            json={
+                "asset_id": asset_id,
+                "row_data": row_data,
+                "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(
+                    SheetOperationResponse,
+                    parse_obj_as(
+                        type_=SheetOperationResponse,  # 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,
         *,