From 2715c69b85a5f0ae667d303c0d61f2c385bfa7fe Mon Sep 17 00:00:00 2001
From: Kaushik <davidkgp@gmail.com>
Date: Fri, 12 Sep 2025 23:41:19 +0200
Subject: [PATCH] Update MCPDiscoveryNode.ts for fixing the issue
 https://github.com/Ironclad/rivet/issues/523

---
 packages/core/src/model/nodes/MCPDiscoveryNode.ts | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/packages/core/src/model/nodes/MCPDiscoveryNode.ts b/packages/core/src/model/nodes/MCPDiscoveryNode.ts
index 0c746fd96..040433d69 100644
--- a/packages/core/src/model/nodes/MCPDiscoveryNode.ts
+++ b/packages/core/src/model/nodes/MCPDiscoveryNode.ts
@@ -195,8 +195,9 @@ class MCPDiscoveryNodeImpl extends NodeImpl<MCPDiscoveryNode> {
           );
         }
 
-        tools = await context.mcpProvider.getHTTPTools({ name, version }, serverUrl);
-        prompts = await context.mcpProvider.getHTTPrompts({ name, version }, serverUrl);
+        tools = this.data.useToolsOutput ? await context.mcpProvider.getHTTPTools({ name, version }, serverUrl) : [];
+        prompts = this.data.usePromptsOutput ? await context.mcpProvider.getHTTPPrompts({ name, version }, serverUrl) : [];
+        
       } else if (transportType === 'stdio') {
         const serverId = this.data.serverId ?? '';
 
@@ -210,8 +211,8 @@ class MCPDiscoveryNodeImpl extends NodeImpl<MCPDiscoveryNode> {
           serverId,
         };
 
-        tools = await context.mcpProvider.getStdioTools({ name, version }, serverConfig);
-        prompts = await context.mcpProvider.getStdioPrompts({ name, version }, serverConfig);
+        tools = this.data.useToolsOutput ? await context.mcpProvider.getStdioTools({ name, version }, serverConfig) : [];
+        prompts = this.data.usePromptsOutput ? await context.mcpProvider.getStdioPrompts({ name, version }, serverConfig) : [];
       }
 
       const output: Outputs = {};