athena-intelligence 0.1.86__py3-none-any.whl → 0.1.88__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 (18) hide show
  1. athena/base_client.py +4 -4
  2. athena/chain/client.py +30 -138
  3. athena/client.py +57 -19
  4. athena/core/client_wrapper.py +13 -3
  5. athena/core/http_client.py +333 -20
  6. athena/core/remove_none_from_dict.py +2 -2
  7. athena/dataset/client.py +8 -64
  8. athena/message/client.py +20 -113
  9. athena/query/client.py +8 -54
  10. athena/report/client.py +10 -66
  11. athena/search/client.py +22 -68
  12. athena/snippet/client.py +10 -116
  13. athena/tools/client.py +112 -542
  14. athena/upload/client.py +2 -58
  15. athena/workflow/client.py +8 -54
  16. {athena_intelligence-0.1.86.dist-info → athena_intelligence-0.1.88.dist-info}/METADATA +1 -1
  17. {athena_intelligence-0.1.86.dist-info → athena_intelligence-0.1.88.dist-info}/RECORD +18 -18
  18. {athena_intelligence-0.1.86.dist-info → athena_intelligence-0.1.88.dist-info}/WHEEL +0 -0
athena/upload/client.py CHANGED
@@ -1,16 +1,12 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
3
  import typing
4
- import urllib.parse
5
4
  from json.decoder import JSONDecodeError
6
5
 
7
6
  from .. import core
8
7
  from ..core.api_error import ApiError
9
8
  from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
10
- from ..core.jsonable_encoder import jsonable_encoder
11
9
  from ..core.pydantic_utilities import pydantic_v1
12
- from ..core.query_encoder import encode_query
13
- from ..core.remove_none_from_dict import remove_none_from_dict
14
10
  from ..core.request_options import RequestOptions
15
11
  from ..errors.unprocessable_entity_error import UnprocessableEntityError
16
12
  from ..types.http_validation_error import HttpValidationError
@@ -51,33 +47,7 @@ class UploadClient:
51
47
  client.upload.upload_documents()
52
48
  """
53
49
  _response = self._client_wrapper.httpx_client.request(
54
- method="POST",
55
- url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/upload"),
56
- params=encode_query(
57
- jsonable_encoder(
58
- request_options.get("additional_query_parameters") if request_options is not None else None
59
- )
60
- ),
61
- data=jsonable_encoder(remove_none_from_dict({}))
62
- if request_options is None or request_options.get("additional_body_parameters") is None
63
- else {
64
- **jsonable_encoder(remove_none_from_dict({})),
65
- **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
66
- },
67
- files=core.convert_file_dict_to_httpx_tuples(remove_none_from_dict({"files": files})),
68
- headers=jsonable_encoder(
69
- remove_none_from_dict(
70
- {
71
- **self._client_wrapper.get_headers(),
72
- **(request_options.get("additional_headers", {}) if request_options is not None else {}),
73
- }
74
- )
75
- ),
76
- timeout=request_options.get("timeout_in_seconds")
77
- if request_options is not None and request_options.get("timeout_in_seconds") is not None
78
- else self._client_wrapper.get_timeout(),
79
- retries=0,
80
- max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
50
+ "api/v0/upload", method="POST", data={}, files={"files": files}, request_options=request_options, omit=OMIT
81
51
  )
82
52
  if 200 <= _response.status_code < 300:
83
53
  return pydantic_v1.parse_obj_as(UploadDocumentsOut, _response.json()) # type: ignore
@@ -123,33 +93,7 @@ class AsyncUploadClient:
123
93
  await client.upload.upload_documents()
124
94
  """
125
95
  _response = await self._client_wrapper.httpx_client.request(
126
- method="POST",
127
- url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/upload"),
128
- params=encode_query(
129
- jsonable_encoder(
130
- request_options.get("additional_query_parameters") if request_options is not None else None
131
- )
132
- ),
133
- data=jsonable_encoder(remove_none_from_dict({}))
134
- if request_options is None or request_options.get("additional_body_parameters") is None
135
- else {
136
- **jsonable_encoder(remove_none_from_dict({})),
137
- **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
138
- },
139
- files=core.convert_file_dict_to_httpx_tuples(remove_none_from_dict({"files": files})),
140
- headers=jsonable_encoder(
141
- remove_none_from_dict(
142
- {
143
- **self._client_wrapper.get_headers(),
144
- **(request_options.get("additional_headers", {}) if request_options is not None else {}),
145
- }
146
- )
147
- ),
148
- timeout=request_options.get("timeout_in_seconds")
149
- if request_options is not None and request_options.get("timeout_in_seconds") is not None
150
- else self._client_wrapper.get_timeout(),
151
- retries=0,
152
- max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
96
+ "api/v0/upload", method="POST", data={}, files={"files": files}, request_options=request_options, omit=OMIT
153
97
  )
