athena-intelligence 0.1.87__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.
- athena/base_client.py +4 -4
- athena/chain/client.py +30 -138
- athena/client.py +57 -19
- athena/core/client_wrapper.py +13 -3
- athena/core/http_client.py +333 -20
- athena/core/remove_none_from_dict.py +2 -2
- athena/dataset/client.py +8 -64
- athena/message/client.py +20 -113
- athena/query/client.py +8 -54
- athena/report/client.py +10 -66
- athena/search/client.py +22 -68
- athena/snippet/client.py +10 -116
- athena/tools/client.py +112 -542
- athena/upload/client.py +2 -58
- athena/workflow/client.py +8 -54
- {athena_intelligence-0.1.87.dist-info → athena_intelligence-0.1.88.dist-info}/METADATA +1 -1
- {athena_intelligence-0.1.87.dist-info → athena_intelligence-0.1.88.dist-info}/RECORD +18 -18
- {athena_intelligence-0.1.87.dist-info → athena_intelligence-0.1.88.dist-info}/WHEEL +0 -0
athena/snippet/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.get_snippet_out import GetSnippetOut
|
@@ -29,7 +25,7 @@ class SnippetClient:
|
|
29
25
|
*,
|
30
26
|
page: typing.Optional[int] = None,
|
31
27
|
page_size: typing.Optional[int] = None,
|
32
|
-
request_options: typing.Optional[RequestOptions] = None
|
28
|
+
request_options: typing.Optional[RequestOptions] = None
|
33
29
|
) -> GetSnippetsResponse:
|
34
30
|
"""
|
35
31
|
Parameters
|
@@ -58,36 +54,10 @@ class SnippetClient:
|
|
58
54
|
client.snippet.get()
|
59
55
|
"""
|
60
56
|
_response = self._client_wrapper.httpx_client.request(
|
57
|
+
"api/v0/snippets",
|
61
58
|
method="GET",
|
62
|
-
|
63
|
-
|
64
|
-
jsonable_encoder(
|
65
|
-
remove_none_from_dict(
|
66
|
-
{
|
67
|
-
"page": page,
|
68
|
-
"page_size": page_size,
|
69
|
-
**(
|
70
|
-
request_options.get("additional_query_parameters", {})
|
71
|
-
if request_options is not None
|
72
|
-
else {}
|
73
|
-
),
|
74
|
-
}
|
75
|
-
)
|
76
|
-
)
|
77
|
-
),
|
78
|
-
headers=jsonable_encoder(
|
79
|
-
remove_none_from_dict(
|
80
|
-
{
|
81
|
-
**self._client_wrapper.get_headers(),
|
82
|
-
**(request_options.get("additional_headers", {}) if request_options is not None else {}),
|
83
|
-
}
|
84
|
-
)
|
85
|
-
),
|
86
|
-
timeout=request_options.get("timeout_in_seconds")
|
87
|
-
if request_options is not None and request_options.get("timeout_in_seconds") is not None
|
88
|
-
else self._client_wrapper.get_timeout(),
|
89
|
-
retries=0,
|
90
|
-
max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
|
59
|
+
params={"page": page, "page_size": page_size},
|
60
|
+
request_options=request_options,
|
91
61
|
)
|
92
62
|
if 200 <= _response.status_code < 300:
|
93
63
|
return pydantic_v1.parse_obj_as(GetSnippetsResponse, _response.json()) # type: ignore
|
@@ -127,32 +97,7 @@ class SnippetClient:
|
|
127
97
|
)
|
128
98
|
"""
|
129
99
|
_response = self._client_wrapper.httpx_client.request(
|
130
|
-
method="POST",
|
131
|
-
url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/snippet"),
|
132
|
-
params=encode_query(
|
133
|
-
jsonable_encoder(
|
134
|
-
request_options.get("additional_query_parameters") if request_options is not None else None
|
135
|
-
)
|
136
|
-
),
|
137
|
-
json=jsonable_encoder({"id": id})
|
138
|
-
if request_options is None or request_options.get("additional_body_parameters") is None
|
139
|
-
else {
|
140
|
-
**jsonable_encoder({"id": id}),
|
141
|
-
**(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
|
142
|
-
},
|
143
|
-
headers=jsonable_encoder(
|
144
|
-
remove_none_from_dict(
|
145
|
-
{
|
146
|
-
**self._client_wrapper.get_headers(),
|
147
|
-
**(request_options.get("additional_headers", {}) if request_options is not None else {}),
|
148
|
-
}
|
149
|
-
)
|
150
|
-
),
|
151
|
-
timeout=request_options.get("timeout_in_seconds")
|
152
|
-
if request_options is not None and request_options.get("timeout_in_seconds") is not None
|
153
|
-
else self._client_wrapper.get_timeout(),
|
154
|
-
retries=0,
|
155
|
-
max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
|
100
|
+
"api/v0/snippet", method="POST", json={"id": id}, request_options=request_options, omit=OMIT
|
156
101
|
)
|
157
102
|
if 200 <= _response.status_code < 300:
|
158
103
|
return pydantic_v1.parse_obj_as(GetSnippetOut, _response.json()) # type: ignore
|
@@ -176,7 +121,7 @@ class AsyncSnippetClient:
|
|
176
121
|
*,
|
177
122
|
page: typing.Optional[int] = None,
|
178
123
|
page_size: typing.Optional[int] = None,
|
179
|
-
request_options: typing.Optional[RequestOptions] = None
|
124
|
+
request_options: typing.Optional[RequestOptions] = None
|
180
125
|
) -> GetSnippetsResponse:
|
181
126
|
"""
|
182
127
|
Parameters
|
@@ -205,36 +150,10 @@ class AsyncSnippetClient:
|
|
205
150
|
await client.snippet.get()
|
206
151
|
"""
|
207
152
|
_response = await self._client_wrapper.httpx_client.request(
|
153
|
+
"api/v0/snippets",
|
208
154
|
method="GET",
|
209
|
-
|
210
|
-
|
211
|
-
jsonable_encoder(
|
212
|
-
remove_none_from_dict(
|
213
|
-
{
|
214
|
-
"page": page,
|
215
|
-
"page_size": page_size,
|
216
|
-
**(
|
217
|
-
request_options.get("additional_query_parameters", {})
|
218
|
-
if request_options is not None
|
219
|
-
else {}
|
220
|
-
),
|
221
|
-
}
|
222
|
-
)
|
223
|
-
)
|
224
|
-
),
|
225
|
-
headers=jsonable_encoder(
|
226
|
-
remove_none_from_dict(
|
227
|
-
{
|
228
|
-
**self._client_wrapper.get_headers(),
|
229
|
-
**(request_options.get("additional_headers", {}) if request_options is not None else {}),
|
230
|
-
}
|
231
|
-
)
|
232
|
-
),
|
233
|
-
timeout=request_options.get("timeout_in_seconds")
|
234
|
-
if request_options is not None and request_options.get("timeout_in_seconds") is not None
|
235
|
-
else self._client_wrapper.get_timeout(),
|
236
|
-
retries=0,
|
237
|
-
max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
|
155
|
+
params={"page": page, "page_size": page_size},
|
156
|
+
request_options=request_options,
|
238
157
|
)
|
239
158
|
if 200 <= _response.status_code < 300:
|
240
159
|
return pydantic_v1.parse_obj_as(GetSnippetsResponse, _response.json()) # type: ignore
|
@@ -274,32 +193,7 @@ class AsyncSnippetClient:
|
|
274
193
|
)
|
275
194
|
"""
|
276
195
|
_response = await self._client_wrapper.httpx_client.request(
|
277
|
-
method="POST",
|
278
|
-
url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/snippet"),
|
279
|
-
params=encode_query(
|
280
|
-
jsonable_encoder(
|
281
|
-
request_options.get("additional_query_parameters") if request_options is not None else None
|
282
|
-
)
|
283
|
-
),
|
284
|
-
json=jsonable_encoder({"id": id})
|
285
|
-
if request_options is None or request_options.get("additional_body_parameters") is None
|
286
|
-
else {
|
287
|
-
**jsonable_encoder({"id": id}),
|
288
|
-
**(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
|
289
|
-
},
|
290
|
-
headers=jsonable_encoder(
|
291
|
-
remove_none_from_dict(
|
292
|
-
{
|
293
|
-
**self._client_wrapper.get_headers(),
|
294
|
-
**(request_options.get("additional_headers", {}) if request_options is not None else {}),
|
295
|
-
}
|
296
|
-
)
|
297
|
-
),
|
298
|
-
timeout=request_options.get("timeout_in_seconds")
|
299
|
-
if request_options is not None and request_options.get("timeout_in_seconds") is not None
|
300
|
-
else self._client_wrapper.get_timeout(),
|
301
|
-
retries=0,
|
302
|
-
max_retries=request_options.get("max_retries") if request_options is not None else 0, # type: ignore
|
196
|
+
"api/v0/snippet", method="POST", json={"id": id}, request_options=request_options, omit=OMIT
|
303
197
|
)
|
304
198
|
if 200 <= _response.status_code < 300:
|
305
199
|
return pydantic_v1.parse_obj_as(GetSnippetOut, _response.json()) # type: ignore
|