telemetry

This commit is contained in:
Xi Yan 2024-12-26 16:48:54 -08:00
parent bb0a3f5c8e
commit b7bc1c6297

View file

@ -12,10 +12,18 @@ import threading
import uuid import uuid
from datetime import datetime from datetime import datetime
from functools import wraps 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 (
from llama_stack.apis.telemetry import * # noqa: F403 LogSeverity,
Span,
SpanEndPayload,
SpanStartPayload,
SpanStatus,
StructuredLogEvent,
Telemetry,
UnstructuredLogEvent,
)
from llama_stack.providers.utils.telemetry.trace_protocol import serialize_value from llama_stack.providers.utils.telemetry.trace_protocol import serialize_value
log = logging.getLogger(__name__) log = logging.getLogger(__name__)