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

Idimatic REST API: Telemetry

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-15 20:54:34 -08:00
parent 8d30ecb91a
commit 749bc74509
3 changed files with 39 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,7 +7,7 @@
from typing import List, Optional
from llama_stack.apis.datasetio import DatasetIO
from llama_stack.apis.telemetry import QueryCondition, Span
from llama_stack.apis.telemetry import QueryCondition, QuerySpansResponse, Span
class TelemetryDatasetMixin:
@ -48,7 +48,7 @@ class TelemetryDatasetMixin:
attribute_filters: List[QueryCondition],
attributes_to_return: List[str],
max_depth: Optional[int] = None,
) -> List[Span]:
) -> QuerySpansResponse:
traces = await self.query_traces(attribute_filters=attribute_filters)
spans = []