athena-intelligence 0.1.31__py3-none-any.whl → 0.1.33__py3-none-any.whl

athena/core/client_wrapper.py

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

athena/message/client.py

@@ -61,8 +61,8 @@ class MessageClient:
         )
         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],
+            model=Model.GPT_35_TURBO,
+            tools=[Tools.SEARCH, Tools.BROWSE],
         )
         """
         _request: typing.Dict[str, typing.Any] = {"content": content}
@@ -193,8 +193,8 @@ class AsyncMessageClient:
         )
         await 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],
+            model=Model.GPT_35_TURBO,
+            tools=[Tools.SEARCH, Tools.BROWSE],
         )
         """
         _request: typing.Dict[str, typing.Any] = {"content": content}

athena/types/model.py

@@ -16,6 +16,12 @@ class Model(str, enum.Enum):
     GPT_4 = "gpt-4"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
+    LLAMA_V_270_B_CHAT = "llama-v2-70b-chat"
+    CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
+    CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
+    CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
+    GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
+    DATABRICKS_DBRX = "databricks-dbrx"
 
     def visit(
         self,
@@ -24,6 +30,12 @@ class Model(str, enum.Enum):
         gpt_4: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
+        llama_v_270_b_chat: typing.Callable[[], T_Result],
+        claude_3_opus_20240229: typing.Callable[[], T_Result],
+        claude_3_sonnet_20240229: typing.Callable[[], T_Result],
+        claude_3_haiku_20240307: typing.Callable[[], T_Result],
+        google_gemini_10_pro_latest: typing.Callable[[], T_Result],
+        databricks_dbrx: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is Model.GPT_35_TURBO:
             return gpt_35_turbo()
@@ -35,3 +47,15 @@ class Model(str, enum.Enum):
             return mixtral_small_8_x_7_b_0211()
         if self is Model.MISTRAL_LARGE_0224:
             return mistral_large_0224()
+        if self is Model.LLAMA_V_270_B_CHAT:
+            return llama_v_270_b_chat()
+        if self is Model.CLAUDE_3_OPUS_20240229:
+            return claude_3_opus_20240229()
+        if self is Model.CLAUDE_3_SONNET_20240229:
+            return claude_3_sonnet_20240229()
+        if self is Model.CLAUDE_3_HAIKU_20240307:
+            return claude_3_haiku_20240307()
+        if self is Model.GOOGLE_GEMINI_10_PRO_LATEST:
+            return google_gemini_10_pro_latest()
+        if self is Model.DATABRICKS_DBRX:
+            return databricks_dbrx()

{athena_intelligence-0.1.31.dist-info → athena_intelligence-0.1.33.dist-info}/METADATA

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

{athena_intelligence-0.1.31.dist-info → athena_intelligence-0.1.33.dist-info}/RECORD

@@ -3,7 +3,7 @@ athena/base_client.py,sha256=0V7k8HeQrYl1dK2lni53I0EkrmMwFiTVW5Pdt-fScuE,4806
 athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
 athena/core/__init__.py,sha256=RWfyDqkzWsf8e3VGc3NV60MovfJbg5XWzNFGB2DZ0hA,790
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=WiHpvO9N7x_BMwVn3_kDlhGzG5T2mx3DJz7EJOwvwDs,1198
+athena/core/client_wrapper.py,sha256=VBjHF6uPLJOCMUri_dfWQE7iTUfsNCShd2zjDPvCLI8,1198
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=LI0yP3jUyE0Ue7oyBcI9nyo1pljOwh9Y5ycTeIpKwOg,4882
@@ -16,7 +16,7 @@ athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
 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=2jSeLD6LBDTWdYIHiz_iqdB78KX-nlzqi0NRosmyTgM,12215
+athena/message/client.py,sha256=NT0E6j-LsjZygrjEzn-Bcy8T_Q_ayNK0jR6z3hQjcnI,12185
 athena/polling_message_client.py,sha256=dmmycImvog2niEFFPo4rE5xMJHUlq9NqAr4xlFK6_Os,3998
 athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 athena/query/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -32,7 +32,7 @@ athena/types/get_snippets_response.py,sha256=Lpn7bHJLpPQozN93unCV-8eByAAfz1MhQWR
 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=sp9NmrUNYebEwK4S8N4YtMEvvfrLUCYjb6-9XwoCGY4,1145
+athena/types/model.py,sha256=w-JHrWXHgMpCtrptPzeHgG8fu9Hle-QmmLP4RG6dCdc,2390
 athena/types/report.py,sha256=QVaqVfHMAV3s9_V2CqjIEMcRrbJhD8zmi82vrk2A8x0,946
 athena/types/snippet.py,sha256=POIVJNV9iQxiVegB_qwQx-PZPPSyoIPhyxTsueNVUGA,1126
 athena/types/sql_results.py,sha256=pNH32nyf1bzoYJs3FgHctLdLO02oOjyGgLkHACACB6k,900
@@ -40,6 +40,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.31.dist-info/METADATA,sha256=6Li2bEpumfkLuVdWu2sfbnAVyyn3p0G8cdMcfCd9RzM,4738
-athena_intelligence-0.1.31.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.31.dist-info/RECORD,,
+athena_intelligence-0.1.33.dist-info/METADATA,sha256=SmQ1tqzbjpOmwbjVYkyvbdiC9RG-xAJQIKWIokqG7QI,4738
+athena_intelligence-0.1.33.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.33.dist-info/RECORD,,