reflect agent

Author: nicoloboschi

hindsight-api/hindsight_api/engine/memory_engine.py

@@ -400,8 +400,10 @@ def __init__(
         self._cross_encoder_reranker = CrossEncoderReranker(cross_encoder=cross_encoder)
 
         # Initialize task backend
-        _task_batch_size = task_batch_size if task_batch_size is not None else config.task_batch_size
-        _task_batch_interval = task_batch_interval if task_batch_interval is not None else config.task_batch_interval
+        _task_batch_size = task_batch_size if task_batch_size is not None else config.task_backend_memory_batch_size
+        _task_batch_interval = (
+            task_batch_interval if task_batch_interval is not None else config.task_backend_memory_batch_interval
+        )
         self._task_backend = task_backend or AsyncIOQueueBackend(
             batch_size=_task_batch_size, batch_interval=_task_batch_interval
         )
@@ -1885,7 +1887,7 @@ def to_tuple_format(results):
             await reranker_instance.ensure_initialized()
 
             # Rerank using cross-encoder
-            scored_results = reranker_instance.rerank(query, merged_candidates)
+            scored_results = await reranker_instance.rerank(query, merged_candidates)
 
             step_duration = time.time() - step_start
             log_buffer.append(f"  [4] Reranking: {len(scored_results)} candidates scored in {step_duration:.3f}s")