athena-intelligence 0.1.118__py3-none-any.whl → 0.1.120__py3-none-any.whl

athena/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.118",
+            "X-Fern-SDK-Version": "0.1.120",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/tools/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import ToolsDataFrameRequestColumnsItem
+from . import calendar, email, structured_data_extractor, tasks
 
-__all__ = ["ToolsDataFrameRequestColumnsItem"]
+__all__ = ["ToolsDataFrameRequestColumnsItem", "calendar", "email", "structured_data_extractor", "tasks"]

athena/tools/calendar/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

athena/tools/calendar/client.py

@@ -0,0 +1,155 @@
+# 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.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+
+
+class CalendarClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list_events(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! List calendar events with optional filtering.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.calendar.list_events()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/calendar/events", method="GET", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create_event(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Create new calendar events.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.calendar.create_event()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/calendar/events", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncCalendarClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list_events(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! List calendar events with optional filtering.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.calendar.list_events()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/calendar/events", method="GET", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create_event(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Create new calendar events.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.calendar.create_event()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/calendar/events", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/tools/client.py

@@ -22,6 +22,10 @@ from ..types.file_chunk_request_out import FileChunkRequestOut
 from ..types.file_too_large_error import FileTooLargeError
 from ..types.parent_folder_error import ParentFolderError
 from ..types.save_asset_request_out import SaveAssetRequestOut
+from .calendar.client import AsyncCalendarClient, CalendarClient
+from .email.client import AsyncEmailClient, EmailClient
+from .structured_data_extractor.client import AsyncStructuredDataExtractorClient, StructuredDataExtractorClient
+from .tasks.client import AsyncTasksClient, TasksClient
 from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
 
 # this is used as the default value for optional parameters
@@ -31,6 +35,10 @@ OMIT = typing.cast(typing.Any, ...)
 class ToolsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
+        self.calendar = CalendarClient(client_wrapper=self._client_wrapper)
+        self.email = EmailClient(client_wrapper=self._client_wrapper)
+        self.structured_data_extractor = StructuredDataExtractorClient(client_wrapper=self._client_wrapper)
+        self.tasks = TasksClient(client_wrapper=self._client_wrapper)
 
     def get_asset_chunks(
         self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
@@ -280,6 +288,10 @@ class ToolsClient:
 class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
+        self.calendar = AsyncCalendarClient(client_wrapper=self._client_wrapper)
+        self.email = AsyncEmailClient(client_wrapper=self._client_wrapper)
+        self.structured_data_extractor = AsyncStructuredDataExtractorClient(client_wrapper=self._client_wrapper)
+        self.tasks = AsyncTasksClient(client_wrapper=self._client_wrapper)
 
     async def get_asset_chunks(
         self, *, asset_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None

athena/tools/email/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

athena/tools/email/client.py

@@ -0,0 +1,223 @@
+# 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.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+
+
+class EmailClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def create_draft(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Create email drafts with specified content and recipients.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.email.create_draft()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/email/draft", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def search(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Search through emails with configurable filters.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.email.search()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/email/search", method="GET", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def send(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Send emails to specified recipients.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.email.send()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/email/send", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncEmailClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def create_draft(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Create email drafts with specified content and recipients.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.email.create_draft()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/email/draft", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def search(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Search through emails with configurable filters.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.email.search()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/email/search", method="GET", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def send(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Send emails to specified recipients.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.email.send()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/email/send", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/tools/structured_data_extractor/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

athena/tools/structured_data_extractor/client.py

@@ -0,0 +1,150 @@
+# 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.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.structured_data_extractor_reponse import StructuredDataExtractorReponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class StructuredDataExtractorClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def invoke(
+        self,
+        *,
+        asset_ids: typing.Sequence[str],
+        json_schema: typing.Dict[str, typing.Any],
+        map_: typing.Optional[bool] = OMIT,
+        reduce: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> StructuredDataExtractorReponse:
+        """
+        Coming soon! Extract structured data from assets of arbitrary length.
+
+        Parameters
+        ----------
+        asset_ids : typing.Sequence[str]
+            The IDs of the assets from which to extract structured data matching `json_schema`.
+
+        json_schema : typing.Dict[str, typing.Any]
+            The JSON schema to use for validation (version draft 2020-12).  See the docs [here](https://json-schema.org/learn/getting-started-step-by-step).
+
+        map_ : typing.Optional[bool]
+            Whether to split the asset into chunks and attempt to extract the schema from each chunk.  Set to false if you know the asset is small.
+
+        reduce : typing.Optional[bool]
+            If `map`, whether to reduce the chunks to a single structured object (true) or return the full list (false).  Use True unless you want to preserve duplicates from each page or expect the object to overflow the output context.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        StructuredDataExtractorReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.structured_data_extractor.invoke(
+            asset_ids=["asset_ids"],
+            json_schema={"key": "value"},
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/structured-data-extractor/invoke",
+            method="POST",
+            json={"asset_ids": asset_ids, "json_schema": json_schema, "map": map_, "reduce": reduce},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(StructuredDataExtractorReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncStructuredDataExtractorClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def invoke(
+        self,
+        *,
+        asset_ids: typing.Sequence[str],
+        json_schema: typing.Dict[str, typing.Any],
+        map_: typing.Optional[bool] = OMIT,
+        reduce: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> StructuredDataExtractorReponse:
+        """
+        Coming soon! Extract structured data from assets of arbitrary length.
+
+        Parameters
+        ----------
+        asset_ids : typing.Sequence[str]
+            The IDs of the assets from which to extract structured data matching `json_schema`.
+
+        json_schema : typing.Dict[str, typing.Any]
+            The JSON schema to use for validation (version draft 2020-12).  See the docs [here](https://json-schema.org/learn/getting-started-step-by-step).
+
+        map_ : typing.Optional[bool]
+            Whether to split the asset into chunks and attempt to extract the schema from each chunk.  Set to false if you know the asset is small.
+
+        reduce : typing.Optional[bool]
+            If `map`, whether to reduce the chunks to a single structured object (true) or return the full list (false).  Use True unless you want to preserve duplicates from each page or expect the object to overflow the output context.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        StructuredDataExtractorReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.structured_data_extractor.invoke(
+            asset_ids=["asset_ids"],
+            json_schema={"key": "value"},
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/structured-data-extractor/invoke",
+            method="POST",
+            json={"asset_ids": asset_ids, "json_schema": json_schema, "map": map_, "reduce": reduce},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(StructuredDataExtractorReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/tools/tasks/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

athena/tools/tasks/client.py

@@ -0,0 +1,87 @@
+# 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.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+
+
+class TasksClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def run_task(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Run a [task](https://resources.athenaintel.com/docs/task-studio/home).
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.tasks.run_task()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/tasks/run", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncTasksClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def run_task(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Run a [task](https://resources.athenaintel.com/docs/task-studio/home).
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.tasks.run_task()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/tasks/run", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)