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.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.
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue