athena-intelligence 0.1.29__tar.gz → 0.1.30__tar.gz

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 (45) hide show
  1. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/PKG-INFO +1 -1
  2. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/pyproject.toml +1 -1
  3. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/client_wrapper.py +1 -1
  4. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/model.py +8 -0
  5. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/README.md +0 -0
  6. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/__init__.py +0 -0
  7. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/base_client.py +0 -0
  8. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/client.py +0 -0
  9. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/__init__.py +0 -0
  10. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/api_error.py +0 -0
  11. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/datetime_utils.py +0 -0
  12. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/file.py +0 -0
  13. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/http_client.py +0 -0
  14. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/jsonable_encoder.py +0 -0
  15. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/remove_none_from_dict.py +0 -0
  16. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/request_options.py +0 -0
  17. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/dataset/__init__.py +0 -0
  18. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/dataset/client.py +0 -0
  19. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/environment.py +0 -0
  20. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/errors/__init__.py +0 -0
  21. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/errors/unprocessable_entity_error.py +0 -0
  22. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/message/__init__.py +0 -0
  23. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/message/client.py +0 -0
  24. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/polling_message_client.py +0 -0
  25. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/py.typed +0 -0
  26. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/query/__init__.py +0 -0
  27. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/query/client.py +0 -0
  28. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/report/__init__.py +0 -0
  29. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/report/client.py +0 -0
  30. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/snippet/__init__.py +0 -0
  31. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/snippet/client.py +0 -0
  32. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/__init__.py +0 -0
  33. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/dataset.py +0 -0
  34. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/get_datasets_response.py +0 -0
  35. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/get_snippets_response.py +0 -0
  36. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/http_validation_error.py +0 -0
  37. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/message_out.py +0 -0
  38. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/message_out_dto.py +0 -0
  39. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/report.py +0 -0
  40. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/snippet.py +0 -0
  41. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/sql_results.py +0 -0
  42. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/status_enum.py +0 -0
  43. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/tools.py +0 -0
  44. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/validation_error.py +0 -0
  45. {athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/validation_error_loc_item.py +0 -0
{athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/PKG-INFO 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 → athena_intelligence-0.1.30}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "athena-intelligence"
3
- version = "0.1.29"
3
+ version = "0.1.30"
4
4
  description = ""
5
5
  readme = "README.md"
6
6
  authors = []
{athena_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/core/client_wrapper.py RENAMED
@@ -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_intelligence-0.1.29 → athena_intelligence-0.1.30}/src/athena/types/model.py RENAMED
@@ -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: