Skip to content

Commit 0ee66f0

Browse files
committed
fix: Add missing token propagation for execute tool
1 parent 86355f6 commit 0ee66f0

File tree

1 file changed

+5
-0
lines changed
  • crates/apollo-mcp-server/src/server/states

1 file changed

+5
-0
lines changed

crates/apollo-mcp-server/src/server/states/running.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,6 +210,11 @@ impl ServerHandler for Running {
210210
EXECUTE_TOOL_NAME => {
211211
let mut headers = self.headers.clone();
212212
if let Some(axum_parts) = context.extensions.get::<axum::http::request::Parts>() {
213+
// Optionally extract the validated token and propagate it to upstream servers if present
214+
if let Some(token) = axum_parts.extensions.get::<ValidToken>() {
215+
headers.typed_insert(token.deref().clone());
216+
}
217+
213218
// Forward the mcp-session-id header if present
214219
if let Some(session_id) = axum_parts.headers.get("mcp-session-id") {
215220
headers.insert("mcp-session-id", session_id.clone());

0 commit comments

Comments
 (0)