mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-04 18:13:44 +00:00
chore(telemetry): code cleanup
# What does this PR do? ## Test Plan # What does this PR do? ## Test Plan
This commit is contained in:
parent
d12e5f0999
commit
0962228c73
36 changed files with 90 additions and 179 deletions
|
|
@ -256,7 +256,7 @@ class LiteLLMOpenAIMixin(
|
|||
params: OpenAIChatCompletionRequestWithExtraBody,
|
||||
) -> OpenAIChatCompletion | AsyncIterator[OpenAIChatCompletionChunk]:
|
||||
# Add usage tracking for streaming when telemetry is active
|
||||
from llama_stack.providers.utils.telemetry.tracing import get_current_span
|
||||
from llama_stack.core.telemetry.tracing import get_current_span
|
||||
|
||||
stream_options = params.stream_options
|
||||
if params.stream and get_current_span() is not None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue