diff --git a/src/backend/base/langflow/services/deps.py b/src/backend/base/langflow/services/deps.py index beafdec30..304fce1bc 100644 --- a/src/backend/base/langflow/services/deps.py +++ b/src/backend/base/langflow/services/deps.py @@ -18,6 +18,7 @@ if TYPE_CHECKING: from langflow.services.storage.service import StorageService from langflow.services.store.service import StoreService from langflow.services.task.service import TaskService + from langflow.services.tracing.service import TracingService 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 +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": """ Retrieves the StateService instance from the service manager. diff --git a/src/backend/base/langflow/services/tracing/factory.py b/src/backend/base/langflow/services/tracing/factory.py index e179d2118..01a561d6a 100644 --- a/src/backend/base/langflow/services/tracing/factory.py +++ b/src/backend/base/langflow/services/tracing/factory.py @@ -5,11 +5,12 @@ from langflow.services.tracing.service import TracingService if TYPE_CHECKING: from langflow.services.settings.service import SettingsService + from langflow.services.monitor.service import MonitorService class TracingServiceFactory(ServiceFactory): def __init__(self): super().__init__(TracingService) - def create(self, settings_service: "SettingsService"): - return TracingService(settings_service) + def create(self, settings_service: "SettingsService", monitor_service: "MonitorService"): + return TracingService(settings_service, monitor_service)