nest extra mcp metadata in metadata object

This commit is contained in:
Luca Beurer-Kellner
2025-05-06 16:20:58 +02:00
parent 6ca07a5802
commit 9cf6aa2607
3 changed files with 9 additions and 7 deletions
+1 -1
View File
@@ -26,7 +26,7 @@ class RequestContext:
guardrails: Optional[GuardrailRuleSet] = None
config: Dict[str, Any] = None
# extra headers to send to invariant services (Explorer, Guardrails, etc.)
# extra parameters available as input.<key> during guardrail evaluation
guardrails_parameters: Optional[Dict[str, Any]] = None
_created_via_factory: bool = field(
+7 -5
View File
@@ -178,11 +178,13 @@ def get_guardrails_check_result(
invariant_authorization="Bearer " + os.getenv("INVARIANT_API_KEY"),
guardrails=ctx.guardrails,
guardrails_parameters={
"session_id": ctx.local_session_id,
"system_user": user_and_host(),
"mcp_client": ctx.mcp_client_name,
"mcp_server": ctx.mcp_server_name,
**(ctx.extra_metadata or {})
"metadata": {
"session_id": ctx.local_session_id,
"system_user": user_and_host(),
"mcp_client": ctx.mcp_client_name,
"mcp_server": ctx.mcp_server_name,
**(ctx.extra_metadata or {})
}
}
)
+1 -1
View File
@@ -39,7 +39,7 @@ class McpContext:
blocking_guardrails=[], logging_guardrails=[]
)
# parsed from CLI
# parsed from CLI (all --metadata-* args)
self.extra_metadata: Dict[str, str] = {}
for arg in extra_args:
assert "=" in arg, f"Invalid extra metadata argument: {arg}"