chore: Update TracingServiceFactory to include MonitorService in create method
This commit is contained in:
parent
a4736fd0b9
commit
9dc6def567
2 changed files with 16 additions and 2 deletions
|
|
@ -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.
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue