highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-04 21:11:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Extremely robustified Index Attempt migration (#2151)

Browse Source 
* account for connector_id edge case

* robustified
This commit is contained in:
pablodanswer
2024-08-16 18:12:18 -07:00
committed by GitHub
parent 3cbc341b60
commit f8e0e6f015
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
backend/alembic/versions/8a87bd6ec550_associate_index_attempts_with_ccpair.py
View File

@@ -35,18 +35,22 @@ def upgrade() -> None:
op.execute( op.execute(
""" """
UPDATE index_attempt ia UPDATE index_attempt ia
SET connector_credential_pair_id = SET connector_credential_pair_id = (
CASE SELECT id FROM connector_credential_pair ccp
WHEN ia.credential_id IS NULL THEN WHERE
(SELECT id FROM connector_credential_pair (ia.connector_id IS NULL OR ccp.connector_id = ia.connector_id)
WHERE connector_id = ia.connector_id AND (ia.credential_id IS NULL OR ccp.credential_id = ia.credential_id)
LIMIT 1) LIMIT 1
ELSE )
(SELECT id FROM connector_credential_pair WHERE ia.connector_id IS NOT NULL OR ia.credential_id IS NOT NULL
WHERE connector_id = ia.connector_id """
AND credential_id = ia.credential_id) )
END
WHERE ia.connector_id IS NOT NULL # For good measure
op.execute(
"""
DELETE FROM index_attempt
WHERE connector_credential_pair_id IS NULL
""" """
) )