From f4576868a63093188412e0ab4acc35ecf0404991 Mon Sep 17 00:00:00 2001
From: phact <estevezsebastian@gmail.com>
Date: Fri, 10 Oct 2025 14:21:39 -0400
Subject: [PATCH] document_service connector fix

---
 src/connectors/service.py | 12 ++++++++++++
 src/models/processors.py  |  3 ++-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/connectors/service.py b/src/connectors/service.py
index 278743d3..96daaf77 100644
--- a/src/connectors/service.py
+++ b/src/connectors/service.py
@@ -271,6 +271,7 @@ class ConnectorService:
 
         # Create custom processor for connector files
         from models.processors import ConnectorFileProcessor
+        from services.document_service import DocumentService
 
         processor = ConnectorFileProcessor(
             self,
@@ -280,6 +281,11 @@ class ConnectorService:
             jwt_token=jwt_token,
             owner_name=owner_name,
             owner_email=owner_email,
+            document_service=(
+                self.task_service.document_service
+                if self.task_service and self.task_service.document_service
+                else DocumentService(session_manager=self.session_manager)
+            ),
         )
 
         # Use file IDs as items (no more fake file paths!)
@@ -366,6 +372,7 @@ class ConnectorService:
 
         # Create custom processor for specific connector files
         from models.processors import ConnectorFileProcessor
+        from services.document_service import DocumentService
 
         # Use expanded_file_ids which has folders already expanded
         processor = ConnectorFileProcessor(
@@ -376,6 +383,11 @@ class ConnectorService:
             jwt_token=jwt_token,
             owner_name=owner_name,
             owner_email=owner_email,
+            document_service=(
+                self.task_service.document_service
+                if self.task_service and self.task_service.document_service
+                else DocumentService(session_manager=self.session_manager)
+            ),
         )
 
         # Create custom task using TaskService
diff --git a/src/models/processors.py b/src/models/processors.py
index f83bb01b..972aa640 100644
--- a/src/models/processors.py
+++ b/src/models/processors.py
@@ -355,8 +355,9 @@ class ConnectorFileProcessor(TaskProcessor):
         jwt_token: str = None,
         owner_name: str = None,
         owner_email: str = None,
+        document_service=None,
     ):
-        super().__init__()
+        super().__init__(document_service=document_service)
         self.connector_service = connector_service
         self.connection_id = connection_id
         self.files_to_process = files_to_process