athena-intelligence 0.1.33__py3-none-any.whl → 0.1.35__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 (6) hide show
  1. athena/core/client_wrapper.py +1 -1
  2. athena/types/model.py +4 -0
  3. athena/types/tools.py +4 -0
  4. {athena_intelligence-0.1.33.dist-info → athena_intelligence-0.1.35.dist-info}/METADATA +1 -1
  5. {athena_intelligence-0.1.33.dist-info → athena_intelligence-0.1.35.dist-info}/RECORD +6 -6
  6. {athena_intelligence-0.1.33.dist-info → athena_intelligence-0.1.35.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.33",
19
+ "X-Fern-SDK-Version": "0.1.35",
20
20
  }
21
21
  headers["X-API-KEY"] = self.api_key
22
22
  return headers
athena/types/model.py CHANGED
@@ -12,6 +12,7 @@ class Model(str, enum.Enum):
12
12
  """
13
13
 
14
14
  GPT_35_TURBO = "gpt-3.5-turbo"
15
+ GPT_4_TURBO = "gpt-4-turbo"
15
16
  GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
16
17
  GPT_4 = "gpt-4"
17
18
  MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
@@ -26,6 +27,7 @@ class Model(str, enum.Enum):
26
27
  def visit(
27
28
  self,
28
29
  gpt_35_turbo: typing.Callable[[], T_Result],
30
+ gpt_4_turbo: typing.Callable[[], T_Result],
29
31
  gpt_4_turbo_preview: typing.Callable[[], T_Result],
30
32
  gpt_4: typing.Callable[[], T_Result],
31
33
  mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
@@ -39,6 +41,8 @@ class Model(str, enum.Enum):
39
41
  ) -> T_Result:
40
42
  if self is Model.GPT_35_TURBO:
41
43
  return gpt_35_turbo()
44
+ if self is Model.GPT_4_TURBO:
45
+ return gpt_4_turbo()
42
46
  if self is Model.GPT_4_TURBO_PREVIEW:
43
47
  return gpt_4_turbo_preview()
44
48
  if self is Model.GPT_4:
athena/types/tools.py CHANGED
@@ -13,6 +13,7 @@ class Tools(str, enum.Enum):
13
13
 
14
14
  SEARCH = "search"
15
15
  BROWSE = "browse"
16
+ WIKIPEDIA = "wikipedia"
16
17
  DATABASE_METADATA = "database_metadata"
17
18
  QUERY = "query"
18
19
  CHART = "chart"
@@ -23,6 +24,7 @@ class Tools(str, enum.Enum):
23
24
  self,
24
25
  search: typing.Callable[[], T_Result],
25
26
  browse: typing.Callable[[], T_Result],
27
+ wikipedia: typing.Callable[[], T_Result],
26
28
  database_metadata: typing.Callable[[], T_Result],
27
29
  query: typing.Callable[[], T_Result],
28
30
  chart: typing.Callable[[], T_Result],
@@ -33,6 +35,8 @@ class Tools(str, enum.Enum):
33
35
  return search()
34
36
  if self is Tools.BROWSE:
35
37
  return browse()
38
+ if self is Tools.WIKIPEDIA:
39
+ return wikipedia()
36
40
  if self is Tools.DATABASE_METADATA:
37
41
  return database_metadata()
38
42
  if self is Tools.QUERY:
{athena_intelligence-0.1.33.dist-info → athena_intelligence-0.1.35.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: athena-intelligence
3
- Version: 0.1.33
3
+ Version: 0.1.35
4
4
  Summary:
5
5
  Requires-Python: >=3.8,<4.0
6
6
  Classifier: Programming Language :: Python :: 3
{athena_intelligence-0.1.33.dist-info → athena_intelligence-0.1.35.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=VBjHF6uPLJOCMUri_dfWQE7iTUfsNCShd2zjDPvCLI8,1198
6
+ athena/core/client_wrapper.py,sha256=K-fBUkqFuZezc4WTb32LxUi0HZfXqDBWDoPubvWlk28,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,14 +32,14 @@ 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=w-JHrWXHgMpCtrptPzeHgG8fu9Hle-QmmLP4RG6dCdc,2390
35
+ athena/types/model.py,sha256=XbXkKXbmnfZ8bPTAn1xnWGjqKK1SVOLdxf1RGk5ON5k,2545
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
39
39
  athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,672
40
- athena/types/tools.py,sha256=Vfb-D9CYcPZH6MBATNODgfXjFyBpCs4qbkqpCMl7eBM,1277
40
+ athena/types/tools.py,sha256=mhRkKAwlsDud-fFOhsx2T3hBD-FAtuCnGHyU9cLPcGU,1422
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.33.dist-info/METADATA,sha256=SmQ1tqzbjpOmwbjVYkyvbdiC9RG-xAJQIKWIokqG7QI,4738
44
- athena_intelligence-0.1.33.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
45
- athena_intelligence-0.1.33.dist-info/RECORD,,
43
+ athena_intelligence-0.1.35.dist-info/METADATA,sha256=ZomGyKNltsSPQX5fsfBHchS7q_gdKxwDO0yaqzuqNAk,4738
44
+ athena_intelligence-0.1.35.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
45
+ athena_intelligence-0.1.35.dist-info/RECORD,,