Allow compressing visible chunks before selection

env/environment.py

@@ -293,9 +293,10 @@ class RagContextOptimizerEnv:
         return reward, {"event": "chunk_deselected", "chunk_id": chunk_id, "hybrid_score": score}
 
     def _handle_compress(self, chunk_id: str, compression_ratio: float) -> tuple[float, dict[str, Any]]:
-        if chunk_id not in self._selected_chunks:
-            self._last_action_feedback = "chunk_not_selected_for_compression"
-            return -0.05, {"event": "chunk_not_selected_for_compression", "chunk_id": chunk_id}
+        chunk = self._chunk_map().get(chunk_id)
+        if chunk is None:
+            self._last_action_feedback = "chunk_not_found"
+            return -0.1, {"event": "chunk_not_found", "chunk_id": chunk_id}
 
         self._compression_ratios[chunk_id] = compression_ratio
         is_relevant, score = self._is_relevant(chunk_id)