[Cosmos] Fix return type to CosmosDict for public APIs (#43254)

* Fix return type to CosmosDict for public APIs

* Update description of return values

Author: allenkim0129

sdk/cosmos/azure-cosmos/azure/cosmos/_cosmos_client_connection.py

@@ -2007,7 +2007,7 @@ def ReadConflict(
         conflict_link: str,
         options: Optional[Mapping[str, Any]] = None,
         **kwargs: Any
-    ) -> Dict[str, Any]:
+    ) -> CosmosDict:
         """Reads a conflict.
 
         :param str conflict_link:
@@ -2017,7 +2017,7 @@ def ReadConflict(
         :return:
             The read Conflict.
         :rtype:
-            dict
+            ~azure.cosmos.CosmosDict[str, Any]
 
         """
         if options is None:

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_container.py

@@ -1642,7 +1642,7 @@ async def get_conflict(
         conflict: Union[str, Mapping[str, Any]],
         partition_key: PartitionKeyType,
         **kwargs: Any,
-    ) -> Dict[str, Any]:
+    ) -> CosmosDict:
         """Get the conflict identified by `conflict`.
 
         :param conflict: The ID (name) or dict representing the conflict to retrieve.
@@ -1656,8 +1656,8 @@ async def get_conflict(
         :keyword response_hook: A callable invoked with the response metadata.
         :paramtype response_hook: Callable[[Dict[str, str], Dict[str, Any]], None]
         :raises ~azure.cosmos.exceptions.CosmosHttpResponseError: The given conflict couldn't be retrieved.
-        :returns: A dict representing the retrieved conflict.
-        :rtype: Dict[str, Any]
+        :returns: A CosmosDict representing the retrieved conflict.
+        :rtype: ~azure.cosmos.CosmosDict[str, Any]
         """
         request_options = _build_options(kwargs)
         request_options["partitionKey"] = await self._set_partition_key(partition_key)

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_cosmos_client_connection_async.py

@@ -1201,7 +1201,7 @@ async def ReadConflict(
         conflict_link: str,
         options: Optional[Mapping[str, Any]] = None,
         **kwargs: Any
-    ) -> Dict[str, Any]:
+    ) -> CosmosDict:
         """Reads a conflict.
 
         :param str conflict_link:
@@ -1211,7 +1211,7 @@ async def ReadConflict(
         :return:
             The read Conflict.
         :rtype:
-            dict
+            ~azure.cosmos.CosmosDict[str, Any]
 
         """
         if options is None:

sdk/cosmos/azure-cosmos/azure/cosmos/container.py

@@ -1828,7 +1828,7 @@ def get_conflict(
         conflict: Union[str, Mapping[str, Any]],
         partition_key: _PartitionKeyType,
         **kwargs: Any
-    ) -> Dict[str, Any]:
+    ) -> CosmosDict:
         """Get the conflict identified by `conflict`.
 
         :param conflict: The ID (name) or dict representing the conflict to retrieve.
@@ -1840,9 +1840,9 @@ def get_conflict(
         :type partition_key: Union[str, int, float, bool, Type[NonePartitionKeyValue], Type[NullPartitionKeyValue],
             None, Sequence[Union[str, int, float, bool, None]]]
         :keyword Callable response_hook: A callable invoked with the response metadata.
-        :returns: A dict representing the retrieved conflict.
         :raises ~azure.cosmos.exceptions.CosmosHttpResponseError: The given conflict couldn't be retrieved.
-        :rtype: Dict[str, Any]
+        :returns: A CosmosDict representing the retrieved conflict.
+        :rtype: ~azure.cosmos.CosmosDict[str, Any]
         """
         request_options = build_options(kwargs)
         request_options["partitionKey"] = self._set_partition_key(partition_key)