Updated xray sdk version

Author: webarch-ai

lib/idp_common_pkg/pyproject.toml

@@ -101,7 +101,7 @@ agents = [
 # This includes all dependencies needed for both backends
 docs_service = [
   "requests==2.32.4", # Required for appsync module (dynamodb only needs boto3 which is core)
-  "aws-xray-sdk>=2.12.0", # Required for X-Ray tracing
+  "aws-xray-sdk>=2.14.0", # Required for X-Ray tracing
 ]
 
 # Testing dependencies

lib/idp_common_pkg/setup.py

@@ -61,7 +61,7 @@
     # Document service factory dependencies (includes both appsync and dynamodb support)
     "docs_service": [
         "requests==2.32.4",
-        "aws-xray-sdk>=2.12.0",
+        "aws-xray-sdk>=2.14.0",
     ],
     # Testing dependencies
     "test": [

src/lambda/queue_processor/index.py

@@ -142,12 +142,13 @@ def process_message(record: Dict[str, Any]) -> Tuple[bool, str]:
         object_key = document.input_key
         logger.info(f"Processing message {message_id} for object {object_key}")
 
-        # Add X-Ray annotations and capture trace_id early
-        xray_recorder.put_annotation('document_id', document.id)
-        current_segment = xray_recorder.current_segment()
-        if current_segment:
-            document.trace_id = current_segment.trace_id
-            logger.info(f"Updated {document.id} trace_id: {document.trace_id}")
+        # X-Ray annotations
+        xray_recorder.put_annotation('document_id', {document.id})
+        xray_recorder.put_annotation('processing_stage', 'queue_processor')
+        # current_segment = xray_recorder.current_segment()
+        # if current_segment:
+        #     document.trace_id = current_segment.trace_id
+        #     logger.info(f"Updated {document.id} trace_id: {document.trace_id}")
 
         # Try to increment counter
         if not update_counter(increment=True):