update: return sublist

dev-diaries41 · dev-diaries41 · commit ca4503e21c98 · 2025-10-14T02:22:34.000+01:00
diff --git a/extensions/src/main/java/com/fpf/smartscansdk/extensions/embeddings/FileEmbeddingRetriever.kt b/extensions/src/main/java/com/fpf/smartscansdk/extensions/embeddings/FileEmbeddingRetriever.kt
@@ -43,7 +43,7 @@ class FileEmbeddingRetriever(
         val e = end.coerceAtMost(ids.size)
         if (s >= e) return emptyList()
 
-        val batch = store.get(ids)
+        val batch = store.get(ids.subList(s, e))
         return batch
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ class FileEmbeddingRetriever(`
`43`	`43`	`val e = end.coerceAtMost(ids.size)`
`44`	`44`	`if (s >= e) return emptyList()`
`45`	`45`
`46`		`- val batch = store.get(ids)`
	`46`	`+ val batch = store.get(ids.subList(s, e))`
`47`	`47`	`return batch`
`48`	`48`	`}`
`49`	`49`	`}`