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

fix index attempt migration where no credential ID

Browse Source
This commit is contained in:
pablodanswer 2024-07-30 08:50:20 -07:00 committed by Chris Weaver
parent 3938a053aa
commit e4b1f5b963
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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