From 9d1bc25ffa7bb627496bac12e05b410b61ab9832 Mon Sep 17 00:00:00 2001
From: Zamil Majdy <zamil.majdy@agpt.co>
Date: Fri, 10 Jan 2025 04:59:22 +0700
Subject: [PATCH] hotfix(backend): Increase statement timeout for the double
 brace migration (#9245)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

### Changes 🏗️


https://github.com/Significant-Gravitas/AutoGPT/actions/runs/12696734339/job/35391431786

### Checklist 📋

#### For code changes:
- [ ] I have clearly listed my changes in the PR description
- [ ] I have made a test plan
- [ ] I have tested my changes according to the test plan:
  <!-- Put your test plan here: -->
  - [ ] ...

<details>
  <summary>Example test plan</summary>

  - [ ] Create from scratch and execute an agent with at least 3 blocks
- [ ] Import an agent from file upload, and confirm it executes
correctly
  - [ ] Upload agent to marketplace
- [ ] Import an agent from marketplace and confirm it executes correctly
  - [ ] Edit an agent from monitor, and confirm it executes correctly
</details>

#### For configuration changes:
- [ ] `.env.example` is updated or already compatible with my changes
- [ ] `docker-compose.yml` is updated or already compatible with my
changes
- [ ] I have included a list of my configuration changes in the PR
description (under **Changes**)

<details>
  <summary>Examples of configuration changes</summary>

  - Changing ports
  - Adding new services that need to communicate with each other
  - Secrets or environment variable changes
  - New or infrastructure changes such as databases
</details>
---
 .../migration.sql                                      | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/autogpt_platform/backend/migrations/20250105254106_migrate_brace_to_double_brace_string_format/migration.sql b/autogpt_platform/backend/migrations/20250105254106_migrate_brace_to_double_brace_string_format/migration.sql
index 64abb735383e..2a7141b0f9c8 100644
--- a/autogpt_platform/backend/migrations/20250105254106_migrate_brace_to_double_brace_string_format/migration.sql
+++ b/autogpt_platform/backend/migrations/20250105254106_migrate_brace_to_double_brace_string_format/migration.sql
@@ -6,11 +6,14 @@
         - AITextGeneratorBlock
         - AIStructuredResponseGeneratorBlock
     with a double brace format.
+   - This migration can be slow for a large updated AgentNode tables.
 */
-SET statement_timeout = '15min';
+BEGIN;
+SET LOCAL statement_timeout = '10min';
 
 WITH to_update AS (
     SELECT
+        "id",
         "agentBlockId",
         "constantInput"::jsonb AS j
     FROM "AgentNode"
@@ -29,6 +32,7 @@ WITH to_update AS (
 ),
 updated_rows AS (
     SELECT
+        "id",
         "agentBlockId",
         (
             j
@@ -77,4 +81,6 @@ updated_rows AS (
 UPDATE "AgentNode" AS an
 SET "constantInput" = ur."newConstantInput"
 FROM updated_rows ur
-WHERE an."agentBlockId" = ur."agentBlockId";
+WHERE an."id" = ur."id";
+
+COMMIT;