From 4a66bd5b9aa923438e4d7fade407cf2af36cff9f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gerg=C5=91=20M=C3=B3ricz?= <mo.geryy@gmail.com>
Date: Tue, 12 Aug 2025 18:45:50 +0200
Subject: [PATCH] fix(v2/map): respect limit when sitemap=only

---
 apps/api/src/__tests__/snips/v2/map.test.ts | 12 ++++++++++++
 apps/api/src/controllers/v2/map.ts          |  4 +---
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/apps/api/src/__tests__/snips/v2/map.test.ts b/apps/api/src/__tests__/snips/v2/map.test.ts
index 562717cc8..1c06b2270 100644
--- a/apps/api/src/__tests__/snips/v2/map.test.ts
+++ b/apps/api/src/__tests__/snips/v2/map.test.ts
@@ -41,4 +41,16 @@ describe("Map tests", () => {
     expect(response.body.success).toBe(true);
     expect(response.body.links.some(x => x.url.match(/^https:\/\/www\.hfea\.gov\.uk\/choose-a-clinic\/clinic-search\/results\/?\?options=\d+$/))).toBe(true);
   }, 60000);
+
+  it.concurrent("sitemap=only respects limit", async () => {
+    const response = await map({
+      url: "https://firecrawl.dev",
+      sitemap: "only",
+      limit: 10,
+    }, identity);
+
+    expect(response.statusCode).toBe(200);
+    expect(response.body.success).toBe(true);
+    expect(response.body.links.length).toBe(10);
+  });
 });
diff --git a/apps/api/src/controllers/v2/map.ts b/apps/api/src/controllers/v2/map.ts
index 7581854b4..e1ee585c0 100644
--- a/apps/api/src/controllers/v2/map.ts
+++ b/apps/api/src/controllers/v2/map.ts
@@ -291,9 +291,7 @@ export async function getMapResults({
     mapResults = dedupeMapDocumentArray(mapResults);
   }
 
-  mapResults = crawlerOptions.sitemapOnly
-    ? mapResults
-    : mapResults.slice(0, limit);
+  mapResults = mapResults.slice(0, limit);
 
   return {
     success: true,