fix: Set readonly flag to false when begin new generic txn (#2946)

lablup · Oct 23, 2024 · e8664f6 · e8664f6
1 parent ff7a4a4
commit e8664f6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/changes/2946.fix.md b/changes/2946.fix.md
@@ -0,0 +1 @@
+Set the `postgres_readonly` flag to `false` when begin generic sessions
diff --git a/src/ai/backend/manager/models/utils.py b/src/ai/backend/manager/models/utils.py
@@ -103,7 +103,10 @@ async def _begin(self, connection: SAConnection) -> AsyncIterator[SAConnection]:
         """
         Begin generic transaction within the given connection.
         """
-        async with connection.begin():
+        conn_with_exec_opts = await connection.execution_options(
+            postgresql_readonly=False,
+        )
+        async with conn_with_exec_opts.begin():
             self._generic_txn_count += 1
             self._check_generic_txn_cnt()
             try: