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

athena/agents/{athena_assistant → general}/client.py

@@ -8,66 +8,70 @@ 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.athena_assistant_reponse import AthenaAssistantReponse
-from ...types.athena_assistant_request import AthenaAssistantRequest
+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 AthenaAssistantClient:
+class GeneralClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
     def batch(
-        self,
-        *,
-        request: typing.Sequence[AthenaAssistantRequest],
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[AthenaAssistantReponse]:
+        self, *, request: typing.Sequence[GeneralAgentRequest], request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[GeneralAgentResponse]:
         """
-        Call the Athena assistant with batched requests and return the results.
+        Coming soon! Call the general agent with batched requests and return the results.
 
         Parameters
         ----------
-        request : typing.Sequence[AthenaAssistantRequest]
+        request : typing.Sequence[GeneralAgentRequest]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[AthenaAssistantReponse]
+        typing.List[GeneralAgentResponse]
             Successful Response
 
         Examples
         --------
-        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
         from athena.client import Athena
 
         client = Athena(
             api_key="YOUR_API_KEY",
         )
-        client.agents.athena_assistant.batch(
+        client.agents.general.batch(
             request=[
-                AthenaAssistantRequest(
-                    config=AthenaAssistantConfig(
+                GeneralAgentRequest(
+                    config=GeneralAgentConfig(
                         enabled_tools=[Tool.SEARCH],
                     ),
-                    messages=[],
+                    messages=[
+                        GeneralAgentRequestMessagesItem(
+                            content="Please call the search tool for AAPL news.",
+                            type=GeneralAgentRequestMessagesItemType.USER,
+                        )
+                    ],
                 )
             ],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/v0/agents/athena-assistant/batch",
-            method="POST",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
+            "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[AthenaAssistantReponse], _response.json())  # type: ignore
+            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:
@@ -77,52 +81,59 @@ class AthenaAssistantClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def invoke(
-        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> AthenaAssistantReponse:
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
         """
-        Call the Athena Assistant agent synchronously.
+        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 : AthenaAssistantRequest
+        request : GeneralAgentRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AthenaAssistantReponse
+        GeneralAgentResponse
             Successful Response
 
         Examples
         --------
-        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
         from athena.client import Athena
 
         client = Athena(
             api_key="YOUR_API_KEY",
         )
-        client.agents.athena_assistant.invoke(
-            request=AthenaAssistantRequest(
-                config=AthenaAssistantConfig(
+        client.agents.general.invoke(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
                     enabled_tools=[Tool.SEARCH],
                 ),
-                messages=[],
+                messages=[
+                    GeneralAgentRequestMessagesItem(
+                        content="Please call the search tool for AAPL news.",
+                        type=GeneralAgentRequestMessagesItemType.USER,
+                    )
+                ],
             ),
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/v0/agents/athena-assistant/invoke",
-            method="POST",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
+            "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(AthenaAssistantReponse, _response.json())  # type: ignore
+            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:
@@ -132,49 +143,60 @@ class AthenaAssistantClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def stream_events(
-        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> AthenaAssistantReponse:
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
         """
-        Call the Athena assistant and stream events.
+        Coming soon! Call the general agent and stream events for real-time chat applications.
 
         Parameters
         ----------
-        request : AthenaAssistantRequest
+        request : GeneralAgentRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AthenaAssistantReponse
+        GeneralAgentResponse
             Successful Response
 
         Examples
         --------
-        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
         from athena.client import Athena
 
         client = Athena(
             api_key="YOUR_API_KEY",
         )
-        client.agents.athena_assistant.stream_events(
-            request=AthenaAssistantRequest(
-                config=AthenaAssistantConfig(
+        client.agents.general.stream_events(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
                     enabled_tools=[Tool.SEARCH],
                 ),
-                messages=[],
+                messages=[
+                    GeneralAgentRequestMessagesItem(
+                        content="Please call the search tool for AAPL news.",
+                        type=GeneralAgentRequestMessagesItemType.USER,
+                    )
+                ],
             ),
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/v0/agents/athena-assistant/stream_events",
+            "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(AthenaAssistantReponse, _response.json())  # type: ignore
+            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:
@@ -184,59 +206,63 @@ class AthenaAssistantClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
 
-class AsyncAthenaAssistantClient:
+class AsyncGeneralClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
     async def batch(
-        self,
-        *,
-        request: typing.Sequence[AthenaAssistantRequest],
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[AthenaAssistantReponse]:
+        self, *, request: typing.Sequence[GeneralAgentRequest], request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[GeneralAgentResponse]:
         """
-        Call the Athena assistant with batched requests and return the results.
+        Coming soon! Call the general agent with batched requests and return the results.
 
         Parameters
         ----------
-        request : typing.Sequence[AthenaAssistantRequest]
+        request : typing.Sequence[GeneralAgentRequest]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[AthenaAssistantReponse]
+        typing.List[GeneralAgentResponse]
             Successful Response
 
         Examples
         --------
-        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
-        await client.agents.athena_assistant.batch(
+        await client.agents.general.batch(
             request=[
-                AthenaAssistantRequest(
-                    config=AthenaAssistantConfig(
+                GeneralAgentRequest(
+                    config=GeneralAgentConfig(
                         enabled_tools=[Tool.SEARCH],
                     ),
-                    messages=[],
+                    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/athena-assistant/batch",
-            method="POST",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
+            "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[AthenaAssistantReponse], _response.json())  # type: ignore
+            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:
@@ -246,52 +272,59 @@ class AsyncAthenaAssistantClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def invoke(
-        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> AthenaAssistantReponse:
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
         """
-        Call the Athena Assistant agent synchronously.
+        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 : AthenaAssistantRequest
+        request : GeneralAgentRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AthenaAssistantReponse
+        GeneralAgentResponse
             Successful Response
 
         Examples
         --------
-        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
-        await client.agents.athena_assistant.invoke(
-            request=AthenaAssistantRequest(
-                config=AthenaAssistantConfig(
+        await client.agents.general.invoke(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
                     enabled_tools=[Tool.SEARCH],
                 ),
-                messages=[],
+                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/athena-assistant/invoke",
-            method="POST",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
+            "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(AthenaAssistantReponse, _response.json())  # type: ignore
+            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:
@@ -301,49 +334,60 @@ class AsyncAthenaAssistantClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def stream_events(
-        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
-    ) -> AthenaAssistantReponse:
+        self, *, request: GeneralAgentRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> GeneralAgentResponse:
         """
-        Call the Athena assistant and stream events.
+        Coming soon! Call the general agent and stream events for real-time chat applications.
 
         Parameters
         ----------
-        request : AthenaAssistantRequest
+        request : GeneralAgentRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AthenaAssistantReponse
+        GeneralAgentResponse
             Successful Response
 
         Examples
         --------
-        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena import (
+            GeneralAgentConfig,
+            GeneralAgentRequest,
+            GeneralAgentRequestMessagesItem,
+            GeneralAgentRequestMessagesItemType,
+            Tool,
+        )
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
-        await client.agents.athena_assistant.stream_events(
-            request=AthenaAssistantRequest(
-                config=AthenaAssistantConfig(
+        await client.agents.general.stream_events(
+            request=GeneralAgentRequest(
+                config=GeneralAgentConfig(
                     enabled_tools=[Tool.SEARCH],
                 ),
-                messages=[],
+                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/athena-assistant/stream_events",
+            "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(AthenaAssistantReponse, _response.json())  # type: ignore
+            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:

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.
+