From adf44f339a576d979f71d00281195f744a96da78 Mon Sep 17 00:00:00 2001
From: claude-bot-go <bot@claude-bot.local>
Date: Wed, 22 Jul 2026 09:24:04 +0800
Subject: [PATCH] Fix #273: [milestone Milestone 2 ] Update
 `on_http_response_headers()` in `crates/proxy-wasm-evidence/src/filter.rs...

---
 crates/proxy-wasm-evidence/src/filter.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/crates/proxy-wasm-evidence/src/filter.rs b/crates/proxy-wasm-evidence/src/filter.rs
index b24e4f8..35d674a 100644
--- a/crates/proxy-wasm-evidence/src/filter.rs
+++ b/crates/proxy-wasm-evidence/src/filter.rs
@@ -156,6 +156,9 @@ impl HttpContext for EvidenceFilter {
             RecordingMode::Full => self.metric_full,
         };
         let _ = increment_metric(metric_id, 1);
+        if let Some(ref risk_str) = evidence.mcp_header_risk {
+            self.set_http_response_header("x-aep-mcp-header-risk", Some(risk_str));
+        }
         let _ = self.evidence_buffer.push(evidence);
         Action::Continue
     }