fix(server.py): allow user to specify custom headers to forward to mcp server

This commit is contained in:
Krrish Dholakia
2025-09-27 17:55:04 -07:00
parent 1c71d4cdbb
commit c6a7e676f6
2 changed files with 8 additions and 0 deletions
@@ -414,6 +414,13 @@ if MCP_AVAILABLE:
if server.auth_type == MCPAuth.oauth2:
extra_headers = oauth2_headers
if server.forwardable_headers and raw_headers:
if extra_headers is None:
extra_headers = {}
for header in server.forwardable_headers:
if header in raw_headers:
extra_headers[header] = raw_headers[header]
# Fall back to deprecated mcp_auth_header if no server-specific header found
if server_auth_header is None:
server_auth_header = mcp_auth_header
+1
View File
@@ -25,6 +25,7 @@ mcp_servers:
client_id: os.environ/GITHUB_OAUTH_CLIENT_ID
client_secret: os.environ/GITHUB_OAUTH_CLIENT_SECRET
scopes: ["public_repo", "user:email"]
forwardable_headers: ["custom_key"]
# allowed_tools: ["list_tools"]
# disallowed_tools: ["repo_delete"]