fix(telemetry): move out of core + fix name for safety

src/llama_stack/core/routers/safety.py

@@ -7,7 +7,6 @@
 from typing import Any
 
 from llama_stack.core.datatypes import SafetyConfig
-from llama_stack.core.telemetry.helpers import safety_request_span_attributes, safety_span_name
 from llama_stack.log import get_logger
 from llama_stack_api import ModerationObject, OpenAIMessageParam, RoutingTable, RunShieldResponse, Safety, Shield
 

src/llama_stack/telemetry/constants.py

@@ -7,7 +7,7 @@
 llama_stack_prefix = "llama_stack"
 
 # Safety Attributes
-SAFETY_SPAN_NAME = "safety.run_shield"
+RUN_SHIELD_OPERATION_NAME = "run_shield"
 
 SAFETY_REQUEST_PREFIX = f"{llama_stack_prefix}.safety.request"
 SAFETY_REQUEST_SHIELD_ID_ATTRIBUTE = f"{SAFETY_REQUEST_PREFIX}.shield_id"

src/llama_stack/telemetry/helpers.py

@@ -12,17 +12,17 @@ from llama_stack.apis.inference import Message
 from llama_stack.apis.safety import RunShieldResponse
 
 from .constants import (
+    RUN_SHIELD_OPERATION_NAME,
     SAFETY_REQUEST_MESSAGES_ATTRIBUTE,
     SAFETY_REQUEST_SHIELD_ID_ATTRIBUTE,
     SAFETY_RESPONSE_METADATA_ATTRIBUTE,
     SAFETY_RESPONSE_USER_MESSAGE_ATTRIBUTE,
     SAFETY_RESPONSE_VIOLATION_LEVEL_ATTRIBUTE,
-    SAFETY_SPAN_NAME,
 )
 
 
 def safety_span_name(shield_id: str) -> str:
-    return f"{SAFETY_SPAN_NAME} {shield_id}"
+    return f"{RUN_SHIELD_OPERATION_NAME} {shield_id}"
 
 
 # TODO: Consider using Wrapt to automatically instrument code