validator usage

Author: CalebCourier

guardrails/validator_service/async_validator_service.py

@@ -9,6 +9,7 @@
     PassResult,
     ValidationResult,
 )
+from guardrails.telemetry.hub_tracing import async_trace
 from guardrails.telemetry.validator_tracing import trace_async_validator
 from guardrails.types import ValidatorMap, OnFailAction
 from guardrails.classes.validation.validator_logs import ValidatorLogs
@@ -23,6 +24,9 @@
 
 
 class AsyncValidatorService(ValidatorServiceBase):
+    @async_trace(
+        name="/validator_usage", origin="AsyncValidatorService.execute_validator"
+    )
     async def execute_validator(
         self,
         validator: Validator,

guardrails/validator_service/validator_service_base.py

@@ -12,8 +12,8 @@
 )
 from guardrails.errors import ValidationError
 from guardrails.merge import merge
+from guardrails.telemetry.hub_tracing import trace
 from guardrails.types import OnFailAction
-from guardrails.utils.hub_telemetry_utils import HubTelemetry
 from guardrails.classes.validation.validator_logs import ValidatorLogs
 from guardrails.actions.reask import FieldReAsk
 from guardrails.telemetry import trace_validator
@@ -43,6 +43,7 @@ def __init__(self, disable_tracer: Optional[bool] = True):
     #       This is a well known issue without any real solutions.
     #       Using `fork` instead of `spawn` may alleviate the symptom for POSIX systems,
     #       but is relatively unsupported on Windows.
+    @trace(name="/validator_usage", origin="ValidatorServiceBase.execute_validator")
     def execute_validator(
         self,
         validator: Validator,
@@ -152,26 +153,6 @@ def after_run_validator(
         validator_logs.validation_result = result
         validator_logs.end_time = end_time
 
-        if not self._disable_tracer:
-            # Get HubTelemetry singleton and create a new span to
-            # log the validator usage
-            _hub_telemetry = HubTelemetry()
-            _hub_telemetry.create_new_span(
-                span_name="/validator_usage",
-                attributes=[
-                    ("validator_name", validator.rail_alias),
-                    ("validator_on_fail", validator.on_fail_descriptor),
-                    (
-                        "validator_result",
-                        result.outcome
-                        if isinstance(result, ValidationResult)
-                        else None,
-                    ),
-                ],
-                is_parent=False,  # This span will have no children
-                has_parent=True,  # This span has a parent
-            )
-
         return validator_logs
 
     def run_validator(