154
98
  if 200 <= _response.status_code < 300:
155
99
  return pydantic_v1.parse_obj_as(UploadDocumentsOut, _response.json()) # type: ignore
athena/workflow/client.py CHANGED
@@ -1,15 +1,11 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
3
  import typing
4
- import urllib.parse
5
4
  from json.decoder import JSONDecodeError
6
5
 
7
6
  from ..core.api_error import ApiError
8
7
  from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
9
- from ..core.jsonable_encoder import jsonable_encoder
10
8
  from ..core.pydantic_utilities import pydantic_v1
11
- from ..core.query_encoder import encode_query
12
- from ..core.remove_none_from_dict import remove_none_from_dict
13
9
  from ..core.request_options import RequestOptions
14
10
  from ..errors.unprocessable_entity_error import UnprocessableEntityError
15
11
  from ..types.http_validation_error import HttpValidationError
@@ -51,32 +47,11 @@ class WorkflowClient:
51
47
  )
52
48
  """
53
49
  _response = self._client_wrapper.httpx_client.request(
50
+ "api/v0/workflow/status",
54
51
  method="POST",
55
- url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/workflow/status"),
56
- params=encode_query(
57
- jsonable_encoder(
58
- request_options.get("additional_query_parameters") if request_options is not None else None
59
- )
60
- ),
61
- json=jsonable_encoder({"workflow_id": workflow_id})
62
- if request_options is None or request_options.get("additional_body_parameters") is None
63
- else {
64
- **jsonable_encoder({"workflow_id": workflow_id}),
65
- **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
66
- },
67
- headers=jsonable_encoder(
68
- remove_none_from_dict(
69
- {
70
- **self._client_wrapper.get_headers(),
71
- **(request_options.get("additional_headers", {}) if request_options is not None else {}),
72
- }
73
- )
74
- ),
75
- timeout=request_options.get("timeout_in_seconds")
76
- if request_options is not None and request_options.get("timeout_in_seconds") is not None
77
- else self._client_wrapper.get_timeout(),
78
- retries=0,
79
- max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
52
+ json={"workflow_id": workflow_id},
53
+ request_options=request_options,
54
+ omit=OMIT,
80
55
  )
81
56
  if 200 <= _response.status_code < 300:
82
57
  return pydantic_v1.parse_obj_as(WorkflowStatusOut, _response.json()) # type: ignore
@@ -123,32 +98,11 @@ class AsyncWorkflowClient:
123
98
  )
124
99
  """
125
100
  _response = await self._client_wrapper.httpx_client.request(
101
+ "api/v0/workflow/status",
126
102
  method="POST",
127
- url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/workflow/status"),
128
- params=encode_query(
129
- jsonable_encoder(
130
- request_options.get("additional_query_parameters") if request_options is not None else None
131
- )
132
- ),
133
- json=jsonable_encoder({"workflow_id": workflow_id})
134
- if request_options is None or request_options.get("additional_body_parameters") is None
135
- else {
136
- **jsonable_encoder({"workflow_id": workflow_id}),
137
- **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
138
- },
139
- headers=jsonable_encoder(
140
- remove_none_from_dict(
141
- {
142
- **self._client_wrapper.get_headers(),
143
- **(request_options.get("additional_headers", {}) if request_options is not None else {}),
144
- }
145
- )
146
- ),
147
- timeout=request_options.get("timeout_in_seconds")
148
- if request_options is not None and request_options.get("timeout_in_seconds") is not None
149
- else self._client_wrapper.get_timeout(),
150
- retries=0,
151
- max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
103
+ json={"workflow_id": workflow_id},
104
+ request_options=request_options,
105
+ omit=OMIT,
152
106
  )
153
107
  if 200 <= _response.status_code < 300:
154
108
  return pydantic_v1.parse_obj_as(WorkflowStatusOut, _response.json()) # type: ignore
{athena_intelligence-0.1.86.dist-info → athena_intelligence-0.1.88.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: athena-intelligence
3
- Version: 0.1.86
3
+ Version: 0.1.88
4
4
  Summary: Athena Intelligence Python Library
5
5
  Requires-Python: >=3.8,<4.0
6
6
  Classifier: Intended Audience :: Developers
{athena_intelligence-0.1.86.dist-info → athena_intelligence-0.1.88.dist-info}/RECORD RENAMED
@@ -1,21 +1,21 @@
1
1
  athena/__init__.py,sha256=vS1F7G3hdiHGM-q24LBoKf77EHpaDBrb3ZzM_sgXoGY,2801
2
- athena/base_client.py,sha256=4HD21xS-Kl2oyy0OQGPVSH8X3aRpKdefKKOGXkAa-1M,7101
2
+ athena/base_client.py,sha256=yP2dIc8ja04pFizBx5xF4qhkeOPJ3V9DMfqYcW5HIGg,7119
3
3
  athena/chain/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
4
- athena/chain/client.py,sha256=3O5tTl9skrhNS--6Jt-Za7CmmBz1MQLiqeNK0Ewtup0,16977
5
- athena/client.py,sha256=gVUvs9Wui80KKYfXOzWQjfcar98ijg8NzNaPLkULFFA,6757
4
+ athena/chain/client.py,sha256=5YJJsXnXLFbPN1y0Qe__QWoQ2QOMp6zK_is_obk7Gg8,11105
5
+ athena/client.py,sha256=evktoequsijpv_bKJP4IPLog2E7VhgDL1AgOgJ2tq_0,7833
6
6
  athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
7
7
  athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
8
- athena/core/client_wrapper.py,sha256=ujw-SZMlg7IKRB5rwW50cLXxVAMNLoMM7e6vQ-2Pcig,1495
8
+ athena/core/client_wrapper.py,sha256=8dklUiRaBWUyJzo_CO2DsA6Syg_QwA5TK65GlyOeOTA,1805
9
9
  athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
10
10
  athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
11
- athena/core/http_client.py,sha256=5ok6hqgZDJhg57EHvMnr0BBaHdG50QxFPKaCZ9aVWTc,5059
11
+ athena/core/http_client.py,sha256=Z4NuAsJD-51yqmoME17O5sxwx5orSp1wsnd6bPyKcgA,17768
12
12
  athena/core/jsonable_encoder.py,sha256=L6G68Py6gEo8n87rXwlkLPUyzHvXftEBjJZNb2tCuOA,3742
13
13
  athena/core/pydantic_utilities.py,sha256=hI3vcpSG47sVlafyPol2T2ICt8HNMIu_rM9amc2zf7w,748
14
14
  athena/core/query_encoder.py,sha256=sI6XiwFby-WRIk4MVN_5wlAlm30aCHE73Uvm09fb9qQ,1266
15
- athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
15
+ athena/core/remove_none_from_dict.py,sha256=EU9SGgYidWq7SexuJbNs4-PZ-5Bl3Vppd864mS6vQZw,342
16
16
  athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
17
17
  athena/dataset/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
18
- athena/dataset/client.py,sha256=qmMfqcDpaXWKhhvH3TNn-wdtzvxA77-tX7Gd22S3rGg,6716
18
+ athena/dataset/client.py,sha256=GJNYHKpb3_NMJ6A1cVObhlLiEKQxNONP5acJw8RA4_E,4108
19
19
  athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
20
20
  athena/errors/__init__.py,sha256=0LicWKR1fNQwwL8_ohKhBkxXTD8StFlBYDFG1jPgqOk,405
21
21
  athena/errors/internal_server_error.py,sha256=WEwVqzsfpBTaqZipvse-kvKbW-3NbpXVvuHXs_64U0M,315
@@ -23,19 +23,19 @@ athena/errors/not_found_error.py,sha256=uBK3JapPPgTkMoSCX9j22C6MDtSuTuSPmjT0lDRi
23
23
  athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
24
24
  athena/errors/unsupported_media_type_error.py,sha256=fQ7TYQ3QYcT_YzFzO4f8_bLger7UQfZmuF22I9jYxFA,340
25
25
  athena/message/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
26
- athena/message/client.py,sha256=-z6g7VMSRyWJhWwsgK9o_Wf5W1YGM1IIm-LHUasfgVc,13210
26
+ athena/message/client.py,sha256=WroRT0SbFM_DfAtdKsvn3lBYWyxeUOn48SVvKPrznMo,8329
27
27
  athena/polling_message_client.py,sha256=dmmycImvog2niEFFPo4rE5xMJHUlq9NqAr4xlFK6_Os,3998
28
28
  athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
29
  athena/query/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
30
- athena/query/client.py,sha256=dAPKPZysuAJRLSyr-0mYQJkZyfnysqecvvQLheG4-a0,6726
30
+ athena/query/client.py,sha256=-ucV9cjOYiosZ1vlOlKzDEE_hsLRA9ygahwHE7mYzE8,4030
31
31
  athena/report/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
32
- athena/report/client.py,sha256=yTOt_NXw5OcBa_2myr7-ZB5O5otFceocQtXRjXkXxZY,7003
32
+ athena/report/client.py,sha256=7m8BGbuPK0nY6qvUDTZBXDzJ3tls0OgtR3kom9pl6F0,4053
33
33
  athena/search/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
34
- athena/search/client.py,sha256=1XD9KW-crpB4U2bGOrYngOxdomjoUInthNe_H9tXdgA,7978
34
+ athena/search/client.py,sha256=Rgnx9gpAYdEcBQfWaCwFQXOxGMzbpG2w6B-ob1JXQCI,5136
35
35
  athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
36
- athena/snippet/client.py,sha256=0_lhr8_IHNah2qdDT8OpwOSHN_SCqE0F7KLx4YHp-8I,12217
36
+ athena/snippet/client.py,sha256=LTSC-XPiv5fuUveErgdnJYsDTP0JtmZImame4ywScuA,7089
37
37
  athena/tools/__init__.py,sha256=3n7oOoMebo06MAQqYRE2CX9Q0fTNnKBYE0cTlh1MPkM,165
38
- athena/tools/client.py,sha256=ur1PDs-ftZDWIB_5-FUKLlHNzvGgN2DgJea_B09lidM,63809
38
+ athena/tools/client.py,sha256=7NISJcJIfcnQzvs3Sebv1T58hL7p_bfqKo88AJcokvo,40373
39
39
  athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
40
40
  athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
41
41
  athena/types/__init__.py,sha256=UKGjYkgCipVr4_CBwGkCL6PdYGC4XsS7qYdRmCGpr8Q,3298
@@ -83,10 +83,10 @@ athena/types/validation_error.py,sha256=zub9I6bl192KHCE8RA0pEkdaEODR3ktTFnZ7E1B7
83
83
  athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
84
84
  athena/types/workflow_status_out.py,sha256=qB-jBg1kMlwi942Xosd2XSGbhXFJUI7LpM04hd3l9UE,1242
85
85
  athena/upload/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
86
- athena/upload/client.py,sha256=L66H0qFpFH1RYIfKJLQgPDZ3HK5ZagRwYQX5BIen1rc,6825
86
+ athena/upload/client.py,sha256=Z44mBN3Vg_HG4uOqrzBkkNhCGETjsAlbU1RA50KDxkQ,3897
87
87
  athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
88
88
  athena/workflow/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
89
- athena/workflow/client.py,sha256=3LCFYbAoOWcvzwME677syefoJ55ukdTUxiQ71jXw7so,6663
90
- athena_intelligence-0.1.86.dist-info/METADATA,sha256=oiAY6HrOZNxyXsjDtH8XB6QZkv5mgS77ouasBMLWlfQ,5273
91
- athena_intelligence-0.1.86.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
92
- athena_intelligence-0.1.86.dist-info/RECORD,,
89
+ athena/workflow/client.py,sha256=1WGH4MCnwae5Wgb7pGfiyZRr--HscVpdHxeouNSyPe0,4023
90
+ athena_intelligence-0.1.88.dist-info/METADATA,sha256=6GXX6FETmic5jhaAoWrnqS7zmrLk0LVu9eqL6yR4B_I,5273
91
+ athena_intelligence-0.1.88.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
92
+ athena_intelligence-0.1.88.dist-info/RECORD,,