chore: Update TracingServiceFactory to include MonitorService in create method

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 18:53:36 -03:00
commit 9dc6def567
2 changed files with 16 additions and 2 deletions

View file

@ -18,6 +18,7 @@ if TYPE_CHECKING:
from langflow.services.storage.service import StorageService from langflow.services.storage.service import StorageService
from langflow.services.store.service import StoreService from langflow.services.store.service import StoreService
from langflow.services.task.service import TaskService from langflow.services.task.service import TaskService
from langflow.services.tracing.service import TracingService
from langflow.services.variable.service import VariableService from langflow.services.variable.service import VariableService
@ -41,6 +42,18 @@ def get_service(service_type: ServiceType, default=None):
return service_manager.get(service_type, default) # type: ignore return service_manager.get(service_type, default) # type: ignore
def get_tracing_service() -> "TracingService":
"""
Retrieves the TracingService instance from the service manager.
Returns:
TracingService: The TracingService instance.
"""
from langflow.services.tracing.factory import TracingServiceFactory
return get_service(ServiceType.TRACING_SERVICE, TracingServiceFactory()) # type: ignore
def get_state_service() -> "StateService": def get_state_service() -> "StateService":
""" """
Retrieves the StateService instance from the service manager. Retrieves the StateService instance from the service manager.

View file

@ -5,11 +5,12 @@ from langflow.services.tracing.service import TracingService
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.settings.service import SettingsService from langflow.services.settings.service import SettingsService
from langflow.services.monitor.service import MonitorService
class TracingServiceFactory(ServiceFactory): class TracingServiceFactory(ServiceFactory):
def __init__(self): def __init__(self):
super().__init__(TracingService) super().__init__(TracingService)
def create(self, settings_service: "SettingsService"): def create(self, settings_service: "SettingsService", monitor_service: "MonitorService"):
return TracingService(settings_service) return TracingService(settings_service, monitor_service)