updates

Author: xiangyan99

sdk/search/azure-search-documents/azure/search/documents/indexes/_search_index_client.py

@@ -133,7 +133,7 @@ def list_index_names(self, **kwargs: Any) -> ItemPaged[str]:
         return cast(ItemPaged[str], names)
 
     @distributed_trace
-    def get_index(self, name: str, **kwargs: Any) -> Optional[SearchIndex]:
+    def get_index(self, name: str, **kwargs: Any) -> SearchIndex:
         """
 
         :param name: The name of the index to retrieve.
@@ -153,7 +153,7 @@ def get_index(self, name: str, **kwargs: Any) -> Optional[SearchIndex]:
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = self._client.indexes.get(name, **kwargs)
-        return SearchIndex._from_generated(result)  # pylint:disable=protected-access
+        return cast(SearchIndex, SearchIndex._from_generated(result))  # pylint:disable=protected-access
 
     @distributed_trace
     def get_index_statistics(self, index_name: str, **kwargs: Any) -> MutableMapping[str, Any]:
@@ -356,7 +356,7 @@ def get_synonym_map_names(self, **kwargs: Any) -> List[str]:
         return [x.name for x in result.synonym_maps]
 
     @distributed_trace
-    def get_synonym_map(self, name: str, **kwargs: Any) -> Optional[SynonymMap]:
+    def get_synonym_map(self, name: str, **kwargs: Any) -> SynonymMap:
         """Retrieve a named Synonym Map in an Azure Search service
 
         :param name: The name of the Synonym Map to get
