athena-intelligence 0.1.65__py3-none-any.whl → 0.1.67__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/llm_model.py +12 -4
  3. athena/types/workflow_status_out.py +1 -0
  4. {athena_intelligence-0.1.65.dist-info → athena_intelligence-0.1.67.dist-info}/METADATA +1 -1
  5. {athena_intelligence-0.1.65.dist-info → athena_intelligence-0.1.67.dist-info}/RECORD +6 -6
  6. {athena_intelligence-0.1.65.dist-info → athena_intelligence-0.1.67.dist-info}/WHEEL +0 -0
athena/core/client_wrapper.py CHANGED
@@ -17,7 +17,7 @@ class BaseClientWrapper:
17
17
  headers: typing.Dict[str, str] = {
18
18
  "X-Fern-Language": "Python",
19
19
  "X-Fern-SDK-Name": "athena-intelligence",
20
- "X-Fern-SDK-Version": "0.1.65",
20
+ "X-Fern-SDK-Version": "0.1.67",
21
21
  }
22
22
  headers["X-API-KEY"] = self.api_key
23
23
  return headers
athena/types/llm_model.py CHANGED
@@ -28,9 +28,11 @@ class LlmModel(str, enum.Enum):
28
28
  GROQ_LLAMA_38_B_8192 = "groq-llama3-8b-8192"
29
29
  GROQ_LLAMA_370_B_8192 = "groq-llama3-70b-8192"
30
30
  GROQ_GEMMA_7_B_IT = "groq-gemma-7b-it"
31
- GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
32
31
  DATABRICKS_DBRX = "databricks-dbrx"
32
+ GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
33
33
  GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
34
+ GOOGLE_GEMINI_15_FLASH_001 = "google-gemini-1.5-flash-001"
35
+ GOOGLE_GEMINI_15_PRO_001 = "google-gemini-1.5-pro-001"
34
36
 
35
37
  def visit(
36
38
  self,
@@ -51,9 +53,11 @@ class LlmModel(str, enum.Enum):
51
53
  groq_llama_38_b_8192: typing.Callable[[], T_Result],
52
54
  groq_llama_370_b_8192: typing.Callable[[], T_Result],
53
55
  groq_gemma_7_b_it: typing.Callable[[], T_Result],
54
- google_gemini_10_pro_latest: typing.Callable[[], T_Result],
55
56
  databricks_dbrx: typing.Callable[[], T_Result],
57
+ google_gemini_10_pro_latest: typing.Callable[[], T_Result],
56
58
  google_gemini_15_pro_latest: typing.Callable[[], T_Result],
59
+ google_gemini_15_flash_001: typing.Callable[[], T_Result],
60
+ google_gemini_15_pro_001: typing.Callable[[], T_Result],
57
61
  ) -> T_Result:
58
62
  if self is LlmModel.GPT_35_TURBO:
59
63
  return gpt_35_turbo()
@@ -89,9 +93,13 @@ class LlmModel(str, enum.Enum):
89
93
  return groq_llama_370_b_8192()
90
94
  if self is LlmModel.GROQ_GEMMA_7_B_IT:
91
95
  return groq_gemma_7_b_it()
92
- if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:
93
- return google_gemini_10_pro_latest()
94
96
  if self is LlmModel.DATABRICKS_DBRX:
95
97
  return databricks_dbrx()
98
+ if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:
99
+ return google_gemini_10_pro_latest()
96
100
  if self is LlmModel.GOOGLE_GEMINI_15_PRO_LATEST:
97
101
  return google_gemini_15_pro_latest()
102
+ if self is LlmModel.GOOGLE_GEMINI_15_FLASH_001:
103
+ return google_gemini_15_flash_001()
104
+ if self is LlmModel.GOOGLE_GEMINI_15_PRO_001:
105
+ return google_gemini_15_pro_001()
athena/types/workflow_status_out.py CHANGED
@@ -10,6 +10,7 @@ from ..core.pydantic_utilities import pydantic_v1
10
10
  class WorkflowStatusOut(pydantic_v1.BaseModel):
11
11
  status: str
12
12
  execution_info: typing.Dict[str, typing.Any]
13
+ workflow_output: typing.Optional[typing.Dict[str, typing.Any]] = None
13
14
 
14
15
  def json(self, **kwargs: typing.Any) -> str:
15
16
  kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
{athena_intelligence-0.1.65.dist-info → athena_intelligence-0.1.67.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: athena-intelligence
3
- Version: 0.1.65
3
+ Version: 0.1.67
4
4
  Summary:
5
5
  Requires-Python: >=3.8,<4.0
6
6
  Classifier: Programming Language :: Python :: 3
{athena_intelligence-0.1.65.dist-info → athena_intelligence-0.1.67.dist-info}/RECORD RENAMED
@@ -5,7 +5,7 @@ athena/chain/client.py,sha256=2vSu7d4RvgbGc7jbWpKkCs5dU-ryCIJ1i0I1EsoCEdQ,16177
5
5
  athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
6
6
  athena/core/__init__.py,sha256=1pNSKkwyQvMl_F0wohBqmoQAITptg3zlvCwsoSSzy7c,853
7
7
  athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
8
- athena/core/client_wrapper.py,sha256=P95jbPaQeXN4l6x3W5D4LAHOECLQKVtkzZUqEhhJ82o,1495
8
+ athena/core/client_wrapper.py,sha256=p_ZgyALumvaaz3yYAtJ6qJju4B4opMEMkmmZNyJ8QTU,1495
9
9
  athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
10
10
  athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
11
11
  athena/core/http_client.py,sha256=5ok6hqgZDJhg57EHvMnr0BBaHdG50QxFPKaCZ9aVWTc,5059
@@ -44,7 +44,7 @@ athena/types/get_snippet_out.py,sha256=AkkF6YJcYysiQVnOvhRerHMsHkBTu1BP9tYZC8wET
44
44
  athena/types/get_snippets_response.py,sha256=pgwYqmddU5shKeVaE4RQSFN9SLsVAeQp3sqIkQlvzoU,969
45
45
  athena/types/http_validation_error.py,sha256=u4t-1U0DI0u3Zj_Oz7AmGmpL4sqBzoS_5nZHImWQbmM,953
46
46
  athena/types/langchain_documents_request_out.py,sha256=O1v7mcgt0ryaY4e8YODpAHYJKyUY7jYFBc0s93A1sgo,892
47
- athena/types/llm_model.py,sha256=bkxs5YcNXyv7YxIkKQIxEGMjRUVkWku5VEnQhHklQAE,4238
47
+ athena/types/llm_model.py,sha256=BoinyNzPV2wMAKxjgf1OLyVCN3mmHZ-N1_q93iLY9Wk,4696
48
48
  athena/types/map_reduce_chain_out.py,sha256=6R-fuxHaww60dhUAuwrdZPp5lV-DyFZh9SGLCc6fp8E,950
49
49
  athena/types/message_out.py,sha256=HJZizmFH7crD3OHm0fdTy3189F2gv5qR8aaUbTTfWFI,845
50
50
  athena/types/message_out_dto.py,sha256=1G8srlYaIYmoYRstLKm97xZGxK87DK57CiO9hYnt3gQ,1031
@@ -61,12 +61,12 @@ athena/types/upload_documents_out.py,sha256=3FJ0QIKl6zGmswAUpgkrVGP2nLdH3AloXrSh
61
61
  athena/types/url_result.py,sha256=lIgnQeyKy_UfFFPe7HMrrRzb-SK089RxcKcKN9Q3DNQ,873
62
62
  athena/types/validation_error.py,sha256=yqombbKLBSzTPFn6CJH_hbo7tpS68T3JvMdd7kBtO1g,972
63
63
  athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
64
- athena/types/workflow_status_out.py,sha256=sGY5oIQVenDzGV2VJAdIbs1gIWTh41Py9_VeOntA2Bo,905
64
+ athena/types/workflow_status_out.py,sha256=q28kJnkuPcRu152btJZW5u6ZQ2wS20-K_TnCmDEIID8,979
65
65
  athena/upload/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
66
66
  athena/upload/client.py,sha256=e5h10wZ7lGBasJ6X907x7nXHRhX600mLSkdw2qz6pmY,6385
67
67
  athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
68
68
  athena/workflow/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
69
69
  athena/workflow/client.py,sha256=uY9IS_v2GDQ-g2nbatpTUP1aT1oHbG_E8WAor8JzxPI,6249
70
- athena_intelligence-0.1.65.dist-info/METADATA,sha256=YpifM-uSfcKYR7HlK5VjkUa-gbXoNbtOyVlLdaRZF4E,4738
71
- athena_intelligence-0.1.65.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
72
- athena_intelligence-0.1.65.dist-info/RECORD,,
70
+ athena_intelligence-0.1.67.dist-info/METADATA,sha256=CNGZz68IUT7GatLSIHdoG2ShlhL4WmTH5VWFOHlA5mM,4738
71
+ athena_intelligence-0.1.67.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
72
+ athena_intelligence-0.1.67.dist-info/RECORD,,