athena-intelligence 0.1.119__py3-none-any.whl → 0.1.121__py3-none-any.whl

athena/agents/sql/client.py

@@ -0,0 +1,134 @@
+# 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.sql_agent_response import SqlAgentResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class SqlClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def invoke(
+        self,
+        *,
+        config: typing.Dict[str, typing.Any],
+        messages: typing.Sequence[typing.Dict[str, typing.Any]],
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> SqlAgentResponse:
+        """
+        Coming soon! Generate, execute, and test SQL queries. Returns an asset ID for the query object.
+
+        Parameters
+        ----------
+        config : typing.Dict[str, typing.Any]
+            Configuration for the SQL agent including database connection details and query parameters
+
+        messages : typing.Sequence[typing.Dict[str, typing.Any]]
+            The messages to send to the SQL agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SqlAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.sql.invoke(
+            config={"key": "value"},
+            messages=[{"key": "value"}],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/sql/invoke",
+            method="POST",
+            json={"config": config, "messages": messages},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(SqlAgentResponse, _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 AsyncSqlClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def invoke(
+        self,
+        *,
+        config: typing.Dict[str, typing.Any],
+        messages: typing.Sequence[typing.Dict[str, typing.Any]],
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> SqlAgentResponse:
+        """
+        Coming soon! Generate, execute, and test SQL queries. Returns an asset ID for the query object.
+
+        Parameters
+        ----------
+        config : typing.Dict[str, typing.Any]
+            Configuration for the SQL agent including database connection details and query parameters
+
+        messages : typing.Sequence[typing.Dict[str, typing.Any]]
+            The messages to send to the SQL agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SqlAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.sql.invoke(
+            config={"key": "value"},
+            messages=[{"key": "value"}],
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/sql/invoke",
+            method="POST",
+            json={"config": config, "messages": messages},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(SqlAgentResponse, _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/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.119",
+            "X-Fern-SDK-Version": "0.1.121",
         }
         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/{agents → tools}/structured_data_extractor/client.py

@@ -36,7 +36,7 @@ class StructuredDataExtractorClient:
             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.
+            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.
@@ -59,13 +59,13 @@ class StructuredDataExtractorClient:
         client = Athena(
             api_key="YOUR_API_KEY",
         )
-        client.agents.structured_data_extractor.invoke(
+        client.tools.structured_data_extractor.invoke(
             asset_ids=["asset_ids"],
             json_schema={"key": "value"},
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/v0/agents/structured-data-extractor/invoke",
+            "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,
@@ -104,7 +104,7 @@ class AsyncStructuredDataExtractorClient:
             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.
+            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.
@@ -127,13 +127,13 @@ class AsyncStructuredDataExtractorClient:
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
-        await client.agents.structured_data_extractor.invoke(
+        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/agents/structured-data-extractor/invoke",
+            "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,

athena/tools/tasks/__init__.py

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