updates

Author: xiangyan99

sdk/search/azure-search-documents/azure/search/documents/_operations/_patch.py

@@ -13,7 +13,7 @@
 import itertools
 import json
 
-from azure.core.paging import ItemPaged, PageIterator
+from azure.core.paging import ItemPaged, PageIterator, ReturnType
 from azure.core.tracing.decorator import distributed_trace
 
 from ._operations import _SearchClientOperationsMixin as _SearchClientOperationsMixinGenerated
@@ -306,7 +306,7 @@ def get_debug_info(self) -> Optional[_models.DebugInfo]:
         return response.debug_info
 
 
-class SearchItemPaged(ItemPaged[Dict]):
+class SearchItemPaged(ItemPaged[ReturnType]):
     """A pageable list of search results with metadata accessors."""
 
     def __init__(self, page_iterator_factory_func) -> None:
@@ -327,7 +327,7 @@ def by_page(self, continuation_token=None):
         """
         return self._page_iterator_factory(continuation_token)
 
-    def __next__(self) -> Dict:
+    def __next__(self) -> ReturnType:
         if self._page_iterator is None:
             first_iterator = self._first_iterator_instance()
             self._page_iterator = itertools.chain.from_iterable(first_iterator)

sdk/search/azure-search-documents/azure/search/documents/aio/_operations/_patch.py

@@ -9,7 +9,7 @@
 """
 from typing import Any, Dict, List, Optional, Union, cast
 
-from azure.core.async_paging import AsyncItemPaged, AsyncPageIterator
+from azure.core.async_paging import AsyncItemPaged, AsyncPageIterator, ReturnType
 from azure.core.tracing.decorator_async import distributed_trace_async
 
 from ._operations import _SearchClientOperationsMixin as _SearchClientOperationsMixinGenerated
@@ -107,7 +107,7 @@ async def get_debug_info(self) -> Optional[_models.DebugInfo]:
         return response.debug_info
 
 
-class AsyncSearchItemPaged(AsyncItemPaged[Dict]):
+class AsyncSearchItemPaged(AsyncItemPaged[ReturnType]):
     """An async pageable list of search results with metadata accessors."""
 
     def __init__(self, page_iterator_factory_func) -> None:
@@ -128,7 +128,7 @@ def by_page(self, continuation_token=None):
         """
         return self._page_iterator_factory(continuation_token)
 
-    async def __anext__(self) -> Dict:
+    async def __anext__(self) -> ReturnType:
         if self._page_iterator is None:
             self._page_iterator = self.by_page()
         if self._page is None: