phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 22:59:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

explicit span management using with

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-25 13:07:43 -08:00
parent 6411007024
commit b3021ea2da
3 changed files with 114 additions and 96 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/utils/telemetry/tracing.py
View file

@ -253,3 +253,11 @@ class SpanContextManager:
def span(name: str, attributes: Dict[str, Any] = None):
return SpanContextManager(name, attributes)
def get_current_span() -> Optional[Span]:
global CURRENT_TRACE_CONTEXT
context = CURRENT_TRACE_CONTEXT
if context:
return context.get_current_span()
return None