From 91717c81b33ee03851a4816f1b75a2b8cb2d58d3 Mon Sep 17 00:00:00 2001
From: Andrew Kane <andrew@ankane.org>
Date: Wed, 10 Jun 2026 12:10:20 -0700
Subject: [PATCH] Improved batch size flexibility [skip ci]

---
 src/hnswscan.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/hnswscan.c b/src/hnswscan.c
index d530795..0a30679 100644
--- a/src/hnswscan.c
+++ b/src/hnswscan.c
@@ -332,11 +332,15 @@ hnswgetbatch(IndexScanDesc scan, IndexScanBatch priorbatch, ScanDirection dir)
 			batch->items[nitems].indexOffset = -1;
 			batch->items[nitems].tupleOffset = 0;
 			nitems++;
+
+			/* Keep batch size flexible */
+			if (nitems == scan->maxitemsbatch)
+				break;
 		}
 
 		/* Needed for strict iterative scans */
 		if (nitems == 0)
-			break;
+			continue;
 
 		MemoryContextSwitchTo(oldCtx);