get_metrics -> query_metrics

This commit is contained in:
Ashwin Bharambe 2025-05-06 17:10:52 -07:00
parent 7073efede5
commit 471745fff3
4 changed files with 331 additions and 331 deletions

View file

@ -240,7 +240,7 @@ class MetricSeries(BaseModel):
values: list[MetricDataPoint]
class GetMetricsResponse(BaseModel):
class QueryMetricsResponse(BaseModel):
data: list[MetricSeries]
@ -290,7 +290,7 @@ class Telemetry(Protocol):
) -> None: ...
@webmethod(route="/telemetry/metrics/{metric_name}", method="POST")
async def get_metrics(
async def query_metrics(
self,
metric_name: str,
start_time: int,
@ -298,4 +298,4 @@ class Telemetry(Protocol):
granularity: str | None = "1d",
query_type: MetricQueryType = MetricQueryType.RANGE,
label_matchers: list[MetricLabelMatcher] | None = None,
) -> GetMetricsResponse: ...
) -> QueryMetricsResponse: ...

View file

@ -19,11 +19,11 @@ from opentelemetry.semconv.resource import ResourceAttributes
from llama_stack.apis.telemetry import (
Event,
GetMetricsResponse,
MetricEvent,
MetricLabelMatcher,
MetricQueryType,
QueryCondition,
QueryMetricsResponse,
QuerySpanTreeResponse,
QueryTracesResponse,
Span,
@ -126,16 +126,16 @@ class TelemetryAdapter(TelemetryDatasetMixin, Telemetry):
else:
raise ValueError(f"Unknown event type: {event}")
async def get_metrics(
async def query_metrics(
self,
metric_name: str,
start_time: int,
end_time: Optional[int] = None,
step: Optional[str] = "1d",
end_time: int | None = None,
step: str | None = "1d",
query_type: MetricQueryType = MetricQueryType.RANGE,
label_matchers: Optional[List[MetricLabelMatcher]] = None,
) -> GetMetricsResponse:
pass
label_matchers: list[MetricLabelMatcher] | None = None,
) -> QueryMetricsResponse:
raise NotImplementedError("Querying metrics is not implemented")
def _log_unstructured(self, event: UnstructuredLogEvent, ttl_seconds: int) -> None:
with self._lock: