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

athena/__init__.py

@@ -3,6 +3,9 @@
 # isort: skip_file
 
 from .types import (
+    AopAsyncExecuteResponseOut,
+    AopExecuteRequestIn,
+    AopExecuteResponseOut,
     AssetContentRequestOut,
     AssetNode,
     AssetNotFoundError,
@@ -40,7 +43,6 @@ from .types import (
     InputMessageContentItem_ImageUrl,
     InputMessageContentItem_Text,
     PaginatedAssetsOut,
-    ParentFolderError,
     PromptMessage,
     PublicAssetOut,
     ResearchAgentResponse,
@@ -60,7 +62,7 @@ from .errors import (
     UnprocessableEntityError,
     UnsupportedMediaTypeError,
 )
-from . import agents, assets, query, tools
+from . import agents, aop, assets, query, tools
 from .client import AsyncAthena, Athena
 from .environment import AthenaEnvironment
 from .query import QueryExecuteRequestDatabaseAssetIds
@@ -68,6 +70,9 @@ from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
+    "AopAsyncExecuteResponseOut",
+    "AopExecuteRequestIn",
+    "AopExecuteResponseOut",
     "AssetContentRequestOut",
     "AssetNode",
     "AssetNotFoundError",
@@ -112,7 +117,6 @@ __all__ = [
     "InternalServerError",
     "NotFoundError",
     "PaginatedAssetsOut",
-    "ParentFolderError",
     "PromptMessage",
     "PublicAssetOut",
     "QueryExecuteRequestDatabaseAssetIds",
@@ -129,6 +133,7 @@ __all__ = [
     "UnsupportedMediaTypeError",
     "__version__",
     "agents",
+    "aop",
     "assets",
     "query",
     "tools",

athena/aop/__init__.py

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

athena/aop/client.py

@@ -0,0 +1,201 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.aop_async_execute_response_out import AopAsyncExecuteResponseOut
+from ..types.aop_execute_request_in import AopExecuteRequestIn
+from ..types.aop_execute_response_out import AopExecuteResponseOut
+from .raw_client import AsyncRawAopClient, RawAopClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class AopClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawAopClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawAopClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawAopClient
+        """
+        return self._raw_client
+
+    def execute(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopExecuteResponseOut:
+        """
+        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        from athena import AopExecuteRequestIn, Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.aop.execute(
+            request=AopExecuteRequestIn(
+                asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+            ),
+        )
+        """
+        _response = self._raw_client.execute(request=request, request_options=request_options)
+        return _response.data
+
+    def execute_async(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopAsyncExecuteResponseOut:
+        """
+        Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopAsyncExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        from athena import AopExecuteRequestIn, Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.aop.execute_async(
+            request=AopExecuteRequestIn(
+                asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+            ),
+        )
+        """
+        _response = self._raw_client.execute_async(request=request, request_options=request_options)
+        return _response.data
+
+
+class AsyncAopClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawAopClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawAopClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawAopClient
+        """
+        return self._raw_client
+
+    async def execute(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopExecuteResponseOut:
+        """
+        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AopExecuteRequestIn, AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.aop.execute(
+                request=AopExecuteRequestIn(
+                    asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                    user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+                ),
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.execute(request=request, request_options=request_options)
+        return _response.data
+
+    async def execute_async(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AopAsyncExecuteResponseOut:
+        """
+        Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AopAsyncExecuteResponseOut
+            AOP execution started successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AopExecuteRequestIn, AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.aop.execute_async(
+                request=AopExecuteRequestIn(
+                    asset_id="asset_9249292-d118-42d3-95b4-00eccfe0754f",
+                    user_inputs={"company": "Acme Corp", "quarter": "Q1 2024"},
+                ),
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.execute_async(request=request, request_options=request_options)
+        return _response.data

athena/aop/raw_client.py

@@ -0,0 +1,433 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+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.bad_request_error import BadRequestError
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.aop_async_execute_response_out import AopAsyncExecuteResponseOut
+from ..types.aop_execute_request_in import AopExecuteRequestIn
+from ..types.aop_execute_response_out import AopExecuteResponseOut
+from ..types.asset_not_found_error import AssetNotFoundError
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawAopClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def execute(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[AopExecuteResponseOut]:
+        """
+        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[AopExecuteResponseOut]
+            AOP execution started successfully
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/aop/execute",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=AopExecuteRequestIn, direction="write"
+            ),
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AopExecuteResponseOut,
+                    parse_obj_as(
+                        type_=AopExecuteResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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 execute_async(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[AopAsyncExecuteResponseOut]:
+        """
+        Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[AopAsyncExecuteResponseOut]
+            AOP execution started successfully
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/aop/execute-async",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=AopExecuteRequestIn, direction="write"
+            ),
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AopAsyncExecuteResponseOut,
+                    parse_obj_as(
+                        type_=AopAsyncExecuteResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)
+
+
+class AsyncRawAopClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def execute(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[AopExecuteResponseOut]:
+        """
+        Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[AopExecuteResponseOut]
+            AOP execution started successfully
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/aop/execute",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=AopExecuteRequestIn, direction="write"
+            ),
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AopExecuteResponseOut,
+                    parse_obj_as(
+                        type_=AopExecuteResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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 execute_async(
+        self, *, request: AopExecuteRequestIn, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[AopAsyncExecuteResponseOut]:
+        """
+        Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
+
+        Parameters
+        ----------
+        request : AopExecuteRequestIn
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[AopAsyncExecuteResponseOut]
+            AOP execution started successfully
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/aop/execute-async",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=AopExecuteRequestIn, direction="write"
+            ),
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AopAsyncExecuteResponseOut,
+                    parse_obj_as(
+                        type_=AopAsyncExecuteResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            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(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    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)

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.203",
+            "User-Agent": "athena-intelligence/0.1.205",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.203",
+            "X-Fern-SDK-Version": "0.1.205",
             **(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/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.203.dist-info → athena_intelligence-0.1.205.dist-info}/METADATA

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

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

@@ -1,4 +1,4 @@
-athena/__init__.py,sha256=nQ-oM62OoQr7TrcL7YqLGCieFFxYmw0Dbf5dFORbyhI,3455
+athena/__init__.py,sha256=VXyrAkqUDbvUpdBa-A7lShZliUPwL00Q0brxX3cS_Hs,3597
 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
@@ -14,14 +14,17 @@ athena/agents/research/raw_client.py,sha256=l0XGwUGn0ZjNZSvDaLLWeYlQsDSaSFCnUcCt
 athena/agents/sql/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 athena/agents/sql/client.py,sha256=Ht7PMJSMqVwihdqk74yyqwFUSXOPcLjxNB-YU6cVGAE,4069
 athena/agents/sql/raw_client.py,sha256=sI7Aq6_Z4AdgXzqBXG2G4l2fYQTWjQ-5IeNZdClmXhY,6017
+athena/aop/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
+athena/aop/client.py,sha256=xdmDUTR_XAKLX3Oo2lRrnpuh34zZtU18yBI0KTsOSGY,6506
+athena/aop/raw_client.py,sha256=plXlJeQ4c9FgCh8cB8tr44kEPHaemm1H_akuDV9Xfms,18246
 athena/assets/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 athena/assets/client.py,sha256=8qxDICl5Dz-cNeUOl0du1spj54g_-bLObSLK_biUXV0,5159
 athena/assets/raw_client.py,sha256=TW6dCb1i68p-0sf4_B2F0H8SK-kk9sIHFAuwfTEg5GQ,7077
-athena/base_client.py,sha256=dWl0sEjfhGQ_Jj_wq0ypDrGnOoi1s5YwCXinP6Qnyjk,6336
+athena/base_client.py,sha256=v0xO5KUQrRW8zlNSeyNWwv8r3Yc5bn3RzRWNTOeeGEY,6523
 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=ncxABrvUNAyODlc4ybw9Oo3AGD-s3RhDojHTAUh609s,2392
+athena/core/client_wrapper.py,sha256=fZWSYnU3HfaLMpPZ75iQcWJJOtpC_vpESfpLKi8Yu9Y,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
@@ -35,7 +38,7 @@ athena/core/request_options.py,sha256=h0QUNCFVdCW_7GclVySCAY2w4NhtXVBUCmHgmzaxpc
 athena/core/serialization.py,sha256=ECL3bvv_0i7U4uvPidZCNel--MUbA0iq0aGcNKi3kws,9818
 athena/environment.py,sha256=_e7YwByXALEk7fNJNJvjF81caYTbsx4c7wwftCJHQ7g,214
 athena/errors/__init__.py,sha256=ouNi-VrkRD37haxSAk1-3mbexWE2qCMtctdE5CEgXvM,675
-athena/errors/bad_request_error.py,sha256=bOLbzeen9VcMrhSZ3zrq11kSNmRdkL1Z34_gIQ6Fj0c,386
+athena/errors/bad_request_error.py,sha256=PnE3v3kETCXm9E3LiNcHLNtjPEUvpe98-r59q-kQb78,338
 athena/errors/content_too_large_error.py,sha256=i4Af_rueEaP5CftotSDwCaMRlkcC76KyT2NsEIjZH0I,392
 athena/errors/internal_server_error.py,sha256=t1-kpoDC2biEuoE-Ne8v1kuQswvsIEwt_xPPoBmGG00,342
 athena/errors/not_found_error.py,sha256=m-s2prtvwM0THnWsIJun5Km57RDrNptxZnxvI5Q7Nhk,388
@@ -56,7 +59,7 @@ athena/tools/client.py,sha256=MoSKdWoj5GP1Ars8vRZOq2XTSxZjgKBnXIoBGKa7x_g,19298
 athena/tools/email/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 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=gUwmZV08uBGmTTNoWFuYGPMnFxPEhswKPDEbLIcVHPo,53027
+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=PFN_qTfL2sNXkSyU8rDbwEqrHW1ACI9hVkGdfzbDdOE,55273
 athena/tools/sheets/raw_client.py,sha256=bi2koBw2Ggct_9ZWZ3kRS50a9sVbT-ZWVtpU9XfMS9k,89950
@@ -68,7 +71,10 @@ 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=yxNEamC14ufn5CUS3y57Qda5qvauEEXEKEY9LruwSDo,3850
+athena/types/__init__.py,sha256=Vip2IwctpaB5yqoT_iZ5djAL9leN-HeNV-0scuGjo9Y,4051
+athena/types/aop_async_execute_response_out.py,sha256=R3oy0RLyD-t4MQCnp3Ll13Ci3D5JUMhPyuetbhrONNs,1798
+athena/types/aop_execute_request_in.py,sha256=mEsMKyNN6e90gZra733lJDC6z0bZWdc72af3B-Z5aqE,889
+athena/types/aop_execute_response_out.py,sha256=sOaPWALEo5hWKSnoq_kXrHKtgqsJvgCWcOWpjCTtTCA,1816
 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
@@ -102,7 +108,6 @@ athena/types/image_url_content.py,sha256=SIlScFxuyfXpSU-xMLypitQK5gu_6ITjZYt64Dq
 athena/types/input_message.py,sha256=MwgLCWH9mfUM5NFEV_AVjhS12ruNOxZxSXLcYKQ0rww,854
 athena/types/input_message_content_item.py,sha256=F-H4SNqrr__t7mPquBMyXnR8-gpyDJC6x48oJO1PXak,1163
 athena/types/paginated_assets_out.py,sha256=NJ9vwjimfaGCMPeYRKw4yEZPWYkdWpSsB60-iW8_yPI,1384
-athena/types/parent_folder_error.py,sha256=yDIQtaAY9XqVCScw-QGmWTHrbi0Zl1Nur15fCEYKdTI,527
 athena/types/prompt_message.py,sha256=5WrlKURJuD0DPhMmP3gpBNuRgGfbE9ArY0BW_OSq0P4,634
 athena/types/public_asset_out.py,sha256=rBPFX3PKM0zxK3Qh8uICE14idg-UkDDob_xFRprO4bo,2775
 athena/types/research_agent_response.py,sha256=BnBRbDcQEh-qNjLfvjoigTVHtkYZjwdjYU5ONjr-OWQ,656
@@ -113,6 +118,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.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,,
+athena_intelligence-0.1.205.dist-info/METADATA,sha256=cxr4yqUDhnPtGoJ148NkSYC9BlTHIvUIJRn5EZwPUbE,5440
+athena_intelligence-0.1.205.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.205.dist-info/RECORD,,