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

athena/agents/general/client.py

@@ -0,0 +1,397 @@
+# 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.general_agent_request import GeneralAgentRequest
+from ...types.general_agent_response import GeneralAgentResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class GeneralClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def batch(
+        self, *, request: typing.Sequence[GeneralAgentRequest], request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[GeneralAgentResponse]:
+        """
+        Coming soon! Call the general agent with batched requests and return the results.
+
+        Parameters
+        ----------
+        request : typing.Sequence[GeneralAgentRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[GeneralAgentResponse]
+            Successful Response
+
+        Examples
+        --------
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.general.batch(
+            request=[
+                GeneralAgentRequest(
+                    config=GeneralAgentConfig(
+                        enabled_tools=[Tool.SEARCH],
+                    ),
+                    messages=[
+                        GeneralAgentRequestMessagesItem(
+                            content="Please call the search tool for AAPL news.",
+                            type=GeneralAgentRequestMessagesItemType.USER,
+                        )
+                    ],
+                )
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/general/batch", method="POST", json=request, request_options=request_options, omit=OMIT
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.List[GeneralAgentResponse], _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)
+
+    def invoke(
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
+        """
+        Call the general Athena agent synchronously.
+
+        Call the agent with the messages list, wait for the agent to complete,
+        and return the result.
+
+        Parameters
+        ----------
+        request : GeneralAgentRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GeneralAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.general.invoke(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[
+                    GeneralAgentRequestMessagesItem(
+                        content="Please call the search tool for AAPL news.",
+                        type=GeneralAgentRequestMessagesItemType.USER,
+                    )
+                ],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/general/invoke", method="POST", json=request, request_options=request_options, omit=OMIT
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(GeneralAgentResponse, _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)
+
+    def stream_events(
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
+        """
+        Coming soon! Call the general agent and stream events for real-time chat applications.
+
+        Parameters
+        ----------
+        request : GeneralAgentRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GeneralAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.general.stream_events(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[
+                    GeneralAgentRequestMessagesItem(
+                        content="Please call the search tool for AAPL news.",
+                        type=GeneralAgentRequestMessagesItemType.USER,
+                    )
+                ],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/general/stream_events",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(GeneralAgentResponse, _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 AsyncGeneralClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def batch(
+        self, *, request: typing.Sequence[GeneralAgentRequest], request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[GeneralAgentResponse]:
+        """
+        Coming soon! Call the general agent with batched requests and return the results.
+
+        Parameters
+        ----------
+        request : typing.Sequence[GeneralAgentRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[GeneralAgentResponse]
+            Successful Response
+
+        Examples
+        --------
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.general.batch(
+            request=[
+                GeneralAgentRequest(
+                    config=GeneralAgentConfig(
+                        enabled_tools=[Tool.SEARCH],
+                    ),
+                    messages=[
+                        GeneralAgentRequestMessagesItem(
+                            content="Please call the search tool for AAPL news.",
+                            type=GeneralAgentRequestMessagesItemType.USER,
+                        )
+                    ],
+                )
+            ],
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/general/batch", method="POST", json=request, request_options=request_options, omit=OMIT
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.List[GeneralAgentResponse], _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)
+
+    async def invoke(
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
+        """
+        Call the general Athena agent synchronously.
+
+        Call the agent with the messages list, wait for the agent to complete,
+        and return the result.
+
+        Parameters
+        ----------
+        request : GeneralAgentRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GeneralAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.general.invoke(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[
+                    GeneralAgentRequestMessagesItem(
+                        content="Please call the search tool for AAPL news.",
+                        type=GeneralAgentRequestMessagesItemType.USER,
+                    )
+                ],
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/general/invoke", method="POST", json=request, request_options=request_options, omit=OMIT
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(GeneralAgentResponse, _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)
+
+    async def stream_events(
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
+        """
+        Coming soon! Call the general agent and stream events for real-time chat applications.
+
+        Parameters
+        ----------
+        request : GeneralAgentRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GeneralAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.general.stream_events(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[
+                    GeneralAgentRequestMessagesItem(
+                        content="Please call the search tool for AAPL news.",
+                        type=GeneralAgentRequestMessagesItemType.USER,
+                    )
+                ],
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/general/stream_events",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(GeneralAgentResponse, _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/agents/research/__init__.py

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

athena/agents/research/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.research_agent_response import ResearchAgentResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class ResearchClient:
+    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
+    ) -> ResearchAgentResponse:
+        """
+        Coming soon! Conduct research using web and other sources.
+
+        Parameters
+        ----------
+        config : typing.Dict[str, typing.Any]
+            Configuration for the research agent including search parameters and sources
+
+        messages : typing.Sequence[typing.Dict[str, typing.Any]]
+            The messages to send to the research agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ResearchAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.research.invoke(
+            config={"key": "value"},
+            messages=[{"key": "value"}],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/research/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(ResearchAgentResponse, _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 AsyncResearchClient:
+    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
+    ) -> ResearchAgentResponse:
+        """
+        Coming soon! Conduct research using web and other sources.
+
+        Parameters
+        ----------
+        config : typing.Dict[str, typing.Any]
+            Configuration for the research agent including search parameters and sources
+
+        messages : typing.Sequence[typing.Dict[str, typing.Any]]
+            The messages to send to the research agent
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ResearchAgentResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.research.invoke(
+            config={"key": "value"},
+            messages=[{"key": "value"}],
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/research/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(ResearchAgentResponse, _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/agents/sql/__init__.py

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

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/base_client.py

@@ -4,6 +4,7 @@ import typing
 
 import httpx
 
+from .agents.client import AgentsClient, AsyncAgentsClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import AthenaEnvironment
 from .query.client import AsyncQueryClient, QueryClient
@@ -68,6 +69,7 @@ class BaseAthena:
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.agents = AgentsClient(client_wrapper=self._client_wrapper)
         self.query = QueryClient(client_wrapper=self._client_wrapper)
         self.tools = ToolsClient(client_wrapper=self._client_wrapper)
 
@@ -130,6 +132,7 @@ class AsyncBaseAthena:
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
         self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
         self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)