Update rag/retriever.py

rag/retriever.py

@@ -234,8 +234,7 @@ def expand_context(top_chunks: list[dict], machine_id: str) -> list[dict]:
                 
         # Traverse forward up to 1 chunk (stricter boundary)
         curr = c
-        for _ in range(8):
-        # for _ in range(2):
+        for _ in range(2):
             next_id = curr.get("next_chunk")
             if next_id and next_id in local_lookup:
                 nxt = local_lookup[next_id]