mirror of
https://github.com/invariantlabs-ai/invariant-gateway.git
synced 2026-05-28 09:21:27 +02:00
Luca Beurer-Kellner
e18c6b5bdb
* use select() before readline * support for setting static metadata for MCP sessions * nest extra mcp metadata in metadata object * unify session metadata * extra metadata tests * use empty object as parameters, if None * list_tools as tool call * offset indices in tests * test: adjust addresses * mcp: make error reporting configurable * line logging * log version * verbose logging + loud exception failure * add server and client name to policy get * append trace even if not pushing * port tools/list message support to SSE * use python -m build * adjust guardrail failure address * support for blocking tools/list in SSE * use error-based failure response format by default * tools/list test * don't list_tools in stdio connect * flaky test: handle second possible result in anthropic streaming case --------- Co-authored-by: knielsen404 <kristian@invariantlabs.ai>
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
"""Common constants used in the gateway."""
|
|
|
|
IGNORED_HEADERS = [
|
|
"accept-encoding",
|
|
"host",
|
|
"invariant-authorization",
|
|
"x-forwarded-for",
|
|
"x-forwarded-host",
|
|
"x-forwarded-port",
|
|
"x-forwarded-proto",
|
|
"x-forwarded-server",
|
|
"x-real-ip",
|
|
]
|
|
|
|
CLIENT_TIMEOUT = 60.0
|
|
|
|
# MCP related constants
|
|
MCP_METHOD = "method"
|
|
MCP_TOOL_CALL = "tools/call"
|
|
MCP_LIST_TOOLS = "tools/list"
|
|
MCP_PARAMS = "params"
|
|
MCP_RESULT = "result"
|
|
MCP_SERVER_INFO = "serverInfo"
|
|
MCP_CLIENT_INFO = "clientInfo"
|
|
INVARIANT_GUARDRAILS_BLOCKED_MESSAGE = """
|
|
[Invariant Guardrails] The MCP tool call was blocked for security reasons.
|
|
Do not attempt to circumvent this block, rather explain to the user based
|
|
on the following output what went wrong: %s
|
|
"""
|
|
INVARIANT_GUARDRAILS_BLOCKED_TOOLS_MESSAGE = """
|
|
[Invariant Guardrails] This server was blocked from advertising its tools due to a security guardrail failure.
|
|
The operation was blocked by Invariant Guardrails (mention this in your user report).
|
|
When users ask about this tool, inform them that it was blocked due to a security guardrail failure.
|
|
%s
|
|
""" |