From 6d0a94e4b44b98fed60c75815c47fbe1d8331372 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 8 Sep 2025 16:42:07 -0300 Subject: [PATCH] Refactor connector service initialization to use LangflowConnectorService This commit updates the service initialization in main.py to replace the existing ConnectorService with LangflowConnectorService. This change enhances the management of connector documents by leveraging the new service's capabilities, aligning with the ongoing improvements in asynchronous processing and code maintainability. --- src/main.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/src/main.py b/src/main.py index 9810ca24..890faa98 100644 --- a/src/main.py +++ b/src/main.py @@ -1,6 +1,7 @@ import sys # Configure structured logging early +from connectors.langflow_connector_service import LangflowConnectorService from utils.logging_config import configure_from_env, get_logger configure_from_env() @@ -49,7 +50,6 @@ from config.settings import ( ) # Existing services -from connectors.service import ConnectorService from services.auth_service import AuthService from services.chat_service import ChatService @@ -301,11 +301,7 @@ async def initialize_services(): document_service.process_pool = process_pool # Initialize connector service - connector_service = ConnectorService( - patched_async_client=clients.patched_async_client, - process_pool=process_pool, - embed_model="text-embedding-3-small", - index_name=INDEX_NAME, + connector_service = LangflowConnectorService( task_service=task_service, session_manager=session_manager, )