phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 10:52:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

passing tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-16 11:56:51 -08:00
parent bb2adef7e4
commit 97d017de1c
7 changed files with 887 additions and 655 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/providers/utils/telemetry/dataset_mixin.py
View file

@ -49,17 +49,17 @@ class TelemetryDatasetMixin:
attributes_to_return: List[str],
max_depth: Optional[int] = None,
) -> QuerySpansResponse:
traces = await self.query_traces(attribute_filters=attribute_filters).data
traces = await self.query_traces(attribute_filters=attribute_filters)
spans = []
for trace in traces:
spans_by_id = await self.query_span_tree(
for trace in traces.data:
spans_by_id_resp = await self.get_span_tree(
span_id=trace.root_span_id,
attributes_to_return=attributes_to_return,
max_depth=max_depth,
)
for span in spans_by_id.values():
for span in spans_by_id_resp.data.values():
if span.attributes and all(
attr in span.attributes and span.attributes[attr] is not None
for attr in attributes_to_return
@ -76,4 +76,4 @@ class TelemetryDatasetMixin:
)
)
return spans
return QuerySpansResponse(data=spans)