athena-intelligence 0.1.29__py3-none-any.whl → 0.1.30__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. athena/core/client_wrapper.py +1 -1
  2. athena/types/model.py +8 -0
  3. {athena_intelligence-0.1.29.dist-info → athena_intelligence-0.1.30.dist-info}/METADATA +1 -1
  4. {athena_intelligence-0.1.29.dist-info → athena_intelligence-0.1.30.dist-info}/RECORD +5 -5
  5. {athena_intelligence-0.1.29.dist-info → athena_intelligence-0.1.30.dist-info}/WHEEL +0 -0
athena/core/client_wrapper.py CHANGED
@@ -16,7 +16,7 @@ class BaseClientWrapper:
16
16
  headers: typing.Dict[str, str] = {
17
17
  "X-Fern-Language": "Python",
18
18
  "X-Fern-SDK-Name": "athena-intelligence",
19
- "X-Fern-SDK-Version": "0.1.29",
19
+ "X-Fern-SDK-Version": "0.1.30",
20
20
  }
21
21
  headers["X-API-KEY"] = self.api_key
22
22
  return headers
athena/types/model.py CHANGED
@@ -13,6 +13,8 @@ class Model(str, enum.Enum):
13
13
 
14
14
  GPT_35_TURBO = "gpt-3.5-turbo"
15
15
  GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
16
+ GPT_4 = "gpt-4"
17
+ GPT_432_K = "gpt-4-32k"
16
18
  MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
17
19
  MISTRAL_LARGE_0224 = "mistral-large-0224"
18
20
 
@@ -20,6 +22,8 @@ class Model(str, enum.Enum):
20
22
  self,
21
23
  gpt_35_turbo: typing.Callable[[], T_Result],
22
24
  gpt_4_turbo_preview: typing.Callable[[], T_Result],
25
+ gpt_4: typing.Callable[[], T_Result],
26
+ gpt_432_k: typing.Callable[[], T_Result],
23
27
  mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
24
28
  mistral_large_0224: typing.Callable[[], T_Result],
25
29
  ) -> T_Result:
@@ -27,6 +31,10 @@ class Model(str, enum.Enum):
27
31
  return gpt_35_turbo()
28
32
  if self is Model.GPT_4_TURBO_PREVIEW:
29
33
  return gpt_4_turbo_preview()
34
+ if self is Model.GPT_4:
35
+ return gpt_4()
36
+ if self is Model.GPT_432_K:
37
+ return gpt_432_k()
30
38
  if self is Model.MIXTRAL_SMALL_8_X_7_B_0211:
31
39
  return mixtral_small_8_x_7_b_0211()
32
40
  if self is Model.MISTRAL_LARGE_0224:
{athena_intelligence-0.1.29.dist-info → athena_intelligence-0.1.30.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: athena-intelligence
3
- Version: 0.1.29
3
+ Version: 0.1.30
4
4
  Summary:
5
5
  Requires-Python: >=3.8,<4.0
6
6
  Classifier: Programming Language :: Python :: 3
{athena_intelligence-0.1.29.dist-info → athena_intelligence-0.1.30.dist-info}/RECORD RENAMED
@@ -3,7 +3,7 @@ athena/base_client.py,sha256=0V7k8HeQrYl1dK2lni53I0EkrmMwFiTVW5Pdt-fScuE,4806
3
3
  athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
4
4
  athena/core/__init__.py,sha256=RWfyDqkzWsf8e3VGc3NV60MovfJbg5XWzNFGB2DZ0hA,790
5
5
  athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
6
- athena/core/client_wrapper.py,sha256=ycEzNClcBiaib7lONhCpSD3x-LeUGEPYduyf2zzEleA,1198
6
+ athena/core/client_wrapper.py,sha256=KZ7FY-ZxhFOGj0zURiA5iNdT6Wndof8UYC71q6wRYKY,1198
7
7
  athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
8
8
  athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
9
9
  athena/core/http_client.py,sha256=LI0yP3jUyE0Ue7oyBcI9nyo1pljOwh9Y5ycTeIpKwOg,4882
@@ -32,7 +32,7 @@ athena/types/get_snippets_response.py,sha256=Lpn7bHJLpPQozN93unCV-8eByAAfz1MhQWR
32
32
  athena/types/http_validation_error.py,sha256=Fcv_CTMMrLvCeTHjF0n5xf5tskMDgt-J6H9gp654eQw,973
33
33
  athena/types/message_out.py,sha256=uvZY_Podv2XccEk8CICug9I_S2hFJTSzCBwcHiauW7A,865
34
34
  athena/types/message_out_dto.py,sha256=qgRibRbDNOWVnVGP7Rribh9WdoCT2CSiPUXeIWECqq4,1051
35
- athena/types/model.py,sha256=yZyzOojHN4mRCd-ITHJNiKZDpQhy7LKZcM9_saJ0qy4,1020
35
+ athena/types/model.py,sha256=hwksocGCn7CTMDUkxkc--sHs8qqm8WzC8urIbeGsX-A,1290
36
36
  athena/types/report.py,sha256=QVaqVfHMAV3s9_V2CqjIEMcRrbJhD8zmi82vrk2A8x0,946
37
37
  athena/types/snippet.py,sha256=POIVJNV9iQxiVegB_qwQx-PZPPSyoIPhyxTsueNVUGA,1126
38
38
  athena/types/sql_results.py,sha256=pNH32nyf1bzoYJs3FgHctLdLO02oOjyGgLkHACACB6k,900
@@ -40,6 +40,6 @@ athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,6
40
40
  athena/types/tools.py,sha256=Vfb-D9CYcPZH6MBATNODgfXjFyBpCs4qbkqpCMl7eBM,1277
41
41
  athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
42
42
  athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
43
- athena_intelligence-0.1.29.dist-info/METADATA,sha256=0jcMT3X67HpEzWOrUJJ0oeNBOsf1BHV9BSOPWPMjnNc,4738
44
- athena_intelligence-0.1.29.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
45
- athena_intelligence-0.1.29.dist-info/RECORD,,
43
+ athena_intelligence-0.1.30.dist-info/METADATA,sha256=_0tdDHwC0b-A4kpjYlilzm-biSIGT7yBlKWeCq2ldYk,4738
44
+ athena_intelligence-0.1.30.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
45
+ athena_intelligence-0.1.30.dist-info/RECORD,,