@@ -126,11 +126,10 @@ def sdk_init(
126
126
self .client = client
127
127
return base_url , client
128
128
129
-
130
129
# pylint: disable=too-many-return-statements
131
130
def before_request (
132
- self , hook_ctx : BeforeRequestContext , request : requests . PreparedRequest
133
- ) -> Union [requests . PreparedRequest , Exception ]:
131
+ self , hook_ctx : BeforeRequestContext , request : httpx . Request
132
+ ) -> Union [httpx . Request , Exception ]:
134
133
"""If `splitPdfPage` is set to `true` in the request, the PDF file is split into
135
134
separate pages. Each page is sent as a separate request in parallel. The last
136
135
page request is returned by this method. It will return the original request
@@ -159,7 +158,9 @@ def before_request(
159
158
nest_asyncio .apply ()
160
159
operation_id = hook_ctx .operation_id
161
160
content_type = request .headers .get ("Content-Type" )
162
- body = request .body
161
+
162
+ request_content = request .read ()
163
+ body = request_content
163
164
if not isinstance (body , bytes ) or content_type is None :
164
165
return request
165
166
@@ -296,9 +297,8 @@ async def call_api_partial(page):
296
297
body = request_utils .create_request_body (
297
298
form_data , last_page_content , file .file_name , last_page_number
298
299
)
299
- last_page_request = request_utils .create_request (request , body )
300
- last_page_prepared_request = self .client .prepare_request (last_page_request )
301
- return last_page_prepared_request
300
+ last_page_request = request_utils .create_httpx_request (request , body )
301
+ return last_page_request
302
302
303
303
def _await_elements (
304
304
self , operation_id : str , response : requests .Response
@@ -355,8 +355,8 @@ def _await_elements(
355
355
return flattened_elements
356
356
357
357
def after_success (
358
- self , hook_ctx : AfterSuccessContext , response : requests .Response
359
- ) -> Union [requests .Response , Exception ]:
358
+ self , hook_ctx : AfterSuccessContext , response : httpx .Response
359
+ ) -> Union [httpx .Response , Exception ]:
360
360
"""Executes after a successful API request. Awaits all parallel requests and
361
361
combines the responses into a single response object.
362
362
@@ -390,9 +390,9 @@ def after_success(
390
390
def after_error (
391
391
self ,
392
392
hook_ctx : AfterErrorContext ,
393
- response : Optional [requests .Response ],
393
+ response : Optional [httpx .Response ],
394
394
error : Optional [Exception ],
395
- ) -> Union [Tuple [Optional [requests .Response ], Optional [Exception ]], Exception ]:
395
+ ) -> Union [Tuple [Optional [httpx .Response ], Optional [Exception ]], Exception ]:
396
396
"""Executes after an unsuccessful API request. Awaits all parallel requests,
397
397
if at least one request was successful, combines the responses into a single
398
398
response object and doesn't throw an error. It will return an error only if
0 commit comments