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

View file

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