phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 15:32:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

explicit type for trace

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-27 09:24:23 -08:00
parent dfe152cb97
commit 2dfbb9744d
3 changed files with 23 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -137,6 +137,13 @@ Event = Annotated[
]
@json_schema_type
class EvalTrace(BaseModel):
step: str
input: str
output: str
@runtime_checkable
class Telemetry(Protocol):
@webmethod(route="/telemetry/log-event")
@ -145,11 +152,11 @@ class Telemetry(Protocol):
@webmethod(route="/telemetry/get-trace", method="POST")
async def get_trace(self, trace_id: str) -> Trace: ...
@webmethod(route="/telemetry/get-traces-for-eval", method="POST")
async def get_traces_for_eval(
@webmethod(route="/telemetry/get-traces-for-agent-eval", method="POST")
async def get_traces_for_agent_eval(
self,
session_ids: List[str],
lookback: str = "1h",
limit: int = 100,
dataset_id: Optional[str] = None,
) -> List[Trace]: ...
) -> List[EvalTrace]: ...