Update the logic to parse Google Spreadsheet in Gdrive connector

Signed-off-by: Alex Co <alex.tuan@mindvalley.com>
mindvalley · Jun 20, 2024 · 6e69ffb · 6e69ffb
1 parent 5757a58
commit 6e69ffb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/backend/danswer/connectors/google_drive/connector.py b/backend/danswer/connectors/google_drive/connector.py
@@ -334,7 +334,10 @@ def extract_text(file: dict[str, str], service: discovery.Resource) -> str:
         response = service.files().get_media(fileId=file["id"]).execute()
         return pptx_to_text(file=io.BytesIO(response))
     elif mime_type == GDriveMimeType.PPT.value:
-        response = service.files().get_media(fileId=file["id"]).execute()
+        response = service.files().export(
+            fileId=file["id"],
+            mimeType="application/vnd.openxmlformats-officedocument.presentationml.presentation"
+        ).execute()
         return pptx_to_text(file=io.BytesIO(response))
 
     return UNSUPPORTED_FILE_TYPE_CONTENT