CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-08-16 14:00:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update metadata in MCP SSE similar to what we do in MCP stdio.

Browse Source
This commit is contained in:
Hemang
2025-05-21 22:45:59 +02:00
committed by Hemang Sarkar
parent 177d247a83
commit 03817b005c
3 changed files with 35 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/mcp/test_mcp.py
+7 -1
View File
@@ -174,6 +174,8 @@ async def test_mcp_with_gateway_and_logging_guardrails(
and metadata["mcp_client"] == "mcp"
and metadata["mcp_server"] == "messenger_server"
)
assert "session_id" in metadata
assert "system_user" in metadata
assert trace["messages"][2]["role"] == "assistant"
assert trace["messages"][2]["tool_calls"][0]["function"] == {
"name": "get_last_message_from_user",
@@ -287,6 +289,8 @@ async def test_mcp_with_gateway_and_blocking_guardrails(
and metadata["mcp_client"] == "mcp"
and metadata["mcp_server"] == "messenger_server"
)
assert "session_id" in metadata
assert "system_user" in metadata
assert trace["messages"][2]["role"] == "assistant"
assert trace["messages"][2]["tool_calls"][0]["function"] == {
"name": "get_last_message_from_user",
@@ -388,6 +392,8 @@ async def test_mcp_with_gateway_hybrid_guardrails(
and metadata["mcp_client"] == "mcp"
and metadata["mcp_server"] == "messenger_server"
)
assert "session_id" in metadata
assert "system_user" in metadata
assert trace["messages"][2]["role"] == "assistant"
assert trace["messages"][2]["tool_calls"][0]["function"] == {
"name": "get_last_message_from_user",
@@ -475,4 +481,4 @@ async def test_mcp_tool_list_blocking(
assert "blocked_get_last_message_from_user" in str(tools_result), (
"Expected the tool names to be renamed and blocked because of the blocking guardrail on the tools/list call. Instead got: "
+ str(tools_result)
)
)