refactor: address feedback

Author: leandrodamascena

src/aws_durable_execution_sdk_python/execution.py

@@ -224,11 +224,10 @@ def wrapper(event: Any, context: LambdaContext) -> MutableMapping[str, Any]:
                 invocation_input = DurableExecutionInvocationInput.from_dict(event)
             except (KeyError, TypeError, AttributeError) as e:
                 msg = (
-                    "The function is not being invoked as a durable function. "
-                    "Please check the function configuration to ensure durability is turned on, "
-                    "or use the testing SDK for local testing."
+                    "The payload is not the correct Durable Function input. "
+                    "Please set DurableConfig on the AWS Lambda to invoke it as a Durable Function."
                 )
-                raise ValueError(msg) from e
+                raise ExecutionError(msg) from e
 
             # Local runner always uses its own client, otherwise use custom or default
             if invocation_input.is_local_runner:

tests/execution_test.py

@@ -2036,8 +2036,8 @@ def test_handler(event: Any, context: DurableContext) -> dict:
     # WHEN the handler is invoked with a non-durable payload
     # THEN it raises a ValueError with a helpful message
     with pytest.raises(
-        ValueError,
-        match="The function is not being invoked as a durable function",
+        ExecutionError,
+        match="The payload is not the correct Durable Function input",
     ):
         test_handler(regular_event, lambda_context)
 
@@ -2064,7 +2064,7 @@ def test_handler(event: Any, context: DurableContext) -> dict:
     # WHEN the handler is invoked with a non-dict event
     # THEN it raises a ValueError with a helpful message
     with pytest.raises(
-        ValueError,
-        match="The function is not being invoked as a durable function",
+        ExecutionError,
+        match="The payload is not the correct Durable Function input",
     ):
         test_handler(non_dict_event, lambda_context)