Auto-populate ACL fields on server startup

backend/scripts/migrate_vespa_to_acl.py

@@ -1,44 +1,9 @@
 """Script which updates Vespa to align with the access described in Postgres.
-Should be run wehn a user who has docs already indexed switches over to the new
-access control system. This allows them to not have to re-index all documents."""
-from sqlalchemy import select
-from sqlalchemy.orm import Session
-
-from danswer.access.models import DocumentAccess
-from danswer.datastores.document_index import get_default_document_index
-from danswer.datastores.interfaces import UpdateRequest
-from danswer.datastores.vespa.store import VespaIndex
-from danswer.db.document import get_acccess_info_for_documents
-from danswer.db.engine import get_sqlalchemy_engine
-from danswer.db.models import Document
-from danswer.utils.logger import setup_logger
-
-logger = setup_logger()
-
-
-def _migrate_vespa_to_acl() -> None:
-    vespa_index = get_default_document_index()
-    if not isinstance(vespa_index, VespaIndex):
-        raise ValueError("This script is only for Vespa indexes")
-
-    with Session(get_sqlalchemy_engine()) as db_session:
-        # for all documents, set the `access_control_list` field apporpriately
-        # based on the state of Postgres
-        documents = db_session.scalars(select(Document)).all()
-        document_access_info = get_acccess_info_for_documents(
-            db_session=db_session,
-            document_ids=[document.id for document in documents],
-        )
-        vespa_index.update(
-            update_requests=[
-                UpdateRequest(
-                    document_ids=[document_id],
-                    access=DocumentAccess.build(user_ids, is_public),
-                )
-                for document_id, user_ids, is_public in document_access_info
-            ],
-        )
+Should be run when a user who has docs already indexed switches over to the new
+access control system. This allows them to not have to re-index all documents.
+NOTE: this is auto-run on server startup, so should not be necessary in most cases."""
+from danswer.utils.acl import set_acl_for_vespa
 
 
 if __name__ == "__main__":
-    _migrate_vespa_to_acl()
+    set_acl_for_vespa()