athena-intelligence 0.1.77__py3-none-any.whl → 0.1.78__py3-none-any.whl

athena/core/client_wrapper.py

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

athena/types/llm_model.py

@@ -22,6 +22,7 @@ class LlmModel(str, enum.Enum):
     MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
     LLAMA_V_38_B_INSTRUCT = "llama-v3-8b-instruct"
     LLAMA_V_370_B_INSTRUCT = "llama-v3-70b-instruct"
+    LLAMA_V_3_P_1405_B_INSTRUCT = "llama-v3p1-405b-instruct"
     CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
     CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
     CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
@@ -30,6 +31,11 @@ class LlmModel(str, enum.Enum):
     GROQ_LLAMA_38_B_8192 = "groq-llama3-8b-8192"
     GROQ_LLAMA_370_B_8192 = "groq-llama3-70b-8192"
     GROQ_GEMMA_7_B_IT = "groq-gemma-7b-it"
+    GROQ_LLAMA_31405_B_REASONING = "groq-llama-3.1-405b-reasoning"
+    GROQ_LLAMA_3170_B_VERSATILE = "groq-llama-3.1-70b-versatile"
+    GROQ_LLAMA_318_B_INSTANT = "groq-llama-3.1-8b-instant"
+    GROQ_LLAMA_3_GROQ_70_B_8192_TOOL_USE_PREVIEW = "groq-llama3-groq-70b-8192-tool-use-preview"
+    GROQ_LLAMA_3_GROQ_8_B_8192_TOOL_USE_PREVIEW = "groq-llama3-groq-8b-8192-tool-use-preview"
     DATABRICKS_DBRX = "databricks-dbrx"
     GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
     GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
@@ -49,6 +55,7 @@ class LlmModel(str, enum.Enum):
         mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
         llama_v_38_b_instruct: typing.Callable[[], T_Result],
         llama_v_370_b_instruct: typing.Callable[[], T_Result],
+        llama_v_3_p_1405_b_instruct: typing.Callable[[], T_Result],
         claude_35_sonnet_20240620: typing.Callable[[], T_Result],
         claude_3_opus_20240229: typing.Callable[[], T_Result],
         claude_3_sonnet_20240229: typing.Callable[[], T_Result],
@@ -57,6 +64,11 @@ class LlmModel(str, enum.Enum):
         groq_llama_38_b_8192: typing.Callable[[], T_Result],
         groq_llama_370_b_8192: typing.Callable[[], T_Result],
         groq_gemma_7_b_it: typing.Callable[[], T_Result],
+        groq_llama_31405_b_reasoning: typing.Callable[[], T_Result],
+        groq_llama_3170_b_versatile: typing.Callable[[], T_Result],
+        groq_llama_318_b_instant: typing.Callable[[], T_Result],
+        groq_llama_3_groq_70_b_8192_tool_use_preview: typing.Callable[[], T_Result],
+        groq_llama_3_groq_8_b_8192_tool_use_preview: typing.Callable[[], T_Result],
         databricks_dbrx: typing.Callable[[], T_Result],
         google_gemini_10_pro_latest: typing.Callable[[], T_Result],
         google_gemini_15_pro_latest: typing.Callable[[], T_Result],
@@ -85,6 +97,8 @@ class LlmModel(str, enum.Enum):
             return llama_v_38_b_instruct()
         if self is LlmModel.LLAMA_V_370_B_INSTRUCT:
             return llama_v_370_b_instruct()
+        if self is LlmModel.LLAMA_V_3_P_1405_B_INSTRUCT:
+            return llama_v_3_p_1405_b_instruct()
         if self is LlmModel.CLAUDE_35_SONNET_20240620:
             return claude_35_sonnet_20240620()
         if self is LlmModel.CLAUDE_3_OPUS_20240229:
@@ -101,6 +115,16 @@ class LlmModel(str, enum.Enum):
             return groq_llama_370_b_8192()
         if self is LlmModel.GROQ_GEMMA_7_B_IT:
             return groq_gemma_7_b_it()
+        if self is LlmModel.GROQ_LLAMA_31405_B_REASONING:
+            return groq_llama_31405_b_reasoning()
+        if self is LlmModel.GROQ_LLAMA_3170_B_VERSATILE:
+            return groq_llama_3170_b_versatile()
+        if self is LlmModel.GROQ_LLAMA_318_B_INSTANT:
+            return groq_llama_318_b_instant()
+        if self is LlmModel.GROQ_LLAMA_3_GROQ_70_B_8192_TOOL_USE_PREVIEW:
+            return groq_llama_3_groq_70_b_8192_tool_use_preview()
+        if self is LlmModel.GROQ_LLAMA_3_GROQ_8_B_8192_TOOL_USE_PREVIEW:
+            return groq_llama_3_groq_8_b_8192_tool_use_preview()
         if self is LlmModel.DATABRICKS_DBRX:
             return databricks_dbrx()
         if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:

athena/types/model.py

@@ -25,6 +25,7 @@ class Model(str, enum.Enum):
     CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
     CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
     CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
+    CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
     GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
     DATABRICKS_DBRX = "databricks-dbrx"
 
