Fix slack error logging (#1800)

This commit is contained in:
hagen-danswer 2024-07-11 08:31:48 -07:00 committed by GitHub
parent 08c6e821e7
commit 3e645a510e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -12,6 +12,7 @@ from danswer.connectors.gmail.constants import (
from danswer.connectors.google_drive.constants import (
DB_CREDENTIALS_DICT_SERVICE_ACCOUNT_KEY,
)
from danswer.db.models import ConnectorCredentialPair
from danswer.db.models import Credential
from danswer.db.models import User
from danswer.server.documents.models import CredentialBase
@ -142,6 +143,18 @@ def delete_credential(
f"Credential by provided id {credential_id} does not exist or does not belong to user"
)
associated_connectors = (
db_session.query(ConnectorCredentialPair)
.filter(ConnectorCredentialPair.credential_id == credential_id)
.all()
)
if associated_connectors:
raise ValueError(
f"Cannot delete credential {credential_id} as it is still associated with {len(associated_connectors)} connector(s). "
"Please delete all associated connectors first."
)
db_session.delete(credential)
db_session.commit()