Fix connector deletion bug

This commit is contained in:
Weves
2023-11-14 09:05:28 -08:00
committed by Chris Weaver
parent 0a0e672b35
commit 81cbcbb403

View File

@@ -11,8 +11,6 @@ connector / credential pair from the access list
(6) delete all relevant entries from postgres (6) delete all relevant entries from postgres
""" """
import time import time
from collections.abc import Callable
from typing import cast
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
@@ -37,7 +35,6 @@ from danswer.document_index.interfaces import DocumentIndex
from danswer.document_index.interfaces import UpdateRequest from danswer.document_index.interfaces import UpdateRequest
from danswer.server.models import ConnectorCredentialPairIdentifier from danswer.server.models import ConnectorCredentialPairIdentifier
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
from danswer.utils.variable_functionality import fetch_versioned_implementation
logger = setup_logger() logger = setup_logger()
@@ -173,14 +170,8 @@ def delete_connector_credential_pair(
# Clean up document sets / access information from Postgres # Clean up document sets / access information from Postgres
# and sync these updates to Vespa # and sync these updates to Vespa
cleanup_synced_entities__versioned = cast( # TODO: add user group cleanup with `fetch_versioned_implementation`
Callable[[ConnectorCredentialPair, Session], None], cleanup_synced_entities(cc_pair, db_session)
fetch_versioned_implementation(
"danswer.background.connector_deletion",
"cleanup_synced_entities",
),
)
cleanup_synced_entities__versioned(cc_pair, db_session)
# clean up the rest of the related Postgres entities # clean up the rest of the related Postgres entities
delete_index_attempts( delete_index_attempts(