highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-05-30 09:40:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix slack error logging (#1800)

Browse Source
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
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/danswer/db/credentials.py
View File

@ -12,6 +12,7 @@ from danswer.connectors.gmail.constants import (
from danswer.connectors.google_drive.constants import ( from danswer.connectors.google_drive.constants import (
DB_CREDENTIALS_DICT_SERVICE_ACCOUNT_KEY, DB_CREDENTIALS_DICT_SERVICE_ACCOUNT_KEY,
) )
from danswer.db.models import ConnectorCredentialPair
from danswer.db.models import Credential from danswer.db.models import Credential
from danswer.db.models import User from danswer.db.models import User
from danswer.server.documents.models import CredentialBase 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" 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.delete(credential)
db_session.commit() db_session.commit()