From 6418d5e56c40bb5056eaf8383fdc4d4af85d43cf Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 1 Sep 2023 17:22:01 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20chore(utils.py):=20add=20type=20?= =?UTF-8?q?hints=20to=20imported=20modules=20for=20better=20code=20readabi?= =?UTF-8?q?lity=20and=20maintainability=20=E2=9C=A8=20feat(utils.py):=20ad?= =?UTF-8?q?d=20get=5Ftask=5Fmanager()=20function=20to=20retrieve=20the=20T?= =?UTF-8?q?askManager=20instance=20from=20the=20service=20manager?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/utils.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py index f1b3a82aa..00c0571c5 100644 --- a/src/backend/langflow/services/utils.py +++ b/src/backend/langflow/services/utils.py @@ -5,6 +5,10 @@ from typing import TYPE_CHECKING if TYPE_CHECKING: from langflow.services.database.manager import DatabaseManager from langflow.services.settings.manager import SettingsManager + from langflow.services.cache.manager import BaseCacheManager + from langflow.services.session.manager import SessionManager + from langflow.services.task.manager import TaskManager + from sqlmodel import Session def get_settings_manager() -> "SettingsManager": @@ -15,14 +19,18 @@ def get_db_manager() -> "DatabaseManager": return service_manager.get(ServiceType.DATABASE_MANAGER) -def get_session(): +def get_session() -> "Session": db_manager = service_manager.get(ServiceType.DATABASE_MANAGER) yield from db_manager.get_session() -def get_cache_manager(): +def get_cache_manager() -> "BaseCacheManager": return service_manager.get(ServiceType.CACHE_MANAGER) -def get_session_manager(): +def get_session_manager() -> "SessionManager": return service_manager.get(ServiceType.SESSION_MANAGER) + + +def get_task_manager() -> "TaskManager": + return service_manager.get(ServiceType.TASK_MANAGER)