From b7bc1c6297888765a601aef371fda80f77911bc3 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 26 Dec 2024 16:48:54 -0800 Subject: [PATCH] telemetry --- llama_stack/providers/utils/telemetry/tracing.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/utils/telemetry/tracing.py b/llama_stack/providers/utils/telemetry/tracing.py index 2846afdc8..f304d58f6 100644 --- a/llama_stack/providers/utils/telemetry/tracing.py +++ b/llama_stack/providers/utils/telemetry/tracing.py @@ -12,10 +12,18 @@ import threading import uuid from datetime import datetime from functools import wraps -from typing import Any, Callable, Dict, List +from typing import Any, Callable, Dict, List, Optional - -from llama_stack.apis.telemetry import * # noqa: F403 +from llama_stack.apis.telemetry import ( + LogSeverity, + Span, + SpanEndPayload, + SpanStartPayload, + SpanStatus, + StructuredLogEvent, + Telemetry, + UnstructuredLogEvent, +) from llama_stack.providers.utils.telemetry.trace_protocol import serialize_value log = logging.getLogger(__name__)