revert from the original file

Author: Ana Falcao

aws_lambda_powertools/utilities/parser/parser.py

@@ -202,4 +202,4 @@ def handler(event: Order, context: LambdaContext):
             f"Error: {str(exc)}. Please ensure the Input model inherits from BaseModel,\n"
             "and your payload adheres to the specified Input model structure.\n"
             f"Model={model}",
-        )
+        ) from exc

tests/functional/parser/test_parser.py

@@ -4,6 +4,7 @@
 
 import pydantic
 import pytest
+from typing_extensions import Annotated
 from pydantic import ValidationError, BaseModel
 
 from aws_lambda_powertools.utilities.parser import event_parser, exceptions, parse
@@ -251,19 +252,18 @@ class SuccessfulCallback(pydantic.BaseModel):
     class FailedCallback(pydantic.BaseModel):
         status: Literal["failed"]
         error: str
+        
+    DogCallback = Annotated[Union[SuccessfulCallback, FailedCallback], pydantic.Field(discriminator="status")]
+    DogCallbackTypeAdapter = pydantic.TypeAdapter(DogCallback)
 
-    class DogCallbackModel(pydantic.BaseModel):
-        data: Union[SuccessfulCallback, FailedCallback] = pydantic.Field(discriminator="status")
-
-    @event_parser(model=DogCallbackModel)
+    @event_parser(model=DogCallbackTypeAdapter)
     def handler(event, _: Any) -> str:
-        if isinstance(event.data, FailedCallback):
-            return f"Uh oh. Had a problem: {event.data.error}"
+        if isinstance(event, FailedCallback):
+            return f"Uh oh. Had a problem: {event.error}"
 
-        return f"Successfully retrieved {event.data.breed} named {event.data.name}"
+        return f"Successfully retrieved {event.breed} named {event.name}"
 
-    wrapped_input = {"data": test_input}
-    ret = handler(wrapped_input, None)
+    ret = handler(test_input, None)
     assert ret == expected