phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-05 12:21:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

bug fixes to make this work, trace creation worked - spans dont yet

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-19 08:56:03 -07:00
parent 84ebed9c9f
commit 6e5ca1350e
3 changed files with 25 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -223,13 +223,11 @@ class SpanContextManager:
def __call__(self, func: Callable):
@wraps(func)
def sync_wrapper(*args, **kwargs):
print("sync wrapper")
with self:
return func(*args, **kwargs)
@wraps(func)
async def async_wrapper(*args, **kwargs):
print("async wrapper")
async with self:
return await func(*args, **kwargs)