From 0f0f85f4a5fb998cdbddc8591d83fa8712201cb6 Mon Sep 17 00:00:00 2001 From: Emilio Garcia Date: Thu, 13 Nov 2025 16:21:41 -0500 Subject: [PATCH] fix(rebase): resolve errors from rebase --- src/llama_stack/telemetry/helpers.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/llama_stack/telemetry/helpers.py b/src/llama_stack/telemetry/helpers.py index 29aadf4a3..2968f10bd 100644 --- a/src/llama_stack/telemetry/helpers.py +++ b/src/llama_stack/telemetry/helpers.py @@ -6,11 +6,9 @@ import json +from llama_stack_api import OpenAIMessageParam, RunShieldResponse from opentelemetry import trace -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, @@ -27,7 +25,9 @@ def safety_span_name(shield_id: str) -> str: # TODO: Consider using Wrapt to automatically instrument code # This is the industry standard way to package automatically instrumentation in python. -def safety_request_span_attributes(shield_id: str, messages: list[Message], response: RunShieldResponse) -> None: +def safety_request_span_attributes( + shield_id: str, messages: list[OpenAIMessageParam], response: RunShieldResponse +) -> None: span = trace.get_current_span() span.set_attribute(SAFETY_REQUEST_SHIELD_ID_ATTRIBUTE, shield_id) messages_json = json.dumps([msg.model_dump() for msg in messages])