athena-intelligence 0.1.59__py3-none-any.whl → 0.1.60__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.59",
+            "X-Fern-SDK-Version": "0.1.60",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/types/llm_model.py

@@ -14,6 +14,7 @@ class LlmModel(str, enum.Enum):
     GPT_35_TURBO = "gpt-3.5-turbo"
     GPT_4_TURBO = "gpt-4-turbo"
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
+    GPT_4_O = "gpt-4o"
     GPT_4 = "gpt-4"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
@@ -36,6 +37,7 @@ class LlmModel(str, enum.Enum):
         gpt_35_turbo: typing.Callable[[], T_Result],
         gpt_4_turbo: typing.Callable[[], T_Result],
         gpt_4_turbo_preview: typing.Callable[[], T_Result],
+        gpt_4_o: typing.Callable[[], T_Result],
         gpt_4: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
@@ -59,6 +61,8 @@ class LlmModel(str, enum.Enum):
             return gpt_4_turbo()
         if self is LlmModel.GPT_4_TURBO_PREVIEW:
             return gpt_4_turbo_preview()
+        if self is LlmModel.GPT_4_O:
+            return gpt_4_o()
         if self is LlmModel.GPT_4:
             return gpt_4()
         if self is LlmModel.MIXTRAL_SMALL_8_X_7_B_0211:

athena/types/model.py

@@ -15,6 +15,7 @@ class Model(str, enum.Enum):
     GPT_4_TURBO = "gpt-4-turbo"
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
     GPT_4 = "gpt-4"
+    GPT_4_O = "gpt-4o"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
     MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
@@ -32,6 +33,7 @@ class Model(str, enum.Enum):
         gpt_4_turbo: typing.Callable[[], T_Result],
         gpt_4_turbo_preview: typing.Callable[[], T_Result],
         gpt_4: typing.Callable[[], T_Result],
+        gpt_4_o: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
         mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
@@ -51,6 +53,8 @@ class Model(str, enum.Enum):
             return gpt_4_turbo_preview()
         if self is Model.GPT_4:
             return gpt_4()
+        if self is Model.GPT_4_O:
+            return gpt_4_o()
         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:

{athena_intelligence-0.1.59.dist-info → athena_intelligence-0.1.60.dist-info}/METADATA

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

{athena_intelligence-0.1.59.dist-info → athena_intelligence-0.1.60.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=21DEFMBtapoi_57LP2xfhiOKWX1PLlEovSMDGU-JF8w,1495
+athena/core/client_wrapper.py,sha256=6WS6Ny57Y_HbybCSpga92o9T6LSQi-UajT7-6BP5IFQ,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
@@ -43,11 +43,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=b1W7CtiusS5Du3jLugdvpd7fUQ5GhcIU-2YxK4UAqvs,4101
+athena/types/llm_model.py,sha256=bkxs5YcNXyv7YxIkKQIxEGMjRUVkWku5VEnQhHklQAE,4238
 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=73jxfZPxAtK0iFex4bluPRrfpF5sY3Y6VFa9iALykBw,2991
+athena/types/model.py,sha256=K2OAggAs9-ljK2TSr1XrlnmfRaLyH374wWrIJGcyU0s,3125
 athena/types/report.py,sha256=km2CgCbHBXQQbPai1y5sGlsQpO7WAlUVvdsRC_7f4KI,926
 athena/types/snippet.py,sha256=Mrc92_hBABJQjCSToAA-FgwhvO-Jn8Kjm-lYI6aMlUY,1106
 athena/types/sql_results.py,sha256=ExPFds4vZ425AxGt0jhykbPhOjkplZPGQwVKb0LHg_g,880
@@ -58,6 +58,6 @@ athena/types/url_result.py,sha256=lIgnQeyKy_UfFFPe7HMrrRzb-SK089RxcKcKN9Q3DNQ,87
 athena/types/validation_error.py,sha256=yqombbKLBSzTPFn6CJH_hbo7tpS68T3JvMdd7kBtO1g,972
 athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
-athena_intelligence-0.1.59.dist-info/METADATA,sha256=qyuSEH8iws7O5tlJcYSSAQgH5ki8GDrOw--8dXFN-Fs,4738
-athena_intelligence-0.1.59.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.59.dist-info/RECORD,,
+athena_intelligence-0.1.60.dist-info/METADATA,sha256=f4JcHRM52V-cstpugvqntPA7g-T_9pQ2PxQhrsqRNNc,4738
+athena_intelligence-0.1.60.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.60.dist-info/RECORD,,