From a214837b1e955e7432c8b26d61dc0a96de2b9d3f Mon Sep 17 00:00:00 2001
From: Hemang <hemang@invariantlabs.ai>
Date: Wed, 14 May 2025 15:44:52 +0200
Subject: [PATCH] Add message to session store for MCP sse in post path before
 returning.

---
 gateway/routes/mcp_sse.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/gateway/routes/mcp_sse.py b/gateway/routes/mcp_sse.py
index 677e342..2411ad7 100644
--- a/gateway/routes/mcp_sse.py
+++ b/gateway/routes/mcp_sse.py
@@ -333,9 +333,7 @@ async def _hook_tool_call(session_id: str, request_json: dict) -> Tuple[dict, bo
             },
         }, True
     # Push trace to the explorer - don't block on its response
-    asyncio.create_task(
-        session_store.add_message_to_session(session_id, message, guardrails_result)
-    )
+    await session_store.add_message_to_session(session_id, message, guardrails_result)
     return request_json, False