Fix drive connector for service accounts with shared files turned off

2025-09-19 20:24:32 +02:00 · 2023-10-08 00:05:16 -07:00
parent 9425ccd043
commit c658ffd0b6
1 changed files with 3 additions and 1 deletions
--- a/backend/danswer/connectors/google_drive/connector.py
+++ b/backend/danswer/connectors/google_drive/connector.py
@@ -75,7 +75,9 @@ def _run_drive_file_query(
        results = (
            service.files()
            .list(
-                corpora="allDrives",  # needed to search through shared drives
+                corpora="allDrives"
+                if include_shared
+                else "user",  # needed to search through shared drives
                pageSize=batch_size,
                supportsAllDrives=include_shared,
                includeItemsFromAllDrives=include_shared,