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 loguru import logger
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from langflow.services.deps import get_monitor_service
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.graph.vertex.base import Vertex 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.chat.service import ChatService
from langflow.services.credentials.service import CredentialService from langflow.services.credentials.service import CredentialService
from langflow.services.database.service import DatabaseService from langflow.services.database.service import DatabaseService
from langflow.services.monitor.service import MonitorService
from langflow.services.plugins.service import PluginService from langflow.services.plugins.service import PluginService
from langflow.services.session.service import SessionService from langflow.services.session.service import SessionService
from langflow.services.settings.service import SettingsService 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 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": def get_task_service() -> "TaskService":
return service_manager.get(ServiceType.TASK_SERVICE) # type: ignore return service_manager.get(ServiceType.TASK_SERVICE) # type: ignore