athena-intelligence 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

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.15",
+            "X-Fern-SDK-Version": "0.1.17",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/polling_message_client.py

@@ -17,7 +17,7 @@ OMIT = typing.cast(typing.Any, ...)
 
 class MessagePollingClient(MessageClient):
 
-    def submit(  # type: ignore
+    def submit_and_poll(
         self,
         *,
         content: str,
@@ -49,7 +49,6 @@ class MessagePollingClient(MessageClient):
 
         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",
@@ -65,8 +64,7 @@ class MessagePollingClient(MessageClient):
           conversation_name=conversation_name,
           request_options=request_options)
         message = self.get(id=response.id)
-        while message.status != StatusEnum.RESPONDED \
-                or message.status != StatusEnum.ERROR:
+        while message.status == StatusEnum.PENDING:
             time.sleep(poll_interval)
             message = self.get(id=message.id)
         return message
@@ -74,7 +72,7 @@ class MessagePollingClient(MessageClient):
 
 class AsyncMessagePollingClient(AsyncMessageClient):
 
-    async def submit(  # type: ignore
+    async def submit_and_poll(
         self,
         *,
         content: str,
@@ -121,8 +119,7 @@ class AsyncMessagePollingClient(AsyncMessageClient):
           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:
+        while message.status == StatusEnum.PENDING:
             time.sleep(poll_interval)
             message = await self.get(id=message.id)
-        return message
+        return message

athena/types/model.py

@@ -14,12 +14,14 @@ class Model(str, enum.Enum):
     GPT_35_TURBO = "gpt-3.5-turbo"
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
+    MISTRAL_LARGE_0224 = "mistral-large-0224"
 
     def visit(
         self,
         gpt_35_turbo: typing.Callable[[], T_Result],
         gpt_4_turbo_preview: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
+        mistral_large_0224: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is Model.GPT_35_TURBO:
             return gpt_35_turbo()
@@ -27,3 +29,5 @@ class Model(str, enum.Enum):
             return gpt_4_turbo_preview()
         if self is Model.MIXTRAL_SMALL_8_X_7_B_0211:
             return mixtral_small_8_x_7_b_0211()
+        if self is Model.MISTRAL_LARGE_0224:
+            return mistral_large_0224()

{athena_intelligence-0.1.15.dist-info → athena_intelligence-0.1.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.15
+Version: 0.1.17
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3
@@ -39,11 +39,16 @@ Simply import `Athena` and start making calls to our API.
 
 ```python
 from athena.client import Athena
+from athena import Model, Tools
 
 client = Athena(
   api_key="YOUR_API_KEY" # Defaults to ATHENA_API_KEY
 )
-message = client.messages.get("message_id")
+message = client.message.submit(
+    content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
+    model=Model.GPT_3_5_TURBO,
+    tools=[Tools.SEARCH, Tools.BROWSE, Tools.SEARCH],
+)
 ```
 
 ## Async Client
@@ -52,18 +57,42 @@ calls to our API.
 
 ```python
 from athena.client import AsyncAthena
+from athena import Model, Tools
 
 client = AsyncAthena(
   api_key="YOUR_API_KEY" # Defaults to ATHENA_API_KEY
 )
 
 async def main() -> None:
-    message = client.messages.get("message_id")
+    message = client.message.submit(
+      content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
+      model=Model.GPT_3_5_TURBO,
+      tools=[Tools.SEARCH, Tools.BROWSE, Tools.SEARCH],
+    )
     print("Received message", message)
 
 asyncio.run(main())
 ```
 
+## Polling
+The SDK provides helper functions that will automatically poll when 
+retrieving a message. Use the `submit_and_poll` method as shown below: 
+
+```python
+from athena.client import Athena
+from athena import Model, Tools
+
+client = Athena(api_key="...")
+message =  client.message.submit_and_poll(
+  content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
+  model=Model.GPT_3_5_TURBO,
+  tools=[Tools.SEARCH, Tools.BROWSE, Tools.SEARCH],
+)
+```
+
+By default, the method will poll every 2 seconds but you can override
+this with the `poll_interval` argument.
+
 ## Athena Module
 All of the models are nested within the Athena module. Let IntelliSense 
 guide you! 

{athena_intelligence-0.1.15.dist-info → athena_intelligence-0.1.17.dist-info}/RECORD

@@ -2,7 +2,7 @@ athena/__init__.py,sha256=Prn1DsBOMu9g_ClmWZR6GIfi_3F-tZJ317hdudgLrQc,595
 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=yjNC9j-BOq7sIqTbzhMGi3pQY6-in2qROV5SFaSpDBE,1089
+athena/core/client_wrapper.py,sha256=nenzqERD-XkoSQKEVXB33q0zDcutfpvVbfYEdlSCsiA,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,17 +13,17 @@ 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/polling_message_client.py,sha256=dmmycImvog2niEFFPo4rE5xMJHUlq9NqAr4xlFK6_Os,3998
 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
 athena/types/message_out.py,sha256=uvZY_Podv2XccEk8CICug9I_S2hFJTSzCBwcHiauW7A,865
 athena/types/message_out_dto.py,sha256=qgRibRbDNOWVnVGP7Rribh9WdoCT2CSiPUXeIWECqq4,1051
-athena/types/model.py,sha256=xyD4lk5YWZTzYQc6f76FUBOQe4_jTHSS8edSzOY5eQc,830
+athena/types/model.py,sha256=yZyzOojHN4mRCd-ITHJNiKZDpQhy7LKZcM9_saJ0qy4,1020
 athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,672
 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.15.dist-info/METADATA,sha256=iRONe2pK-3_nLA5ZNzVk4YPmO0LSnQPW-TYq4X_O4KE,3742
-athena_intelligence-0.1.15.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.15.dist-info/RECORD,,
+athena_intelligence-0.1.17.dist-info/METADATA,sha256=HcVpNxrLjXUYWCGsEofNrYTTBS-k0jl5amJKLzeYov4,4738
+athena_intelligence-0.1.17.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.17.dist-info/RECORD,,