highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-19 12:30:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update force deletion script

Browse Source
This commit is contained in:
Weves 2024-08-14 23:20:01 -07:00 committed by Chris Weaver
parent f6547a64a0
commit 0853d1a8f1
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/scripts/force_delete_connector_by_id.py Normal file → Executable file
View File

@ -143,9 +143,9 @@ def _delete_connector(cc_pair_id: int, db_session: Session) -> None:
logger.error(f"Connector credential pair with ID {cc_pair_id} not found")
return
if cc_pair.status != ConnectorCredentialPairStatus.PAUSED:
if cc_pair.status == ConnectorCredentialPairStatus.ACTIVE:
logger.error(
f"Connector {cc_pair.connector.name} is not paused, cannot continue. \
f"Connector {cc_pair.connector.name} is active, cannot continue. \
Please navigate to the connector and pause before attempting again"
)
return
@ -177,6 +177,11 @@ def _delete_connector(cc_pair_id: int, db_session: Session) -> None:
f"{connector_id} and Credential ID: {credential_id} does not exist."
)
file_names: list[str] = (
cc_pair.connector.connector_specific_config["file_locations"]
if cc_pair.connector.source == DocumentSource.FILE
else []
)
try:
logger.info("Deleting information from Vespa and Postgres")
curr_ind_name, sec_ind_name = get_both_index_names(db_session)
@ -195,11 +200,10 @@ def _delete_connector(cc_pair_id: int, db_session: Session) -> None:
except Exception as e:
logger.error(f"Failed to delete connector due to {e}")
if cc_pair.connector.source == DocumentSource.FILE:
connector = cc_pair.connector
if file_names:
logger.info("Deleting stored files!")
file_store = get_default_file_store(db_session)
for file_name in connector.connector_specific_config["file_locations"]:
for file_name in file_names:
logger.info(f"Deleting file {file_name}")
file_store.delete_file(file_name)