From def847af5fc5f89535a93066f06223772a85c011 Mon Sep 17 00:00:00 2001
From: MaheshtheDev <38828053+MaheshtheDev@users.noreply.github.com>
Date: Wed, 17 Jun 2026 17:16:22 +0000
Subject: [PATCH] Revert "fix(mcp): clarify memory graph pagination (#1032)"
 (#1133)

This reverts commit feca81f69e8db94b11f64c673e3b546b23907732.
---
 apps/mcp/mcp-app.html             |   2 -
 apps/mcp/src/memory-graph-data.ts |  48 ------------
 apps/mcp/src/server.ts            |  42 +++++-----
 apps/mcp/src/ui/mcp-app.css       |  26 -------
 apps/mcp/src/ui/mcp-app.ts        | 123 ++++--------------------------
 5 files changed, 36 insertions(+), 205 deletions(-)
 delete mode 100644 apps/mcp/src/memory-graph-data.ts

diff --git a/apps/mcp/mcp-app.html b/apps/mcp/mcp-app.html
index 07ae39d6..1c1f179f 100644
--- a/apps/mcp/mcp-app.html
+++ b/apps/mcp/mcp-app.html
@@ -16,8 +16,6 @@
 
   <div id="stats"></div>
 
-  <button type="button" id="load-more-btn" hidden>Load more</button>
-
   <div id="popup">
     <span id="popup-type"></span>
     <div id="popup-title"></div>
diff --git a/apps/mcp/src/memory-graph-data.ts b/apps/mcp/src/memory-graph-data.ts
deleted file mode 100644
index e7e966c1..00000000
--- a/apps/mcp/src/memory-graph-data.ts
+++ /dev/null
@@ -1,48 +0,0 @@
-import type { DocumentWithMemories, DocumentsApiResponse } from "./client"
-
-export const GRAPH_PAGE_LIMIT = 10
-
-export interface MemoryGraphStructuredContent {
-	containerTag?: string
-	documents: DocumentWithMemories[]
-	totalCount: number
-	loadedCount: number
-	pagination: DocumentsApiResponse["pagination"]
-}
-
-export function createMemoryGraphStructuredContent(
-	result: DocumentsApiResponse,
-	containerTag?: string,
-): MemoryGraphStructuredContent {
-	return {
-		containerTag,
-		documents: result.documents,
-		totalCount: result.pagination.totalItems,
-		loadedCount: result.documents.length,
-		pagination: result.pagination,
-	}
-}
-
-export function summarizeMemoryGraphResult(
-	result: DocumentsApiResponse,
-	containerTag?: string,
-): string {
-	let memoryCount = 0
-	for (const document of result.documents) {
-		memoryCount += document.memoryEntries.length
-	}
-
-	const textParts = [
-		`Memory Graph: showing ${result.documents.length} of ${result.pagination.totalItems} documents, ${memoryCount} loaded memories`,
-	]
-	if (result.pagination.totalPages > result.pagination.currentPage) {
-		textParts.push(
-			`More documents available (${result.pagination.totalPages} pages total)`,
-		)
-	}
-	if (containerTag) {
-		textParts.push(`Project: ${containerTag}`)
-	}
-
-	return textParts.join(". ")
-}
diff --git a/apps/mcp/src/server.ts b/apps/mcp/src/server.ts
index b758b892..de54deff 100644
--- a/apps/mcp/src/server.ts
+++ b/apps/mcp/src/server.ts
@@ -5,12 +5,7 @@ import {
 	registerAppResource,
 	RESOURCE_MIME_TYPE,
 } from "@modelcontextprotocol/ext-apps/server"
-import { SupermemoryClient, getMemoryText } from "./client"
-import {
-	GRAPH_PAGE_LIMIT,
-	createMemoryGraphStructuredContent,
-	summarizeMemoryGraphResult,
-} from "./memory-graph-data"
+import { SupermemoryClient } from "./client"
 import { formatMemories } from "./format"
 import { initPosthog, posthog } from "./posthog"
 import { z } from "zod"
@@ -326,23 +321,26 @@ export class SupermemoryMCP extends McpAgent<Env, unknown, Props> {
 						? [effectiveContainerTag]
 						: undefined
 
-					const result = await client.getDocuments(
-						containerTags,
-						1,
-						GRAPH_PAGE_LIMIT,
+					const result = await client.getDocuments(containerTags, 1, 10)
+
+					const memoryCount = result.documents.reduce(
+						(sum, d) => sum + d.memoryEntries.length,
+						0,
 					)
+					const textParts = [
+						`Memory Graph: ${result.documents.length} documents, ${memoryCount} memories`,
+					]
+					if (effectiveContainerTag) {
+						textParts.push(`Project: ${effectiveContainerTag}`)
+					}
 
 					return {
-						content: [
-							{
-								type: "text" as const,
-								text: summarizeMemoryGraphResult(result, effectiveContainerTag),
-							},
-						],
-						structuredContent: createMemoryGraphStructuredContent(
-							result,
-							effectiveContainerTag,
-						),
+						content: [{ type: "text" as const, text: textParts.join(". ") }],
+						structuredContent: {
+							containerTag: effectiveContainerTag,
+							documents: result.documents,
+							totalCount: result.pagination.totalItems,
+						},
 					}
 				} catch (error) {
 					const message =
@@ -371,7 +369,7 @@ export class SupermemoryMCP extends McpAgent<Env, unknown, Props> {
 				inputSchema: z.object({
 					containerTag: z.string().optional(),
 					page: z.number().optional().default(1),
-					limit: z.number().optional().default(GRAPH_PAGE_LIMIT),
+					limit: z.number().optional().default(10),
 				}),
 				_meta: {
 					ui: {
@@ -696,7 +694,7 @@ export class SupermemoryMCP extends McpAgent<Env, unknown, Props> {
 					mcp_client_name: clientInfo?.name,
 					mcp_client_version: clientInfo?.version,
 					sessionId: this.getMcpSessionId(),
-					containerTag: effectiveContainerTag,
+					containerTag: containerTag || this.props?.containerTag,
 				})
 				.catch((error) => console.error("PostHog tracking error:", error))
 
diff --git a/apps/mcp/src/ui/mcp-app.css b/apps/mcp/src/ui/mcp-app.css
index 5d71fa0f..65ea3de0 100644
--- a/apps/mcp/src/ui/mcp-app.css
+++ b/apps/mcp/src/ui/mcp-app.css
@@ -57,32 +57,6 @@
 		0 2px 4px -2px rgba(0, 0, 0, 0.1);
 }
 
-#load-more-btn {
-	position: fixed;
-	top: 12px;
-	right: 12px;
-	z-index: 10;
-	padding: 6px 12px;
-	border: 1px solid var(--border);
-	border-radius: 10px;
-	background: var(--bg-secondary);
-	color: var(--text-muted);
-	font-size: 12px;
-	cursor: pointer;
-	box-shadow:
-		0 4px 6px -1px rgba(0, 0, 0, 0.1),
-		0 2px 4px -2px rgba(0, 0, 0, 0.1);
-}
-
-#load-more-btn:hover:not(:disabled) {
-	opacity: 0.85;
-}
-
-#load-more-btn:disabled {
-	cursor: wait;
-	opacity: 0.7;
-}
-
 /* Popup */
 #popup {
 	display: none;
diff --git a/apps/mcp/src/ui/mcp-app.ts b/apps/mcp/src/ui/mcp-app.ts
index f6ffdddd..4c0c7d34 100644
--- a/apps/mcp/src/ui/mcp-app.ts
+++ b/apps/mcp/src/ui/mcp-app.ts
@@ -55,13 +55,6 @@ interface ToolResultData {
 	containerTag?: string
 	documents: GraphApiDocument[]
 	totalCount: number
-	loadedCount?: number
-	pagination?: {
-		currentPage: number
-		limit: number
-		totalItems: number
-		totalPages: number
-	}
 }
 
 interface MemoryNode extends NodeObject {
@@ -136,8 +129,6 @@ const CLUSTER_SPREAD = 120
 let isDark = true
 let selectedNode: GraphNode | null = null
 let hoveredNode: GraphNode | null = null
-let activeToolData: ToolResultData | null = null
-let isLoadingMore = false
 
 // =============================================================================
 // DOM References (elements are guaranteed to exist in mcp-app.html)
@@ -164,9 +155,6 @@ const zoomInBtn = document.getElementById("zoom-in")!
 const zoomOutBtn = document.getElementById("zoom-out")!
 // biome-ignore lint/style/noNonNullAssertion: DOM element guaranteed to exist in HTML
 const fitBtn = document.getElementById("fit-btn")!
-const loadMoreBtn = document.getElementById(
-	"load-more-btn",
-) as HTMLButtonElement
 
 // =============================================================================
 // Helpers
@@ -869,99 +857,6 @@ prefersDark.addEventListener("change", (e) =>
 	applyTheme(e.matches ? "dark" : "light"),
 )
 
-function updateLoadMore(data: ToolResultData) {
-	const pagination = data.pagination
-	const loadedCount = data.loadedCount ?? data.documents.length
-	const hasMore = pagination
-		? pagination.currentPage < pagination.totalPages
-		: loadedCount < data.totalCount
-
-	loadMoreBtn.hidden = !hasMore
-	loadMoreBtn.disabled = isLoadingMore
-	if (hasMore) {
-		loadMoreBtn.textContent = isLoadingMore
-			? "Loading..."
-			: `Load more (${loadedCount}/${data.totalCount})`
-	}
-}
-
-function renderGraphData(data: ToolResultData) {
-	const { nodes, links } = transformData(data)
-	const memCount = nodes.filter((n) => n.nodeType === "memory").length
-	const docCount = nodes.filter((n) => n.nodeType === "document").length
-	const loadedCount = data.loadedCount ?? data.documents.length
-	const totalCount = data.totalCount
-	const countLabel =
-		totalCount > loadedCount
-			? `${docCount}/${totalCount} docs`
-			: `${docCount} docs`
-
-	statsEl.textContent = `${countLabel} · ${memCount} memories · ${links.length} connections`
-
-	// Update legend counts
-	const docCountEl = document.getElementById("legend-doc-count")
-	const memCountEl = document.getElementById("legend-mem-count")
-	if (docCountEl) docCountEl.textContent = String(docCount)
-	if (memCountEl) memCountEl.textContent = String(memCount)
-
-	graph.graphData({ nodes, links })
-	updateLoadMore(data)
-
-	// Fit to view after layout stabilizes
-	setTimeout(() => graph.zoomToFit(400, 40), 600)
-}
-
-async function loadNextPage() {
-	const data = activeToolData
-	const pagination = data?.pagination
-	if (!data || !pagination || pagination.currentPage >= pagination.totalPages) {
-		return
-	}
-
-	isLoadingMore = true
-	updateLoadMore(data)
-	try {
-		const result = await app.callServerTool({
-			name: "fetch-graph-data",
-			arguments: {
-				containerTag: data.containerTag,
-				page: pagination.currentPage + 1,
-				limit: pagination.limit,
-			},
-		})
-		if (result.isError) {
-			statsEl.textContent = "Error loading more graph data"
-			return
-		}
-		const nextData = result.structuredContent as unknown as {
-			documents?: GraphApiDocument[]
-			pagination?: ToolResultData["pagination"]
-		}
-		if (!nextData?.documents || !nextData.pagination) {
-			statsEl.textContent = "No additional graph data available"
-			return
-		}
-		activeToolData = {
-			...data,
-			documents: [...data.documents, ...nextData.documents],
-			loadedCount: data.documents.length + nextData.documents.length,
-			pagination: nextData.pagination,
-			totalCount: nextData.pagination.totalItems,
-		}
-		renderGraphData(activeToolData)
-	} catch (error) {
-		statsEl.textContent = "Error loading more graph data"
-		console.error(error)
-	} finally {
-		isLoadingMore = false
-		if (activeToolData) updateLoadMore(activeToolData)
-	}
-}
-
-loadMoreBtn.addEventListener("click", () => {
-	void loadNextPage()
-})
-
 // =============================================================================
 // MCP App SDK
 // =============================================================================
@@ -986,8 +881,22 @@ app.ontoolresult = (result: CallToolResult) => {
 		return
 	}
 
-	activeToolData = data
-	renderGraphData(data)
+	const { nodes, links } = transformData(data)
+	const memCount = nodes.filter((n) => n.nodeType === "memory").length
+	const docCount = nodes.filter((n) => n.nodeType === "document").length
+
+	statsEl.textContent = `${docCount} docs \u00b7 ${memCount} memories \u00b7 ${links.length} connections`
+
+	// Update legend counts
+	const docCountEl = document.getElementById("legend-doc-count")
+	const memCountEl = document.getElementById("legend-mem-count")
+	if (docCountEl) docCountEl.textContent = String(docCount)
+	if (memCountEl) memCountEl.textContent = String(memCount)
+
+	graph.graphData({ nodes, links })
+
+	// Fit to view after layout stabilizes
+	setTimeout(() => graph.zoomToFit(400, 40), 600)
 }
 
 app.ontoolcancelled = () => {