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

add new endpoints

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-16 11:03:34 -08:00
parent a9192e252c
commit bb2adef7e4
3 changed files with 31 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/inline/telemetry/meta_reference/telemetry.py
View file

@ -23,11 +23,13 @@ from llama_stack.apis.telemetry import (
QueryCondition,
QuerySpanTreeResponse,
QueryTracesResponse,
Span,
SpanEndPayload,
SpanStartPayload,
SpanStatus,
StructuredLogEvent,
Telemetry,
Trace,
UnstructuredLogEvent,
)
from llama_stack.distribution.datatypes import Api
@ -251,6 +253,12 @@ class TelemetryAdapter(TelemetryDatasetMixin, Telemetry):
)
)
async def get_trace(self, trace_id: str) -> Trace:
return await self.trace_store.get_trace(trace_id)
async def get_span(self, trace_id: str, span_id: str) -> Span:
return await self.trace_store.get_span(trace_id, span_id)
async def query_span_tree(
self,
span_id: str,