athena-intelligence 0.1.204__py3-none-any.whl → 0.1.206__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)