@@ -44,6 +45,7 @@ class Model(str, enum.Enum):
         claude_3_opus_20240229: typing.Callable[[], T_Result],
         claude_3_sonnet_20240229: typing.Callable[[], T_Result],
         claude_3_haiku_20240307: typing.Callable[[], T_Result],
+        claude_35_sonnet_20240620: typing.Callable[[], T_Result],
         google_gemini_10_pro_latest: typing.Callable[[], T_Result],
         databricks_dbrx: typing.Callable[[], T_Result],
     ) -> T_Result:
@@ -75,6 +77,8 @@ class Model(str, enum.Enum):
             return claude_3_sonnet_20240229()
         if self is Model.CLAUDE_3_HAIKU_20240307:
             return claude_3_haiku_20240307()
+        if self is Model.CLAUDE_35_SONNET_20240620:
+            return claude_35_sonnet_20240620()
         if self is Model.GOOGLE_GEMINI_10_PRO_LATEST:
             return google_gemini_10_pro_latest()
         if self is Model.DATABRICKS_DBRX:

{athena_intelligence-0.1.77.dist-info → athena_intelligence-0.1.78.dist-info}/METADATA

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

{athena_intelligence-0.1.77.dist-info → athena_intelligence-0.1.78.dist-info}/RECORD

@@ -5,7 +5,7 @@ athena/chain/client.py,sha256=2vSu7d4RvgbGc7jbWpKkCs5dU-ryCIJ1i0I1EsoCEdQ,16177
 athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
 athena/core/__init__.py,sha256=1pNSKkwyQvMl_F0wohBqmoQAITptg3zlvCwsoSSzy7c,853
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=cVRzoQd4Ne-59pvfCqIXXUPMcdBsOBOTc3QS-CElRxM,1495
+athena/core/client_wrapper.py,sha256=ktlcne8pQKwjBz4oYtzrS1z1cArOoXgscTKdhAp8yfo,1495
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=5ok6hqgZDJhg57EHvMnr0BBaHdG50QxFPKaCZ9aVWTc,5059
@@ -47,11 +47,11 @@ athena/types/get_snippet_out.py,sha256=AkkF6YJcYysiQVnOvhRerHMsHkBTu1BP9tYZC8wET
 athena/types/get_snippets_response.py,sha256=pgwYqmddU5shKeVaE4RQSFN9SLsVAeQp3sqIkQlvzoU,969
 athena/types/http_validation_error.py,sha256=u4t-1U0DI0u3Zj_Oz7AmGmpL4sqBzoS_5nZHImWQbmM,953
 athena/types/langchain_documents_request_out.py,sha256=O1v7mcgt0ryaY4e8YODpAHYJKyUY7jYFBc0s93A1sgo,892
-athena/types/llm_model.py,sha256=YFj5S6Wcp7CV75YChhz0CTMFTlvmf_q6vVwhmLxwYvw,5087
+athena/types/llm_model.py,sha256=kGCC7WEAe-Lq7r6kiXhy5UCn6vdSdb7ydbq5gIjeye0,6666
 athena/types/map_reduce_chain_out.py,sha256=6R-fuxHaww60dhUAuwrdZPp5lV-DyFZh9SGLCc6fp8E,950
 athena/types/message_out.py,sha256=HJZizmFH7crD3OHm0fdTy3189F2gv5qR8aaUbTTfWFI,845
 athena/types/message_out_dto.py,sha256=1G8srlYaIYmoYRstLKm97xZGxK87DK57CiO9hYnt3gQ,1031
-athena/types/model.py,sha256=uUaC03YGJxMcgQzBf30jVC0NsLCICDxSYS4un0VYlME,3284
+athena/types/model.py,sha256=qUqa8Tcmb3o0Cxul_1UGiNlJGPlbr4OGqYNLtxf8oEQ,3510
 athena/types/publish_formats.py,sha256=1_F5vyEwDtxshFG0S2gNx05V8jZHFEK6ZoZkjIJVhyQ,885
 athena/types/query_model.py,sha256=ZRuFxbgYi6X9nLNQvmrOV594OaSpKmqdGBNdrmj8pZ4,1318
 athena/types/report.py,sha256=km2CgCbHBXQQbPai1y5sGlsQpO7WAlUVvdsRC_7f4KI,926
@@ -73,6 +73,6 @@ athena/upload/client.py,sha256=e5h10wZ7lGBasJ6X907x7nXHRhX600mLSkdw2qz6pmY,6385
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
 athena/workflow/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/workflow/client.py,sha256=uY9IS_v2GDQ-g2nbatpTUP1aT1oHbG_E8WAor8JzxPI,6249
-athena_intelligence-0.1.77.dist-info/METADATA,sha256=q2w7ZJqhgjdYKks1uZoBYEbf9n2DeZ9wQKhlBJRjTew,4738
-athena_intelligence-0.1.77.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.77.dist-info/RECORD,,
+athena_intelligence-0.1.78.dist-info/METADATA,sha256=gMOUYGf_9BzAsRTwr4S4XUBLauoiiqQqNujR6WPjloM,4738
+athena_intelligence-0.1.78.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.78.dist-info/RECORD,,