levelfour 0.1.0__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 (208) hide show
  1. levelfour/__init__.py +495 -0
  2. levelfour/api_keys/__init__.py +34 -0
  3. levelfour/api_keys/client.py +337 -0
  4. levelfour/api_keys/raw_client.py +621 -0
  5. levelfour/api_keys/types/__init__.py +34 -0
  6. levelfour/api_keys/types/create_api_key_request_scope.py +5 -0
  7. levelfour/async_levelfour_client.py +39 -0
  8. levelfour/base_client.py +346 -0
  9. levelfour/client.py +54 -0
  10. levelfour/core/__init__.py +127 -0
  11. levelfour/core/api_error.py +23 -0
  12. levelfour/core/client_wrapper.py +105 -0
  13. levelfour/core/datetime_utils.py +70 -0
  14. levelfour/core/file.py +67 -0
  15. levelfour/core/force_multipart.py +18 -0
  16. levelfour/core/http_client.py +776 -0
  17. levelfour/core/http_response.py +59 -0
  18. levelfour/core/http_sse/__init__.py +42 -0
  19. levelfour/core/http_sse/_api.py +112 -0
  20. levelfour/core/http_sse/_decoders.py +61 -0
  21. levelfour/core/http_sse/_exceptions.py +7 -0
  22. levelfour/core/http_sse/_models.py +17 -0
  23. levelfour/core/jsonable_encoder.py +108 -0
  24. levelfour/core/logging.py +107 -0
  25. levelfour/core/pagination.py +82 -0
  26. levelfour/core/pydantic_utilities.py +577 -0
  27. levelfour/core/query_encoder.py +58 -0
  28. levelfour/core/remove_none_from_dict.py +11 -0
  29. levelfour/core/request_options.py +35 -0
  30. levelfour/core/serialization.py +276 -0
  31. levelfour/errors/__init__.py +40 -0
  32. levelfour/errors/forbidden_error.py +11 -0
  33. levelfour/errors/unauthorized_error.py +11 -0
  34. levelfour/errors/unprocessable_entity_error.py +11 -0
  35. levelfour/evaluation/__init__.py +4 -0
  36. levelfour/evaluation/client.py +102 -0
  37. levelfour/evaluation/raw_client.py +141 -0
  38. levelfour/exceptions.py +63 -0
  39. levelfour/health/__init__.py +4 -0
  40. levelfour/health/client.py +163 -0
  41. levelfour/health/raw_client.py +164 -0
  42. levelfour/integrations/__init__.py +0 -0
  43. levelfour/integrations/django.py +26 -0
  44. levelfour/integrations/fastapi.py +39 -0
  45. levelfour/integrations/telemetry.py +66 -0
  46. levelfour/levelfour_client.py +47 -0
  47. levelfour/pagination.py +84 -0
  48. levelfour/pr_analysis/__init__.py +4 -0
  49. levelfour/pr_analysis/client.py +408 -0
  50. levelfour/pr_analysis/raw_client.py +620 -0
  51. levelfour/pr_analysis_actions/__init__.py +4 -0
  52. levelfour/pr_analysis_actions/client.py +588 -0
  53. levelfour/pr_analysis_actions/raw_client.py +939 -0
  54. levelfour/pr_analysis_installations/__init__.py +4 -0
  55. levelfour/pr_analysis_installations/client.py +470 -0
  56. levelfour/pr_analysis_installations/raw_client.py +868 -0
  57. levelfour/providers/__init__.py +34 -0
  58. levelfour/providers/client.py +1504 -0
  59. levelfour/providers/raw_client.py +2398 -0
  60. levelfour/providers/types/__init__.py +36 -0
  61. levelfour/providers/types/list_spending_breakdown_providers_response.py +11 -0
  62. levelfour/py.typed +0 -0
  63. levelfour/recommendations/__init__.py +4 -0
  64. levelfour/recommendations/client.py +1018 -0
  65. levelfour/recommendations/raw_client.py +1810 -0
  66. levelfour/savings/__init__.py +4 -0
  67. levelfour/savings/client.py +401 -0
  68. levelfour/savings/raw_client.py +624 -0
  69. levelfour/spending/__init__.py +34 -0
  70. levelfour/spending/client.py +437 -0
  71. levelfour/spending/raw_client.py +684 -0
  72. levelfour/spending/types/__init__.py +34 -0
  73. levelfour/spending/types/list_breakdown_spending_response.py +12 -0
  74. levelfour/types/__init__.py +413 -0
  75. levelfour/types/accept_terms_data.py +25 -0
  76. levelfour/types/accept_terms_response.py +27 -0
  77. levelfour/types/action_request.py +42 -0
  78. levelfour/types/api_key_created_data.py +28 -0
  79. levelfour/types/api_key_data.py +27 -0
  80. levelfour/types/authentication_error.py +5 -0
  81. levelfour/types/authorization_error.py +5 -0
  82. levelfour/types/bad_request_error.py +5 -0
  83. levelfour/types/comparison_item.py +32 -0
  84. levelfour/types/complete_wizard_data.py +27 -0
  85. levelfour/types/complete_wizard_response.py +27 -0
  86. levelfour/types/conflict_error.py +5 -0
  87. levelfour/types/cost_breakdown_data.py +35 -0
  88. levelfour/types/cost_breakdown_item.py +71 -0
  89. levelfour/types/cost_breakdown_raw_data.py +37 -0
  90. levelfour/types/cost_breakdown_raw_response.py +27 -0
  91. levelfour/types/cost_breakdown_response.py +27 -0
  92. levelfour/types/costs_breakdown_chart_data.py +48 -0
  93. levelfour/types/costs_breakdown_chart_response.py +27 -0
  94. levelfour/types/costs_breakdown_data_point.py +46 -0
  95. levelfour/types/costs_breakdown_grouped_chart_data.py +52 -0
  96. levelfour/types/costs_breakdown_grouped_chart_response.py +27 -0
  97. levelfour/types/costs_breakdown_grouped_data_point.py +36 -0
  98. levelfour/types/costs_breakdown_totals.py +41 -0
  99. levelfour/types/create_api_key_response.py +23 -0
  100. levelfour/types/daily_spending_breakdown_data.py +42 -0
  101. levelfour/types/daily_spending_breakdown_response.py +27 -0
  102. levelfour/types/daily_spending_data_point.py +31 -0
  103. levelfour/types/dashboard_summary_data.py +72 -0
  104. levelfour/types/dashboard_summary_response.py +27 -0
  105. levelfour/types/delete_endpoint_data.py +20 -0
  106. levelfour/types/delete_endpoint_response.py +23 -0
  107. levelfour/types/endpoint_data.py +23 -0
  108. levelfour/types/error_response.py +23 -0
  109. levelfour/types/error_response_error.py +21 -0
  110. levelfour/types/evaluation_status_data.py +71 -0
  111. levelfour/types/evaluation_status_response.py +27 -0
  112. levelfour/types/execution_log_command.py +21 -0
  113. levelfour/types/execution_log_entry.py +23 -0
  114. levelfour/types/execution_step_schema.py +23 -0
  115. levelfour/types/http_validation_error.py +20 -0
  116. levelfour/types/implementation_method_data.py +25 -0
  117. levelfour/types/implementation_method_response.py +27 -0
  118. levelfour/types/implementation_step.py +27 -0
  119. levelfour/types/internal_server_error.py +5 -0
  120. levelfour/types/list_api_keys_data.py +21 -0
  121. levelfour/types/list_api_keys_response.py +23 -0
  122. levelfour/types/list_endpoints_response.py +23 -0
  123. levelfour/types/metric_data_point.py +27 -0
  124. levelfour/types/metric_series.py +40 -0
  125. levelfour/types/metric_series_entry.py +25 -0
  126. levelfour/types/monthly_spending_data.py +27 -0
  127. levelfour/types/monthly_spending_data_point.py +31 -0
  128. levelfour/types/monthly_spending_response.py +27 -0
  129. levelfour/types/not_found_error.py +5 -0
  130. levelfour/types/pagination_meta.py +51 -0
  131. levelfour/types/processing_recommendation_item.py +61 -0
  132. levelfour/types/processing_recommendations_data.py +28 -0
  133. levelfour/types/processing_recommendations_response.py +27 -0
  134. levelfour/types/provider.py +46 -0
  135. levelfour/types/provider_breakdown_data.py +31 -0
  136. levelfour/types/provider_breakdown_item.py +77 -0
  137. levelfour/types/provider_breakdown_response.py +27 -0
  138. levelfour/types/provider_filter_options_data.py +51 -0
  139. levelfour/types/provider_filter_options_response.py +27 -0
  140. levelfour/types/provider_potential_savings_summary_data.py +46 -0
  141. levelfour/types/provider_potential_savings_summary_response.py +27 -0
  142. levelfour/types/provider_recommendations_summary_data.py +37 -0
  143. levelfour/types/provider_recommendations_summary_response.py +27 -0
  144. levelfour/types/provider_saved_summary_data.py +56 -0
  145. levelfour/types/provider_saved_summary_response.py +27 -0
  146. levelfour/types/provider_service_breakdown_data.py +65 -0
  147. levelfour/types/provider_service_breakdown_item.py +72 -0
  148. levelfour/types/provider_service_breakdown_raw_data.py +62 -0
  149. levelfour/types/provider_service_breakdown_raw_response.py +27 -0
  150. levelfour/types/provider_service_breakdown_response.py +27 -0
  151. levelfour/types/provider_service_costs_breakdown_data.py +62 -0
  152. levelfour/types/provider_service_costs_breakdown_response.py +27 -0
  153. levelfour/types/provider_service_costs_data_point.py +36 -0
  154. levelfour/types/provider_spending.py +61 -0
  155. levelfour/types/provider_spending_summary_data.py +67 -0
  156. levelfour/types/provider_spending_summary_response.py +27 -0
  157. levelfour/types/provider_timeline_data.py +57 -0
  158. levelfour/types/provider_timeline_response.py +27 -0
  159. levelfour/types/provider_top_service.py +36 -0
  160. levelfour/types/providers_summary_data.py +24 -0
  161. levelfour/types/providers_summary_response.py +27 -0
  162. levelfour/types/rate_limit_error.py +5 -0
  163. levelfour/types/recommendation_actions.py +60 -0
  164. levelfour/types/recommendation_detail.py +42 -0
  165. levelfour/types/recommendation_detail_response.py +27 -0
  166. levelfour/types/recommendation_filter_options_data.py +41 -0
  167. levelfour/types/recommendation_filter_options_response.py +27 -0
  168. levelfour/types/recommendation_item.py +36 -0
  169. levelfour/types/recommendation_wizard_data.py +173 -0
  170. levelfour/types/recommendation_wizard_data_metrics.py +7 -0
  171. levelfour/types/recommendation_wizard_response.py +27 -0
  172. levelfour/types/recommendations_breakdown_data.py +26 -0
  173. levelfour/types/recommendations_breakdown_response.py +27 -0
  174. levelfour/types/recommendations_overview_data.py +51 -0
  175. levelfour/types/recommendations_overview_response.py +27 -0
  176. levelfour/types/register_endpoint_response.py +23 -0
  177. levelfour/types/resource_change.py +42 -0
  178. levelfour/types/resource_snapshot.py +21 -0
  179. levelfour/types/revoke_api_key_data.py +20 -0
  180. levelfour/types/revoke_api_key_response.py +23 -0
  181. levelfour/types/risk_assessment.py +29 -0
  182. levelfour/types/risk_factor.py +27 -0
  183. levelfour/types/saved_breakdown_data.py +31 -0
  184. levelfour/types/saved_breakdown_response.py +27 -0
  185. levelfour/types/saved_by_provider_breakdown_data.py +31 -0
  186. levelfour/types/saved_by_provider_breakdown_response.py +27 -0
  187. levelfour/types/saved_by_provider_item.py +76 -0
  188. levelfour/types/saved_item.py +35 -0
  189. levelfour/types/saved_provider_summary.py +36 -0
  190. levelfour/types/saved_summary_data.py +57 -0
  191. levelfour/types/saved_summary_response.py +27 -0
  192. levelfour/types/savings_decision_data.py +28 -0
  193. levelfour/types/savings_decision_response.py +27 -0
  194. levelfour/types/spending_by_date_item.py +36 -0
  195. levelfour/types/top_recommendation.py +56 -0
  196. levelfour/types/trigger_info.py +22 -0
  197. levelfour/types/update_tag_data.py +20 -0
  198. levelfour/types/update_tag_response.py +23 -0
  199. levelfour/types/validation_error.py +24 -0
  200. levelfour/types/validation_error_loc_item.py +5 -0
  201. levelfour/webhooks/__init__.py +4 -0
  202. levelfour/webhooks/client.py +273 -0
  203. levelfour/webhooks/raw_client.py +483 -0
  204. levelfour/webhooks/verifier.py +61 -0
  205. levelfour-0.1.0.dist-info/METADATA +12 -0
  206. levelfour-0.1.0.dist-info/RECORD +208 -0
  207. levelfour-0.1.0.dist-info/WHEEL +5 -0
  208. levelfour-0.1.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,776 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ import asyncio
