1616from azure .core .tracing .decorator_async import distributed_trace_async
1717
1818from ... import models as _models
19+ from ..._vendor import _convert_request
1920from ...operations ._documents_operations import build_autocomplete_get_request , build_autocomplete_post_request , build_count_request , build_get_request , build_index_request , build_search_get_request , build_search_post_request , build_suggest_get_request , build_suggest_post_request
2021
2122T = TypeVar ('T' )
@@ -71,7 +72,8 @@ async def count(
7172 request = build_count_request (
7273 x_ms_client_request_id = _x_ms_client_request_id ,
7374 template_url = self .count .metadata ['url' ],
74- )._to_pipeline_transport_request ()
75+ )
76+ request = _convert_request (request )
7577 path_format_arguments = {
7678 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
7779 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -83,7 +85,7 @@ async def count(
8385
8486 if response .status_code not in [200 ]:
8587 map_error (status_code = response .status_code , response = response , error_map = error_map )
86- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
88+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
8789 raise HttpResponseError (response = response , model = error )
8890
8991 deserialized = self ._deserialize ('long' , pipeline_response )
@@ -202,7 +204,8 @@ async def search_get(
202204 semantic_fields = _semantic_fields ,
203205 x_ms_client_request_id = _x_ms_client_request_id ,
204206 template_url = self .search_get .metadata ['url' ],
205- )._to_pipeline_transport_request ()
207+ )
208+ request = _convert_request (request )
206209 path_format_arguments = {
207210 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
208211 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -214,7 +217,7 @@ async def search_get(
214217
215218 if response .status_code not in [200 ]:
216219 map_error (status_code = response .status_code , response = response , error_map = error_map )
217- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
220+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
218221 raise HttpResponseError (response = response , model = error )
219222
220223 deserialized = self ._deserialize ('SearchDocumentsResult' , pipeline_response )
@@ -263,7 +266,8 @@ async def search_post(
263266 x_ms_client_request_id = _x_ms_client_request_id ,
264267 json = json ,
265268 template_url = self .search_post .metadata ['url' ],
266- )._to_pipeline_transport_request ()
269+ )
270+ request = _convert_request (request )
267271 path_format_arguments = {
268272 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
269273 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -275,7 +279,7 @@ async def search_post(
275279
276280 if response .status_code not in [200 ]:
277281 map_error (status_code = response .status_code , response = response , error_map = error_map )
278- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
282+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
279283 raise HttpResponseError (response = response , model = error )
280284
281285 deserialized = self ._deserialize ('SearchDocumentsResult' , pipeline_response )
@@ -325,7 +329,8 @@ async def get(
325329 selected_fields = selected_fields ,
326330 x_ms_client_request_id = _x_ms_client_request_id ,
327331 template_url = self .get .metadata ['url' ],
328- )._to_pipeline_transport_request ()
332+ )
333+ request = _convert_request (request )
329334 path_format_arguments = {
330335 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
331336 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -337,7 +342,7 @@ async def get(
337342
338343 if response .status_code not in [200 ]:
339344 map_error (status_code = response .status_code , response = response , error_map = error_map )
340- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
345+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
341346 raise HttpResponseError (response = response , model = error )
342347
343348 deserialized = self ._deserialize ('object' , pipeline_response )
@@ -419,7 +424,8 @@ async def suggest_get(
419424 top = _top ,
420425 x_ms_client_request_id = _x_ms_client_request_id ,
421426 template_url = self .suggest_get .metadata ['url' ],
422- )._to_pipeline_transport_request ()
427+ )
428+ request = _convert_request (request )
423429 path_format_arguments = {
424430 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
425431 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -431,7 +437,7 @@ async def suggest_get(
431437
432438 if response .status_code not in [200 ]:
433439 map_error (status_code = response .status_code , response = response , error_map = error_map )
434- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
440+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
435441 raise HttpResponseError (response = response , model = error )
436442
437443 deserialized = self ._deserialize ('SuggestDocumentsResult' , pipeline_response )
@@ -480,7 +486,8 @@ async def suggest_post(
480486 x_ms_client_request_id = _x_ms_client_request_id ,
481487 json = json ,
482488 template_url = self .suggest_post .metadata ['url' ],
483- )._to_pipeline_transport_request ()
489+ )
490+ request = _convert_request (request )
484491 path_format_arguments = {
485492 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
486493 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -492,7 +499,7 @@ async def suggest_post(
492499
493500 if response .status_code not in [200 ]:
494501 map_error (status_code = response .status_code , response = response , error_map = error_map )
495- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
502+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
496503 raise HttpResponseError (response = response , model = error )
497504
498505 deserialized = self ._deserialize ('SuggestDocumentsResult' , pipeline_response )
@@ -542,7 +549,8 @@ async def index(
542549 x_ms_client_request_id = _x_ms_client_request_id ,
543550 json = json ,
544551 template_url = self .index .metadata ['url' ],
545- )._to_pipeline_transport_request ()
552+ )
553+ request = _convert_request (request )
546554 path_format_arguments = {
547555 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
548556 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -554,7 +562,7 @@ async def index(
554562
555563 if response .status_code not in [200 , 207 ]:
556564 map_error (status_code = response .status_code , response = response , error_map = error_map )
557- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
565+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
558566 raise HttpResponseError (response = response , model = error )
559567
560568 if response .status_code == 200 :
@@ -636,7 +644,8 @@ async def autocomplete_get(
636644 search_fields = _search_fields ,
637645 top = _top ,
638646 template_url = self .autocomplete_get .metadata ['url' ],
639- )._to_pipeline_transport_request ()
647+ )
648+ request = _convert_request (request )
640649 path_format_arguments = {
641650 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
642651 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -648,7 +657,7 @@ async def autocomplete_get(
648657
649658 if response .status_code not in [200 ]:
650659 map_error (status_code = response .status_code , response = response , error_map = error_map )
651- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
660+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
652661 raise HttpResponseError (response = response , model = error )
653662
654663 deserialized = self ._deserialize ('AutocompleteResult' , pipeline_response )
@@ -697,7 +706,8 @@ async def autocomplete_post(
697706 x_ms_client_request_id = _x_ms_client_request_id ,
698707 json = json ,
699708 template_url = self .autocomplete_post .metadata ['url' ],
700- )._to_pipeline_transport_request ()
709+ )
710+ request = _convert_request (request )
701711 path_format_arguments = {
702712 "endpoint" : self ._serialize .url ("self._config.endpoint" , self ._config .endpoint , 'str' , skip_quote = True ),
703713 "indexName" : self ._serialize .url ("self._config.index_name" , self ._config .index_name , 'str' ),
@@ -709,7 +719,7 @@ async def autocomplete_post(
709719
710720 if response .status_code not in [200 ]:
711721 map_error (status_code = response .status_code , response = response , error_map = error_map )
712- error = self ._deserialize .failsafe_deserialize (_models .SearchError , response )
722+ error = self ._deserialize .failsafe_deserialize (_models .SearchError , pipeline_response )
713723 raise HttpResponseError (response = response , model = error )
714724
715725 deserialized = self ._deserialize ('AutocompleteResult' , pipeline_response )
0 commit comments