From 9d89c01b97b5b0ed36808f830bb6dacd89152caf Mon Sep 17 00:00:00 2001
From: timothycarambat <rambat1010@gmail.com>
Date: Thu, 17 Apr 2025 11:27:21 -0700
Subject: [PATCH] fix MCP missing refs on reload

---
 frontend/src/pages/Admin/Agents/MCPServers/index.jsx |  7 ++++++-
 frontend/src/pages/Admin/Agents/index.jsx            | 10 ++++++++--
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/frontend/src/pages/Admin/Agents/MCPServers/index.jsx b/frontend/src/pages/Admin/Agents/MCPServers/index.jsx
index e10df55c..7c6b4445 100644
--- a/frontend/src/pages/Admin/Agents/MCPServers/index.jsx
+++ b/frontend/src/pages/Admin/Agents/MCPServers/index.jsx
@@ -3,8 +3,13 @@ import { titleCase } from "text-case";
 import { BookOpenText, ArrowClockwise } from "@phosphor-icons/react";
 import MCPLogo from "@/media/agents/mcp-logo.svg";
 import MCPServers from "@/models/mcpServers";
+import showToast from "@/utils/toast";
 
-export function MCPServerHeader({ setMcpServers, children }) {
+export function MCPServerHeader({
+  setMcpServers,
+  setSelectedMcpServer,
+  children,
+}) {
   const [loadingMcpServers, setLoadingMcpServers] = useState(false);
   useEffect(() => {
     async function fetchMCPServers() {
diff --git a/frontend/src/pages/Admin/Agents/index.jsx b/frontend/src/pages/Admin/Agents/index.jsx
index 69781dd3..a322be5a 100644
--- a/frontend/src/pages/Admin/Agents/index.jsx
+++ b/frontend/src/pages/Admin/Agents/index.jsx
@@ -315,7 +315,10 @@ export default function AdminAgents() {
               id="active_agent_flows"
               value={activeFlowIds.join(",")}
             />
-            <MCPServerHeader setMcpServers={setMcpServers}>
+            <MCPServerHeader
+              setMcpServers={setMcpServers}
+              setSelectedMcpServer={setSelectedMcpServer}
+            >
               {({ loadingMcpServers }) => {
                 return (
                   <MCPServersList
@@ -519,7 +522,10 @@ export default function AdminAgents() {
                 handleClick={handleFlowClick}
               />
 
-              <MCPServerHeader setMcpServers={setMcpServers}>
+              <MCPServerHeader
+                setMcpServers={setMcpServers}
+                setSelectedMcpServer={setSelectedMcpServer}
+              >
                 {({ loadingMcpServers }) => {
                   return (
                     <MCPServersList