4
+ import email.utils
5
+ import re
6
+ import time
7
+ import typing
8
+ from contextlib import asynccontextmanager, contextmanager
9
+ from random import random
10
+
11
+ import httpx
12
+ from .file import File, convert_file_dict_to_httpx_tuples
13
+ from .force_multipart import FORCE_MULTIPART
14
+ from .jsonable_encoder import jsonable_encoder
15
+ from .logging import LogConfig, Logger, create_logger
16
+ from .query_encoder import encode_query
17
+ from .remove_none_from_dict import remove_none_from_dict as remove_none_from_dict
18
+ from .request_options import RequestOptions
19
+ from httpx._types import RequestFiles
20
+
21
+ INITIAL_RETRY_DELAY_SECONDS = 1.0
22
+ MAX_RETRY_DELAY_SECONDS = 60.0
23
+ JITTER_FACTOR = 0.2 # 20% random jitter
24
+
25
+
26
+ def _parse_retry_after(response_headers: httpx.Headers) -> typing.Optional[float]:
27
+ """
28
+ This function parses the `Retry-After` header in a HTTP response and returns the number of seconds to wait.
29
+
30
+ Inspired by the urllib3 retry implementation.
31
+ """
32
+ retry_after_ms = response_headers.get("retry-after-ms")
33
+ if retry_after_ms is not None:
34
+ try:
35
+ return int(retry_after_ms) / 1000 if retry_after_ms > 0 else 0
36
+ except Exception:
37
+ pass
38
+
39
+ retry_after = response_headers.get("retry-after")
40
+ if retry_after is None:
41
+ return None
42
+
43
+ # Attempt to parse the header as an int.
44
+ if re.match(r"^\s*[0-9]+\s*$", retry_after):
45
+ seconds = float(retry_after)
46
+ # Fallback to parsing it as a date.
47
+ else:
48
+ retry_date_tuple = email.utils.parsedate_tz(retry_after)
49
+ if retry_date_tuple is None:
50
+ return None
51
+ if retry_date_tuple[9] is None: # Python 2
52
+ # Assume UTC if no timezone was specified
53
+ # On Python2.7, parsedate_tz returns None for a timezone offset
54
+ # instead of 0 if no timezone is given, where mktime_tz treats
55
+ # a None timezone offset as local time.
56
+ retry_date_tuple = retry_date_tuple[:9] + (0,) + retry_date_tuple[10:]
57
+
58
+ retry_date = email.utils.mktime_tz(retry_date_tuple)
59
+ seconds = retry_date - time.time()
60
+
61
+ if seconds < 0:
62
+ seconds = 0
63
+
64
+ return seconds
65
+
66
+
67
+ def _add_positive_jitter(delay: float) -> float:
68
+ """Add positive jitter (0-20%) to prevent thundering herd."""
69
+ jitter_multiplier = 1 + random() * JITTER_FACTOR
70
+ return delay * jitter_multiplier
71
+
72
+
73
+ def _add_symmetric_jitter(delay: float) -> float:
74
+ """Add symmetric jitter (±10%) for exponential backoff."""
75
+ jitter_multiplier = 1 + (random() - 0.5) * JITTER_FACTOR
76
+ return delay * jitter_multiplier
77
+
78
+
79
+ def _parse_x_ratelimit_reset(response_headers: httpx.Headers) -> typing.Optional[float]:
80
+ """
81
+ Parse the X-RateLimit-Reset header (Unix timestamp in seconds).
82
+ Returns seconds to wait, or None if header is missing/invalid.
83
+ """
84
+ reset_time_str = response_headers.get("x-ratelimit-reset")
85
+ if reset_time_str is None:
86
+ return None
87
+
88
+ try:
89
+ reset_time = int(reset_time_str)
90
+ delay = reset_time - time.time()
91
+ if delay > 0:
92
+ return delay
93
+ except (ValueError, TypeError):
94
+ pass
95
+
96
+ return None
97
+
98
+
99
+ def _retry_timeout(response: httpx.Response, retries: int) -> float:
100
+ """
101
+ Determine the amount of time to wait before retrying a request.
102
+ This function begins by trying to parse a retry-after header from the response, and then proceeds to use exponential backoff
103
+ with a jitter to determine the number of seconds to wait.
104
+ """
105
+
106
+ # 1. Check Retry-After header first
107
+ retry_after = _parse_retry_after(response.headers)
108
+ if retry_after is not None and retry_after > 0:
109
+ return min(retry_after, MAX_RETRY_DELAY_SECONDS)
110
+
111
+ # 2. Check X-RateLimit-Reset header (with positive jitter)
112
+ ratelimit_reset = _parse_x_ratelimit_reset(response.headers)
113
+ if ratelimit_reset is not None:
114
+ return _add_positive_jitter(min(ratelimit_reset, MAX_RETRY_DELAY_SECONDS))
115
+
116
+ # 3. Fall back to exponential backoff (with symmetric jitter)
117
+ backoff = min(INITIAL_RETRY_DELAY_SECONDS * pow(2.0, retries), MAX_RETRY_DELAY_SECONDS)
118
+ return _add_symmetric_jitter(backoff)
119
+
120
+
121
+ def _should_retry(response: httpx.Response) -> bool:
122
+ retryable_400s = [429, 408, 409]
123
+ return response.status_code >= 500 or response.status_code in retryable_400s
124
+
125
+
126
+ _SENSITIVE_HEADERS = frozenset(
127
+ {
128
+ "authorization",
129
+ "www-authenticate",
130
+ "x-api-key",
131
+ "api-key",
132
+ "apikey",
133
+ "x-api-token",
134
+ "x-auth-token",
135
+ "auth-token",
136
+ "cookie",
137
+ "set-cookie",
138
+ "proxy-authorization",
139
+ "proxy-authenticate",
140
+ "x-csrf-token",
141
+ "x-xsrf-token",
142
+ "x-session-token",
143
+ "x-access-token",
144
+ }
145
+ )
146
+
147
+
148
+ def _redact_headers(headers: typing.Dict[str, str]) -> typing.Dict[str, str]:
149
+ return {k: ("[REDACTED]" if k.lower() in _SENSITIVE_HEADERS else v) for k, v in headers.items()}
150
+
151
+
152
+ def _build_url(base_url: str, path: typing.Optional[str]) -> str:
153
+ """
154
+ Build a full URL by joining a base URL with a path.
155
+
156
+ This function correctly handles base URLs that contain path prefixes (e.g., tenant-based URLs)
157
+ by using string concatenation instead of urllib.parse.urljoin(), which would incorrectly
158
+ strip path components when the path starts with '/'.
159
+
160
+ Example:
161
+ >>> _build_url("https://cloud.example.com/org/tenant/api", "/users")
162
+ 'https://cloud.example.com/org/tenant/api/users'
163
+
164
+ Args:
165
+ base_url: The base URL, which may contain path prefixes.
166
+ path: The path to append. Can be None or empty string.
167
+
168
+ Returns:
169
+ The full URL with base_url and path properly joined.
170
+ """
171
+ if not path:
172
+ return base_url
173
+ return f"{base_url.rstrip('/')}/{path.lstrip('/')}"
174
+
175
+
176
+ def _maybe_filter_none_from_multipart_data(
177
+ data: typing.Optional[typing.Any],
178
+ request_files: typing.Optional[RequestFiles],
179
+ force_multipart: typing.Optional[bool],
180
+ ) -> typing.Optional[typing.Any]:
181
+ """
182
+ Filter None values from data body for multipart/form requests.
183
+ This prevents httpx from converting None to empty strings in multipart encoding.
184
+ Only applies when files are present or force_multipart is True.
185
+ """
186
+ if data is not None and isinstance(data, typing.Mapping) and (request_files or force_multipart):
187
+ return remove_none_from_dict(data)
188
+ return data
189
+
190
+
191
+ def remove_omit_from_dict(
192
+ original: typing.Dict[str, typing.Optional[typing.Any]],
193
+ omit: typing.Optional[typing.Any],
194
+ ) -> typing.Dict[str, typing.Any]:
195
+ if omit is None:
196
+ return original
197
+ new: typing.Dict[str, typing.Any] = {}
198
+ for key, value in original.items():
199
+ if value is not omit:
200
+ new[key] = value
201
+ return new
202
+
203
+
204
+ def maybe_filter_request_body(
205
+ data: typing.Optional[typing.Any],
206
+ request_options: typing.Optional[RequestOptions],
207
+ omit: typing.Optional[typing.Any],
208
+ ) -> typing.Optional[typing.Any]:
209
+ if data is None:
210
+ return (
211
+ jsonable_encoder(request_options.get("additional_body_parameters", {})) or {}
212
+ if request_options is not None
213
+ else None
214
+ )
215
+ elif not isinstance(data, typing.Mapping):
216
+ data_content = jsonable_encoder(data)
217
+ else:
218
+ data_content = {
219
+ **(jsonable_encoder(remove_omit_from_dict(data, omit))), # type: ignore
220
+ **(
221
+ jsonable_encoder(request_options.get("additional_body_parameters", {})) or {}
222
+ if request_options is not None
223
+ else {}
224
+ ),
225
+ }
226
+ return data_content
227
+
228
+
229
+ # Abstracted out for testing purposes
230
+ def get_request_body(
231
+ *,
232
+ json: typing.Optional[typing.Any],
233
+ data: typing.Optional[typing.Any],
234
+ request_options: typing.Optional[RequestOptions],
235
+ omit: typing.Optional[typing.Any],
236
+ ) -> typing.Tuple[typing.Optional[typing.Any], typing.Optional[typing.Any]]:
237
+ json_body = None
238
+ data_body = None
239
+ if data is not None:
240
+ data_body = maybe_filter_request_body(data, request_options, omit)
241
+ else:
242
+ # If both data and json are None, we send json data in the event extra properties are specified
243
+ json_body = maybe_filter_request_body(json, request_options, omit)
244
+
245
+ has_additional_body_parameters = bool(
246
+ request_options is not None and request_options.get("additional_body_parameters")
247
+ )
248
+
249
+ # Only collapse empty dict to None when the body was not explicitly provided
250
+ # and there are no additional body parameters. This preserves explicit empty
251
+ # bodies (e.g., when an endpoint has a request body type but all fields are optional).
252
+ if json_body == {} and json is None and not has_additional_body_parameters:
253
+ json_body = None
254
+ if data_body == {} and data is None and not has_additional_body_parameters:
255
+ data_body = None
256
+
257
+ return json_body, data_body
258
+
259
+
260
+ class HttpClient:
261
+ def __init__(
262
+ self,
263
+ *,
264
+ httpx_client: httpx.Client,
265
+ base_timeout: typing.Callable[[], typing.Optional[float]],
266
+ base_headers: typing.Callable[[], typing.Dict[str, str]],
267
+ base_url: typing.Optional[typing.Callable[[], str]] = None,
268
+ logging_config: typing.Optional[typing.Union[LogConfig, Logger]] = None,
269
+ ):
270
+ self.base_url = base_url
271
+ self.base_timeout = base_timeout
272
+ self.base_headers = base_headers
273
+ self.httpx_client = httpx_client
274
+ self.logger = create_logger(logging_config)
275
+
276
+ def get_base_url(self, maybe_base_url: typing.Optional[str]) -> str:
277
+ base_url = maybe_base_url
278
+ if self.base_url is not None and base_url is None:
279
+ base_url = self.base_url()
280
+
281
+ if base_url is None:
282
+ raise ValueError("A base_url is required to make this request, please provide one and try again.")
283
+ return base_url
284
+
285
+ def request(
286
+ self,
287
+ path: typing.Optional[str] = None,
288
+ *,
289
+ method: str,
290
+ base_url: typing.Optional[str] = None,
291
+ params: typing.Optional[typing.Dict[str, typing.Any]] = None,
292
+ json: typing.Optional[typing.Any] = None,
293
+ data: typing.Optional[typing.Any] = None,
294
+ content: typing.Optional[typing.Union[bytes, typing.Iterator[bytes], typing.AsyncIterator[bytes]]] = None,
295
+ files: typing.Optional[
296
+ typing.Union[
297
+ typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]],
298
+ typing.List[typing.Tuple[str, File]],
299
+ ]
300
+ ] = None,
301
+ headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
302
+ request_options: typing.Optional[RequestOptions] = None,
303
+ retries: int = 0,
304
+ omit: typing.Optional[typing.Any] = None,
305
+ force_multipart: typing.Optional[bool] = None,
306
+ ) -> httpx.Response:
307
+ base_url = self.get_base_url(base_url)
308
+ timeout = (
309
+ request_options.get("timeout_in_seconds")
310
+ if request_options is not None and request_options.get("timeout_in_seconds") is not None
311
+ else self.base_timeout()
312
+ )
313
+
314
+ json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
315
+
316
+ request_files: typing.Optional[RequestFiles] = (
317
+ convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
318
+ if (files is not None and files is not omit and isinstance(files, dict))
319
+ else None
320
+ )
321
+
322
+ if (request_files is None or len(request_files) == 0) and force_multipart:
323
+ request_files = FORCE_MULTIPART
324
+
325
+ data_body = _maybe_filter_none_from_multipart_data(data_body, request_files, force_multipart)
326
+
327
+ # Compute encoded params separately to avoid passing empty list to httpx
328
+ # (httpx strips existing query params from URL when params=[] is passed)
329
+ _encoded_params = encode_query(
330
+ jsonable_encoder(
331
+ remove_none_from_dict(
332
+ remove_omit_from_dict(
333
+ {
334
+ **(params if params is not None else {}),
335
+ **(
336
+ request_options.get("additional_query_parameters", {}) or {}
337
+ if request_options is not None
338
+ else {}
339
+ ),
340
+ },
341
+ omit,
342
+ )
343
+ )
344
+ )
345
+ )
346
+
347
+ _request_url = _build_url(base_url, path)
348
+ _request_headers = jsonable_encoder(
349
+ remove_none_from_dict(
350
+ {
351
+ **self.base_headers(),
352
+ **(headers if headers is not None else {}),
353
+ **(request_options.get("additional_headers", {}) or {} if request_options is not None else {}),
354
+ }
355
+ )
356
+ )
357
+
358
+ if self.logger.is_debug():
359
+ self.logger.debug(
360
+ "Making HTTP request",
361
+ method=method,
362
+ url=_request_url,
363
+ headers=_redact_headers(_request_headers),
364
+ has_body=json_body is not None or data_body is not None,
365
+ )
366
+
367
+ response = self.httpx_client.request(
368
+ method=method,
369
+ url=_request_url,
370
+ headers=_request_headers,
371
+ params=_encoded_params if _encoded_params else None,
372
+ json=json_body,
373
+ data=data_body,
374
+ content=content,
375
+ files=request_files,
376
+ timeout=timeout,
377
+ )
378
+
379
+ max_retries: int = request_options.get("max_retries", 2) if request_options is not None else 2
380
+ if _should_retry(response=response):
381
+ if retries < max_retries:
382
+ time.sleep(_retry_timeout(response=response, retries=retries))
383
+ return self.request(
384
+ path=path,
385
+ method=method,
386
+ base_url=base_url,
387
+ params=params,
388
+ json=json,
389
+ content=content,
390
+ files=files,
391
+ headers=headers,
392
+ request_options=request_options,
393
+ retries=retries + 1,
394
+ omit=omit,
395
+ )
396
+
397
+ if self.logger.is_debug():
398
+ if 200 <= response.status_code < 400:
399
+ self.logger.debug(
400
+ "HTTP request succeeded",
401
+ method=method,
402
+ url=_request_url,
403
+ status_code=response.status_code,
404
+ )
405
+
406
+ if self.logger.is_error():
407
+ if response.status_code >= 400:
408
+ self.logger.error(
409
+ "HTTP request failed with error status",
410
+ method=method,
411
+ url=_request_url,
412
+ status_code=response.status_code,
413
+ )
414
+
415
+ return response
416
+
417
+ @contextmanager
418
+ def stream(
419
+ self,
420
+ path: typing.Optional[str] = None,
421
+ *,
422
+ method: str,
423
+ base_url: typing.Optional[str] = None,
424
+ params: typing.Optional[typing.Dict[str, typing.Any]] = None,
425
+ json: typing.Optional[typing.Any] = None,
426
+ data: typing.Optional[typing.Any] = None,
427
+ content: typing.Optional[typing.Union[bytes, typing.Iterator[bytes], typing.AsyncIterator[bytes]]] = None,
428
+ files: typing.Optional[
429
+ typing.Union[
430
+ typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]],
431
+ typing.List[typing.Tuple[str, File]],
432
+ ]
433
+ ] = None,
434
+ headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
435
+ request_options: typing.Optional[RequestOptions] = None,
436
+ retries: int = 0,
437
+ omit: typing.Optional[typing.Any] = None,
438
+ force_multipart: typing.Optional[bool] = None,
439
+ ) -> typing.Iterator[httpx.Response]:
440
+ base_url = self.get_base_url(base_url)
441
+ timeout = (
442
+ request_options.get("timeout_in_seconds")
443
+ if request_options is not None and request_options.get("timeout_in_seconds") is not None
444
+ else self.base_timeout()
445
+ )
446
+
447
+ request_files: typing.Optional[RequestFiles] = (
448
+ convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
449
+ if (files is not None and files is not omit and isinstance(files, dict))
450
+ else None
451
+ )
452
+
453
+ if (request_files is None or len(request_files) == 0) and force_multipart:
454
+ request_files = FORCE_MULTIPART
455
+
456
+ json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
457
+
458
+ data_body = _maybe_filter_none_from_multipart_data(data_body, request_files, force_multipart)
459
+
460
+ # Compute encoded params separately to avoid passing empty list to httpx
461
+ # (httpx strips existing query params from URL when params=[] is passed)
462
+ _encoded_params = encode_query(
463
+ jsonable_encoder(
464
+ remove_none_from_dict(
465
+ remove_omit_from_dict(
466
+ {
467
+ **(params if params is not None else {}),
468
+ **(
469
+ request_options.get("additional_query_parameters", {})
470
+ if request_options is not None
471
+ else {}
472
+ ),
473
+ },
474
+ omit,
475
+ )
476
+ )
477
+ )
478
+ )
479
+
480
+ _request_url = _build_url(base_url, path)
481
+ _request_headers = jsonable_encoder(
482
+ remove_none_from_dict(
483
+ {
484
+ **self.base_headers(),
485
+ **(headers if headers is not None else {}),
486
+ **(request_options.get("additional_headers", {}) if request_options is not None else {}),
487
+ }
488
+ )
489
+ )
490
+
491
+ if self.logger.is_debug():
492
+ self.logger.debug(
493
+ "Making streaming HTTP request",
494
+ method=method,
495
+ url=_request_url,
496
+ headers=_redact_headers(_request_headers),
497
+ )
498
+
499
+ with self.httpx_client.stream(
500
+ method=method,
501
+ url=_request_url,
502
+ headers=_request_headers,
503
+ params=_encoded_params if _encoded_params else None,
504
+ json=json_body,
505
+ data=data_body,
506
+ content=content,
507
+ files=request_files,
508
+ timeout=timeout,
509
+ ) as stream:
510
+ yield stream
511
+
512
+
513
+ class AsyncHttpClient:
514
+ def __init__(
515
+ self,
516
+ *,
517
+ httpx_client: httpx.AsyncClient,
518
+ base_timeout: typing.Callable[[], typing.Optional[float]],
519
+ base_headers: typing.Callable[[], typing.Dict[str, str]],
520
+ base_url: typing.Optional[typing.Callable[[], str]] = None,
521
+ async_base_headers: typing.Optional[typing.Callable[[], typing.Awaitable[typing.Dict[str, str]]]] = None,
522
+ logging_config: typing.Optional[typing.Union[LogConfig, Logger]] = None,
523
+ ):
524
+ self.base_url = base_url
525
+ self.base_timeout = base_timeout
526
+ self.base_headers = base_headers
527
+ self.async_base_headers = async_base_headers
528
+ self.httpx_client = httpx_client
529
+ self.logger = create_logger(logging_config)
530
+
531
+ async def _get_headers(self) -> typing.Dict[str, str]:
532
+ if self.async_base_headers is not None:
533
+ return await self.async_base_headers()
534
+ return self.base_headers()
535
+
536
+ def get_base_url(self, maybe_base_url: typing.Optional[str]) -> str:
537
+ base_url = maybe_base_url
538
+ if self.base_url is not None and base_url is None:
539
+ base_url = self.base_url()
540
+
541
+ if base_url is None:
542
+ raise ValueError("A base_url is required to make this request, please provide one and try again.")
543
+ return base_url
544
+
545
+ async def request(
546
+ self,
547
+ path: typing.Optional[str] = None,
548
+ *,
549
+ method: str,
550
+ base_url: typing.Optional[str] = None,
551
+ params: typing.Optional[typing.Dict[str, typing.Any]] = None,
552
+ json: typing.Optional[typing.Any] = None,
553
+ data: typing.Optional[typing.Any] = None,
554
+ content: typing.Optional[typing.Union[bytes, typing.Iterator[bytes], typing.AsyncIterator[bytes]]] = None,
555
+ files: typing.Optional[
556
+ typing.Union[
557
+ typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]],
558
+ typing.List[typing.Tuple[str, File]],
559
+ ]
560
+ ] = None,
561
+ headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
562
+ request_options: typing.Optional[RequestOptions] = None,
563
+ retries: int = 0,
564
+ omit: typing.Optional[typing.Any] = None,
565
+ force_multipart: typing.Optional[bool] = None,
566
+ ) -> httpx.Response:
567
+ base_url = self.get_base_url(base_url)
568
+ timeout = (
569
+ request_options.get("timeout_in_seconds")
570
+ if request_options is not None and request_options.get("timeout_in_seconds") is not None
571
+ else self.base_timeout()
572
+ )
573
+
574
+ request_files: typing.Optional[RequestFiles] = (
575
+ convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
576
+ if (files is not None and files is not omit and isinstance(files, dict))
577
+ else None
578
+ )
579
+
580
+ if (request_files is None or len(request_files) == 0) and force_multipart:
581
+ request_files = FORCE_MULTIPART
582
+
583
+ json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
584
+
585
+ data_body = _maybe_filter_none_from_multipart_data(data_body, request_files, force_multipart)
586
+
587
+ # Get headers (supports async token providers)
588
+ _headers = await self._get_headers()
589
+
590
+ # Compute encoded params separately to avoid passing empty list to httpx
591
+ # (httpx strips existing query params from URL when params=[] is passed)
592
+ _encoded_params = encode_query(
593
+ jsonable_encoder(
594
+ remove_none_from_dict(
595
+ remove_omit_from_dict(
596
+ {
597
+ **(params if params is not None else {}),
598
+ **(
599
+ request_options.get("additional_query_parameters", {}) or {}
600
+ if request_options is not None
601
+ else {}
602
+ ),
603
+ },
604
+ omit,
605
+ )
606
+ )
607
+ )
608
+ )
609
+
610
+ _request_url = _build_url(base_url, path)
611
+ _request_headers = jsonable_encoder(
612
+ remove_none_from_dict(
613
+ {
614
+ **_headers,
615
+ **(headers if headers is not None else {}),
616
+ **(request_options.get("additional_headers", {}) or {} if request_options is not None else {}),
617
+ }
618
+ )
619
+ )
620
+
621
+ if self.logger.is_debug():
622
+ self.logger.debug(
623
+ "Making HTTP request",
624
+ method=method,
625
+ url=_request_url,
626
+ headers=_redact_headers(_request_headers),
627
+ has_body=json_body is not None or data_body is not None,
628
+ )
629
+
630
+ response = await self.httpx_client.request(
631
+ method=method,
632
+ url=_request_url,
633
+ headers=_request_headers,
634
+ params=_encoded_params if _encoded_params else None,
635
+ json=json_body,
636
+ data=data_body,
637
+ content=content,
638
+ files=request_files,
639
+ timeout=timeout,
640
+ )
641
+
642
+ max_retries: int = request_options.get("max_retries", 2) if request_options is not None else 2
643
+ if _should_retry(response=response):
644
+ if retries < max_retries:
645
+ await asyncio.sleep(_retry_timeout(response=response, retries=retries))
646
+ return await self.request(
647
+ path=path,
648
+ method=method,
649
+ base_url=base_url,
650
+ params=params,
651
+ json=json,
652
+ content=content,
653
+ files=files,
654
+ headers=headers,
655
+ request_options=request_options,
656
+ retries=retries + 1,
657
+ omit=omit,
658
+ )
659
+
660
+ if self.logger.is_debug():
661
+ if 200 <= response.status_code < 400:
662
+ self.logger.debug(
663
+ "HTTP request succeeded",
664
+ method=method,
665
+ url=_request_url,
666
+ status_code=response.status_code,
667
+ )
668
+
669
+ if self.logger.is_error():
670
+ if response.status_code >= 400:
671
+ self.logger.error(
672
+ "HTTP request failed with error status",
673
+ method=method,
674
+ url=_request_url,
675
+ status_code=response.status_code,
676
+ )
677
+
678
+ return response
679
+
680
+ @asynccontextmanager
681
+ async def stream(
682
+ self,
683
+ path: typing.Optional[str] = None,
684
+ *,
685
+ method: str,
686
+ base_url: typing.Optional[str] = None,
687
+ params: typing.Optional[typing.Dict[str, typing.Any]] = None,
688
+ json: typing.Optional[typing.Any] = None,
689
+ data: typing.Optional[typing.Any] = None,
690
+ content: typing.Optional[typing.Union[bytes, typing.Iterator[bytes], typing.AsyncIterator[bytes]]] = None,
691
+ files: typing.Optional[
692
+ typing.Union[
693
+ typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]],
694
+ typing.List[typing.Tuple[str, File]],
695
+ ]
696
+ ] = None,
697
+ headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
698
+ request_options: typing.Optional[RequestOptions] = None,
699
+ retries: int = 0,
700
+ omit: typing.Optional[typing.Any] = None,
701
+ force_multipart: typing.Optional[bool] = None,
702
+ ) -> typing.AsyncIterator[httpx.Response]:
703
+ base_url = self.get_base_url(base_url)
704
+ timeout = (
705
+ request_options.get("timeout_in_seconds")
706
+ if request_options is not None and request_options.get("timeout_in_seconds") is not None
707
+ else self.base_timeout()
708
+ )
709
+
710
+ request_files: typing.Optional[RequestFiles] = (
711
+ convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
712
+ if (files is not None and files is not omit and isinstance(files, dict))
713
+ else None
714
+ )
715
+
716
+ if (request_files is None or len(request_files) == 0) and force_multipart:
717
+ request_files = FORCE_MULTIPART
718
+
719
+ json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
720
+
721
+ data_body = _maybe_filter_none_from_multipart_data(data_body, request_files, force_multipart)
722
+
723
+ # Get headers (supports async token providers)
724
+ _headers = await self._get_headers()
725
+
726
+ # Compute encoded params separately to avoid passing empty list to httpx
727
+ # (httpx strips existing query params from URL when params=[] is passed)
728
+ _encoded_params = encode_query(
729
+ jsonable_encoder(
730
+ remove_none_from_dict(
731
+ remove_omit_from_dict(
732
+ {
733
+ **(params if params is not None else {}),
734
+ **(
735
+ request_options.get("additional_query_parameters", {})
736
+ if request_options is not None
737
+ else {}
738
+ ),
739
+ },
740
+ omit=omit,
741
+ )
742
+ )
743
+ )
744
+ )
745
+
746
+ _request_url = _build_url(base_url, path)
747
+ _request_headers = jsonable_encoder(
748
+ remove_none_from_dict(
749
+ {
750
+ **_headers,
751
+ **(headers if headers is not None else {}),
752
+ **(request_options.get("additional_headers", {}) if request_options is not None else {}),
753
+ }
754
+ )
755
+ )
756
+
757
+ if self.logger.is_debug():
758
+ self.logger.debug(
759
+ "Making streaming HTTP request",
760
+ method=method,
761
+ url=_request_url,
762
+ headers=_redact_headers(_request_headers),
763
+ )
764
+
765
+ async with self.httpx_client.stream(
766
+ method=method,
767
+ url=_request_url,
768
+ headers=_request_headers,
769
+ params=_encoded_params if _encoded_params else None,
770
+ json=json_body,
771
+ data=data_body,
772
+ content=content,
773
+ files=request_files,
774
+ timeout=timeout,
775
+ ) as stream:
776
+ yield stream