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

Fix sleep + handle duplicates

Browse Source
This commit is contained in:
Weves 2024-05-14 17:43:56 -07:00 committed by Chris Weaver
parent 10232c7c54
commit f087d3eac0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/danswer/db/document.py
View File

@ -303,7 +303,7 @@ def acquire_document_locks(db_session: Session, document_ids: list[str]) -> bool
documents = db_session.scalars(stmt).all()
# make sure we found every document
if len(documents) != len(document_ids):
if len(documents) != len(set(document_ids)):
logger.warning("Didn't find row for all specified document IDs. Aborting.")
return False
@ -341,7 +341,8 @@ def prepare_to_modify_documents(
break
except OperationalError as e:
logger.info(f"Failed to acquire locks for documents, retrying. Error: {e}")
time.sleep(retry_delay)
time.sleep(retry_delay)
if not lock_acquired:
raise RuntimeError(