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

append trace even if not pushing

Browse Source
This commit is contained in:
Luca Beurer-Kellner
2025-05-16 14:05:23 +02:00
parent d57e3f8740
commit f05731b55f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gateway/mcp/mcp.py
+4 -1
View File
@@ -305,7 +305,6 @@ async def hook_tool_call(ctx: McpContext, request: dict) -> tuple[dict, bool]:
and check_if_new_errors(ctx, guardrailing_result)
):
if ctx.push_explorer:
await append_and_push_trace(ctx, message, guardrailing_result)
return json_rpc_error_response(
@@ -356,6 +355,8 @@ async def hook_tool_result(ctx: McpContext, result: dict) -> dict:
if ctx.push_explorer:
await append_and_push_trace(ctx, message, guardrailing_result)
else:
ctx.trace.append(message)
return result
elif method == MCP_LIST_TOOLS:
@@ -392,6 +393,8 @@ async def hook_tool_result(ctx: McpContext, result: dict) -> dict:
# add it to the session trace (and run logging guardrails)
if ctx.push_explorer:
await append_and_push_trace(ctx, message, guardrailing_result)
else:
ctx.trace.append(message)
return result
else: