athena-intelligence 0.1.204__py3-none-any.whl → 0.1.206__py3-none-any.whl

athena/base_client.py

@@ -4,6 +4,7 @@ import typing
 
 import httpx
 from .agents.client import AgentsClient, AsyncAgentsClient
+from .aop.client import AopClient, AsyncAopClient
 from .assets.client import AssetsClient, AsyncAssetsClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import AthenaEnvironment
@@ -77,6 +78,7 @@ class BaseAthena:
             timeout=_defaulted_timeout,
         )
         self.agents = AgentsClient(client_wrapper=self._client_wrapper)
+        self.aop = AopClient(client_wrapper=self._client_wrapper)
         self.assets = AssetsClient(client_wrapper=self._client_wrapper)
         self.query = QueryClient(client_wrapper=self._client_wrapper)
         self.tools = ToolsClient(client_wrapper=self._client_wrapper)
@@ -148,6 +150,7 @@ class AsyncBaseAthena:
             timeout=_defaulted_timeout,
         )
         self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
+        self.aop = AsyncAopClient(client_wrapper=self._client_wrapper)
         self.assets = AsyncAssetsClient(client_wrapper=self._client_wrapper)
         self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
         self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)

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

athena/errors/bad_request_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.parent_folder_error import ParentFolderError
 
 
 class BadRequestError(ApiError):
-    def __init__(self, body: ParentFolderError, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=400, headers=headers, body=body)

athena/tools/raw_client.py

@@ -25,7 +25,6 @@ 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.folder_response import FolderResponse
-from ..types.parent_folder_error import ParentFolderError
 from ..types.save_asset_request_out import SaveAssetRequestOut
 from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
 
@@ -337,9 +336,9 @@ class RawToolsClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ParentFolderError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -645,9 +644,9 @@ class RawToolsClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ParentFolderError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -984,9 +983,9 @@ class AsyncRawToolsClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ParentFolderError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -1293,9 +1292,9 @@ class AsyncRawToolsClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ParentFolderError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ParentFolderError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),

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/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,
         *,

athena/types/__init__.py

@@ -2,6 +2,9 @@
 
 # isort: skip_file
 
+from .aop_async_execute_response_out import AopAsyncExecuteResponseOut
+from .aop_execute_request_in import AopExecuteRequestIn
+from .aop_execute_response_out import AopExecuteResponseOut
 from .asset_content_request_out import AssetContentRequestOut
 from .asset_node import AssetNode
 from .asset_not_found_error import AssetNotFoundError
@@ -39,7 +42,6 @@ from .input_message_content_item import (
     InputMessageContentItem_Text,
 )
 from .paginated_assets_out import PaginatedAssetsOut
-from .parent_folder_error import ParentFolderError
 from .prompt_message import PromptMessage
 from .public_asset_out import PublicAssetOut
 from .research_agent_response import ResearchAgentResponse
@@ -51,6 +53,9 @@ from .text_content import TextContent
 from .type import Type
 
 __all__ = [
+    "AopAsyncExecuteResponseOut",
+    "AopExecuteRequestIn",
+    "AopExecuteResponseOut",
     "AssetContentRequestOut",
     "AssetNode",
     "AssetNotFoundError",
@@ -88,7 +93,6 @@ __all__ = [
     "InputMessageContentItem_ImageUrl",
     "InputMessageContentItem_Text",
     "PaginatedAssetsOut",
-    "ParentFolderError",
     "PromptMessage",
     "PublicAssetOut",
     "ResearchAgentResponse",

athena/types/aop_async_execute_response_out.py

@@ -0,0 +1,76 @@
+# 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 AopAsyncExecuteResponseOut(UniversalBaseModel):
+    """
+    Response model for async AOP execution.
+    """
+
+    aop_asset_id: str = pydantic.Field()
+    """
+    ID of the AOP asset that was executed
+    """
+
+    aop_config: typing.Dict[str, typing.Optional[typing.Any]] = pydantic.Field()
+    """
+    Full configuration of the AOP asset
+    """
+
+    aop_title: str = pydantic.Field()
+    """
+    Title of the AOP asset
+    """
+
+    base_prompt: str = pydantic.Field()
+    """
+    Base prompt of the AOP before user inputs were added
+    """
+
+    enabled_tools: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of tools that were enabled for this execution
+    """
+
+    final_prompt: str = pydantic.Field()
+    """
+    Final prompt used for execution including user inputs
+    """
+
+    message: str = pydantic.Field()
+    """
+    Status message about the async execution
+    """
+
+    status: str = pydantic.Field()
+    """
+    Status of the execution (always 'started' for async)
+    """
+
+    sync_server: str = pydantic.Field()
+    """
+    Sync server URL used for execution
+    """
+
+    thread_id: str = pydantic.Field()
+    """
+    Unique thread ID for tracking the execution
+    """
+
+    trigger_type: str = pydantic.Field()
+    """
+    Type of trigger that initiated the execution
+    """
+
+    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/types/{parent_folder_error.py → aop_execute_request_in.py}

@@ -6,8 +6,20 @@ import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class ParentFolderError(UniversalBaseModel):
-    message: str
+class AopExecuteRequestIn(UniversalBaseModel):
+    """
+    Request model for executing an AOP (Agent Operating Procedure).
+    """
+
+    asset_id: str = pydantic.Field()
+    """
+    ID of the existing AOP asset to execute
+    """
+
+    user_inputs: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(default=None)
+    """
+    Optional user inputs to append to the AOP's prompt as key-value pairs
+    """
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

athena/types/aop_execute_response_out.py

@@ -0,0 +1,76 @@
+# 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 AopExecuteResponseOut(UniversalBaseModel):
+    """
+    Response model for AOP execution.
+    """
+
+    aop_asset_id: str = pydantic.Field()
+    """
+    ID of the AOP asset that was executed
+    """
+
+    aop_config: typing.Dict[str, typing.Optional[typing.Any]] = pydantic.Field()
+    """
+    Full configuration of the AOP asset
+    """
+
+    aop_title: str = pydantic.Field()
+    """
+    Title of the AOP asset
+    """
+
+    base_prompt: str = pydantic.Field()
+    """
+    Base prompt of the AOP before user inputs were added
+    """
+
+    enabled_tools: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of tools that were enabled for this execution
+    """
+
+    final_prompt: str = pydantic.Field()
+    """
+    Final prompt used for execution including user inputs
+    """
+
+    response: typing.Dict[str, typing.Optional[typing.Any]] = pydantic.Field()
+    """
+    The execution response from the AOP
+    """
+
+    status: str = pydantic.Field()
+    """
+    Status of the execution (e.g., 'submitted')
+    """
+
+    sync_server: str = pydantic.Field()
+    """
+    Sync server URL used for execution
+    """
+
+    thread_id: str = pydantic.Field()
+    """
+    Unique thread ID for tracking the execution
+    """
+
+    trigger_type: str = pydantic.Field()
+    """
+    Type of trigger that initiated the execution
+    """
+
+    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.204.dist-info → athena_intelligence-0.1.206.dist-info}/METADATA

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