1 files changed, 2 insertions, 2 deletions

diff --git a/rag/rag.go b/rag/rag.go
index b29b9eb..d8b6978 100644
--- a/rag/rag.go
+++ b/rag/rag.go
@@ -107,7 +107,7 @@ func (r *RAG) LoadRAG(fpath string) error {
 	}
 
 	// Adjust batch size if needed
-	if len(paragraphs) < int(r.cfg.RAGBatchSize) && len(paragraphs) > 0 {
+	if len(paragraphs) < r.cfg.RAGBatchSize && len(paragraphs) > 0 {
 		r.cfg.RAGBatchSize = len(paragraphs)
 	}
 
@@ -133,7 +133,7 @@ func (r *RAG) LoadRAG(fpath string) error {
 	ctn := 0
 	totalParagraphs := len(paragraphs)
 	for {
-		if int(right) > totalParagraphs {
+		if right > totalParagraphs {
 			batchCh <- map[int][]string{left: paragraphs[left:]}
 			break
 		}