Update comments

Author: lym953

datadog_lambda/durable.py

@@ -41,7 +41,6 @@ def extract_durable_function_tags(event):
     if not parsed:
         logger.error("Failed to parse DurableExecutionArn: %s", durable_execution_arn)
         return {}
-    print(f"event: {event}")
     execution_name, execution_id = parsed
     # Use the number of operations to determine if it's the first invocation. This is
     # what the durable execution SDK does to determine the replay status.
@@ -50,5 +49,5 @@ def extract_durable_function_tags(event):
     return {
         "durable_function_execution_name": execution_name,
         "durable_function_execution_id": execution_id,
-        "durable_function_first_invocation": str(is_first_invocation).lower(),
+        "aws.lambda.durable_function.first_invocation": str(is_first_invocation).lower(),
     }

tests/test_durable.py

@@ -46,7 +46,7 @@ def test_works_with_numeric_version_qualifier(self):
 
 class TestExtractDurableFunctionTags(unittest.TestCase):
     def test_sets_first_invocation_true_when_only_execution_operation(self):
-        # One operation → not replaying → first invocation
+        # One operation (the current EXECUTION operation itself) → not replaying → first invocation
         event = {
             "DurableExecutionArn": "arn:aws:lambda:us-east-1:123456789012:function:my-func:1/durable-execution/my-execution/550e8400-e29b-41d4-a716-446655440004",
             "CheckpointToken": "some-token",
@@ -58,7 +58,7 @@ def test_sets_first_invocation_true_when_only_execution_operation(self):
             {
                 "durable_function_execution_name": "my-execution",
                 "durable_function_execution_id": "550e8400-e29b-41d4-a716-446655440004",
-                "durable_function_first_invocation": "true",
+                "aws.lambda.durable_function.first_invocation": "true",
             },
         )
 
@@ -80,7 +80,7 @@ def test_sets_first_invocation_false_when_multiple_operations(self):
             {
                 "durable_function_execution_name": "my-execution",
                 "durable_function_execution_id": "550e8400-e29b-41d4-a716-446655440004",
-                "durable_function_first_invocation": "false",
+                "aws.lambda.durable_function.first_invocation": "false",
             },
         )