From 00dafbe5353eab70f08bdd763aa300ddc1da0a54 Mon Sep 17 00:00:00 2001
From: Aliyan Ishfaq <aliyanishfaq200@gmail.com>
Date: Thu, 24 Jul 2025 08:59:45 -0700
Subject: [PATCH] fix: checks for doc cache on mcp tool call (#514)

---
 apps/open-swe/src/utils/tool-output-processing.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/apps/open-swe/src/utils/tool-output-processing.ts b/apps/open-swe/src/utils/tool-output-processing.ts
index 2ddceff8..e5858bbb 100644
--- a/apps/open-swe/src/utils/tool-output-processing.ts
+++ b/apps/open-swe/src/utils/tool-output-processing.ts
@@ -38,6 +38,14 @@ export async function processToolCallContent(
     const url = toolCall.args?.url || toolCall.args?.uri || toolCall.args?.path;
     const parsedResult = typeof url === "string" ? parseUrl(url) : null;
     const parsedUrl = parsedResult?.success ? parsedResult.url.href : undefined;
+
+    // avoid generating TOC again if it's already in the cache
+    if (parsedUrl && state.documentCache[parsedUrl]) {
+      return {
+        content: state.documentCache[parsedUrl],
+      };
+    }
+
     const processedContent = await handleMcpDocumentationOutput(
       result,
       config,