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

Fix Remove Index Attempt (#1091)

Browse Source
This commit is contained in:
Yuhong Sun
2024-02-17 19:53:17 -08:00
committed by GitHub
parent e505486ca4
commit 6e8acdb20d
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
backend/danswer/db/index_attempt.py
View File

@@ -244,20 +244,15 @@ def cancel_indexing_attempts_for_connector(
db_session: Session,
include_secondary_index: bool = False,
) -> None:
subquery = select(EmbeddingModel.id).where(
EmbeddingModel.status != IndexModelStatus.FUTURE
)
stmt = (
update(IndexAttempt)
.where(
IndexAttempt.connector_id == connector_id,
IndexAttempt.status == IndexingStatus.NOT_STARTED,
)
.values(status=IndexingStatus.FAILED)
stmt = delete(IndexAttempt).where(
IndexAttempt.connector_id == connector_id,
IndexAttempt.status == IndexingStatus.NOT_STARTED,
)
if not include_secondary_index:
subquery = select(EmbeddingModel.id).where(
EmbeddingModel.status != IndexModelStatus.FUTURE
)
stmt = stmt.where(IndexAttempt.embedding_model_id.in_(subquery))
db_session.execute(stmt)