phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 09:53:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(rebase): resolve errors from rebase

Browse source
This commit is contained in:
Emilio Garcia 2025-11-13 16:21:41 -05:00
parent 532389ddc4
commit 0f0f85f4a5
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/llama_stack/telemetry/helpers.py
View file

@ -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])