endpoint to bulk export traces for eval

This commit is contained in:
Dinesh Yeduguru 2024-11-26 22:09:49 -08:00
parent b3e149334a
commit dfe152cb97
2 changed files with 133 additions and 51 deletions

View file

@ -142,10 +142,14 @@ class Telemetry(Protocol):
@webmethod(route="/telemetry/log-event")
async def log_event(self, event: Event) -> None: ...
@webmethod(route="/telemetry/get-trace", method="GET")
@webmethod(route="/telemetry/get-trace", method="POST")
async def get_trace(self, trace_id: str) -> Trace: ...
@webmethod(route="/telemetry/get-traces-for-session", method="POST")
async def get_traces_for_session(
self, session_id: str, lookback: str = "1h", limit: int = 100
@webmethod(route="/telemetry/get-traces-for-eval", method="POST")
async def get_traces_for_eval(
self,
session_ids: List[str],
lookback: str = "1h",
limit: int = 100,
dataset_id: Optional[str] = None,
) -> List[Trace]: ...