mirror of
https://github.com/invariantlabs-ai/invariant-gateway.git
synced 2026-05-25 00:04:01 +02:00
nest extra mcp metadata in metadata object
This commit is contained in:
@@ -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
@@ -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 {})
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@@ -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}"
|
||||
|
||||
Reference in New Issue
Block a user