From 5b68e80be59e6a9e7155a16d1912ff51a017ce24 Mon Sep 17 00:00:00 2001
From: Hemang <hemang@invariantlabs.ai>
Date: Thu, 22 May 2025 09:36:18 +0200
Subject: [PATCH] Add MCP client and server name when calling
 fetch_guardrails_from_explorer in SSE.

---
 gateway/common/mcp_sessions_manager.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gateway/common/mcp_sessions_manager.py b/gateway/common/mcp_sessions_manager.py
index fa53a5a..fe24bc8 100644
--- a/gateway/common/mcp_sessions_manager.py
+++ b/gateway/common/mcp_sessions_manager.py
@@ -69,6 +69,9 @@ class McpSession(BaseModel):
         self.guardrails = await fetch_guardrails_from_explorer(
             self.explorer_dataset,
             "Bearer " + os.getenv("INVARIANT_API_KEY"),
+            # pylint: disable=no-member
+            self.metadata.get("mcp_client"),
+            self.metadata.get("mcp_server"),
         )
 
     def _deduplicate_annotations(self, new_annotations: list) -> list: