athena-intelligence 0.1.13__py3-none-any.whl → 0.1.14__py3-none-any.whl

athena/client.py

@@ -6,7 +6,8 @@ import httpx
 
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import AthenaEnvironment
-from .message.client import AsyncMessageClient, MessageClient
+from .polling_message_client import \
+        MessagePollingClient, AsyncMessagePollingClient
 
 
 class Athena:
@@ -47,7 +48,7 @@ class Athena:
             api_key=api_key,
             httpx_client=httpx.Client(timeout=timeout) if httpx_client is None else httpx_client,
         )
-        self.message = MessageClient(client_wrapper=self._client_wrapper)
+        self.message = MessagePollingClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncAthena:
@@ -88,7 +89,7 @@ class AsyncAthena:
             api_key=api_key,
             httpx_client=httpx.AsyncClient(timeout=timeout) if httpx_client is None else httpx_client,
         )
-        self.message = AsyncMessageClient(client_wrapper=self._client_wrapper)
+        self.message = AsyncMessagePollingClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:
@@ -97,4 +98,4 @@ def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaE
     elif environment is not None:
         return environment.value
     else:
-        raise Exception("Please pass in either base_url or environment to construct the client")
+        raise Exception("Please pass in either base_url or environment to construct the client")

athena/core/client_wrapper.py

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

athena/polling_message_client.py

@@ -0,0 +1,128 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import time
+import typing
+
+from athena.types.message_out_dto import MessageOutDto
+
+from .core import RequestOptions
+from .message.client import AsyncMessageClient, MessageClient
+from .types.model import Model
+from .types.tools import Tools
+from .types.status_enum import StatusEnum
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class MessagePollingClient(MessageClient):
+
+    def submit(  # type: ignore
+        self,
+        *,
+        content: str,
+        model: typing.Optional[Model] = OMIT,
+        tools: typing.Optional[typing.Sequence[Tools]] = OMIT,
+        conversation_id: typing.Optional[str] = OMIT,
+        conversation_name: typing.Optional[str] = OMIT,
+        poll_interval: int = 2,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> MessageOutDto:
+        """
+        Parameters:
+            - content: str.
+
+            - model: typing.Optional[Model].
+
+            - tools: typing.Optional[typing.Sequence[Tools]].
+
+            - conversation_id: typing.Optional[str].
+
+            - conversation_name: typing.Optional[str].
+
+            - poll_interval: int. The polling interval in seconds.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena import Model, Tools
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+            token="YOUR_TOKEN",
+        )
+        client.message.submit(
+            content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
+            model=Model.GPT_4_TURBO_PREVIEW,
+            tools=[Tools.SEARCH, Tools.BROWSE],
+        )
+        """
+        response = super().submit(
+          content=content,
+          model=model,
+          tools=tools,
+          conversation_id=conversation_id, 
+          conversation_name=conversation_name,
+          request_options=request_options)
+        message = self.get(id=response.id)
+        while message.status != StatusEnum.RESPONDED \
+                or message.status != StatusEnum.ERROR:
+            time.sleep(poll_interval)
+            message = self.get(id=message.id)
+        return message
+
+
+class AsyncMessagePollingClient(AsyncMessageClient):
+
+    async def submit(  # type: ignore
+        self,
+        *,
+        content: str,
+        model: typing.Optional[Model] = OMIT,
+        tools: typing.Optional[typing.Sequence[Tools]] = OMIT,
+        conversation_id: typing.Optional[str] = OMIT,
+        conversation_name: typing.Optional[str] = OMIT,
+        poll_interval: int = 2,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> MessageOutDto: 
+        """
+        Parameters:
+            - content: str.
+
+            - model: typing.Optional[Model].
+
+            - tools: typing.Optional[typing.Sequence[Tools]].
+
+            - conversation_id: typing.Optional[str].
+
+            - conversation_name: typing.Optional[str].
+
+            - poll_interval: int. The polling interval in seconds.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena import Model, Tools
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY"
+        )
+        client.message.submit(
+            content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
+            model=Model.GPT_4_TURBO_PREVIEW,
+            tools=[Tools.SEARCH, Tools.BROWSE],
+        )
+        """
+        response = await super().submit(
+          content=content,
+          model=model,
+          tools=tools,
+          conversation_id=conversation_id, 
+          conversation_name=conversation_name,
+          request_options=request_options)
+        message = await self.get(id=response.id)
+        while message.status != StatusEnum.RESPONDED \
+                or message.status != StatusEnum.ERROR:
+            time.sleep(poll_interval)
+            message = await self.get(id=message.id)
+        return message

{athena_intelligence-0.1.13.dist-info → athena_intelligence-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.13
+Version: 0.1.14
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3

{athena_intelligence-0.1.13.dist-info → athena_intelligence-0.1.14.dist-info}/RECORD

@@ -1,8 +1,8 @@
 athena/__init__.py,sha256=Prn1DsBOMu9g_ClmWZR6GIfi_3F-tZJ317hdudgLrQc,595
-athena/client.py,sha256=BcVF0FxjrXV27p24hcs1N4bxOpTcSQ8gsKyVX_O2bKE,3959
+athena/client.py,sha256=nFVJkayT6k0-DeqX-Ef7eVB-VHLHEq6wsqaZHy5Pp48,4004
 athena/core/__init__.py,sha256=95onSWXymaL0iV-nBsuKNgjh1LbsymAkRkx7POn1nc0,696
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=Ry6TGgXmEbAVLvqfry5Lejn_KxuzbXKhynoyrZgEqm8,1089
+athena/core/client_wrapper.py,sha256=BvopfvFC4H0qyWgsGVywhx0fiJME0ucn-s_FantFg84,1089
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
@@ -13,6 +13,7 @@ athena/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
 athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
 athena/message/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/message/client.py,sha256=KUdDnXGBMxWsr9FQ_GjzJGprOBd1_rOwAuLY7QJNjJQ,11741
+athena/polling_message_client.py,sha256=cWkF7nacXRtEj_r3Nud_Qq3XVAFbHoqK1EYJwbXiNRU,4159
 athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 athena/types/__init__.py,sha256=tcRJeTt7Xlrribx1EwOB3mKiUsHUKFP0vojjIWYHE74,571
 athena/types/http_validation_error.py,sha256=Fcv_CTMMrLvCeTHjF0n5xf5tskMDgt-J6H9gp654eQw,973
@@ -23,6 +24,6 @@ athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,6
 athena/types/tools.py,sha256=Vfb-D9CYcPZH6MBATNODgfXjFyBpCs4qbkqpCMl7eBM,1277
 athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
 athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
-athena_intelligence-0.1.13.dist-info/METADATA,sha256=L89SxGPQX_7X6bQG-HJbUU4Udr8wX5wK7EqR3W_zXDU,2258
-athena_intelligence-0.1.13.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.13.dist-info/RECORD,,
+athena_intelligence-0.1.14.dist-info/METADATA,sha256=iVwFlweS0hfqBwpONRN_07dA-sFMHClxhHLWoHDht84,2258
+athena_intelligence-0.1.14.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.14.dist-info/RECORD,,