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

Use a Dict instead of tree for spans

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-16 11:39:33 -08:00
parent cb8a28c128
commit 5bbdcf70f6
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/apis/telemetry/telemetry.py
View file

@ -150,8 +150,7 @@ class EvalTrace(BaseModel):
@json_schema_type @json_schema_type
class SpanWithChildren(Span): class SpanWithStatus(Span):
children: List["SpanWithChildren"] = Field(default_factory=list)
status: Optional[SpanStatus] = None status: Optional[SpanStatus] = None
@ -192,7 +191,7 @@ class Telemetry(Protocol):
span_id: str, span_id: str,
attributes_to_return: Optional[List[str]] = None, attributes_to_return: Optional[List[str]] = None,
max_depth: Optional[int] = None, max_depth: Optional[int] = None,
) -> SpanWithChildren: ... ) -> Dict[str, SpanWithStatus]: ...
@webmethod(route="/telemetry/query-spans", method="POST") @webmethod(route="/telemetry/query-spans", method="POST")
async def query_spans( async def query_spans(