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

rebase

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-16 10:43:09 -08:00
parent 749bc74509
commit a9192e252c
2 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/utils/telemetry/dataset_mixin.py
View file

@ -49,7 +49,7 @@ class TelemetryDatasetMixin:
attributes_to_return: List[str],
max_depth: Optional[int] = None,
) -> QuerySpansResponse:
traces = await self.query_traces(attribute_filters=attribute_filters)
traces = await self.query_traces(attribute_filters=attribute_filters).data
spans = []
for trace in traces:

8
tests/client-sdk/agents/test_agents.py
View file

@ -178,6 +178,14 @@ def test_builtin_tool_web_search(llama_stack_client, agent_config):
assert "Tool:brave_search Response:" in logs_str
assert "mark zuckerberg" in logs_str.lower()
assert "No Violation" in logs_str
spans = llama_stack_client.telemetry.query_spans(
attribute_filters=[
{"key": "session_id", "op": "eq", "value": session_id},
],
attributes_to_return=["input", "output"],
)
breakpoint()
assert len(spans) > 1
def test_builtin_tool_code_execution(llama_stack_client, agent_config):