gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

document_service connector fix

Browse source
This commit is contained in:
phact 2025-10-10 14:21:39 -04:00
parent 6266e5c18d
commit f4576868a6
2 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/connectors/service.py
View file

@ -271,6 +271,7 @@ class ConnectorService:
# Create custom processor for connector files # Create custom processor for connector files
from models.processors import ConnectorFileProcessor from models.processors import ConnectorFileProcessor
from services.document_service import DocumentService
processor = ConnectorFileProcessor( processor = ConnectorFileProcessor(
self, self,
@ -280,6 +281,11 @@ class ConnectorService:
jwt_token=jwt_token, jwt_token=jwt_token,
owner_name=owner_name, owner_name=owner_name,
owner_email=owner_email, 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!) # Use file IDs as items (no more fake file paths!)
@ -366,6 +372,7 @@ class ConnectorService:
# Create custom processor for specific connector files # Create custom processor for specific connector files
from models.processors import ConnectorFileProcessor from models.processors import ConnectorFileProcessor
from services.document_service import DocumentService
# Use expanded_file_ids which has folders already expanded # Use expanded_file_ids which has folders already expanded
processor = ConnectorFileProcessor( processor = ConnectorFileProcessor(
@ -376,6 +383,11 @@ class ConnectorService:
jwt_token=jwt_token, jwt_token=jwt_token,
owner_name=owner_name, owner_name=owner_name,
owner_email=owner_email, 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 # Create custom task using TaskService

3
src/models/processors.py
View file

@ -355,8 +355,9 @@ class ConnectorFileProcessor(TaskProcessor):
jwt_token: str = None, jwt_token: str = None,
owner_name: str = None, owner_name: str = None,
owner_email: str = None, owner_email: str = None,
document_service=None,
): ):
super().__init__() super().__init__(document_service=document_service)
self.connector_service = connector_service self.connector_service = connector_service
self.connection_id = connection_id self.connection_id = connection_id
self.files_to_process = files_to_process self.files_to_process = files_to_process