@@ -377,7 +377,7 @@ def get_synonym_map(self, name: str, **kwargs: Any) -> Optional[SynonymMap]:
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = self._client.synonym_maps.get(name, **kwargs)
-        return SynonymMap._from_generated(result)  # pylint:disable=protected-access
+        return cast(SynonymMap, SynonymMap._from_generated(result))  # pylint:disable=protected-access
 
     @distributed_trace
     def delete_synonym_map(

sdk/search/azure-search-documents/azure/search/documents/indexes/_search_indexer_client.py

@@ -353,7 +353,7 @@ def create_or_update_data_source_connection(
         return cast(SearchIndexerDataSourceConnection, SearchIndexerDataSourceConnection._from_generated(result))
 
     @distributed_trace
-    def get_data_source_connection(self, name: str, **kwargs: Any) -> Optional[SearchIndexerDataSourceConnection]:
+    def get_data_source_connection(self, name: str, **kwargs: Any) -> SearchIndexerDataSourceConnection:
         """Retrieves a data source connection definition.
 
         :param name: The name of the data source connection to retrieve.
@@ -372,7 +372,8 @@ def get_data_source_connection(self, name: str, **kwargs: Any) -> Optional[Searc
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = self._client.data_sources.get(name, **kwargs)
-        return SearchIndexerDataSourceConnection._from_generated(result)  # pylint:disable=protected-access
+        # pylint:disable=protected-access
+        return cast(SearchIndexerDataSourceConnection, SearchIndexerDataSourceConnection._from_generated(result))
 
     @distributed_trace
     def get_data_source_connections(
@@ -488,7 +489,7 @@ def get_skillset_names(self, **kwargs: Any) -> List[str]:
         return [x.name for x in result.skillsets]
 
     @distributed_trace
-    def get_skillset(self, name: str, **kwargs: Any) -> Optional[SearchIndexerSkillset]:
+    def get_skillset(self, name: str, **kwargs: Any) -> SearchIndexerSkillset:
         """Retrieve a named SearchIndexerSkillset in an Azure Search service
 
         :param name: The name of the SearchIndexerSkillset to get
@@ -499,7 +500,8 @@ def get_skillset(self, name: str, **kwargs: Any) -> Optional[SearchIndexerSkills
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = self._client.skillsets.get(name, **kwargs)
-        return SearchIndexerSkillset._from_generated(result)  # pylint:disable=protected-access
+        # pylint:disable=protected-access
+        return cast(SearchIndexerSkillset, SearchIndexerSkillset._from_generated(result))
 
     @distributed_trace
     def delete_skillset(

sdk/search/azure-search-documents/azure/search/documents/indexes/aio/_search_index_client.py

@@ -131,7 +131,7 @@ def list_index_names(self, **kwargs: Any) -> AsyncItemPaged[str]:
         return cast(AsyncItemPaged[str], names)
 
     @distributed_trace_async
-    async def get_index(self, name: str, **kwargs: Any) -> Optional[SearchIndex]:
+    async def get_index(self, name: str, **kwargs: Any) -> SearchIndex:
         """
 
         :param name: The name of the index to retrieve.
@@ -151,7 +151,7 @@ async def get_index(self, name: str, **kwargs: Any) -> Optional[SearchIndex]:
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = await self._client.indexes.get(name, **kwargs)
-        return SearchIndex._from_generated(result)  # pylint:disable=protected-access
+        return cast(SearchIndex, SearchIndex._from_generated(result))  # pylint:disable=protected-access
 
     @distributed_trace_async
     async def get_index_statistics(self, index_name: str, **kwargs: Any) -> MutableMapping[str, Any]:
@@ -353,7 +353,7 @@ async def get_synonym_map_names(self, **kwargs: Any) -> List[str]:
         return [x.name for x in result.synonym_maps]
 
     @distributed_trace_async
-    async def get_synonym_map(self, name: str, **kwargs: Any) -> Optional[SynonymMap]:
+    async def get_synonym_map(self, name: str, **kwargs: Any) -> SynonymMap:
         """Retrieve a named Synonym Map in an Azure Search service
 
         :param name: The name of the Synonym Map to get
@@ -374,7 +374,7 @@ async def get_synonym_map(self, name: str, **kwargs: Any) -> Optional[SynonymMap
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = await self._client.synonym_maps.get(name, **kwargs)
-        return SynonymMap._from_generated(result)  # pylint:disable=protected-access
+        return cast(SynonymMap, SynonymMap._from_generated(result))  # pylint:disable=protected-access
 
     @distributed_trace_async
     async def delete_synonym_map(

sdk/search/azure-search-documents/azure/search/documents/indexes/aio/_search_indexer_client.py

@@ -381,7 +381,7 @@ async def delete_data_source_connection(
     @distributed_trace_async
     async def get_data_source_connection(
         self, name: str, *, select: Optional[List[str]] = None, **kwargs: Any
-    ) -> Optional[SearchIndexerDataSourceConnection]:
+    ) -> SearchIndexerDataSourceConnection:
         """Retrieves a data source connection definition.
 
         :keyword select: Selects which top-level properties of the skillsets to retrieve. Specified as a
@@ -405,7 +405,7 @@ async def get_data_source_connection(
             kwargs["select"] = ",".join(select)
         result = await self._client.data_sources.get(name, **kwargs)
         # pylint:disable=protected-access
-        return SearchIndexerDataSourceConnection._from_generated(result)
+        return cast(SearchIndexerDataSourceConnection, SearchIndexerDataSourceConnection._from_generated(result))
 
     @distributed_trace_async
     async def get_data_source_connections(self, **kwargs: Any) -> Sequence[SearchIndexerDataSourceConnection]:
@@ -477,7 +477,7 @@ async def get_skillset_names(self, **kwargs) -> List[str]:
         return [x.name for x in result.skillsets]
 
     @distributed_trace_async
-    async def get_skillset(self, name: str, **kwargs) -> Optional[SearchIndexerSkillset]:
+    async def get_skillset(self, name: str, **kwargs) -> SearchIndexerSkillset:
         """Retrieve a named SearchIndexerSkillset in an Azure Search service
 
         :param name: The name of the SearchIndexerSkillset to get
@@ -488,7 +488,8 @@ async def get_skillset(self, name: str, **kwargs) -> Optional[SearchIndexerSkill
         """
         kwargs["headers"] = self._merge_client_headers(kwargs.get("headers"))
         result = await self._client.skillsets.get(name, **kwargs)
-        return SearchIndexerSkillset._from_generated(result)  # pylint:disable=protected-access
+        # pylint:disable=protected-access
+        return cast(SearchIndexerSkillset, SearchIndexerSkillset._from_generated(result))
 
     @distributed_trace_async
     async def delete_skillset(