From 80a445c7ae691798f090591b9fd6539a6ac30e9e Mon Sep 17 00:00:00 2001
From: Olcan <olcans@gmail.com>
Date: Tue, 3 Jun 2025 14:40:36 -0700
Subject: bandaid for broken mcp calls (#732)

---
 packages/core/src/core/coreToolScheduler.ts | 4 ++++
 packages/core/src/tools/mcp-tool.ts         | 3 +--
 2 files changed, 5 insertions(+), 2 deletions(-)

(limited to 'packages/core/src')

diff --git a/packages/core/src/core/coreToolScheduler.ts b/packages/core/src/core/coreToolScheduler.ts
index 3fe6562e..c3735366 100644
--- a/packages/core/src/core/coreToolScheduler.ts
+++ b/packages/core/src/core/coreToolScheduler.ts
@@ -128,6 +128,10 @@ export function formatLlmContentForFunctionResponse(
     additionalParts.push(contentToProcess);
   } else if (contentToProcess.text !== undefined) {
     functionResponseJson = { output: contentToProcess.text };
+  } else if (contentToProcess.functionResponse) {
+    functionResponseJson = JSON.parse(
+      JSON.stringify(contentToProcess.functionResponse),
+    );
   } else {
     functionResponseJson = { status: 'Tool execution succeeded.' };
     additionalParts.push(contentToProcess);
diff --git a/packages/core/src/tools/mcp-tool.ts b/packages/core/src/tools/mcp-tool.ts
index 819dc48d..65c0cae8 100644
--- a/packages/core/src/tools/mcp-tool.ts
+++ b/packages/core/src/tools/mcp-tool.ts
@@ -96,10 +96,9 @@ This MCP tool named '${serverToolName}' was discovered from '${serverName}' MCP
 
     const responseParts: Part[] = await this.mcpTool.callTool(functionCalls);
 
-    const output = getStringifiedResultForDisplay(responseParts);
     return {
       llmContent: responseParts,
-      returnDisplay: output,
+      returnDisplay: getStringifiedResultForDisplay(responseParts),
     };
   }
 }
-- 
cgit v1.2.3