From 2fbc852eb3eb2ab9b4ed12abee8d4ed26cc336f2 Mon Sep 17 00:00:00 2001 From: phact Date: Mon, 8 Sep 2025 23:58:26 -0400 Subject: [PATCH] drive convert to pdf and include file extension --- src/connectors/google_drive/connector.py | 5 +++-- src/connectors/langflow_connector_service.py | 6 ++++-- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/src/connectors/google_drive/connector.py b/src/connectors/google_drive/connector.py index c0ef2ff3..71eb24e0 100644 --- a/src/connectors/google_drive/connector.py +++ b/src/connectors/google_drive/connector.py @@ -400,8 +400,9 @@ class GoogleDriveConnector(BaseConnector): export_mime = self._pick_export_mime(mime_type) if mime_type.startswith("application/vnd.google-apps."): # default fallback if not overridden - if not export_mime: - export_mime = "application/pdf" + #if not export_mime: + # export_mime = "application/pdf" + export_mime = "application/pdf" # NOTE: export_media does not accept supportsAllDrives/includeItemsFromAllDrives request = self.service.files().export_media(fileId=file_id, mimeType=export_mime) else: diff --git a/src/connectors/langflow_connector_service.py b/src/connectors/langflow_connector_service.py index eda9d1fd..12fada87 100644 --- a/src/connectors/langflow_connector_service.py +++ b/src/connectors/langflow_connector_service.py @@ -53,9 +53,11 @@ class LangflowConnectorService: filename=document.filename, ) + suffix = self._get_file_extension(document.mimetype) + # Create temporary file from document content with tempfile.NamedTemporaryFile( - delete=False, suffix=self._get_file_extension(document.mimetype) + delete=False, suffix=suffix ) as tmp_file: tmp_file.write(document.content) tmp_file.flush() @@ -65,7 +67,7 @@ class LangflowConnectorService: logger.debug("Uploading file to Langflow", filename=document.filename) content = document.content file_tuple = ( - document.filename, + document.filename+suffix, content, document.mimetype or "application/octet-stream", )