Updated xray trace id within queue processor

Author: webarch-ai

src/lambda/queue_processor/index.py

@@ -145,10 +145,10 @@ def process_message(record: Dict[str, Any]) -> Tuple[bool, str]:
         # 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}")
+        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):