[change] Moved AtlanResponse directly to _call_api method

Author: Aryamanz29

pyatlan/client/asset.py

@@ -70,7 +70,6 @@
     AssetRequest,
     AssetResponse,
     AtlanObject,
-    AtlanResponse,
     AtlanTag,
     AtlanTagName,
     BulkRequest,
@@ -247,10 +246,7 @@ def search(self, criteria: IndexSearchRequest, bulk=False) -> IndexSearchResults
                     unflatten_custom_metadata_for_entity(
                         entity=entity, attributes=criteria.attributes
                     )
-                translated_entities = AtlanResponse(
-                    raw_json=raw_json["entities"], client=self._client
-                ).to_dict()
-                assets = parse_obj_as(List[Asset], translated_entities)
+                assets = parse_obj_as(List[Asset], raw_json["entities"])
             except ValidationError as err:
                 raise ErrorCode.JSON_ERROR.exception_with_parameters(
                     raw_json, 200, str(err)
@@ -322,10 +318,7 @@ def get_lineage_list(
                     unflatten_custom_metadata_for_entity(
                         entity=entity, attributes=lineage_request.attributes
                     )
-                translated_entities = AtlanResponse(
-                    raw_json=raw_json["entities"], client=self._client
-                ).to_dict()
-                assets = parse_obj_as(List[Asset], translated_entities)
+                assets = parse_obj_as(List[Asset], raw_json["entities"])
                 has_more = parse_obj_as(bool, raw_json["hasMore"])
             except ValidationError as err:
                 raise ErrorCode.JSON_ERROR.exception_with_parameters(
@@ -553,10 +546,7 @@ def _handle_relationships(self, raw_json):
                 raw_json["entity"]["relationshipAttributes"]
             )
         raw_json["entity"]["relationshipAttributes"] = {}
-        translated_raw_json = AtlanResponse(
-            raw_json=raw_json, client=self._client
-        ).to_dict()
-        asset = AssetResponse[A](**translated_raw_json).entity
+        asset = AssetResponse[A](**raw_json).entity
         asset.is_incomplete = False
         return asset
 
@@ -640,10 +630,7 @@ def save(
             asset.validate_required()
         request = BulkRequest[Asset](entities=entities)
         raw_json = self._client._call_api(BULK_UPDATE, query_params, request)
-        translated_raw_json = AtlanResponse(
-            raw_json=raw_json, client=self._client
-        ).to_dict()
-        response = AssetMutationResponse(**translated_raw_json)
+        response = AssetMutationResponse(**raw_json)
         if connections_created := response.assets_created(Connection):
             self._wait_for_connections_to_be_created(connections_created)
         return response
@@ -2095,10 +2082,7 @@ def _process_entities(self, entities):
             unflatten_custom_metadata_for_entity(
                 entity=entity, attributes=self._criteria.attributes
             )
-        translated_entities = AtlanResponse(
-            raw_json=entities, client=self._client
-        ).to_dict()
-        self._assets = parse_obj_as(List[Asset], translated_entities)
+        self._assets = parse_obj_as(List[Asset], entities)
 
     def _update_first_last_record_creation_times(self):
         self._first_record_creation_time = self._last_record_creation_time = -2

pyatlan/client/atlan.py

@@ -81,7 +81,7 @@
     Purpose,
 )
 from pyatlan.model.atlan_image import AtlanImage
-from pyatlan.model.core import Announcement, AtlanObject
+from pyatlan.model.core import Announcement, AtlanObject, AtlanResponse
 from pyatlan.model.custom_metadata import CustomMetadataDict
 from pyatlan.model.enums import AtlanConnectorType, AtlanTypeCategory, CertificateStatus
 from pyatlan.model.group import AtlanGroup, CreateGroupResponse, GroupResponse
@@ -511,6 +511,9 @@ def _call_api_internal(
                         response_ = response.text
                     else:
                         response_ = events if events else response.json()
+                        response_ = AtlanResponse(
+                            raw_json=response.json(), client=self
+                        ).to_dict()
                     LOGGER.debug("response: %s", response_)
                     return response_
                 except (