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, )