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

minor fixes

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-30 17:23:24 -08:00
parent 94b4113c63
commit e5c85b9dba
3 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -190,5 +190,5 @@ class Telemetry(Protocol):
async def export_agent_trace( async def export_agent_trace(
self, self,
session_ids: List[str], session_ids: List[str],
dataset_id: str = None, dataset_id: str,
) -> None: ... ) -> None: ...

2
llama_stack/providers/inline/meta_reference/telemetry/console.py
View file

@ -63,7 +63,7 @@ class ConsoleTelemetryImpl(Telemetry):
async def export_agent_trace( async def export_agent_trace(
self, self,
session_ids: List[str], session_ids: List[str],
dataset_id: str = None, dataset_id: str,
) -> None: ) -> None:
raise NotImplementedError( raise NotImplementedError(
"Console telemetry does not support agent trace export" "Console telemetry does not support agent trace export"

4
llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
View file

@ -223,9 +223,7 @@ class OpenTelemetryAdapter(Telemetry):
return traces return traces
async def export_agent_trace( async def export_agent_trace(self, session_ids: List[str], dataset_id: str) -> None:
self, session_ids: List[str], dataset_id: str = None
) -> None:
traces = await self.get_agent_trace(session_ids) traces = await self.get_agent_trace(session_ids)
traces_dict = [ traces_dict = [
{ {