From db825b605331b619bc5e54331b9011a8f8d0cf54 Mon Sep 17 00:00:00 2001 From: Matthew Zhou Date: Sun, 3 Aug 2025 01:45:27 -0700 Subject: [PATCH] fix: Fix duplicate agents archives relationship migration error (#3716) --- .../74e860718e0d_add_archival_memory_sharing.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/alembic/versions/74e860718e0d_add_archival_memory_sharing.py b/alembic/versions/74e860718e0d_add_archival_memory_sharing.py index 567b8061..61c41631 100644 --- a/alembic/versions/74e860718e0d_add_archival_memory_sharing.py +++ b/alembic/versions/74e860718e0d_add_archival_memory_sharing.py @@ -262,9 +262,8 @@ def upgrade() -> None: JOIN agents a ON ap.agent_id = a.id WHERE ap.is_deleted = FALSE AND NOT EXISTS ( - SELECT 1 FROM archives ar - WHERE ar.organization_id = a.organization_id - AND ar.name = COALESCE(a.name, 'Agent ' || a.id) || '''s Archive' + SELECT 1 FROM archives_agents aa + WHERE aa.agent_id = a.id ) LIMIT :batch_size ), @@ -328,11 +327,9 @@ def upgrade() -> None: sa.text( """ UPDATE agent_passages ap - SET archive_id = ar.id - FROM agents a - JOIN archives ar ON ar.organization_id = a.organization_id - AND ar.name = COALESCE(a.name, 'Agent ' || a.id) || '''s Archive' - WHERE ap.agent_id = a.id + SET archive_id = aa.archive_id + FROM archives_agents aa + WHERE ap.agent_id = aa.agent_id AND ap.archive_id IS NULL AND ap.is_deleted = FALSE AND ap.id IN (