new types and working

This commit is contained in:
Dinesh Yeduguru 2024-11-29 18:53:52 -08:00
parent b7c4997e91
commit 94b4113c63
6 changed files with 252 additions and 157 deletions

View file

@ -7,8 +7,6 @@
import json
from typing import List, Optional
from llama_stack.apis.telemetry.telemetry import Trace
from .config import LogFormat
from llama_stack.apis.telemetry import * # noqa: F403
@ -51,13 +49,25 @@ class ConsoleTelemetryImpl(Telemetry):
if formatted:
print(formatted)
async def get_trace(self, trace_id: str) -> Trace:
raise NotImplementedError()
async def get_trace(self, trace_id: str) -> TraceTree:
raise NotImplementedError("Console telemetry does not support trace retrieval")
async def get_traces_for_agent_eval(
self, session_ids: List[str], lookback: str = "1h", limit: int = 100
async def get_agent_trace(
self,
session_ids: List[str],
) -> List[EvalTrace]:
raise NotImplementedError()
raise NotImplementedError(
"Console telemetry does not support agent trace retrieval"
)
async def export_agent_trace(
self,
session_ids: List[str],
dataset_id: str = None,
) -> None:
raise NotImplementedError(
"Console telemetry does not support agent trace export"
)
COLORS = {