Rebase

SunnySoldier357 · SunnySoldier357 · commit a17c9b8fb9d7 · 2026-04-24T14:54:19.000-07:00
diff --git a/eval_protocol/pytest/remote_rollout_processor.py b/eval_protocol/pytest/remote_rollout_processor.py
@@ -140,9 +140,9 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
                     )
 
                     # Backfill message/details/extras from the full Logs table (one-shot)
-                    completed_logs = await asyncio.to_thread(
-                        self._tracing_adapter.search_logs,
-                        tags=[f"rollout_id:{row.execution_metadata.rollout_id}"],
+                    session = self._get_or_create_session()
+                    completed_logs = await self._tracing_adapter.async_search_logs(
+                        session, tags=[f"rollout_id:{row.execution_metadata.rollout_id}"],
                     )
                     status_message = ""
                     status_details: list = []

Original file line number	Diff line number	Diff line change
`@@ -140,9 +140,9 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:`
`140`	`140`	`)`
`141`	`141`
`142`	`142`	`# Backfill message/details/extras from the full Logs table (one-shot)`
`143`		`- completed_logs = await asyncio.to_thread(`
`144`		`- self._tracing_adapter.search_logs,`
`145`		`- tags=[f"rollout_id:{row.execution_metadata.rollout_id}"],`
	`143`	`+ session = self._get_or_create_session()`
	`144`	`+ completed_logs = await self._tracing_adapter.async_search_logs(`
	`145`	`+ session, tags=[f"rollout_id:{row.execution_metadata.rollout_id}"],`
`146`	`146`	`)`
`147`	`147`	`status_message = ""`
`148`	`148`	`status_details: list = []`