refactor(core): Centralize tool response formatting (#743)

1 files changed, 8 insertions, 1 deletions

diff --git a/packages/core/src/tools/mcp-tool.ts b/packages/core/src/tools/mcp-tool.ts
index 65c0cae8..8a7694d8 100644
--- a/packages/core/src/tools/mcp-tool.ts
+++ b/packages/core/src/tools/mcp-tool.ts
@@ -149,6 +149,13 @@ function getStringifiedResultForDisplay(result: Part[]) {
     return part; // Fallback for unexpected structure or non-FunctionResponsePart
   };
 
-  const processedResults = result.map(processFunctionResponse);
+  const processedResults =
+    result.length === 1
+      ? processFunctionResponse(result[0])
+      : result.map(processFunctionResponse);
+  if (typeof processedResults === 'string') {
+    return processedResults;
+  }
+
   return '```json\n' + JSON.stringify(processedResults, null, 2) + '\n```';
 }