apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-04 05:12:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Make migration 52 idempotent

Browse source
This commit is contained in:
CREDO23 2026-01-13 20:13:30 +02:00
parent 6f15655552
commit dcc6e067bd
1 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
surfsense_backend/alembic/versions/52_rename_llm_preference_columns.py
View file

@ -39,7 +39,7 @@ def upgrade():
""" """
) )
# Rename columns (only if they exist with old names) # Rename columns (only if source exists and target doesn't already exist)
op.execute( op.execute(
""" """
DO $$ DO $$
@ -47,6 +47,9 @@ def upgrade():
IF EXISTS ( IF EXISTS (
SELECT 1 FROM information_schema.columns SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'fast_llm_id' WHERE table_name = 'searchspaces' AND column_name = 'fast_llm_id'
) AND NOT EXISTS (
SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'agent_llm_id'
) THEN ) THEN
ALTER TABLE searchspaces RENAME COLUMN fast_llm_id TO agent_llm_id; ALTER TABLE searchspaces RENAME COLUMN fast_llm_id TO agent_llm_id;
END IF; END IF;
@ -61,6 +64,9 @@ def upgrade():
IF EXISTS ( IF EXISTS (
SELECT 1 FROM information_schema.columns SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'long_context_llm_id' WHERE table_name = 'searchspaces' AND column_name = 'long_context_llm_id'
) AND NOT EXISTS (
SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'document_summary_llm_id'
) THEN ) THEN
ALTER TABLE searchspaces RENAME COLUMN long_context_llm_id TO document_summary_llm_id; ALTER TABLE searchspaces RENAME COLUMN long_context_llm_id TO document_summary_llm_id;
END IF; END IF;
@ -100,7 +106,7 @@ def downgrade():
""" """
) )
# Rename columns back # Rename columns back (only if source exists and target doesn't already exist)
op.execute( op.execute(
""" """
DO $$ DO $$
@ -108,6 +114,9 @@ def downgrade():
IF EXISTS ( IF EXISTS (
SELECT 1 FROM information_schema.columns SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'agent_llm_id' WHERE table_name = 'searchspaces' AND column_name = 'agent_llm_id'
) AND NOT EXISTS (
SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'fast_llm_id'
) THEN ) THEN
ALTER TABLE searchspaces RENAME COLUMN agent_llm_id TO fast_llm_id; ALTER TABLE searchspaces RENAME COLUMN agent_llm_id TO fast_llm_id;
END IF; END IF;
@ -122,6 +131,9 @@ def downgrade():
IF EXISTS ( IF EXISTS (
SELECT 1 FROM information_schema.columns SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'document_summary_llm_id' WHERE table_name = 'searchspaces' AND column_name = 'document_summary_llm_id'
) AND NOT EXISTS (
SELECT 1 FROM information_schema.columns
WHERE table_name = 'searchspaces' AND column_name = 'long_context_llm_id'
) THEN ) THEN
ALTER TABLE searchspaces RENAME COLUMN document_summary_llm_id TO long_context_llm_id; ALTER TABLE searchspaces RENAME COLUMN document_summary_llm_id TO long_context_llm_id;
END IF; END IF;