No transaction in TaskRunRecorder (#16829)

PrefectHQ · Jan 23, 2025 · 525dfb8 · 525dfb8
1 parent 753ae01
commit 525dfb8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/prefect/server/services/task_run_recorder.py b/src/prefect/server/services/task_run_recorder.py
@@ -161,12 +161,13 @@ async def record_task_run_event(event: ReceivedEvent) -> None:
     }
 
     db = provide_database_interface()
-    async with db.session_context(begin_transaction=True) as session:
+    async with db.session_context() as session:
         await _insert_task_run(session, task_run, task_run_attributes)
         await _insert_task_run_state(session, task_run)
         await _update_task_run_with_state(
             session, task_run, denormalized_state_attributes
         )
+        await session.commit()
 
     logger.debug(
         "Recorded task run state change",