Add import for MonitorService in deps.py

This commit is contained in:
anovazzi1 2024-01-19 21:59:47 -03:00
commit 46ed96f20f
2 changed files with 7 additions and 0 deletions

View file

@ -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

View file

@ -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