From 46ed96f20f276a0199d0e34652f925855976aed2 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Fri, 19 Jan 2024 21:59:47 -0300 Subject: [PATCH] Add import for MonitorService in deps.py --- src/backend/langflow/graph/edge/base.py | 2 ++ src/backend/langflow/services/deps.py | 5 +++++ 2 files changed, 7 insertions(+) diff --git a/src/backend/langflow/graph/edge/base.py b/src/backend/langflow/graph/edge/base.py index 25474164c..ef00afaa9 100644 --- a/src/backend/langflow/graph/edge/base.py +++ b/src/backend/langflow/graph/edge/base.py @@ -3,6 +3,8 @@ from typing import TYPE_CHECKING, List, Optional from loguru import logger from pydantic import BaseModel, Field +from langflow.services.deps import get_monitor_service + if TYPE_CHECKING: from langflow.graph.vertex.base import Vertex diff --git a/src/backend/langflow/services/deps.py b/src/backend/langflow/services/deps.py index 1ac06738b..28e7d2977 100644 --- a/src/backend/langflow/services/deps.py +++ b/src/backend/langflow/services/deps.py @@ -7,6 +7,7 @@ if TYPE_CHECKING: from langflow.services.chat.service import ChatService from langflow.services.credentials.service import CredentialService from langflow.services.database.service import DatabaseService + from langflow.services.monitor.service import MonitorService from langflow.services.plugins.service import PluginService from langflow.services.session.service import SessionService from langflow.services.settings.service import SettingsService @@ -51,6 +52,10 @@ def get_session_service() -> "SessionService": return service_manager.get(ServiceType.SESSION_SERVICE) # type: ignore +def get_monitor_service() -> "MonitorService": + return service_manager.get(ServiceType.MONITOR_SERVICE) # type: ignore + + def get_task_service() -> "TaskService": return service_manager.get(ServiceType.TASK_SERVICE) # type: ignore