phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 14:08:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 143cb85cbd into 46ff302d87

Browse source
This commit is contained in:
Sébastien Han 2025-08-14 13:18:00 +08:00 • committed by GitHub
commit 72f928f8da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 28 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

29
llama_stack/providers/inline/telemetry/meta_reference/console_span_processor.py
View file

@ -16,6 +16,29 @@ from llama_stack.log import get_logger
logger = get_logger(name="console_span_processor", category="telemetry") logger = get_logger(name="console_span_processor", category="telemetry")
def _mask_sensitive_value(value: str) -> str:
"""Mask sensitive information like API keys in string values."""
# These should cover OpenAI, Anthropic, Google at least
api_key_prefixes = [
"sk-", # OpenAI, Anthropic
"AIza", # Google
]
for prefix in api_key_prefixes:
if value.startswith(prefix):
logger.debug("Masking potentially sensitive API key value")
return "********"
# Also check for long alphanumeric strings that might be API keys
# This is best effort, as it's not always possible to determine if a string is an API key from a
# known prefix.
if len(value) >= 32 and value.isalnum():
logger.debug("Masking potentially sensitive long alphanumeric API key value")
return "********"
return value
class ConsoleSpanProcessor(SpanProcessor): class ConsoleSpanProcessor(SpanProcessor):
def __init__(self, print_attributes: bool = False): def __init__(self, print_attributes: bool = False):
self.print_attributes = print_attributes self.print_attributes = print_attributes
@ -43,6 +66,7 @@ class ConsoleSpanProcessor(SpanProcessor):
if key.startswith("__"): if key.startswith("__"):
continue continue
str_value = str(value) str_value = str(value)
str_value = _mask_sensitive_value(str_value)
if len(str_value) > 1000: if len(str_value) > 1000:
str_value = str_value[:997] + "..." str_value = str_value[:997] + "..."
logger.info(f" [dim]{key}[/dim]: {str_value}") logger.info(f" [dim]{key}[/dim]: {str_value}")
@ -64,7 +88,10 @@ class ConsoleSpanProcessor(SpanProcessor):
for key, value in event.attributes.items(): for key, value in event.attributes.items():
if key.startswith("__") or key in ["message", "severity"]: if key.startswith("__") or key in ["message", "severity"]:
continue continue
logger.info(f"[dim]{key}[/dim]: {value}")
str_value = str(value)
str_value = _mask_sensitive_value(str_value)
logger.info(f"[dim]{key}[/dim]: {str_value}")
def shutdown(self) -> None: def shutdown(self) -> None:
"""Shutdown the processor.""" """Shutdown the processor."""